@TOC问题背景
在实际项目中，希望用n个goroutine去一个带缓冲区的channel中读取数据，并且每个goroutine每次消费1000个数据，如果不足则按具体个数消费即可。为什么每次要消费1000个数据呢，因为希望在每个goroutine里面使用redis pipeline去获取这1000个数据的对应信息，这样就可以减少redis的访问次数。 而Channel中的数据量是未知的有限数据。

解决思路
在上述的背景下，因为本人之前是Java开发者，所以想到了Java里面的Semaphore共享锁，共享锁就是可以一个有n个凭证去访问共享资源的共享锁，是不是一说这里，大家就都想到了解决办法，n个goroutine并行执行，其实就相当于有n个凭证可以去访问共享资源。所以下面我们要做的就是用go去实现一个共享锁，这样问题就解决了。

具体实现
go实现共享锁，我用了一种比较简单的方式，借助缓冲区大小为n的channel去实现，当缓冲区个数达到n个之后，channel会阻塞。所以每启动一个goroutine，就向channel里面插入一个数据，每个goroutine完成，channel就输出一个数据，这样可以保证程序中最多只有n个goroutine在运行。下面看下代码具体实现。

semaphore.go

package semaphore

// Semaphore 借助channel实现的共享锁
type Semaphore struct {
Channel chan int
}

// NewSemaphore 创建一个共享锁 有parallelism个共享凭证
func NewSemaphore(parallelism int) *Semaphore {
return &Semaphore{Channel: make(chan int, parallelism)}
}

// Acquire 获取一个凭证
func (s *Semaphore) Acquire() {
s.Channel <- 1
}

// Release 归还一个凭证
func (s *Semaphore) Release() {
<-s.Channel
}
定义一个chan int类型，用来缓冲凭证个数，每获取一个凭证，就向channel里面加入一个int数据。

当一个goroutine运行完成之后，通过Release方法归还一个凭证，即channel输出一个数据，这样channel就不阻塞了，
此时就又可以启动一个新的goroutine了。具体的实现逻辑大概就是这样的了。

测试案例
下面通过一个小程序来测试一个整个流程。

main.go

package main

import (
“fmt”
“sync”
“time”

"jianshu.com/mozi/semaphore"

)

func main() {
data := make(chan int, 1000)
go func() {
defer close(data) //一定要close channel
for i := 1; i < 11000; i++ {
data <- i
time.Sleep(1)
}
}()
g := sync.WaitGroup{}
sem := semaphore.NewSemaphore(5)
flag := false //用来标志是否跳出外围死循环

i := 0 //在运行的goroutine个数

for {
    sem.Acquire()
    g.Add(1)
    i++
    fmt.Println(fmt.Sprintf("goroutine-%d", i))
    go func() {
        defer g.Done()
        defer sem.Release() //defer感觉可以理解为java中的finally
        var items []int
        for item := range data {
            items = append(items, item)
            if len(items) == 1000 {
                break
            }
        }
        if len(items) == 0 {
            flag = true
        }
        consumer(items)
        i--
    }()

    if flag {
        break
    }
}

g.Wait()
fmt.Println("exit program")

}

func consumer(data []int) {
for _, v := range data {
if v%1000 == 0 {
fmt.Println(v)
}
}
}
定义了1个缓冲区为1000的channel去存储数据，然后在每个goroutine里面的去消费数据，将能整除1000的数打印出来，每个goroutine每次处理1000个数据。

通过Smaphore共享锁控制程序中最多有5个goroutine在执行task。

通过len(items) == 0，判断 channel中的数据是否消费完了。

运行结果

semaphore-go.png
通过运行结果可以看出，程序中每次确实最多有5个goroutine在并行。 最后消费完所有数据后,退出程序。

也可以看到输出中确实没问题，1000到10000都正常输出。都是1000的倍数。

本文来自：简书

感谢作者：tommy_yang

查看原文：golang-基于channel实现的共享锁

欢迎使用Markdown编辑器

新的改变

1. 全新的界面设计 ，将会带来全新的写作体验；
2. 在创作中心设置你喜爱的代码高亮样式，Markdown 将代码片显示选择的高亮样式 进行展示；
3. 增加了 图片拖拽 功能，你可以将本地的图片直接拖拽到编辑区域直接展示；
4. 全新的 KaTeX数学公式 语法；
5. 增加了支持甘特图的mermaid语法1 功能；
6. 增加了 多屏幕编辑 Markdown文章功能；
7. 增加了 焦点写作模式、预览模式、简洁写作模式、左右区域同步滚轮设置 等功能，功能按钮位于编辑区域与预览区域中间；
8. 增加了 检查列表 功能。

功能快捷键

合理的创建标题，有助于目录的生成

如何改变文本的样式

插入链接与图片

链接: link.

图片:

带尺寸的图片:

居中的图片:

居中并且带尺寸的图片:

当然，我们为了让用户更加便捷，我们增加了图片拖拽功能。

如何插入一段漂亮的代码片

去博客设置页面，选择一款你喜欢的代码片高亮样式，下面展示同样高亮的 代码片.

// An highlighted block
var foo = 'bar';

生成一个适合你的列表

创建一个表格

设定内容居中、居左、居右

SmartyPants

创建一个自定义列表

如何创建一个注脚

一个具有注脚的文本。2

注释也是必不可少的

KaTeX数学公式

您可以使用渲染LaTeX数学表达式 KaTeX:

Gamma公式展示 $Gamma(n) = (n-1)!quadforall ninmathbb N$

$Gamma(z) = int_0^infty t^{z-1}e^{-t}dt,.$

你可以找到更多关于的信息 LaTeX 数学表达式here.

新的甘特图功能，丰富你的文章

• 关于 甘特图 语法，参考 这儿,

UML 图表

可以使用UML图表进行渲染。 Mermaid. 例如下面产生的一个序列图：:

这将产生一个流程图。:

• 关于 Mermaid 语法，参考 这儿,

FLowchart流程图

• 关于 Flowchart流程图 语法，参考 这儿.

导出与导入

导出

导入