channel实现并发控制

前言

怎么用channel来控制并发呢？我们最直观的想法就是构造一个带缓存的channel，一个协程在执行前先往里面占一个坑，执行完后再释放这个坑位，这样一来始终只有固定数量的协程在同时执行，从而达到并发控制的效果。

 

实现

open-falcon里使用第三方包 github.com/toolkits/concurrent/semaphore 来实现这种效果，先上代码：

 

package main
import(
    "fmt"
    "time"
    nsema "github.com/toolkits/concurrent/semaphore"  //查看该源码，其实就是利用了channel来做并发控制
)

func main(){

    concurrentNum := 2 //设置并发数2
    sema := nsema.NewSemaphore(concurrentNum)

    for i:=0; i<8; i++ {
        sema.Acquire()  //先占一个坑，往channel里写一个数据
        go func(num int){
            defer sema.Release()   //释放坑位
            fmt.Printf("%d, get sema, ok\n", num)
            time.Sleep(1*time.Second)

        }(i)
    }
    // keep alive
    time.Sleep(5*time.Second)
}

 

 

结构体

type Semaphore struct {
    bufSize int
    channel chan int8    //带缓存的channel
}

 

Acquire()方法

 

func (this *Semaphore) Acquire() {
    this.channel <- int8(0)  //往cannel里写入一条数据
}

 

 

Release()方法

func (this *Semaphore) Release() {
    <-this.channel   //从channel里拿出一条数据
}