golang worker pool , 工作池,线程池
資深大佬 : xie1xiao1jun 55
gowp
golang worker pool ,线程池 , 工作池
-
并发限制 goroutine 池。
-
限制任务执行的并发性,而不是排队的任务数。
-
无论排队多少任务,都不会阻止提交任务。
-
通过队列支持
-
golang 工作池公共库
支持最大任务数, 放到工作池里面 并等待全部完成
package main import ( "fmt" "time" "github.com/xxjwxc/gowp/workpool" ) func main() { wp := workpool.New(10) //设置最大线程数 for i := 0; i < 20; i++ { //开启 20 个请求 ii := i wp.Do(func() error { for j := 0; j < 10; j++ { //每次打印 0-10 的值 fmt.Println(fmt.Sprintf("%v->t%v", ii, j)) time.Sleep(1 * time.Second) } //time.Sleep(1 * time.Second) return nil }) } wp.Wait() fmt.Println("down") }
支持错误返回
package main import ( "fmt" "time" "github.com/xxjwxc/gowp/workpool" ) func main() { wp := workpool.New(10) //设置最大线程数 for i := 0; i < 20; i++ { //开启 20 个请求 ii := i wp.Do(func() error { for j := 0; j < 10; j++ { //每次打印 0-10 的值 fmt.Println(fmt.Sprintf("%v->t%v", ii, j)) if ii == 1 { return errors.Cause(errors.New("my test err")) //有 err 立即返回 } time.Sleep(1 * time.Second) } return nil }) } err := wp.Wait() if err != nil { fmt.Println(err) } fmt.Println("down") }
支持判断是否完成 (非阻塞)
package main import ( "fmt" "time" "github.com/xxjwxc/gowp/workpool" ) func main() { wp := workpool.New(5) //设置最大线程数 for i := 0; i < 10; i++ { //开启 20 个请求 // ii := i wp.Do(func() error { for j := 0; j < 5; j++ { //fmt.Println(fmt.Sprintf("%v->t%v", ii, j)) time.Sleep(1 * time.Second) } return nil }) fmt.Println(wp.IsDone())//判断是否完成 } wp.Wait() fmt.Println(wp.IsDone()) fmt.Println("down") }
支持同步等待结果
package main import ( "fmt" "time" "github.com/xxjwxc/gowp/workpool" ) func main() { wp := workpool.New(5) //设置最大线程数 for i := 0; i < 10; i++ { //开启 20 个请求 ii := i wp.DoWait(func() error { for j := 0; j < 5; j++ { fmt.Println(fmt.Sprintf("%v->t%v", ii, j)) // if ii == 1 { // return errors.New("my test err") // } time.Sleep(1 * time.Second) } return nil //time.Sleep(1 * time.Second) //return errors.New("my test err") }) } err := wp.Wait() if err != nil { fmt.Println(err) } fmt.Println("down") }
代码地址:gowp 喜欢请给星
大佬有話說 (6)