123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163 |
- // 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
- }
|