golang 简单线程池

golang简单线程池实现

在并发编程中，线程池是一个非常常见且重要的概念。它可以帮助我们管理线程的数量，优化系统资源的使用，提高程序性能。在golang中，我们可以通过基本的goroutine和channel来实现一个简单的线程池。

下面是一个简单的golang线程池的实现示例：

```go package main import ( "fmt" "sync" ) type Job struct { id int num int } type Result struct { job Job sum int sqrt float64 } func worker(wg *sync.WaitGroup, jobs <-chan Job, results chan<- Result) { for job := range jobs { sum := 0 for i := 1; i <= job.num; i++ { sum += i } results <- Result{job: job, sum: sum, sqrt: sqrt(float64(sum))} } wg.Done() } func sqrt(num float64) float64 { // 对num进行平方根计算 return num // 假设这里是一个真实的计算 } func main() { const numJobs = 10 const numWorkers = 4 var wg sync.WaitGroup jobs := make(chan Job, numJobs) results := make(chan Result, numJobs) for w := 1; w <= numWorkers; w++ { wg.Add(1) go worker(&wg, jobs, results) } for j := 1; j <= numJobs; j++ { jobs <- Job{id: j, num: j * 2} } close(jobs) wg.Wait() close(results) for result := range results { fmt.Printf("Job %d: sum=%d, sqrt=%f\n", result.job.id, result.sum, result.sqrt) } } ```

在这个示例中，我们定义了两个结构体：Job和Result。Job结构体表示要执行的任务，Result结构体表示任务的执行结果。线程池的每个工作协程将从一个jobs通道中读取Job，并将执行结果写入results通道。

由于我们不知道任务的数量，因此我们使用了两个通道：jobs用于传递待执行的任务，results用于接收任务的执行结果。我们在main函数中将所有任务发送到jobs通道中，在工作协程中，我们从jobs通道中接收任务，并将执行结果写入results通道。

在main函数中，我们使用sync.WaitGroup来等待所有的工作协程完成工作。在每个工作协程完成工作后，我们调用wg.Done()来通知WaitGroup该协程已完成。最后，我们在关闭jobs通道后，从results通道中接收并打印所有的执行结果。

通过使用线程池，我们可以灵活地控制并发协程的数量，以提高系统的性能和资源利用率。此外，通过使用goroutine和channel，我们可以轻松地实现并发安全和数据同步。

总之，golang提供了强大的并发编程工具，通过简单的goroutine和channel的组合，我们可以实现一个简单的线程池，提高程序的性能和可维护性。