scheduler

type Constructor func(taskMeta []byte, step int64) (Scheduler, error)

type InternalScheduler interface {
	Start()
	Stop()
	Run(context.Context, *proto.Task) error
	Rollback(context.Context, *proto.Task) error
}

type InternalSchedulerImpl struct {
	// contains filtered or unexported fields
}

type Manager struct {
	// contains filtered or unexported fields
}

type ManagerBuilder struct {
	// contains filtered or unexported fields
}

type MockInternalScheduler struct {
	mock.Mock
}

type MockMinimalTask struct{}

type MockPool struct {
	mock.Mock
}

type MockScheduler struct {
	mock.Mock
}

type MockSubtaskExecutor struct {
	mock.Mock
}

type MockTaskTable struct {
	mock.Mock
}

type Pool interface {
	Run(func()) error
	RunWithConcurrency(chan func(), uint32) error
	ReleaseAndWait()
}

type RegisterOption func(opts *schedulerRegisterOptions)

type Scheduler interface {
	// InitSubtaskExecEnv is used to initialize the environment for the subtask executor.
	InitSubtaskExecEnv(context.Context) error
	// SplitSubtask is used to split the subtask into multiple minimal tasks.
	SplitSubtask(ctx context.Context, subtask []byte) ([]proto.MinimalTask, error)
	// CleanupSubtaskExecEnv is used to clean up the environment for the subtask executor.
	CleanupSubtaskExecEnv(context.Context) error
	// OnSubtaskFinished is used to handle the subtask when it is finished.
	OnSubtaskFinished(ctx context.Context, subtask []byte) error
	// Rollback is used to rollback all subtasks.
	Rollback(context.Context) error
}

type SubtaskExecutor interface {
	Run(ctx context.Context) error
}

type SubtaskExecutorConstructor func(minimalTask proto.MinimalTask, step int64) (SubtaskExecutor, error)

type SubtaskExecutorRegisterOption func(opts *subtaskExecutorRegisterOptions)

type TaskTable interface {
	GetGlobalTasksInStates(states ...interface{}) (task []*proto.Task, err error)
	GetGlobalTaskByID(taskID int64) (task *proto.Task, err error)
	GetSubtaskInStates(instanceID string, taskID int64, states ...interface{}) (*proto.Subtask, error)
	UpdateSubtaskStateAndError(id int64, state string, err string) error
	HasSubtasksInStates(instanceID string, taskID int64, states ...interface{}) (bool, error)
}