mirror of
https://codeberg.org/forgejo/forgejo.git
synced 2024-09-01 23:34:17 +00:00
164 lines
3.6 KiB
Go
164 lines
3.6 KiB
Go
|
// Copyright 2022 The Gitea Authors. All rights reserved.
|
||
|
// SPDX-License-Identifier: MIT
|
||
|
|
||
|
package actions
|
||
|
|
||
|
import (
|
||
|
"context"
|
||
|
"fmt"
|
||
|
"time"
|
||
|
|
||
|
"code.gitea.io/gitea/models/db"
|
||
|
"code.gitea.io/gitea/modules/timeutil"
|
||
|
"code.gitea.io/gitea/modules/util"
|
||
|
|
||
|
"xorm.io/builder"
|
||
|
)
|
||
|
|
||
|
// ActionRunJob represents a job of a run
|
||
|
type ActionRunJob struct {
|
||
|
ID int64
|
||
|
RunID int64 `xorm:"index"`
|
||
|
Run *ActionRun `xorm:"-"`
|
||
|
RepoID int64 `xorm:"index"`
|
||
|
OwnerID int64 `xorm:"index"`
|
||
|
CommitSHA string `xorm:"index"`
|
||
|
IsForkPullRequest bool
|
||
|
Name string `xorm:"VARCHAR(255)"`
|
||
|
Attempt int64
|
||
|
WorkflowPayload []byte
|
||
|
JobID string `xorm:"VARCHAR(255)"` // job id in workflow, not job's id
|
||
|
Needs []string `xorm:"JSON TEXT"`
|
||
|
RunsOn []string `xorm:"JSON TEXT"`
|
||
|
TaskID int64 // the latest task of the job
|
||
|
Status Status `xorm:"index"`
|
||
|
Started timeutil.TimeStamp
|
||
|
Stopped timeutil.TimeStamp
|
||
|
Created timeutil.TimeStamp `xorm:"created"`
|
||
|
Updated timeutil.TimeStamp `xorm:"updated index"`
|
||
|
}
|
||
|
|
||
|
func init() {
|
||
|
db.RegisterModel(new(ActionRunJob))
|
||
|
}
|
||
|
|
||
|
func (job *ActionRunJob) Duration() time.Duration {
|
||
|
return calculateDuration(job.Started, job.Stopped, job.Status)
|
||
|
}
|
||
|
|
||
|
func (job *ActionRunJob) LoadRun(ctx context.Context) error {
|
||
|
if job.Run == nil {
|
||
|
run, err := GetRunByID(ctx, job.RunID)
|
||
|
if err != nil {
|
||
|
return err
|
||
|
}
|
||
|
job.Run = run
|
||
|
}
|
||
|
return nil
|
||
|
}
|
||
|
|
||
|
// LoadAttributes load Run if not loaded
|
||
|
func (job *ActionRunJob) LoadAttributes(ctx context.Context) error {
|
||
|
if job == nil {
|
||
|
return nil
|
||
|
}
|
||
|
|
||
|
if err := job.LoadRun(ctx); err != nil {
|
||
|
return err
|
||
|
}
|
||
|
|
||
|
return job.Run.LoadAttributes(ctx)
|
||
|
}
|
||
|
|
||
|
func GetRunJobByID(ctx context.Context, id int64) (*ActionRunJob, error) {
|
||
|
var job ActionRunJob
|
||
|
has, err := db.GetEngine(ctx).Where("id=?", id).Get(&job)
|
||
|
if err != nil {
|
||
|
return nil, err
|
||
|
} else if !has {
|
||
|
return nil, fmt.Errorf("run job with id %d: %w", id, util.ErrNotExist)
|
||
|
}
|
||
|
|
||
|
return &job, nil
|
||
|
}
|
||
|
|
||
|
func GetRunJobsByRunID(ctx context.Context, runID int64) ([]*ActionRunJob, error) {
|
||
|
var jobs []*ActionRunJob
|
||
|
if err := db.GetEngine(ctx).Where("run_id=?", runID).OrderBy("id").Find(&jobs); err != nil {
|
||
|
return nil, err
|
||
|
}
|
||
|
return jobs, nil
|
||
|
}
|
||
|
|
||
|
func UpdateRunJob(ctx context.Context, job *ActionRunJob, cond builder.Cond, cols ...string) (int64, error) {
|
||
|
e := db.GetEngine(ctx)
|
||
|
|
||
|
sess := e.ID(job.ID)
|
||
|
if len(cols) > 0 {
|
||
|
sess.Cols(cols...)
|
||
|
}
|
||
|
|
||
|
if cond != nil {
|
||
|
sess.Where(cond)
|
||
|
}
|
||
|
|
||
|
affected, err := sess.Update(job)
|
||
|
if err != nil {
|
||
|
return 0, err
|
||
|
}
|
||
|
|
||
|
if affected == 0 || (!util.SliceContains(cols, "status") && job.Status == 0) {
|
||
|
return affected, nil
|
||
|
}
|
||
|
|
||
|
if job.RunID == 0 {
|
||
|
var err error
|
||
|
if job, err = GetRunJobByID(ctx, job.ID); err != nil {
|
||
|
return affected, err
|
||
|
}
|
||
|
}
|
||
|
|
||
|
jobs, err := GetRunJobsByRunID(ctx, job.RunID)
|
||
|
if err != nil {
|
||
|
return affected, err
|
||
|
}
|
||
|
|
||
|
runStatus := aggregateJobStatus(jobs)
|
||
|
|
||
|
run := &ActionRun{
|
||
|
ID: job.RunID,
|
||
|
Status: runStatus,
|
||
|
}
|
||
|
if runStatus.IsDone() {
|
||
|
run.Stopped = timeutil.TimeStampNow()
|
||
|
}
|
||
|
return affected, UpdateRun(ctx, run)
|
||
|
}
|
||
|
|
||
|
func aggregateJobStatus(jobs []*ActionRunJob) Status {
|
||
|
allDone := true
|
||
|
allWaiting := true
|
||
|
hasFailure := false
|
||
|
for _, job := range jobs {
|
||
|
if !job.Status.IsDone() {
|
||
|
allDone = false
|
||
|
}
|
||
|
if job.Status != StatusWaiting {
|
||
|
allWaiting = false
|
||
|
}
|
||
|
if job.Status == StatusFailure || job.Status == StatusCancelled {
|
||
|
hasFailure = true
|
||
|
}
|
||
|
}
|
||
|
if allDone {
|
||
|
if hasFailure {
|
||
|
return StatusFailure
|
||
|
}
|
||
|
return StatusSuccess
|
||
|
}
|
||
|
if allWaiting {
|
||
|
return StatusWaiting
|
||
|
}
|
||
|
return StatusRunning
|
||
|
}
|