Golang 코루틴 풀 관리 및 최적화

package main import ( "fmt" "sync" "time" ) type Job struct { Data int Result chan int } func main() { // 创建一个有缓冲的通道用于处理任务结果 result := make(chan int, 10) // 创建一个协程池 var wg sync.WaitGroup pool := make(chan *Job) for i := 0; i < 4; i++ { wg.Add(1) go func(pool chan *Job, wg *sync.WaitGroup) { defer wg.Done() for { job := <-pool job.Result <- job.Data * job.Data } }(pool, &wg) } // 模拟任务处理 for i := 0; i < 10; i++ { job := Job{ Data: i, Result: result, } pool <- &job } close(pool) wg.Wait() close(result) // 打印任务结果 for r := range result { fmt.Println(r) } }

package main import ( "fmt" "sync" "time" "image" "image/jpeg" "os" ) type Job struct { ImageFile string ResultImage chan<- image.Image } func main() { resultChan := make(chan image.Image) // 创建一个协程池 var wg sync.WaitGroup pool := make(chan *Job) for i := 0; i < 4; i++ { wg.Add(1) go func(pool chan *Job, wg *sync.WaitGroup) { defer wg.Done() for { job := <-pool image, err := loadAndProcessImage(job.ImageFile) if err != nil { fmt.Println(err) continue } job.ResultImage <- image } }(pool, &wg) } // 将图像处理任务提交给协程池 for { imageFile, ok := <-filesChan // 从文件通道取文件 if !ok { break } job := Job{ ImageFile: imageFile, ResultImage: resultChan, } pool <- &job } close(pool) wg.Wait() close(resultChan) // 保存处理后的图像 for img := range resultChan { outputFile, err := os.Create("processed_" + imgFile) if err != nil { fmt.Println(err) continue } if err := jpeg.Encode(outputFile, img, &jpeg.Options{Quality: 95}); err != nil { fmt.Println(err) outputFile.Close() continue } outputFile.Close() } }