refactor(计划): 增加非固定参数

goexl · Dec 5, 2024 · b85ffbd · b85ffbd
1 parent d56fc4b
commit b85ffbd
Show file tree

Hide file tree

Showing 2 changed files with 5 additions and 12 deletions.
diff --git a/internal/internal/core/processor.go b/internal/internal/core/processor.go
@@ -29,17 +29,10 @@ func NewProcessor(tasker kernel.Tasker, params *param.Agent) *Processor {
 
 func (p *Processor) Process(selector kernel.Selector) {
 	for {
-		tasks := p.tasker.Pop(p.params.Retries)
-		if 0 == len(tasks) { // 让出时间切片
-			time.Sleep(0)
-			continue
-		}
-
-		for _, _task := range tasks {
-			if _, exists := p.progresses.Load(_task.Id()); exists {
-				continue
-			}
-
+		_task := p.tasker.Pop(p.params.Retries)
+		if _, exists := p.progresses.Load(_task.Id()); exists {
+			time.Sleep(0) // 让出时间片
+		} else {
 			go func() {
 				_ = p.process(_task, selector) // 错误已经处理，纯接收
 			}()

diff --git a/internal/kernel/tasker.go b/internal/kernel/tasker.go
@@ -22,7 +22,7 @@ type Tasker interface {
 	Update(id uint64, status Status, runtime time.Time) error
 
 	// Pop 取出任务并执行
-	Pop(retries uint32) []Task
+	Pop(retries uint32) Task
 
 	// Archive 存档
 	Archive(task Task) error