78 lines
1.9 KiB
Go
78 lines
1.9 KiB
Go
|
// Package routineCtrlUtil
|
|||
|
|
//
|
|||
|
|
// @description: 协程控制-用于海量循环中，限制并发的最大同时执行协程数
|
|||
|
|
// @author:
|
|||
|
|
// @revision history:
|
|||
|
|
// @create date: 2022-02-23 17:23:06
|
|||
|
|
package routineCtrlUtil
|
|||
|
|
|
|||
|
|
import "sync"
|
|||
|
|
|
|||
|
|
// RoutineCtrl
|
|||
|
|
//
|
|||
|
|
// @description: 控制同时运行的协程数
|
|||
|
|
type RoutineCtrl struct {
|
|||
|
|
wg *sync.WaitGroup
|
|||
|
|
chNum chan struct{}
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
// Run
|
|||
|
|
//
|
|||
|
|
// @description: 循环内待执行的协程，封装在f函数参数内执行。f函数的最大并发运行数受New参数maxNum限制
|
|||
|
|
// 注意：*** 待执行函数引用外变变量，程序可能会出现不符合逻辑的结果(外部变量变化时，闭包函数获取到的是变化后的变量值) ***
|
|||
|
|
// *** 闭包函数避坑说明 ***
|
|||
|
|
// 方式1)将待执行函数内使用的外部变量，用参数传入，可避免不符合逻辑的结果
|
|||
|
|
// 方式2)将待执行函数内使用的外部变量，使用一个临时局部变量将其值“固定”下来，闭包内使用临时局部变量（临时局部变量不会再被修改）
|
|||
|
|
// 示例：routine_ctrl_test.go -> TestRoutineCtrl
|
|||
|
|
//
|
|||
|
|
// parameter:
|
|||
|
|
//
|
|||
|
|
// @receiver rtCtrl:
|
|||
|
|
// @f: 循环内待执行的协程
|
|||
|
|
//
|
|||
|
|
// return:
|
|||
|
|
func (rtCtrl *RoutineCtrl) Run(f func(interface{}), arg interface{}) {
|
|||
|
|
rtCtrl.wg.Add(1)
|
|||
|
|
rtCtrl.chNum <- struct{}{}
|
|||
|
|
go func() {
|
|||
|
|
defer func() {
|
|||
|
|
<-rtCtrl.chNum
|
|||
|
|
rtCtrl.wg.Done()
|
|||
|
|
recover() // 异常捕获
|
|||
|
|
}()
|
|||
|
|
|
|||
|
|
f(arg) // 调用实际函数体
|
|||
|
|
}()
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
// Wait
|
|||
|
|
//
|
|||
|
|
// @description: 在循环外面等待所有协程执行结束后返回
|
|||
|
|
//
|
|||
|
|
// parameter:
|
|||
|
|
//
|
|||
|
|
// @receiver rtCtrl:
|
|||
|
|
//
|
|||
|
|
// return:
|
|||
|
|
func (rtCtrl *RoutineCtrl) Wait() {
|
|||
|
|
rtCtrl.wg.Wait()
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
// New
|
|||
|
|
//
|
|||
|
|
// @description: 产生一个协程控制对象
|
|||
|
|
//
|
|||
|
|
// parameter:
|
|||
|
|
//
|
|||
|
|
// @maxNum: 限制调用Run时的f函数的最大运行协程数
|
|||
|
|
//
|
|||
|
|
// return:
|
|||
|
|
//
|
|||
|
|
// @*RoutineCtrl:
|
|||
|
|
func New(maxNum int) *RoutineCtrl {
|
|||
|
|
return &RoutineCtrl{
|
|||
|
|
wg: &sync.WaitGroup{},
|
|||
|
|
chNum: make(chan struct{}, maxNum),
|
|||
|
|
}
|
|||
|
|
}
|