Go实现线程池

谁践踏了优雅 2022-09-26 02:19 186阅读 0赞

本文通过代码讲解如何实现一个线程池。代码及注释如下：

<table> 
 <tbody> 
 <tr> 
 <td> 
 <div>
 1 
 2 
 3 
 4 
 5 
 6 
 7 
 8 
 9 
 10 
 11 
 12 
 13 
 14 
 15 
 16 
 17 
 18 
 19 
 20 
 21 
 22 
 23 
 24 
 25 
 26 
 27 
 28 
 29 
 30 
 31 
 32 
 33 
 34 
 35 
 36 
 37 
 
 </div> </td> 
 <td> 
 <div>
 package main 
 
 import 
 "fmt" 
 import 
 "time" 
 
 
 //这个是工作线程，处理具体的业务逻辑，将jobs中的任务取出，处理后将处理结果放置在results中。 
 func worker 
 (id 
 int 
 , jobs 
 &lt;-chan 
 int 
 , results chan 
 &lt;- 
 int 
 ) 
 { 
 &nbsp; &nbsp; 
 for j 
 := range jobs 
 { 
 &nbsp; &nbsp; &nbsp; &nbsp; fmt. 
 Println 
 ( 
 "worker" 
 , id 
 , 
 "processing job" 
 , j 
 ) 
 &nbsp; &nbsp; &nbsp; &nbsp; 
 <a href="http://www.opengroup.org/onlinepubs/009695399/functions/time.html" rel="nofollow"> time</a>. 
 Sleep 
 ( 
 <a href="http://www.opengroup.org/onlinepubs/009695399/functions/time.html" rel="nofollow">time</a>. 
 Second 
 ) 
 &nbsp; &nbsp; &nbsp; &nbsp; results 
 &lt;- j 
 * 
 2 
 &nbsp; &nbsp; 
 } 
 
 } 
 
 func main 
 ( 
 ) 
 { 
 
 &nbsp; &nbsp; 
 //两个channel，一个用来放置工作项，一个用来存放处理结果。 
 &nbsp; &nbsp; jobs 
 := make 
 (chan 
 int 
 , 
 100 
 ) 
 &nbsp; &nbsp; results 
 := make 
 (chan 
 int 
 , 
 100 
 ) 
 
 &nbsp; &nbsp; 
 // 开启三个线程，也就是说线程池中只有3个线程，实际情况下，我们可以根据需要动态增加或减少线程。 
 &nbsp; &nbsp; 
 for w 
 := 
 1 
 ; w 
 &lt;= 
 3 
 ; w 
 ++ 
 { 
 &nbsp; &nbsp; &nbsp; &nbsp; go worker 
 (w 
 , jobs 
 , results 
 ) 
 &nbsp; &nbsp; 
 } 
 
 &nbsp; &nbsp; 
 // 添加9个任务后关闭Channel 
 &nbsp; &nbsp; 
 // channel to indicate that's all the work we have. 
 &nbsp; &nbsp; 
 for j 
 := 
 1 
 ; j 
 &lt;= 
 9 
 ; j 
 ++ 
 { 
 &nbsp; &nbsp; &nbsp; &nbsp; jobs 
 &lt;- j 
 &nbsp; &nbsp; 
 } 
 &nbsp; &nbsp; close 
 (jobs 
 ) 
 
 &nbsp; &nbsp; 
 //获取所有的处理结果 
 &nbsp; &nbsp; 
 for a 
 := 
 1 
 ; a 
 &lt;= 
 9 
 ; a 
 ++ 
 { 
 &nbsp; &nbsp; &nbsp; &nbsp; 
 &lt;-results 
 &nbsp; &nbsp; 
 } 
 
 } 
 </div> </td> 
 </tr> 
 </tbody> 
</table>

输出结果：  
worker 1 processing job 1  
worker 2 processing job 2  
worker 3 processing job 3  
worker 1 processing job 4  
worker 3 processing job 5  
worker 2 processing job 6  
worker 1 processing job 7  
worker 2 processing job 8  
worker 3 processing job 9

从中可以看出，多个线程轮流处理了9个任务。  
通过这个例子，我们可以学习到：  
1、GO中多线程应用开发非常简单。  
2、Channel是不同线程间数据交互的利器。 上面的例子中，主线程向jobs中写数据，三个工作线程同时从一个Channel中取数据。