golang控制协程数量

/**
 * @Author: zhangsan
 * @Description:
 * @File:  num
 * @Version: 1.0.0
 * @Date: 2020/12/8 上午9:42
 */

package main

import (
	"fmt"
	"runtime"

	"sync"
	"time"
)

type Golimit struct {
	n int
	c chan struct{}
}

// initialization Glimit struct
func New(n int) *Golimit {
	return &Golimit{
		n: n,
		c: make(chan struct{}, n),
	}
}

// Number of released goroutines
func (g *Golimit) Released(n int) {
	if n > g.n {
		g.c = make(chan struct{}, 1)

	} else {
		g.c = make(chan struct{}, g.n-n+1)
	}

}

// Run f in a new goroutine but with limit.
func (g *Golimit) Run(f func()) {
	g.c <- struct{}{}
	go func() {
		f()
		<-g.c
	}()
}

var wg = sync.WaitGroup{}

func main() {

	number := 10
	g := New(5)
	g.Released(6) //释放协程数量
	//os.Exit(1)
	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)
	}
	fmt.Println(runtime.NumGoroutine())
	wg.Wait()
}