refactor(task): execute single task with gRPC data

This commit is contained in:
Bo-Yi.Wu 2022-09-04 15:44:29 +08:00 committed by Jason Song
parent 5051e4aebd
commit 20c3d85ba9
3 changed files with 18 additions and 22 deletions

View File

@ -76,7 +76,7 @@ func runRoot(ctx context.Context, task *runtime.Task) func(cmd *cobra.Command, a
}
task.BuildID, _ = strconv.ParseInt(jobID, 10, 64)
task.Run(ctx)
task.Run(ctx, nil)
return nil
}
}

View File

@ -3,7 +3,6 @@ package runtime
import (
"context"
"errors"
"fmt"
"gitea.com/gitea/act_runner/client"
runnerv1 "gitea.com/gitea/proto-go/runner/v1"
@ -59,18 +58,5 @@ func (s *Runner) Run(ctx context.Context, stage *runnerv1.Stage) error {
}
func (s *Runner) run(ctx context.Context, data *runnerv1.DetailResponse) error {
_, exist := globalTaskMap.Load(data.Build.Id)
if exist {
return fmt.Errorf("task %d already exists", data.Build.Id)
}
task := NewTask(data.Build.Id, s.Client)
// set task ve to global map
// when task is done or canceled, it will be removed from the map
globalTaskMap.Store(data.Build.Id, task)
go task.Run(ctx)
return nil
return NewTask(data.Build.Id, s.Client).Run(ctx, data)
}

View File

@ -216,18 +216,27 @@ func (t *Task) reportSuccess(ctx context.Context) {
_ = t.client.UpdateStep(ctx, stepRequest)
}
func (t *Task) Run(ctx context.Context) {
func (t *Task) Run(ctx context.Context, data *runnerv1.DetailResponse) error {
_, exist := globalTaskMap.Load(data.Build.Id)
if exist {
return fmt.Errorf("task %d already exists", data.Build.Id)
}
// set task ve to global map
// when task is done or canceled, it will be removed from the map
globalTaskMap.Store(data.Build.Id, t)
workflowsPath, err := getWorkflowsPath(t.Input.repoDirectory)
if err != nil {
t.reportFailure(ctx, err)
return
return err
}
t.log.Debugf("workflows path: %s", workflowsPath)
planner, err := model.NewWorkflowPlanner(workflowsPath, false)
if err != nil {
t.reportFailure(ctx, err)
return
return err
}
var eventName string
@ -260,7 +269,7 @@ func (t *Task) Run(ctx context.Context) {
curDir, err := os.Getwd()
if err != nil {
t.reportFailure(ctx, err)
return
return err
}
// run the plan
@ -298,7 +307,7 @@ func (t *Task) Run(ctx context.Context) {
r, err := runner.New(config)
if err != nil {
t.reportFailure(ctx, err)
return
return err
}
cancel := artifacts.Serve(ctx, input.artifactServerPath, input.artifactServerPort)
@ -318,8 +327,9 @@ func (t *Task) Run(ctx context.Context) {
if err := executor(ctx); err != nil {
t.reportFailure(ctx, err)
return
return err
}
t.reportSuccess(ctx)
return nil
}