https://github.com/wychl/worker
https://github.com/wychl/worker
Last synced: 7 months ago
JSON representation
- Host: GitHub
- URL: https://github.com/wychl/worker
- Owner: wychl
- Created: 2021-05-13T02:44:03.000Z (over 4 years ago)
- Default Branch: master
- Last Pushed: 2021-05-13T05:44:11.000Z (over 4 years ago)
- Last Synced: 2025-01-29T13:08:13.494Z (8 months ago)
- Language: Go
- Size: 1.95 KB
- Stars: 0
- Watchers: 2
- Forks: 0
- Open Issues: 0
-
Metadata Files:
- Readme: README.md
Awesome Lists containing this project
README
# worker
通用的任务处理器
## 案例
### 计算`1-10000`和
```go
package main
import (
"context"
"fmt"
"sync"
"github.com/wychl/worker"
)
func main() {
wg := sync.WaitGroup{}
batch := 5
sum := 0
mux := sync.RWMutex{}
dataChan := make(chan interface{}, 10)
// goroutine写入数据
go func() {
step := 500 // 步长
max := 10000 //最大值
for i := 0; i < 20; i++ {
// 截止值
endValue := (i + 1) * step
if i == 19 {
endValue = max + 1
}
// 数据写入channel
dataChan <- sumPayload{
startValue: i * step,
endValue: endValue,
}
}
// 数据已经写完，关闭channel
close(dataChan)
}()
worker := worker.New(batch, newExe(&sum, &mux), &wg)
ctx := context.Background()
worker.Run(dataChan, ctx)
// 等待计算任务完成
<-worker.Finished()
fmt.Println("sum:", sum)
}
// 返回和计算函数
func newExe(sum *int, mux *sync.RWMutex) worker.Exec {
return func(data interface{}) {
p := data.(sumPayload)
for i := p.startValue; i < p.endValue; i++ {
mux.Lock()
*sum += i
mux.Unlock()
}
}
}
// 和计算函数，输入参数类型
type sumPayload struct {
startValue int
endValue int
}
```