package task import ( model "github.com/HFO4/cloudreve/models" "github.com/HFO4/cloudreve/pkg/util" ) // TaskPoll 要使用的任务池 var TaskPoll *Pool // Pool 带有最大配额的任务池 type Pool struct { // 容量 idleWorker chan int } // Add 增加可用Worker数量 func (pool *Pool) Add(num int) { for i := 0; i < num; i++ { pool.idleWorker <- 1 } } // ObtainWorker 阻塞直到获取新的Worker func (pool *Pool) ObtainWorker() Worker { select { case <-pool.idleWorker: // 有空闲Worker名额时,返回新Worker return &GeneralWorker{} } } // FreeWorker 添加空闲Worker func (pool *Pool) FreeWorker() { pool.Add(1) } // Submit 开始提交任务 func (pool *Pool) Submit(job Job) { go func() { util.Log().Debug("等待获取Worker") worker := pool.ObtainWorker() util.Log().Debug("获取到Worker") worker.Do(job) util.Log().Debug("释放Worker") pool.FreeWorker() }() } // Init 初始化任务池 func Init() { maxWorker := model.GetIntSetting("max_worker_num", 10) TaskPoll = &Pool{ idleWorker: make(chan int, maxWorker), } TaskPoll.Add(maxWorker) util.Log().Info("初始化任务队列,WorkerNum = %d", maxWorker) Resume() }