Go语言中如何控制并发

下面这个例子并发打印slice中的每个值（可以查看上一篇介绍的for…range了解其他细节）

var wg sync.WaitGroup
var s []int = []int{1, 2, 3, 4, 5, 6}

for _, value := range s {
	wg.Add(1)
	go func(value int) {
		defer func() {
			wg.Done()
		}()
		t.Logf("go routine value %d", value)
	}(value)

}
wg.Wait()

但有时因为数据库IO或者其他原因，不可能无限制的并发。这里给并发设置一个上限。

var wg sync.WaitGroup
var s []int = []int{1, 2, 3, 4, 5, 6}

var pipe = make(chan struct{}, 2)
for _, value := range s {
	wg.Add(1)
	go func(value int) {
		defer func() {
			wg.Done()
			<-pipe
		}()
		pipe <- struct{}{}
		t.Logf("go routine value %d", value)
	}(value)

}
wg.Wait()

如上述代码，我们通过一个带缓冲的channel来控制go的并发，首先我们声明了一个pipe的channel，来控制程序最多可以并发两个goroutine

var pipe = make(chan struct{}, 2)

之后再每个goroutine函数中做如下处理，充分利用channel阻塞的特性

1. 在函数开始时，尝试写入该channel中一个值。如果写入成功，表示当前可以执行该并发，否则阻塞等待。
2. 当函数结束时从该channel中读取一个值。

defer func() {
	<-pipe
}()
pipe <- struct{}{}

如何查看当前程序是否真正只有2个并发，现在确实我还没有了解到可以查看的方法。不过，我们可以使用panic语句来简单查看一下。

panic执行的时候程序默认会打印出当前程序的所有调用栈信息。我们做一下调整，主动让程序panic

var wg sync.WaitGroup
var s []int = []int{1, 2, 3, 4, 5, 6}

var pipe = make(chan struct{}, 2)
for _, value := range s {
	wg.Add(1)
	go func(value int) {
		defer func() {
			wg.Done()
			<-pipe
			panic("for check")
		}()
		pipe <- struct{}{}
		t.Logf("go routine value %d", value)
	}(value)

}
wg.Wait()

输出的结果如下：

panic: for check

goroutine 15 [running]:
speed-push/tests.TestForRange.func1.1(0xc0420f2e60, 0xc0420fe660)
	E:/go-source/kidsguard/src/speed-push/tests/blog_test.go:19 +0xa0
speed-push/tests.TestForRange.func1(0xc0420f2e60, 0xc0420fe660, 0xc042041450, 0x6)
	E:/go-source/kidsguard/src/speed-push/tests/blog_test.go:23 +0x10c
created by speed-push/tests.TestForRange
	E:/go-source/kidsguard/src/speed-push/tests/blog_test.go:23 +0x118
panic: for check

goroutine 10 [running]:
speed-push/tests.TestForRange.func1.1(0xc0420f2e60, 0xc0420fe660)
	E:/go-source/kidsguard/src/speed-push/tests/blog_test.go:19 +0xa0
speed-push/tests.TestForRange.func1(0xc0420f2e60, 0xc0420fe660, 0xc042041450, 0x1)
	E:/go-source/kidsguard/src/speed-push/tests/blog_test.go:23 +0x10c
created by speed-push/tests.TestForRange
	E:/go-source/kidsguard/src/speed-push/tests/blog_test.go:23 +0x118
exit status 2