12345678910111213141516171819202122232425262728293031323334353637383940414243444546474849505152535455565758596061626364656667686970717273747576777879808182838485868788 |
- // Copyright 2024 Gitea. All rights reserved.
- // SPDX-License-Identifier: MIT
-
- package git
-
- import (
- "context"
-
- "code.gitea.io/gitea/models/db"
- "code.gitea.io/gitea/modules/setting"
- api "code.gitea.io/gitea/modules/structs"
-
- "xorm.io/builder"
- )
-
- // CommitStatusSummary holds the latest commit Status of a single Commit
- type CommitStatusSummary struct {
- ID int64 `xorm:"pk autoincr"`
- RepoID int64 `xorm:"INDEX UNIQUE(repo_id_sha)"`
- SHA string `xorm:"VARCHAR(64) NOT NULL INDEX UNIQUE(repo_id_sha)"`
- State api.CommitStatusState `xorm:"VARCHAR(7) NOT NULL"`
- TargetURL string `xorm:"TEXT"`
- }
-
- func init() {
- db.RegisterModel(new(CommitStatusSummary))
- }
-
- type RepoSHA struct {
- RepoID int64
- SHA string
- }
-
- func GetLatestCommitStatusForRepoAndSHAs(ctx context.Context, repoSHAs []RepoSHA) ([]*CommitStatus, error) {
- cond := builder.NewCond()
- for _, rs := range repoSHAs {
- cond = cond.Or(builder.Eq{"repo_id": rs.RepoID, "sha": rs.SHA})
- }
-
- var summaries []CommitStatusSummary
- if err := db.GetEngine(ctx).Where(cond).Find(&summaries); err != nil {
- return nil, err
- }
-
- commitStatuses := make([]*CommitStatus, 0, len(repoSHAs))
- for _, summary := range summaries {
- commitStatuses = append(commitStatuses, &CommitStatus{
- RepoID: summary.RepoID,
- SHA: summary.SHA,
- State: summary.State,
- TargetURL: summary.TargetURL,
- })
- }
- return commitStatuses, nil
- }
-
- func UpdateCommitStatusSummary(ctx context.Context, repoID int64, sha string) error {
- commitStatuses, _, err := GetLatestCommitStatus(ctx, repoID, sha, db.ListOptionsAll)
- if err != nil {
- return err
- }
- state := CalcCommitStatus(commitStatuses)
- // mysql will return 0 when update a record which state hasn't been changed which behaviour is different from other database,
- // so we need to use insert in on duplicate
- if setting.Database.Type.IsMySQL() {
- _, err := db.GetEngine(ctx).Exec("INSERT INTO commit_status_summary (repo_id,sha,state,target_url) VALUES (?,?,?,?) ON DUPLICATE KEY UPDATE state=?",
- repoID, sha, state.State, state.TargetURL, state.State)
- return err
- }
-
- if cnt, err := db.GetEngine(ctx).Where("repo_id=? AND sha=?", repoID, sha).
- Cols("state, target_url").
- Update(&CommitStatusSummary{
- State: state.State,
- TargetURL: state.TargetURL,
- }); err != nil {
- return err
- } else if cnt == 0 {
- _, err = db.GetEngine(ctx).Insert(&CommitStatusSummary{
- RepoID: repoID,
- SHA: sha,
- State: state.State,
- TargetURL: state.TargetURL,
- })
- return err
- }
- return nil
- }
|