Go工作流中的任务状态追踪

在Go语言的工作流（workflow）中，任务状态追踪是一个重要的功能，它可以帮助我们了解任务的执行进度、状态变化以及可能存在的问题。为了实现这个功能，我们可以使用Go的并发特性（如goroutines和channels）来处理任务的状态变化，并使用合适的数据结构（如map或slice）来存储和管理这些状态信息。

以下是一个简单的示例，展示了如何在Go中实现任务状态追踪：

1. 首先，定义一个表示任务状态的枚举类型：

type TaskStatus int

const (
    StatusPending TaskStatus = iota
    StatusRunning
    StatusCompleted
    StatusFailed
)

2. 接下来，创建一个结构体来表示任务，包括任务ID、状态以及其他相关信息：

type Task struct {
    ID            int
    Status        TaskStatus
    StartTime     time.Time
    EndTime       time.Time
    Progress      float64 // 0.0到1.0之间的值，表示任务的完成度
    Error         error
}

3. 创建一个函数来更新任务的状态：

func UpdateTaskStatus(task *Task, newStatus TaskStatus) {
    task.Status = newStatus
    switch newStatus {
    case StatusCompleted:
        task.EndTime = time.Now()
        task.Progress = 1.0
    case StatusFailed:
        task.EndTime = time.Now()
    }
}

4. 使用goroutines和channels来处理并发任务状态更新：

func main() {
    // 创建一个任务列表
    tasks := []Task{
        {ID: 1, Status: StatusPending},
        {ID: 2, Status: StatusRunning},
        {ID: 3, Status: StatusPending},
    }

    // 创建一个channel来接收任务状态更新的信号
    statusUpdates := make(chan TaskStatusUpdate)

    // 启动一个goroutine来处理任务状态更新
    go func() {
        for update := range statusUpdates {
            task := findTaskByID(tasks, update.TaskID)
            if task != nil {
                UpdateTaskStatus(task, update.NewStatus)
            }
        }
    }()

    // 模拟任务状态更新
    go func() {
        time.Sleep(time.Second)
        statusUpdates <- TaskStatusUpdate{TaskID: 1, NewStatus: StatusRunning}
        time.Sleep(time.Second)
        statusUpdates <- TaskStatusUpdate{TaskID: 1, NewStatus: StatusCompleted}
        time.Sleep(time.Second)
        statusUpdates <- TaskStatusUpdate{TaskID: 2, NewStatus: StatusFailed}
        close(statusUpdates)
    }()

    // 等待所有任务状态更新完成
    time.Sleep(5 * time.Second)

    // 打印任务状态
    for _, task := range tasks {
        fmt.Printf("Task %d: Status: %s, Start: %v, End: %v, Progress: %.2f\n",
            task.ID, task.Status, task.StartTime, task.EndTime, task.Progress)
    }
}

// 辅助函数：根据任务ID查找任务
func findTaskByID(tasks []Task, taskID int) *Task {
    for _, task := range tasks {
        if task.ID == taskID {
            return &task
        }
    }
    return nil
}

// 任务状态更新结构体
type TaskStatusUpdate struct {
    TaskID       int
    NewStatus    TaskStatus
}

在这个示例中，我们创建了一个任务列表，并使用一个goroutine来处理并发任务状态更新。我们还模拟了一些任务状态更新，并在主函数中打印了任务的最终状态。这只是一个简单的示例，实际的工作流可能需要更复杂的逻辑和功能。