分享每一天-golang如何控制goroutine的并发数量

golang如何控制goroutine的并发数量

GoLang 937 0 创建时间：2022-12-09 10:46:22

golang goroutine

#### golang如何控制goroutine的并发数量

通过channel和sync方式限制协程数量。
```go
package main

import (
	"fmt"
	"runtime"
	"sync"
	"time"
)

func main() {
	var wg sync.WaitGroup
	
	ch := make(chan struct{}, 5) //控制并发数量
	for i := 0; i < len(s); i++ {
		ch <- struct{}{}
		wg.Add(1)
		go func(i int) {
			//TODO
 
			defer wg.Done() //建议用defer保证调用
			time.Sleep(time.Second)
			<-ch
		}(i)

}
	wg.Wait()

}
```

#### Channel

Goroutine运行在相同的地址空间，因此访问共享内存必须做好同步。那么Goroutine之间如何进行数据的通信呢？Go提供了一个很好的通信机制channel，channel可以与 Unix shell 中的双向管道做类比：可以通过它发送或者接收值。这些值只能是特定的类型：channel类型。定义一个channel时，也需要定义发送到channel的值的类型。注意，必须使用make创建channel。

####  Sync

Go语言中有一个sync.WaitGroup，WaitGroup 对象内部有一个计数器，最初从0开始，它有三个方法：Add(), Done(), Wait() 用来控制计数器的数量。下面示例代码中wg.Wati会阻塞代码的运行，直到计数器值为0。

通过Golang自带的channel和sync，可以实现需求，下面代码中通过channel控制Goroutine数量。

```go
package main
import (
  "fmt"
  "sync"
  "time"
)
type Glimit struct {
  n int
  c chan struct{}
}
// initialization Glimit struct
func New(n int) *Glimit {
  return &Glimit{
    n: n,
    c: make(chan struct{}, n),
  }
}
// Run f in a new goroutine but with limit.
func (g *Glimit) Run(f func()) {
  g.c <- struct{}{}
  go func() {
    f()
    <-g.c
  }()
}
var wg = sync.WaitGroup{}
func main() {
  number := 10
  g := New(2)
  for i := 0; i < number; i++ {
    wg.Add(1)
    value :=i
    goFunc := func() {
      // 做一些业务逻辑处理
      fmt.Printf("go func: %d\n", value)
      time.Sleep(time.Second)
      wg.Done()
    }
    g.Run(goFunc)
  }
  wg.Wait()
}
```

#### 思考1：为什么我们要使用sync.WaitGroup？

这里如果我们不使用sync.WaitGroup控制的话，原因出在当主程序结束时，子协程也是会被终止掉的。因此剩余的 goroutine 没来及把值输出，程序就已经中断了

#### 思考2：代码中channel数据结构为什么定义struct，而不定义成bool这种类型呢？

因为空结构体变量的内存占用大小为0，而bool类型内存占用大小为1，这样可以更加最大化利用我们服务器的内存空间。

golang如何控制goroutine的并发数量

商品推荐

广告推荐

推荐文章