forked from gitea/gitea
62 lines
1.5 KiB
Go
62 lines
1.5 KiB
Go
|
// Copyright 2023 The Gitea Authors. All rights reserved.
|
||
|
|
// SPDX-License-Identifier: MIT
|
||
|
|
|
||
|
|
package release
|
||
|
|
|
||
|
|
import (
|
||
|
|
"context"
|
||
|
|
"errors"
|
||
|
|
"fmt"
|
||
|
|
|
||
|
|
"code.gitea.io/gitea/models/db"
|
||
|
|
repo_model "code.gitea.io/gitea/models/repo"
|
||
|
|
"code.gitea.io/gitea/modules/graceful"
|
||
|
|
"code.gitea.io/gitea/modules/log"
|
||
|
|
"code.gitea.io/gitea/modules/queue"
|
||
|
|
repo_module "code.gitea.io/gitea/modules/repository"
|
||
|
|
|
||
|
|
"xorm.io/builder"
|
||
|
|
)
|
||
|
|
|
||
|
|
type TagSyncOptions struct {
|
||
|
|
RepoID int64
|
||
|
|
}
|
||
|
|
|
||
|
|
// tagSyncQueue represents a queue to handle tag sync jobs.
|
||
|
|
var tagSyncQueue *queue.WorkerPoolQueue[*TagSyncOptions]
|
||
|
|
|
||
|
|
func handlerTagSync(items ...*TagSyncOptions) []*TagSyncOptions {
|
||
|
|
for _, opts := range items {
|
||
|
|
err := repo_module.SyncRepoTags(graceful.GetManager().ShutdownContext(), opts.RepoID)
|
||
|
|
if err != nil {
|
||
|
|
log.Error("syncRepoTags [%d] failed: %v", opts.RepoID, err)
|
||
|
|
}
|
||
|
|
}
|
||
|
|
return nil
|
||
|
|
}
|
||
|
|
|
||
|
|
func addRepoToTagSyncQueue(repoID int64) error {
|
||
|
|
return tagSyncQueue.Push(&TagSyncOptions{
|
||
|
|
RepoID: repoID,
|
||
|
|
})
|
||
|
|
}
|
||
|
|
|
||
|
|
func initTagSyncQueue(ctx context.Context) error {
|
||
|
|
tagSyncQueue = queue.CreateUniqueQueue(ctx, "tag_sync", handlerTagSync)
|
||
|
|
if tagSyncQueue == nil {
|
||
|
|
return errors.New("unable to create tag_sync queue")
|
||
|
|
}
|
||
|
|
go graceful.GetManager().RunWithCancel(tagSyncQueue)
|
||
|
|
|
||
|
|
return nil
|
||
|
|
}
|
||
|
|
|
||
|
|
func AddAllRepoTagsToSyncQueue(ctx context.Context) error {
|
||
|
|
if err := db.Iterate(ctx, builder.Eq{"is_empty": false}, func(ctx context.Context, repo *repo_model.Repository) error {
|
||
|
|
return addRepoToTagSyncQueue(repo.ID)
|
||
|
|
}); err != nil {
|
||
|
|
return fmt.Errorf("run sync all tags failed: %v", err)
|
||
|
|
}
|
||
|
|
return nil
|
||
|
|
}
|