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里拿出一条数据
}

 

channel实现并发控制

上一篇:python中的 dir()内置函数的作用以及使用方法


下一篇:PDF 补丁丁 0.4.2.1013 测试版发布:修复替换PDF文档字体等问题