Go 如何应对高并发 | 高并发详解

在Go中处理高并发通常涉及到使用goroutines和channels。以下是一个简单的例子，展示了如何使用goroutines处理并发请求，以及如何使用channels来安全地同步和通信。

package main
 
import (
    "fmt"
    "sync"
    "sync/atomic"
    "time"
)
 
var (
    wg sync.WaitGroup
    counter int32
)
 
func worker(id int, c chan bool) {
    defer wg.Done()
    for {
        // 从channel接收任务
        <-c
        // 原子递增计数器
        atomic.AddInt32(&counter, 1)
        // 模拟耗时操作
        time.Sleep(10 * time.Millisecond)
    }
}
 
func main() {
    maxWorkers := 10
    tasks := make(chan bool, maxWorkers)
 
    // 启动worker goroutines
    for i := 0; i < maxWorkers; i++ {
        wg.Add(1)
        go worker(i, tasks)
    }
 
    // 发送任务
    start := time.Now()
    for i := 0; i < 1000; i++ {
        tasks <- true
    }
    close(tasks) // 关闭channel，通知所有worker退出
 
    // 等待所有worker完成
    wg.Wait()
 
    // 输出结果
    elapsed := time.Since(start)
    fmt.Printf("处理了 %d 个任务。用时: %s\n", counter, elapsed)
}

这段代码创建了一个worker池，worker goroutines会从channel接收任务并处理。主goroutine负责生成任务并发送到channel，然后关闭channel来通知所有worker退出。使用sync.WaitGroup确保主goroutine等待所有worker完成任务后再退出。使用atomic.AddInt32确保计数器的操作是线程安全的。这个例子演示了如何管理并发任务的一个简单方法。