60 lines
1.2 KiB
Go
60 lines
1.2 KiB
Go
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()
|
||
}
|