分享一個非常好用的GO並行控制庫 !

2022-12-21 22:01:12
本篇文章給大家介紹有關Golang的相關知識,聊聊一個非常好用的golang並行控制庫,希望對大家有所幫助。

Concurrency

install

GOPROXY=https://goproxy.cn go get -v github.com/lxzan/concurrency@latest
登入後複製

Feature

  • 並行協程數量限制
  • 支援 contex.Contex
  • 支援 panic recover, 返回包含錯誤堆疊的 error
  • 遞迴實現的任務排程, 不依賴 time.Tickerchannel

Usage

  • WorkerGroup 工作組, 新增一組任務, 等待執行完成, 可以很好的替代WaitGroup.
package mainimport (
    "fmt"
    "github.com/lxzan/concurrency"
    "sync/atomic")func main() {
    sum := int64(0)
    w := concurrency.NewWorkerGroup()
    for i := int64(1); i <= 10; i++ {
        w.AddJob(concurrency.Job{
            Args: i,
            Do: func(args interface{}) error {
                fmt.Printf("%v ", args)
                atomic.AddInt64(&sum, args.(int64))
                return nil
            },
        })
    }
    w.StartAndWait()
    fmt.Printf("sum=%d\n", sum)}
登入後複製
4 5 6 7 8 9 10 1 3 2 sum=55
登入後複製
  • WorkerQueue 工作佇列, 可以不斷往裡面新增任務, 一旦有CPU資源空閒就去執行【推薦:】
package mainimport (
    "fmt"
    "github.com/lxzan/concurrency"
    "time")func Add(args interface{}) error {
    arr := args.([]int)
    ans := 0
    for _, item := range arr {
        ans += item    }
    fmt.Printf("args=%v, ans=%d\n", args, ans)
    return nil}func Mul(args interface{}) error {
    arr := args.([]int)
    ans := 1
    for _, item := range arr {
        ans *= item    }
    fmt.Printf("args=%v, ans=%d\n", args, ans)
    return nil}func main() {
    args1 := []int{1, 3}
    args2 := []int{1, 3, 5}
    w := concurrency.NewWorkerQueue()
    w.AddJob(
        concurrency.Job{Args: args1, Do: Add},
        concurrency.Job{Args: args1, Do: Mul},
        concurrency.Job{Args: args2, Do: Add},
        concurrency.Job{Args: args2, Do: Mul},
    )
    w.StopAndWait(30*time.Second)}
登入後複製
args=[1 3], ans=4args=[1 3 5], ans=15args=[1 3], ans=3args=[1 3 5], ans=9
登入後複製

以上就是分享一個非常好用的GO並行控制庫 !的詳細內容,更多請關注TW511.COM其它相關文章!