Платформа ЦРНП "Мирокод" для разработки проектов
https://git.mirocod.ru
You can not select more than 25 topics
Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
49 lines
1.2 KiB
49 lines
1.2 KiB
// Copyright 2020 The Gitea Authors. All rights reserved. |
|
// Use of this source code is governed by a MIT-style |
|
// license that can be found in the LICENSE file. |
|
|
|
package stats |
|
|
|
import ( |
|
"fmt" |
|
|
|
"code.gitea.io/gitea/models" |
|
"code.gitea.io/gitea/modules/graceful" |
|
"code.gitea.io/gitea/modules/log" |
|
"code.gitea.io/gitea/modules/queue" |
|
) |
|
|
|
// statsQueue represents a queue to handle repository stats updates |
|
var statsQueue queue.UniqueQueue |
|
|
|
// handle passed PR IDs and test the PRs |
|
func handle(data ...queue.Data) { |
|
for _, datum := range data { |
|
opts := datum.(int64) |
|
if err := indexer.Index(opts); err != nil { |
|
log.Error("stats queue indexer.Index(%d) failed: %v", opts, err) |
|
} |
|
} |
|
} |
|
|
|
func initStatsQueue() error { |
|
statsQueue = queue.CreateUniqueQueue("repo_stats_update", handle, int64(0)) |
|
if statsQueue == nil { |
|
return fmt.Errorf("Unable to create repo_stats_update Queue") |
|
} |
|
|
|
go graceful.GetManager().RunWithShutdownFns(statsQueue.Run) |
|
|
|
return nil |
|
} |
|
|
|
// UpdateRepoIndexer update a repository's entries in the indexer |
|
func UpdateRepoIndexer(repo *models.Repository) error { |
|
if err := statsQueue.Push(repo.ID); err != nil { |
|
if err != queue.ErrAlreadyInQueue { |
|
return err |
|
} |
|
log.Debug("Repo ID: %d already queued", repo.ID) |
|
} |
|
return nil |
|
}
|
|
|