【go每日一题】：并发任务调度器

问题描述

需要实现一个并发的任务调度器，能够处理多个并发任务并限制同时执行的任务数量。
每个任务是一个函数，任务执行可能需要一段时间。
调度器的目的是控制同时执行的任务数量（即并发度），以防止任务数过多导致系统资源过载。

代码

package testimport ("fmt""math/rand""sync""testing""time""trpc.group/trpc-go/trpc-go/log"
)/**
你需要实现一个并发的任务调度器，能够处理多个并发任务并限制同时执行的任务数量。
每个任务是一个函数，任务执行可能需要一段时间。
调度器的目的是控制同时执行的任务数量（即并发度），以防止任务数过多导致系统资源过载。
*/type TaskDispatcher struct {maxTask intc       chan func()wg      *sync.WaitGroup // 指针不进行初始化是nil
}func NewTaskDispatcher(maxTaskNum int, chanCap int) *TaskDispatcher {return &TaskDispatcher{maxTask: maxTaskNum,c:       make(chan func(), chanCap),wg:      new(sync.WaitGroup),}
}func (td *TaskDispatcher) AddNewTask(f func()) bool {select {case td.c <- f:td.wg.Add(1) // 这里加，消费完毕减少return truecase <-time.After(3 * time.Second):return false}
}func (td *TaskDispatcher) RunDispatcher() {for i := 0; i < td.maxTask; i++ {go func(i int) {for f := range td.c {log.Info(fmt.Sprintf("当前第%d个协程正在执行任务", i))if f != nil {f()}time.Sleep(time.Second * time.Duration(rand.Intn(3)+1))fmt.Println("当前函数执行完毕")td.wg.Done()}}(i)}
}func (td *TaskDispatcher) StopDispatcher() {defer close(td.c)td.wg.Wait()
}func TestTaskDispatcher(t *testing.T) {td := NewTaskDispatcher(3, 10)td.RunDispatcher() //channel先有接收方，才向channel中发数据for i := 0; i < 20; i++ {td.AddNewTask(func() {fmt.Printf("这里是任务%v\n", i)})}td.StopDispatcher()
}