JCS-pub/agent/internal/task/task.go

package task

import (
	"gitlink.org.cn/cloudream/common/pkgs/distlock"             // 引入分布式锁服务
	"gitlink.org.cn/cloudream/common/pkgs/task"                 // 引入任务处理相关的包
	"gitlink.org.cn/cloudream/storage/common/pkgs/connectivity" // 引入网络连接状态收集器
	"gitlink.org.cn/cloudream/storage/common/pkgs/ioswitch"     // 引入IO开关服务
)

// TaskContext 定义了任务执行的上下文环境，包含分布式锁服务、IO开关和网络连接状态收集器
type TaskContext struct {
	distlock     *distlock.Service
	sw           *ioswitch.Switch
	connectivity *connectivity.Collector
}

// CompleteFn 类型定义了任务完成时需要执行的函数，用于设置任务的执行结果
type CompleteFn = task.CompleteFn

// Manager 类型代表任务管理器，用于创建、管理和调度任务
type Manager = task.Manager[TaskContext]

// TaskBody 类型定义了任务体，包含了任务的具体执行逻辑
type TaskBody = task.TaskBody[TaskContext]

// Task 类型代表一个具体的任务，包含了任务的上下文、执行体和其它相关信息
type Task = task.Task[TaskContext]

// CompleteOption 类型定义了任务完成时的选项，可用于定制化任务完成的处理方式
type CompleteOption = task.CompleteOption

// NewManager 创建并返回一个新的任务管理器实例，需要提供分布式锁服务、IO开关和网络连接状态收集器
func NewManager(distlock *distlock.Service, sw *ioswitch.Switch, connectivity *connectivity.Collector) Manager {
	return task.NewManager(TaskContext{
		distlock:     distlock,
		sw:           sw,
		connectivity: connectivity,
	})
}