You cannot select more than 25 topics
Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
32 lines
652 B
Go
32 lines
652 B
Go
package threading
|
|
|
|
import (
|
|
"github.com/zeromicro/go-zero/core/lang"
|
|
"github.com/zeromicro/go-zero/core/rescue"
|
|
)
|
|
|
|
// A TaskRunner is used to control the concurrency of goroutines.
|
|
type TaskRunner struct {
|
|
limitChan chan lang.PlaceholderType
|
|
}
|
|
|
|
// NewTaskRunner returns a TaskRunner.
|
|
func NewTaskRunner(concurrency int) *TaskRunner {
|
|
return &TaskRunner{
|
|
limitChan: make(chan lang.PlaceholderType, concurrency),
|
|
}
|
|
}
|
|
|
|
// Schedule schedules a task to run under concurrency control.
|
|
func (rp *TaskRunner) Schedule(task func()) {
|
|
rp.limitChan <- lang.Placeholder
|
|
|
|
go func() {
|
|
defer rescue.Recover(func() {
|
|
<-rp.limitChan
|
|
})
|
|
|
|
task()
|
|
}()
|
|
}
|