新手问题 golang work pool

xie1xiao1jun · 2019年11月23日 · 69 次阅读

gowp

golang worker pool , 线程池,工作池

  • 并发限制 goroutine 池。
  • 限制任务执行的并发性,而不是排队的任务数。
  • 无论排队多少任务,都不会阻止提交任务。
  • 通过队列支持

  • golang 工作池公共库

支持最大任务数,放到工作池里面 并等待全部完成

package main

import ( "fmt" "time"

"github.com/xxjwxc/gowp/workerpool" )

func main() { wp := workerpool.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) } return nil }) }

wp.Wait() fmt.Println("down") }

支持错误返回

package main

import ( "fmt" "time"

"github.com/xxjwxc/gowp/workerpool" )

func main() { wp := workerpool.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/workerpool" )

func main() { wp := workerpool.New(5) // 设置最大线程数 for i := 0; i < 10; i++ { //开启20个请求 // ii := i wp.Do(func() error { for j := 0; j < 5; j++ { //每次打印0-10的值 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/workerpool" )

func main() { wp := workerpool.New(5) // 设置最大线程数 for i := 0; i < 10; i++ { // 开启20个请求 ii := i wp.DoWait(func() error { for j := 0; j < 5; j++ { // 每次打印0-10的值 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 }) }

err := wp.Wait() if err != nil { fmt.Println(err) } fmt.Println("down") }

代码地址:gowp 喜欢请给星

更多原创文章干货分享,请关注公众号
  • 加微信实战群请加微信(注明:实战群):gocnio
暂无回复。
需要 登录 后方可回复, 如果你还没有账号请点击这里 注册