40 lines
1.6 KiB
Go
40 lines
1.6 KiB
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,
|
||
})
|
||
}
|