code.gitea.io/gitea@v1.22.3/models/git/commit_status_summary.go (about)

     1  // Copyright 2024 Gitea. All rights reserved.
     2  // SPDX-License-Identifier: MIT
     3  
     4  package git
     5  
     6  import (
     7  	"context"
     8  
     9  	"code.gitea.io/gitea/models/db"
    10  	"code.gitea.io/gitea/modules/setting"
    11  	api "code.gitea.io/gitea/modules/structs"
    12  
    13  	"xorm.io/builder"
    14  )
    15  
    16  // CommitStatusSummary holds the latest commit Status of a single Commit
    17  type CommitStatusSummary struct {
    18  	ID        int64                 `xorm:"pk autoincr"`
    19  	RepoID    int64                 `xorm:"INDEX UNIQUE(repo_id_sha)"`
    20  	SHA       string                `xorm:"VARCHAR(64) NOT NULL INDEX UNIQUE(repo_id_sha)"`
    21  	State     api.CommitStatusState `xorm:"VARCHAR(7) NOT NULL"`
    22  	TargetURL string                `xorm:"TEXT"`
    23  }
    24  
    25  func init() {
    26  	db.RegisterModel(new(CommitStatusSummary))
    27  }
    28  
    29  type RepoSHA struct {
    30  	RepoID int64
    31  	SHA    string
    32  }
    33  
    34  func GetLatestCommitStatusForRepoAndSHAs(ctx context.Context, repoSHAs []RepoSHA) ([]*CommitStatus, error) {
    35  	cond := builder.NewCond()
    36  	for _, rs := range repoSHAs {
    37  		cond = cond.Or(builder.Eq{"repo_id": rs.RepoID, "sha": rs.SHA})
    38  	}
    39  
    40  	var summaries []CommitStatusSummary
    41  	if err := db.GetEngine(ctx).Where(cond).Find(&summaries); err != nil {
    42  		return nil, err
    43  	}
    44  
    45  	commitStatuses := make([]*CommitStatus, 0, len(repoSHAs))
    46  	for _, summary := range summaries {
    47  		commitStatuses = append(commitStatuses, &CommitStatus{
    48  			RepoID:    summary.RepoID,
    49  			SHA:       summary.SHA,
    50  			State:     summary.State,
    51  			TargetURL: summary.TargetURL,
    52  		})
    53  	}
    54  	return commitStatuses, nil
    55  }
    56  
    57  func UpdateCommitStatusSummary(ctx context.Context, repoID int64, sha string) error {
    58  	commitStatuses, _, err := GetLatestCommitStatus(ctx, repoID, sha, db.ListOptionsAll)
    59  	if err != nil {
    60  		return err
    61  	}
    62  	state := CalcCommitStatus(commitStatuses)
    63  	// mysql will return 0 when update a record which state hasn't been changed which behaviour is different from other database,
    64  	// so we need to use insert in on duplicate
    65  	if setting.Database.Type.IsMySQL() {
    66  		_, err := db.GetEngine(ctx).Exec("INSERT INTO commit_status_summary (repo_id,sha,state,target_url) VALUES (?,?,?,?) ON DUPLICATE KEY UPDATE state=?",
    67  			repoID, sha, state.State, state.TargetURL, state.State)
    68  		return err
    69  	}
    70  
    71  	if cnt, err := db.GetEngine(ctx).Where("repo_id=? AND sha=?", repoID, sha).
    72  		Cols("state, target_url").
    73  		Update(&CommitStatusSummary{
    74  			State:     state.State,
    75  			TargetURL: state.TargetURL,
    76  		}); err != nil {
    77  		return err
    78  	} else if cnt == 0 {
    79  		_, err = db.GetEngine(ctx).Insert(&CommitStatusSummary{
    80  			RepoID:    repoID,
    81  			SHA:       sha,
    82  			State:     state.State,
    83  			TargetURL: state.TargetURL,
    84  		})
    85  		return err
    86  	}
    87  	return nil
    88  }