Support bot site

This commit is contained in:
Lunny Xiao
2022-11-25 17:48:47 +08:00
committed by Jason Song
parent 7059cd7265
commit 9a45572ce2
6 changed files with 24 additions and 20 deletions
+9 -5
View File
@@ -136,17 +136,21 @@ func updateRepoRunsNumbers(ctx context.Context, repo *repo_model.Repository) err
// InsertRun inserts a bot run
func InsertRun(run *Run, jobs []*jobparser.SingleWorkflow) error {
index, err := db.GetNextResourceIndex("bots_run_index", run.RepoID)
ctx, commiter, err := db.TxContext(db.DefaultContext)
if err != nil {
return err
}
defer commiter.Close()
index, err := db.GetNextResourceIndex(ctx, "bots_run_index", run.RepoID)
if err != nil {
return err
}
run.Index = index
ctx, commiter, err := db.TxContext()
if err != nil {
return err
if run.Status == StatusUnknown {
run.Status = StatusWaiting
}
defer commiter.Close()
if err := db.Insert(ctx, run); err != nil {
return err
+3 -3
View File
@@ -291,7 +291,7 @@ func GetTaskByToken(ctx context.Context, token string) (*Task, error) {
}
func CreateTaskForRunner(ctx context.Context, runner *Runner) (*Task, bool, error) {
dbCtx, commiter, err := db.TxContext()
dbCtx, commiter, err := db.TxContext(ctx)
if err != nil {
return nil, false, err
}
@@ -422,7 +422,7 @@ func UpdateTaskByState(state *runnerv1.TaskState) (*Task, error) {
stepStates[v.Id] = v
}
ctx, commiter, err := db.TxContext()
ctx, commiter, err := db.TxContext(db.DefaultContext)
if err != nil {
return nil, err
}
@@ -485,7 +485,7 @@ func UpdateTaskByState(state *runnerv1.TaskState) (*Task, error) {
}
func StopTask(ctx context.Context, task *Task, result runnerv1.Result) (*Task, error) {
ctx, commiter, err := db.TxContext()
ctx, commiter, err := db.TxContext(ctx)
if err != nil {
return nil, err
}