移动文件位置

internal/app/service/task/delay_task.go

@@ -0,0 +1,53 @@
+package task
+
+import (
+	"fmt"
+	"time"
+)
+
+// DelayTask 是一个用于模拟延迟的 Task 实现
+type DelayTask struct {
+	id       string
+	duration time.Duration
+	priority int
+	done     bool
+}
+
+// NewDelayTask 创建一个新的 DelayTask 实例
+func NewDelayTask(id string, duration time.Duration, priority int) *DelayTask {
+	return &DelayTask{
+		id:       id,
+		duration: duration,
+		priority: priority,
+		done:     false,
+	}
+}
+
+// Execute 执行延迟任务，等待指定的时间
+func (d *DelayTask) Execute() error {
+	fmt.Printf("任务 %s (%s): 开始延迟 %s...\n", d.id, d.GetDescription(), d.duration)
+	time.Sleep(d.duration)
+	fmt.Printf("任务 %s (%s): 延迟结束。\n", d.id, d.GetDescription())
+	d.done = true
+	return nil
+}
+
+// GetID 获取任务ID
+func (d *DelayTask) GetID() string {
+	return d.id
+}
+
+// GetPriority 获取任务优先级
+func (d *DelayTask) GetPriority() int {
+	return d.priority
+}
+
+// IsDone 检查任务是否已完成
+func (d *DelayTask) IsDone() bool {
+	return d.done
+}
+
+// GetDescription 获取任务说明，根据任务ID和延迟时间生成
+func (d *DelayTask) GetDescription() string {
+	return fmt.Sprintf("延迟任务，ID: %s，延迟时间: %s", d.id, d.duration)
+}

internal/app/service/task/delay_task_test.go

@@ -0,0 +1,61 @@
+package task_test
+
+import (
+	"fmt"
+	"testing"
+	"time"
+
+	"git.huangwc.com/pig/pig-farm-controller/internal/infra/task"
+)
+
+func TestNewDelayTask(t *testing.T) {
+	id := "test-delay-task-1"
+	duration := 100 * time.Millisecond
+	priority := 1
+
+	dt := task.NewDelayTask(id, duration, priority)
+
+	if dt.GetID() != id {
+		t.Errorf("期望任务ID为 %s, 实际为 %s", id, dt.GetID())
+	}
+	if dt.GetPriority() != priority {
+		t.Errorf("期望任务优先级为 %d, 实际为 %d", priority, dt.GetPriority())
+	}
+	if dt.IsDone() != false {
+		t.Error("任务初始状态不应为已完成")
+	}
+	// 动态生成的描述，需要匹配 GetDescription 的实现
+	expectedDesc := fmt.Sprintf("延迟任务，ID: %s，延迟时间: %s", id, duration)
+	if dt.GetDescription() != expectedDesc {
+		t.Errorf("期望任务描述为 %s, 实际为 %s", expectedDesc, dt.GetDescription())
+	}
+}
+
+func TestDelayTaskExecute(t *testing.T) {
+	id := "test-delay-task-execute"
+	duration := 50 * time.Millisecond // 使用较短的延迟以加快测试速度
+	priority := 1
+
+	dt := task.NewDelayTask(id, duration, priority)
+
+	if dt.IsDone() {
+		t.Error("任务执行前不应为已完成状态")
+	}
+
+	startTime := time.Now()
+	err := dt.Execute()
+	endTime := time.Now()
+
+	if err != nil {
+		t.Errorf("Execute 方法返回错误: %v", err)
+	}
+	if !dt.IsDone() {
+		t.Error("任务执行后应为已完成状态")
+	}
+
+	// 验证延迟时间大致正确，允许一些误差
+	elapsed := endTime.Sub(startTime)
+	if elapsed < duration || elapsed > duration*2 {
+		t.Errorf("期望执行时间在 %v 左右, 但实际耗时 %v", duration, elapsed)
+	}
+}