Go再学习_4.Go 最简单的协程池

客官°小女子只卖身不卖艺 2023-03-06 03:24 29阅读 0赞

## 说明 ##

在 Golang 中要创建一个协程是一件很简单的事情，你只要定义一个函数，并使用 go 关键字去执行它就行了。  
如果你接触过其他语言，会发现你在使用使用线程时，为了减少线程频繁创建销毁还来的开销，通常我们会使用线程池来复用线程。

池化技术就是利用复用来提升性能的，**那在 Golang 中需要协程池吗**？  
在 Golang 中，goroutine 是一个轻量级的线程，他的创建、调度都是在用户态进行，并不需要进入内核，这意味着创建销毁协程带来的开销是非常小的。因此，在大多数情况下，开发人员是不太需要使用协程池的。

## Code ##

package main
    
    import (
    	"fmt"
    	"time"
    )
    
    type Pool struct {
        
    	work_func chan func()
    	work_pool chan struct{
        }
    }
    
    func NewPool(size int) *Pool {
        
    	return &Pool{
        
    		work_func: make(chan func()),
    		work_pool: make(chan struct{
        }, size),
    	}
    }
    func (p *Pool) RunWork(task func()) {
        
    	println("Run task in work runtime")
    	defer func() {
         <-p.work_pool }()
    	for {
        
    		task()
    		task = <-p.work_func
    		println("get a new task from work_func")
    	}
    }
    func (p *Pool) NewTask(task func()) {
        
    	select {
        
    	case p.work_func <- task:
    		println("push func to task")
    	case p.work_pool <- struct{
        }{
        }:
    		go p.RunWork(task)
    		// Note: 这里不能有break和default，目的就是为了死循环查询状态
    	}
    }
    
    func main() {
        
    	fmt.Println("main run")
    	pool := NewPool(2)
    
    	for i := 1; i < 5; i++ {
        
    		var idx = i
    		pool.NewTask(func() {
        
    			time.Sleep(2 * time.Second)
    			fmt.Println(idx, time.Now())
    		})
    	}
    
    	// wait all
    	time.Sleep(8 * time.Second)
    	/*
    	main run
    	Run task in work runtime
    	Run task in work runtime
    	2 2020-08-01 10:27:02.9386028 +0800 CST m=+2.002617801
    	1 2020-08-01 10:27:02.9386028 +0800 CST m=+2.002617801
    	get a new task from work_func
    	push func to task
    	push func to task
    	get a new task from work_func
    	3 2020-08-01 10:27:04.9522576 +0800 CST m=+4.016272601
    	4 2020-08-01 10:27:04.9522576 +0800 CST m=+4.016272601
    	 */
    }