Golang并发控制技巧全解析

semaphore := make(chan struct{}, 10)
<p>for i := 0; i < 100; i++ {
semaphore <- struct{}{} // 获取信号量
go func(id int) {
defer func() { <-semaphore }() // 释放信号量
// 执行耗时操作，如HTTP请求或文件读写
fmt.Printf("处理任务 %d\n", id)
}(i)
}</p>

var bufferPool = sync.Pool{
    New: func() interface{} {
        return &bytes.Buffer{}
    },
}
<p>func process(data []byte) string {
buf := bufferPool.Get().(*bytes.Buffer)
defer bufferPool.Put(buf)
buf.Reset()
buf.Write(data)
return buf.String()
}</p>

ctx, cancel := context.WithTimeout(context.Background(), 2*time.Second)
defer cancel()
<p>for i := 0; i < 5; i++ {
go func(id int) {
select {
case <-time.After(3 * time.Second):
fmt.Printf("任务 %d 完成\n", id)
case <-ctx.Done():
fmt.Printf("任务 %d 被取消: %v\n", id, ctx.Err())
}
}(i)
}</p><p>// 等待所有任务结束或超时
<-ctx.Done()</p>

limiter := rate.NewLimiter(10, 1) // 每秒10个token，burst为1
<p>for i := 0; i < 20; i++ {
if err := limiter.Wait(ctx); err != nil {
fmt.Printf("请求被拒绝: %v\n", err)
continue
}
// 处理请求
fmt.Printf("执行请求 %d\n", i)
}</p>