ants/README.md

246 lines
6.6 KiB
Markdown
Raw Normal View History

2018-05-19 04:13:39 +03:00
# ants
2018-05-19 04:15:54 +03:00
2018-06-08 09:54:31 +03:00
<div align="center"><img src="ants_logo.png"/></div>
2018-05-19 04:15:54 +03:00
2018-05-20 18:34:34 +03:00
<p align="center">A goroutine pool for Go</p>
2018-05-20 18:35:28 +03:00
2018-05-20 18:27:20 +03:00
2018-06-29 05:45:58 +03:00
[![Build Status][1]][2]
2018-07-15 17:45:55 +03:00
[![codecov][3]][4]
2018-06-29 05:45:58 +03:00
[![goreportcard for panjf2000/ants][5]][6]
2018-07-15 17:45:55 +03:00
[![godoc for panjf2000/ants][7]][8]
[![MIT Licence][9]][10]
2018-05-20 18:27:20 +03:00
2018-07-23 17:19:03 +03:00
[中文项目说明](README_ZH.md) | [Project Tutorial](http://blog.taohuawu.club/article/goroutine-pool)
2018-06-08 09:50:56 +03:00
2018-05-20 18:27:20 +03:00
Package ants implements a fixed goroutine pool for managing and recycling a massive number of goroutines, allowing developers to limit the number of goroutines that created by your concurrent programs.
## Features:
2018-05-21 05:45:25 +03:00
- Automatically managing and recycling a massive number of goroutines.
2018-07-09 11:34:44 +03:00
- Periodically clearing overdue goroutines.
2018-07-09 11:32:33 +03:00
- Friendly interfaces: submitting tasks, getting the number of running goroutines, readjusting capacity of pool dynamically, closing pool.
2018-05-21 05:45:25 +03:00
- Efficient in memory usage and it even achieves higher performance than unlimited goroutines in golang.
2018-05-20 18:27:20 +03:00
## How to install
``` sh
go get -u github.com/panjf2000/ants
```
Or, using glide:
``` sh
glide get github.com/panjf2000/ants
```
## How to use
If your program will generate a massive number of goroutines and you don't want them to consume a vast amount of memory, with ants, all you need to do is to import ants package and submit all your tasks to the default limited pool created when ants was imported:
``` go
package main
import (
"fmt"
"sync"
2018-05-26 03:42:10 +03:00
"sync/atomic"
2018-07-02 09:45:25 +03:00
"time"
2018-05-26 03:42:10 +03:00
"github.com/panjf2000/ants"
2018-05-20 18:27:20 +03:00
)
2018-05-26 03:42:10 +03:00
var sum int32
2018-12-03 06:23:37 +03:00
func myFunc(i interface{}) {
2018-07-02 09:45:25 +03:00
n := i.(int32)
atomic.AddInt32(&sum, n)
2018-05-26 03:42:10 +03:00
fmt.Printf("run with %d\n", n)
}
2018-12-03 06:23:37 +03:00
func demoFunc() {
2018-05-26 03:42:10 +03:00
time.Sleep(10 * time.Millisecond)
2018-05-20 18:27:20 +03:00
fmt.Println("Hello World!")
}
func main() {
2018-07-02 09:45:25 +03:00
defer ants.Release()
2018-05-26 03:42:10 +03:00
runTimes := 1000
// use the common pool
2018-05-20 18:27:20 +03:00
var wg sync.WaitGroup
for i := 0; i < runTimes; i++ {
wg.Add(1)
2018-12-01 14:41:02 +03:00
ants.Submit(func() {
2018-05-26 03:42:10 +03:00
demoFunc()
2018-05-20 18:27:20 +03:00
wg.Done()
})
}
wg.Wait()
2018-05-26 03:42:10 +03:00
fmt.Printf("running goroutines: %d\n", ants.Running())
fmt.Printf("finish all tasks.\n")
// use the pool with a function
2018-07-06 15:39:23 +03:00
// set 10 the size of goroutine pool and 1 second for expired duration
2018-12-01 14:41:02 +03:00
p, _ := ants.NewPoolWithFunc(10, func(i interface{}) {
2018-05-26 03:42:10 +03:00
myFunc(i)
wg.Done()
})
2018-07-02 09:45:25 +03:00
defer p.Release()
2018-05-26 03:42:10 +03:00
// submit tasks
for i := 0; i < runTimes; i++ {
wg.Add(1)
2018-07-02 09:45:25 +03:00
p.Serve(int32(i))
2018-05-26 03:42:10 +03:00
}
wg.Wait()
fmt.Printf("running goroutines: %d\n", p.Running())
fmt.Printf("finish all tasks, result is %d\n", sum)
2018-05-20 18:27:20 +03:00
}
```
2018-12-02 08:48:44 +03:00
## Integrate with http server
```go
package main
import (
"io/ioutil"
"net/http"
"github.com/panjf2000/ants"
)
type Request struct {
Param []byte
Result chan []byte
}
2018-12-02 08:48:44 +03:00
func main() {
pool, _ := ants.NewPoolWithFunc(100, func(payload interface{}) {
2018-12-03 05:13:53 +03:00
request, ok := payload.(*Request)
2018-12-02 08:48:44 +03:00
if !ok {
return
2018-12-02 08:48:44 +03:00
}
reverseParam := func(s []byte) []byte {
for i, j := 0, len(s)-1; i < j; i, j = i+1, j-1 {
s[i], s[j] = s[j], s[i]
}
return s
}(request.Param)
2018-12-02 08:48:44 +03:00
request.Result <- reverseParam
2018-12-02 08:48:44 +03:00
})
defer pool.Release()
http.HandleFunc("/reverse", func(w http.ResponseWriter, r *http.Request) {
param, err := ioutil.ReadAll(r.Body)
2018-12-02 08:48:44 +03:00
if err != nil {
http.Error(w, "request error", http.StatusInternalServerError)
}
defer r.Body.Close()
2018-12-03 05:13:53 +03:00
request := &Request{Param: param, Result: make(chan []byte)}
2018-12-02 08:48:44 +03:00
// Throttle the requests with ants pool. This process is asynchronous and
// you can receive a result from the channel defined outside.
if err := pool.Serve(request); err != nil {
2018-12-02 08:48:44 +03:00
http.Error(w, "throttle limit error", http.StatusInternalServerError)
}
w.Write(<-request.Result)
2018-12-02 08:48:44 +03:00
})
http.ListenAndServe(":8080", nil)
}
```
2018-05-21 07:45:15 +03:00
## Submit tasks
2018-05-26 12:26:36 +03:00
Tasks can be submitted by calling `ants.Submit(func())`
2018-05-21 07:45:15 +03:00
```go
2018-12-02 08:48:44 +03:00
ants.Submit(func(){})
2018-05-21 07:45:15 +03:00
```
## Custom limited pool
2018-05-21 07:48:03 +03:00
Ants also supports custom limited pool. You can use the `NewPool` method to create a pool with the given capacity, as following:
2018-05-20 18:27:20 +03:00
``` go
2018-07-24 16:53:34 +03:00
// set 10000 the size of goroutine pool
p, _ := ants.NewPool(10000)
2018-05-21 07:45:15 +03:00
// submit a task
2018-12-02 08:48:44 +03:00
p.Submit(func(){})
2018-05-20 18:27:20 +03:00
```
## Readjusting pool capacity
You can change ants pool capacity at any time with `ReSize(int)`:
``` go
pool.ReSize(1000) // Readjust its capacity to 1000
pool.ReSize(100000) // Readjust its capacity to 100000
```
Don't worry about the synchronous problems in this case, this method is thread-safe.
## About sequence
All the tasks submitted to ants pool will not be guaranteed to be processed in order, because those tasks distribute among a series of concurrent workers, thus those tasks are processed concurrently.
## Benchmarks
2018-07-06 15:39:23 +03:00
```
OS : macOS High Sierra
Processor : 2.7 GHz Intel Core i5
Memory : 8 GB 1867 MHz DDR3
Go1.9
```
2018-06-08 09:54:31 +03:00
<div align="center"><img src="ants_benchmarks.png"/></div>
2018-05-20 18:27:20 +03:00
2018-07-22 04:32:55 +03:00
In that benchmark-picture, the first and second benchmarks performed test with 1M tasks and the rest of benchmarks performed test with 10M tasks, both unlimited goroutines and ants pool, and the capacity of this ants goroutine-pool was limited to 50K.
2018-05-20 18:34:34 +03:00
2018-05-21 05:45:25 +03:00
- BenchmarkGoroutine-4 represent the benchmarks with unlimited goroutines in golang.
2018-05-20 18:27:20 +03:00
2018-05-21 05:45:25 +03:00
- BenchmarkPoolGroutine-4 represent the benchmarks with a ants pool.
2018-05-20 18:27:20 +03:00
The test data above is a basic benchmark and the more detailed benchmarks will be uploaded later.
2018-05-27 17:41:55 +03:00
### Benchmarks with Pool
2018-06-08 09:54:31 +03:00
![](benchmark_pool.png)
2018-05-27 17:41:55 +03:00
2018-07-22 04:32:55 +03:00
In that benchmark-picture, the first and second benchmarks performed test with 1M tasks and the rest of benchmarks performed test with 10M tasks, both unlimited goroutines and ants pool, and the capacity of this ants goroutine-pool was limited to 50K.
2018-05-27 17:41:55 +03:00
2018-07-22 04:32:55 +03:00
**As you can see, `ants` can up to 2x faster than goroutines without pool (10M tasks) and it only consumes half memory comparing with goroutines without pool. (both 1M and 10M tasks)**
2018-05-27 17:41:55 +03:00
### Benchmarks with PoolWithFunc
2018-06-08 09:54:31 +03:00
![](ants_bench_poolwithfunc.png)
2018-05-27 17:41:55 +03:00
2018-06-28 05:52:20 +03:00
### Throughput ( situation for only submitting tasks and need not waiting for all the tasks to be completed)
2018-05-27 17:41:55 +03:00
2018-07-22 04:32:55 +03:00
#### 100K tasks
2018-05-27 17:41:55 +03:00
2018-06-08 09:54:31 +03:00
![](ants_bench_10w.png)
2018-05-27 17:41:55 +03:00
2018-07-22 04:32:55 +03:00
#### 1M tasks
2018-05-27 17:41:55 +03:00
2018-06-08 09:54:31 +03:00
![](ants_bench_100w.png)
2018-05-27 17:41:55 +03:00
2018-07-22 04:32:55 +03:00
#### 10M tasks
2018-05-27 17:41:55 +03:00
2018-06-08 09:54:31 +03:00
![](ants_bench_1000w.png)
2018-05-27 17:41:55 +03:00
2018-08-04 05:34:35 +03:00
There was only the test of `ants` Pool because my computer was crash when it reached 10M goroutines without pool.
2018-05-27 17:41:55 +03:00
2018-06-28 05:52:20 +03:00
**As you can see, `ants` can up to 2x~6x faster than goroutines without pool and the memory consumption is reduced by 10 to 20 times.**
2018-12-01 13:43:46 +03:00
[1]: https://travis-ci.com/panjf2000/ants.svg?branch=master
2018-06-29 05:45:58 +03:00
[2]: https://travis-ci.com/panjf2000/ants
2018-12-01 13:43:46 +03:00
[3]: https://codecov.io/gh/panjf2000/ants/branch/master/graph/badge.svg
2018-07-15 17:49:47 +03:00
[4]: https://codecov.io/gh/panjf2000/ants
2018-06-29 05:45:58 +03:00
[5]: https://goreportcard.com/badge/github.com/panjf2000/ants
[6]: https://goreportcard.com/report/github.com/panjf2000/ants
2018-07-15 17:45:55 +03:00
[7]: https://godoc.org/github.com/panjf2000/ants?status.svg
[8]: https://godoc.org/github.com/panjf2000/ants
[9]: https://badges.frapsoft.com/os/mit/mit.svg?v=103
[10]: https://opensource.org/licenses/mit-license.php