2014-04-12 19:30:09 -06:00
// Copyright 2014 The Gogs Authors. All rights reserved.
2019-07-15 18:13:03 -06:00
// Copyright 2019 The Gitea Authors. All rights reserved.
2014-04-12 19:30:09 -06:00
// Use of this source code is governed by a MIT-style
// license that can be found in the LICENSE file.
package cron
import (
2014-06-13 11:01:52 -06:00
"time"
2016-11-10 09:24:48 -07:00
"code.gitea.io/gitea/models"
"code.gitea.io/gitea/modules/log"
"code.gitea.io/gitea/modules/setting"
2019-07-15 18:13:03 -06:00
"code.gitea.io/gitea/modules/sync"
2019-10-01 07:40:17 -06:00
mirror_service "code.gitea.io/gitea/services/mirror"
2019-07-15 18:13:03 -06:00
"github.com/gogs/cron"
)
const (
mirrorUpdate = "mirror_update"
gitFsck = "git_fsck"
checkRepos = "check_repos"
archiveCleanup = "archive_cleanup"
syncExternalUsers = "sync_external_users"
deletedBranchesCleanup = "deleted_branches_cleanup"
2016-02-20 13:58:09 -07:00
)
2014-06-13 11:01:52 -06:00
2016-02-20 13:58:09 -07:00
var c = cron . New ( )
2019-07-15 18:13:03 -06:00
// Prevent duplicate running tasks.
var taskStatusTable = sync . NewStatusTable ( )
// Func defines a cron function body
type Func func ( )
// WithUnique wrap a cron func with an unique running check
func WithUnique ( name string , body Func ) Func {
return func ( ) {
if ! taskStatusTable . StartIfNotRunning ( name ) {
return
}
defer taskStatusTable . Stop ( name )
body ( )
}
}
2016-11-25 01:19:24 -07:00
// NewContext begins cron tasks
2016-02-20 13:58:09 -07:00
func NewContext ( ) {
var (
entry * cron . Entry
err error
)
if setting . Cron . UpdateMirror . Enabled {
2019-10-01 07:40:17 -06:00
entry , err = c . AddFunc ( "Update mirrors" , setting . Cron . UpdateMirror . Schedule , WithUnique ( mirrorUpdate , mirror_service . Update ) )
2016-02-20 13:58:09 -07:00
if err != nil {
2019-04-02 01:48:31 -06:00
log . Fatal ( "Cron[Update mirrors]: %v" , err )
2016-02-20 13:58:09 -07:00
}
if setting . Cron . UpdateMirror . RunAtStart {
entry . Prev = time . Now ( )
entry . ExecTimes ++
2019-10-01 07:40:17 -06:00
go WithUnique ( mirrorUpdate , mirror_service . Update ) ( )
2016-02-20 13:58:09 -07:00
}
2014-06-13 11:01:52 -06:00
}
2016-02-20 13:58:09 -07:00
if setting . Cron . RepoHealthCheck . Enabled {
2019-07-15 18:13:03 -06:00
entry , err = c . AddFunc ( "Repository health check" , setting . Cron . RepoHealthCheck . Schedule , WithUnique ( gitFsck , models . GitFsck ) )
2016-02-20 13:58:09 -07:00
if err != nil {
2019-04-02 01:48:31 -06:00
log . Fatal ( "Cron[Repository health check]: %v" , err )
2016-02-20 13:58:09 -07:00
}
if setting . Cron . RepoHealthCheck . RunAtStart {
entry . Prev = time . Now ( )
entry . ExecTimes ++
2019-07-15 18:13:03 -06:00
go WithUnique ( gitFsck , models . GitFsck ) ( )
2016-02-20 13:58:09 -07:00
}
2014-06-13 11:01:52 -06:00
}
2016-02-20 13:58:09 -07:00
if setting . Cron . CheckRepoStats . Enabled {
2019-07-15 18:13:03 -06:00
entry , err = c . AddFunc ( "Check repository statistics" , setting . Cron . CheckRepoStats . Schedule , WithUnique ( checkRepos , models . CheckRepoStats ) )
2016-02-20 13:58:09 -07:00
if err != nil {
2019-04-02 01:48:31 -06:00
log . Fatal ( "Cron[Check repository statistics]: %v" , err )
2014-06-13 11:01:52 -06:00
}
2016-02-20 13:58:09 -07:00
if setting . Cron . CheckRepoStats . RunAtStart {
entry . Prev = time . Now ( )
entry . ExecTimes ++
2019-07-15 18:13:03 -06:00
go WithUnique ( checkRepos , models . CheckRepoStats ) ( )
2014-06-13 11:01:52 -06:00
}
}
2017-02-10 21:00:46 -07:00
if setting . Cron . ArchiveCleanup . Enabled {
2019-07-15 18:13:03 -06:00
entry , err = c . AddFunc ( "Clean up old repository archives" , setting . Cron . ArchiveCleanup . Schedule , WithUnique ( archiveCleanup , models . DeleteOldRepositoryArchives ) )
2017-02-10 21:00:46 -07:00
if err != nil {
2019-04-02 01:48:31 -06:00
log . Fatal ( "Cron[Clean up old repository archives]: %v" , err )
2017-02-10 21:00:46 -07:00
}
if setting . Cron . ArchiveCleanup . RunAtStart {
entry . Prev = time . Now ( )
entry . ExecTimes ++
2019-07-15 18:13:03 -06:00
go WithUnique ( archiveCleanup , models . DeleteOldRepositoryArchives ) ( )
2017-02-10 21:00:46 -07:00
}
}
2017-05-10 07:10:18 -06:00
if setting . Cron . SyncExternalUsers . Enabled {
2019-07-15 18:13:03 -06:00
entry , err = c . AddFunc ( "Synchronize external users" , setting . Cron . SyncExternalUsers . Schedule , WithUnique ( syncExternalUsers , models . SyncExternalUsers ) )
2017-05-10 07:10:18 -06:00
if err != nil {
2019-04-02 01:48:31 -06:00
log . Fatal ( "Cron[Synchronize external users]: %v" , err )
2017-05-10 07:10:18 -06:00
}
if setting . Cron . SyncExternalUsers . RunAtStart {
entry . Prev = time . Now ( )
entry . ExecTimes ++
2019-07-15 18:13:03 -06:00
go WithUnique ( syncExternalUsers , models . SyncExternalUsers ) ( )
2017-05-10 07:10:18 -06:00
}
}
2017-10-25 18:49:16 -06:00
if setting . Cron . DeletedBranchesCleanup . Enabled {
2019-07-15 18:13:03 -06:00
entry , err = c . AddFunc ( "Remove old deleted branches" , setting . Cron . DeletedBranchesCleanup . Schedule , WithUnique ( deletedBranchesCleanup , models . RemoveOldDeletedBranches ) )
2017-10-25 18:49:16 -06:00
if err != nil {
2019-04-02 01:48:31 -06:00
log . Fatal ( "Cron[Remove old deleted branches]: %v" , err )
2017-10-25 18:49:16 -06:00
}
if setting . Cron . DeletedBranchesCleanup . RunAtStart {
entry . Prev = time . Now ( )
entry . ExecTimes ++
2019-07-15 18:13:03 -06:00
go WithUnique ( deletedBranchesCleanup , models . RemoveOldDeletedBranches ) ( )
2017-10-25 18:49:16 -06:00
}
}
2016-02-20 13:58:09 -07:00
c . Start ( )
2014-06-13 11:01:52 -06:00
}
2016-02-20 13:58:09 -07:00
// ListTasks returns all running cron tasks.
func ListTasks ( ) [ ] * cron . Entry {
return c . Entries ( )
2014-04-12 19:30:09 -06:00
}