|
@@ -17,6 +17,7 @@ import (
|
|
|
"regexp"
|
|
|
"sort"
|
|
|
"strings"
|
|
|
+ "sync"
|
|
|
"time"
|
|
|
"unicode/utf8"
|
|
|
|
|
@@ -1377,20 +1378,54 @@ func RewriteRepositoryUpdateHook() error {
|
|
|
})
|
|
|
}
|
|
|
|
|
|
-var (
|
|
|
- // Prevent duplicate running tasks.
|
|
|
- isMirrorUpdating = false
|
|
|
- isGitFscking = false
|
|
|
- isCheckingRepos = false
|
|
|
+// statusPool represents a pool of status with true/false.
|
|
|
+type statusPool struct {
|
|
|
+ lock sync.RWMutex
|
|
|
+ pool map[string]bool
|
|
|
+}
|
|
|
+
|
|
|
+// Start sets value of given name to true in the pool.
|
|
|
+func (p *statusPool) Start(name string) {
|
|
|
+ p.lock.Lock()
|
|
|
+ defer p.lock.Unlock()
|
|
|
+
|
|
|
+ p.pool[name] = true
|
|
|
+}
|
|
|
+
|
|
|
+// Stop sets value of given name to false in the pool.
|
|
|
+func (p *statusPool) Stop(name string) {
|
|
|
+ p.lock.Lock()
|
|
|
+ defer p.lock.Unlock()
|
|
|
+
|
|
|
+ p.pool[name] = false
|
|
|
+}
|
|
|
+
|
|
|
+// IsRunning checks if value of given name is set to true in the pool.
|
|
|
+func (p *statusPool) IsRunning(name string) bool {
|
|
|
+ p.lock.RLock()
|
|
|
+ defer p.lock.RUnlock()
|
|
|
+
|
|
|
+ return p.pool[name]
|
|
|
+}
|
|
|
+
|
|
|
+// Prevent duplicate running tasks.
|
|
|
+var taskStatusPool = &statusPool{
|
|
|
+ pool: make(map[string]bool),
|
|
|
+}
|
|
|
+
|
|
|
+const (
|
|
|
+ _MIRROR_UPDATE = "mirror_update"
|
|
|
+ _GIT_FSCK = "git_fsck"
|
|
|
+ _CHECK_REPOs = "check_repos"
|
|
|
)
|
|
|
|
|
|
// MirrorUpdate checks and updates mirror repositories.
|
|
|
func MirrorUpdate() {
|
|
|
- if isMirrorUpdating {
|
|
|
+ if taskStatusPool.IsRunning(_MIRROR_UPDATE) {
|
|
|
return
|
|
|
}
|
|
|
- isMirrorUpdating = true
|
|
|
- defer func() { isMirrorUpdating = false }()
|
|
|
+ taskStatusPool.Start(_MIRROR_UPDATE)
|
|
|
+ defer taskStatusPool.Stop(_MIRROR_UPDATE)
|
|
|
|
|
|
log.Trace("Doing: MirrorUpdate")
|
|
|
|
|
@@ -1438,11 +1473,11 @@ func MirrorUpdate() {
|
|
|
|
|
|
// GitFsck calls 'git fsck' to check repository health.
|
|
|
func GitFsck() {
|
|
|
- if isGitFscking {
|
|
|
+ if taskStatusPool.IsRunning(_GIT_FSCK) {
|
|
|
return
|
|
|
}
|
|
|
- isGitFscking = true
|
|
|
- defer func() { isGitFscking = false }()
|
|
|
+ taskStatusPool.Start(_GIT_FSCK)
|
|
|
+ defer taskStatusPool.Stop(_GIT_FSCK)
|
|
|
|
|
|
log.Trace("Doing: GitFsck")
|
|
|
|
|
@@ -1507,11 +1542,11 @@ func repoStatsCheck(checker *repoChecker) {
|
|
|
}
|
|
|
|
|
|
func CheckRepoStats() {
|
|
|
- if isCheckingRepos {
|
|
|
+ if taskStatusPool.IsRunning(_CHECK_REPOs) {
|
|
|
return
|
|
|
}
|
|
|
- isCheckingRepos = true
|
|
|
- defer func() { isCheckingRepos = false }()
|
|
|
+ taskStatusPool.Start(_CHECK_REPOs)
|
|
|
+ defer taskStatusPool.Stop(_CHECK_REPOs)
|
|
|
|
|
|
log.Trace("Doing: CheckRepoStats")
|
|
|
|