golang如何实现并发量控制以及精细化秒并发量控制

前言：golang如何实现并发量控制以及精细化秒并发量控制

参考文章：https://github.com/remeh/sizedwaitgroup
参考文章：https://github.com/uber-go/ratelimit

package main

import (
	"fmt"
	"sync"
	"time"

	"github.com/go-redis/sizedwaitgroup"
	"go.uber.org/ratelimit"
)

func main() {
	// 创建一个每秒允许一次操作的速率限制器
	rl := ratelimit.New(1) // per second

	// 定义并发数量
	maxConcurrency := 5
	totalTasks := 10

	// 创建一个带大小限制的等待组
	swg := sizedwaitgroup.New(maxConcurrency)

	// 启动多个任务
	for i := 0; i < totalTasks; i++ {
		swg.Add()
		go func(id int) {
			defer swg.Done()

			// 记录开始时间
			prev := time.Now()
			for j := 0; j < 10; j++ {
				now := rl.Take() // 控制速率
				if j > 0 {
					fmt.Printf("Task %d, Iteration %d: %v\n", id, j, now.Sub(prev))
				}
				prev = now
			}
		}(i)
	}

	// 等待所有任务完成
	swg.Wait()
}