当前位置: 首页 > article >正文

【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()
}


http://www.kler.cn/a/448034.html

相关文章:

  • 【机器学习与数据挖掘实战】案例04:基于K-Means算法的信用卡高风险客户识别
  • How to run Flutter on an Embedded Device
  • 算法设计期末复习
  • Vue Web开发(十)
  • 【C#】实现Json转Lua (Json2Lua)
  • Godot RPG 游戏开发指南
  • 科技查新对专利申请有什么帮助?
  • 考前倒计时98天
  • 【华为OD-E卷-通信误码 100分(python、java、c++、js、c)】
  • Vue3之性能优化
  • 一起学Git【第二节:创建版本库】
  • Solaris-IPMP配置
  • 概率论得学习和整理30: 用EXCEL 描述泊松分布 poisson distribution
  • 高效准确的PDF解析工具,赋能企业非结构化数据治理
  • 【计算机网络篇】计算机网络期末复习题库详解
  • C语言习题2.0
  • PyCharm 中打印完整的 DataFrame
  • ubuntu 上怎么设置应用开机自动启动
  • 复习打卡大数据篇——Hadoop HDFS 01
  • 全志H618 Android12修改doucmentsui进入后左边动态加载的item
  • BUUCTF Pwn picoctf_2018_rop chain WP
  • Firecrawl教程①:自动化抓取与数据转化,赋能AI应用
  • idea中打补丁包
  • 音视频入门基础:AAC专题(13)——FFmpeg源码中,获取ADTS格式的AAC裸流音频信息的实现
  • Linux之用户相关命令
  • 深度学习:解锁智能时代的钥匙与实战案例