mirror of
https://codeberg.org/forgejo/forgejo.git
synced 2024-12-27 13:39:19 -05:00
f5c7d4cfdd
Close #24544 Changes: - Create `action_tasks_version` table to store the latest version of each scope (global, org and repo). - When a job with the status of `waiting` is created, the tasks version of the scopes it belongs to will increase. - When the status of a job already in the database is updated to `waiting`, the tasks version of the scopes it belongs to will increase. - On Gitea side, in `FeatchTask()`, will try to query the `action_tasks_version` record of the scope of the runner that call `FetchTask()`. If the record does not exist, will insert a row. Then, Gitea will compare the version passed from runner to Gitea with the version in database, if inconsistent, try pick task. Gitea always returns the latest version from database to the runner. Related: - Protocol: https://gitea.com/gitea/actions-proto-def/pulls/10 - Runner: https://gitea.com/gitea/act_runner/pulls/219
105 lines
2.8 KiB
Go
105 lines
2.8 KiB
Go
// Copyright 2023 The Gitea Authors. All rights reserved.
|
|
// SPDX-License-Identifier: MIT
|
|
|
|
package actions
|
|
|
|
import (
|
|
"context"
|
|
|
|
"code.gitea.io/gitea/models/db"
|
|
"code.gitea.io/gitea/modules/log"
|
|
"code.gitea.io/gitea/modules/timeutil"
|
|
)
|
|
|
|
// ActionTasksVersion
|
|
// If both ownerID and repoID is zero, its scope is global.
|
|
// If ownerID is not zero and repoID is zero, its scope is org (there is no user-level runner currrently).
|
|
// If ownerID is zero and repoID is not zero, its scope is repo.
|
|
type ActionTasksVersion struct {
|
|
ID int64 `xorm:"pk autoincr"`
|
|
OwnerID int64 `xorm:"UNIQUE(owner_repo)"`
|
|
RepoID int64 `xorm:"INDEX UNIQUE(owner_repo)"`
|
|
Version int64
|
|
CreatedUnix timeutil.TimeStamp `xorm:"created"`
|
|
UpdatedUnix timeutil.TimeStamp `xorm:"updated"`
|
|
}
|
|
|
|
func init() {
|
|
db.RegisterModel(new(ActionTasksVersion))
|
|
}
|
|
|
|
func GetTasksVersionByScope(ctx context.Context, ownerID, repoID int64) (int64, error) {
|
|
var tasksVersion ActionTasksVersion
|
|
has, err := db.GetEngine(ctx).Where("owner_id = ? AND repo_id = ?", ownerID, repoID).Get(&tasksVersion)
|
|
if err != nil {
|
|
return 0, err
|
|
} else if !has {
|
|
return 0, nil
|
|
}
|
|
return tasksVersion.Version, err
|
|
}
|
|
|
|
func insertTasksVersion(ctx context.Context, ownerID, repoID int64) (*ActionTasksVersion, error) {
|
|
tasksVersion := &ActionTasksVersion{
|
|
OwnerID: ownerID,
|
|
RepoID: repoID,
|
|
Version: 1,
|
|
}
|
|
if _, err := db.GetEngine(ctx).Insert(tasksVersion); err != nil {
|
|
return nil, err
|
|
}
|
|
return tasksVersion, nil
|
|
}
|
|
|
|
func increaseTasksVersionByScope(ctx context.Context, ownerID, repoID int64) error {
|
|
result, err := db.GetEngine(ctx).Exec("UPDATE action_tasks_version SET version = version + 1 WHERE owner_id = ? AND repo_id = ?", ownerID, repoID)
|
|
if err != nil {
|
|
return err
|
|
}
|
|
affected, err := result.RowsAffected()
|
|
if err != nil {
|
|
return err
|
|
}
|
|
|
|
if affected == 0 {
|
|
// if update sql does not affect any rows, the database may be broken,
|
|
// so re-insert the row of version data here.
|
|
if _, err := insertTasksVersion(ctx, ownerID, repoID); err != nil {
|
|
return err
|
|
}
|
|
}
|
|
|
|
return nil
|
|
}
|
|
|
|
func IncreaseTaskVersion(ctx context.Context, ownerID, repoID int64) error {
|
|
ctx, commiter, err := db.TxContext(ctx)
|
|
if err != nil {
|
|
return err
|
|
}
|
|
defer commiter.Close()
|
|
|
|
// 1. increase global
|
|
if err := increaseTasksVersionByScope(ctx, 0, 0); err != nil {
|
|
log.Error("IncreaseTasksVersionByScope(Global): %v", err)
|
|
return err
|
|
}
|
|
|
|
// 2. increase owner
|
|
if ownerID > 0 {
|
|
if err := increaseTasksVersionByScope(ctx, ownerID, 0); err != nil {
|
|
log.Error("IncreaseTasksVersionByScope(Owner): %v", err)
|
|
return err
|
|
}
|
|
}
|
|
|
|
// 3. increase repo
|
|
if repoID > 0 {
|
|
if err := increaseTasksVersionByScope(ctx, 0, repoID); err != nil {
|
|
log.Error("IncreaseTasksVersionByScope(Repo): %v", err)
|
|
return err
|
|
}
|
|
}
|
|
|
|
return commiter.Commit()
|
|
}
|