代码拉取完成，页面将自动刷新
OpenMix 出品：https://openmix.org
通用动态工作池、协程池
A dynamic golang worker pool, coroutine pool
go get github.com/mix-go/xwp
适合处理数据计算、转换等场景
先创建一个结构体用来处理任务，使用类型断言转换任务数据类型，例如：i := data.(int)
type Foo struct {
}
func (t *Foo) Do(data interface{}) {
// do something
}
调度任务
RunF 采用闭包来处理任务p.Start() 启动jobQueue := make(chan interface{}, 200)
p := &xwp.WorkerPool{
JobQueue: jobQueue,
MaxWorkers: 1000,
InitWorkers: 100,
MaxIdleWorkers: 100,
RunI: &Foo{},
}
go func() {
// 投放任务
for i := 0; i < 10000; i++ {
jobQueue <- i
}
// 投放完停止调度
p.Stop()
}()
p.Run() // 阻塞等待
适合处理 MQ 队列的异步消费
以 Redis 作为 MQ 为例：
jobQueue := make(chan interface{}, 200)
p := &xwp.WorkerPool{
JobQueue: jobQueue,
MaxWorkers: 1000,
InitWorkers: 100,
MaxIdleWorkers: 100,
RunI: &Foo{},
}
ch := make(chan os.Signal)
signal.Notify(ch, syscall.SIGHUP, syscall.SIGINT, syscall.SIGTERM)
go func() {
<-ch
p.Stop()
}()
go func() {
for {
res, err := redis.BRPop(context.Background(), 3*time.Second, "foo").Result()
if err != nil {
if strings.Contains(err.Error(), "redis: nil") {
continue
}
fmt.Println(fmt.Sprintf("Redis Error: %s", err))
p.Stop()
return
}
// brPop命令最后一个键才是值
jobQueue <- res[1]
}
}()
p.Run() // 阻塞等待
Do 方法中执行的代码，可能会出现 panic 异常，我们可以通过 recover 获取异常信息记录到日志或者执行其他处理
func (t *Foo) Do(data interface{}) {
defer func() {
if err := recover(); err != nil {
// handle error
}
}()
// do something
}
Stats() 可以返回 Workers 实时执行状态，通常可以使用一个定时器，定时打印或者告警处理
go func() {
ticker := time.NewTicker(1000 * time.Millisecond)
for {
<-ticker.C
log.Printf("%+v", p.Stats()) // 2021/04/26 14:32:53 &{Active:5 Idle:95 Total:100}
}
}()
Apache License Version 2.0, http://www.apache.org/licenses/
此处可能存在不合适展示的内容，页面不予展示。您可通过相关编辑功能自查并修改。
如您确认内容无涉及 不当用语 / 纯广告导流 / 暴力 / 低俗色情 / 侵权 / 盗版 / 虚假 / 无价值内容或违法国家有关法律法规的内容，可点击提交进行申诉，我们将尽快为您处理。
马建仓 AI 助手