开源推荐 golang 限流器,控制并发,线程池

xie1xiao1jun · 2021年05月26日 · 322 次阅读
本帖已被设为精华帖!

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)
            }
            //time.Sleep(1 * time.Second)
            return nil
        })
    }

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

限流器 (cache)

package main

import (
    "fmt"
    "sync"
    "time"

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

func main() {
    limiter := limiter.NewLimiter(limiter.WithLimit(10), limiter.WithNamespace("test"), limiter.WithTsTimeout(true) /*, limiter.WithRedis(res)*/)
    var wg sync.WaitGroup
    for i := 0; i < 20; i++ {
        wg.Add(1)
        go func() {
            defer wg.Done()
            token, _ := limiter.Acquire(10) // 获取一个
            fmt.Println(token)

            time.Sleep(1 * time.Second)
            limiter.Release(token) // 回退
        }()
    }
    wg.Wait()
    fmt.Println("down")
}

限流器 (redis)

package main

import (
    "fmt"
    "sync"
    "time"

    "github.com/xxjwxc/gowp/limiter"
    "github.com/xxjwxc/public/myredis"
)

func main() {
    conf := myredis.InitRedis(myredis.WithAddr("127.0.0.1:6379"), myredis.WithPwd("123456"), myredis.WithGroupName("test"))
    res, err := myredis.NewRedis(conf)
    if err != nil {
        fmt.Println(err)
        return
    }

    limiter := limiter.NewLimiter(limiter.WithRedis(res), limiter.WithLimit(10), limiter.WithNamespace("test") /*, limiter.WithRedis(res)*/)
    var wg sync.WaitGroup
    for i := 0; i < 20; i++ {
        wg.Add(1)
        go func() {
            defer wg.Done()
            token, _ := limiter.Acquire(10) // 获取一个
            fmt.Println(token)

            time.Sleep(1 * time.Second)
            limiter.Release(token) // 回退
        }()
    }
    wg.Wait()
    fmt.Println("down")
}


代码地址 gowp

更多原创文章干货分享,请关注公众号
  • 加微信实战群请加微信(注明:实战群):gocnio
astaxie 将本帖设为了精华贴 05月26日 10:59
需要 登录 后方可回复, 如果你还没有账号请点击这里 注册