golang 限流器,控制并发,线程池

2021-05-19 18:09:00 +08:00
 xxjwxc

gowp

golang worker pool ,线程池 , 工作池

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

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

1919 次点击
所在节点    程序员
7 条回复
lrvy
2021-05-19 19:33:44 +08:00
.DS_Store?
wellsc
2021-05-19 19:38:20 +08:00
为什么不在网关层做这些呢
Hyvi
2021-05-19 19:50:36 +08:00
@wellsc +1
jziwenchen
2021-05-19 20:36:34 +08:00
赞一个
xxjwxc
2021-05-19 23:28:53 +08:00
@wellsc 很多可以用的,比如连接池,控制并发数,最大并发数!
xxjwxc
2021-05-19 23:30:42 +08:00
@Hyvi 数据库连接池,可以用这个工具,或者分布式并发数,比如控制 MQ 的消费者数量!
xxjwxc
2021-05-20 09:07:07 +08:00
@lrvy ??

这是一个专为移动设备优化的页面(即为了让你能够在 Google 搜索结果里秒开这个页面),如果你希望参与 V2EX 社区的讨论,你可以继续到 V2EX 上打开本讨论主题的完整版本。

https://www.v2ex.com/t/777961

V2EX 是创意工作者们的社区,是一个分享自己正在做的有趣事物、交流想法,可以遇见新朋友甚至新机会的地方。

V2EX is a community of developers, designers and creative people.

© 2021 V2EX