mirror of
https://github.com/go-gitea/gitea.git
synced 2024-08-26 21:54:19 -04:00
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
|
||
|
}
|