Go语言中工作负载类型对并发的影响

package main
 
import (
    "fmt"
    "runtime"
    "sync"
)
 
// 定义不同的工作负载类型
type WorkloadType int
 
const (
    IOBound WorkloadType = iota // IO密集型
    CPUBound                     // 计算密集型
)
 
// 根据工作负载类型调整Go程池的大小
func SetWorkerPoolSize(workloadType WorkloadType, workerCount int) {
    switch workloadType {
    case IOBound:
        // IO密集型工作负载可以增加Go程池的大小来利用多核
        runtime.GOMAXPROCS(runtime.NumCPU())
    case CPUBound:
        // 计算密集型工作负载可以减少Go程池的大小避免过度的上下文切换
        runtime.GOMAXPROCS(workerCount)
    }
}
 
// 模拟工作负载的例程
func worker(wg *sync.WaitGroup, workloadType WorkloadType) {
    defer wg.Done()
    switch workloadType {
    case IOBound:
        // IO密集型操作
        time.Sleep(100 * time.Millisecond)
    case CPUBound:
        // 计算密集型操作
        for i := 0; i < 1000000000; i++ {
        }
    }
}
 
func main() {
    // 设置工作负载类型并调整Go程池大小
    SetWorkerPoolSize(CPUBound, 4)
 
    // 启动并发任务
    var wg sync.WaitGroup
    for i := 0; i < 10; i++ {
        wg.Add(1)
        go worker(&wg, CPUBound)
    }
    wg.Wait()
}

这段代码根据工作负载类型调整了Go程池的大小，并模拟了两种类型的工作负载。IO密集型工作负载可以增加Go程池的大小来利用多核，而计算密集型工作负载可以减少Go程池的大小来避免过度的上下文切换。