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

问题描述

需要实现一个并发的任务调度器,能够处理多个并发任务并限制同时执行的任务数量。

每个任务是一个函数,任务执行可能需要一段时间。

调度器的目的是控制同时执行的任务数量(即并发度),以防止任务数过多导致系统资源过载。

代码

go 复制代码
package test

import (
	"fmt"
	"math/rand"
	"sync"
	"testing"
	"time"
	"trpc.group/trpc-go/trpc-go/log"
)

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

type TaskDispatcher struct {
	maxTask int
	c       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 true
	case <-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()
}
相关推荐
雷达学弱狗1 小时前
链式法则解释上游梯度应用
开发语言·前端·javascript
小清兔5 小时前
c#基础知识
开发语言·数据库·学习·unity·c#·游戏引擎·.net
快乐就是哈哈哈5 小时前
《一文带你搞懂ElasticSearch:从零到上手搜索引擎》
后端·elasticsearch
大鸡腿同学5 小时前
身弱:修炼之路
后端
bobz9655 小时前
cpu 调度 和 gpu 调度
后端
AirMan5 小时前
深入揭秘 ConcurrentHashMap:JDK7 到 JDK8 并发优化的演进之路
后端·面试
bobz9655 小时前
Linux CPU 调度模型
后端
计算机学姐5 小时前
基于SpringBoot的社团管理系统【2026最新】
java·vue.js·spring boot·后端·mysql·spring·mybatis
奇某人6 小时前
【语法】【C+V】本身常用图表类型用法快查【CSDN不支持,VSCODE可用】
开发语言·vscode·markdown·mermaid
做一位快乐的码农6 小时前
php程序设计之基于PHP的手工艺品销售网站/基于php在线销售系统/基于php在线购物商城系统
开发语言·php