mirror of
https://github.com/garethgeorge/backrest.git
synced 2025-12-12 08:45:38 +00:00
120 lines
3.1 KiB
Go
120 lines
3.1 KiB
Go
package orchestrator
|
|
|
|
import (
|
|
"time"
|
|
|
|
v1 "github.com/garethgeorge/backrest/gen/go/v1"
|
|
"github.com/garethgeorge/backrest/internal/hook"
|
|
"github.com/garethgeorge/backrest/internal/oplog"
|
|
"github.com/garethgeorge/backrest/internal/orchestrator/repo"
|
|
"github.com/garethgeorge/backrest/internal/orchestrator/tasks"
|
|
)
|
|
|
|
// taskRunnerImpl is an implementation of TaskRunner for the default orchestrator.
|
|
type taskRunnerImpl struct {
|
|
orchestrator *Orchestrator
|
|
t tasks.Task
|
|
op *v1.Operation
|
|
repo *v1.Repo // cache, populated on first call to Repo()
|
|
plan *v1.Plan // cache, populated on first call to Plan()
|
|
config *v1.Config // cache, populated on first call to Config()
|
|
}
|
|
|
|
var _ tasks.TaskRunner = &taskRunnerImpl{}
|
|
|
|
func (t *taskRunnerImpl) FindRepo() (*v1.Repo, error) {
|
|
if t.repo != nil {
|
|
return t.repo, nil
|
|
}
|
|
var err error
|
|
t.repo, err = t.orchestrator.GetRepo(t.t.RepoID())
|
|
return t.repo, err
|
|
}
|
|
|
|
func (t *taskRunnerImpl) FindPlan() (*v1.Plan, error) {
|
|
if t.plan != nil {
|
|
return t.plan, nil
|
|
}
|
|
var err error
|
|
t.plan, err = t.orchestrator.GetPlan(t.t.PlanID())
|
|
return t.plan, err
|
|
}
|
|
|
|
func newTaskRunnerImpl(orchestrator *Orchestrator, task tasks.Task, op *v1.Operation) *taskRunnerImpl {
|
|
return &taskRunnerImpl{
|
|
orchestrator: orchestrator,
|
|
t: task,
|
|
op: op,
|
|
}
|
|
}
|
|
|
|
func (t *taskRunnerImpl) CreateOperation(op *v1.Operation) error {
|
|
op.InstanceId = t.orchestrator.config.Instance
|
|
return t.orchestrator.OpLog.Add(op)
|
|
}
|
|
|
|
func (t *taskRunnerImpl) UpdateOperation(op *v1.Operation) error {
|
|
op.InstanceId = t.orchestrator.config.Instance
|
|
return t.orchestrator.OpLog.Update(op)
|
|
}
|
|
|
|
func (t *taskRunnerImpl) Orchestrator() *Orchestrator {
|
|
return t.orchestrator
|
|
}
|
|
|
|
func (t *taskRunnerImpl) OpLog() *oplog.OpLog {
|
|
return t.orchestrator.OpLog
|
|
}
|
|
|
|
func (t *taskRunnerImpl) ExecuteHooks(events []v1.Hook_Condition, vars hook.HookVars) error {
|
|
vars.Task = t.t.Name()
|
|
if t.op != nil {
|
|
vars.Duration = time.Since(time.UnixMilli(t.op.UnixTimeStartMs))
|
|
}
|
|
|
|
repoID := t.t.RepoID()
|
|
planID := t.t.PlanID()
|
|
var repo *v1.Repo
|
|
var plan *v1.Plan
|
|
if repoID != "" {
|
|
var err error
|
|
repo, err = t.FindRepo()
|
|
if err != nil {
|
|
return err
|
|
}
|
|
}
|
|
if planID != "" {
|
|
plan, _ = t.FindPlan()
|
|
}
|
|
var flowID int64
|
|
if t.op != nil {
|
|
flowID = t.op.FlowId
|
|
}
|
|
executor := hook.NewHookExecutor(t.Config(), t.orchestrator.OpLog, t.orchestrator.logStore)
|
|
return executor.ExecuteHooks(flowID, repo, plan, events, vars)
|
|
}
|
|
|
|
func (t *taskRunnerImpl) GetRepo(repoID string) (*v1.Repo, error) {
|
|
return t.orchestrator.GetRepo(repoID)
|
|
}
|
|
|
|
func (t *taskRunnerImpl) GetPlan(planID string) (*v1.Plan, error) {
|
|
return t.orchestrator.GetPlan(planID)
|
|
}
|
|
|
|
func (t *taskRunnerImpl) GetRepoOrchestrator(repoID string) (*repo.RepoOrchestrator, error) {
|
|
return t.orchestrator.GetRepoOrchestrator(repoID)
|
|
}
|
|
|
|
func (t *taskRunnerImpl) ScheduleTask(task tasks.Task, priority int) error {
|
|
return t.orchestrator.ScheduleTask(task, priority)
|
|
}
|
|
|
|
func (t *taskRunnerImpl) Config() *v1.Config {
|
|
if t.config != nil {
|
|
return t.config
|
|
}
|
|
t.config = t.orchestrator.Config()
|
|
return t.config
|
|
}
|