2014-04-13 05:30:09 +04:00
// Copyright 2014 The Gogs Authors. All rights reserved.
2019-07-16 03:13:03 +03:00
// Copyright 2019 The Gitea Authors. All rights reserved.
2014-04-13 05:30:09 +04:00
// Use of this source code is governed by a MIT-style
// license that can be found in the LICENSE file.
package cron
import (
2019-12-15 12:51:28 +03:00
"context"
2014-06-13 21:01:52 +04:00
"time"
2019-12-15 12:51:28 +03:00
"code.gitea.io/gitea/modules/graceful"
2019-07-16 03:13:03 +03:00
"code.gitea.io/gitea/modules/sync"
"github.com/gogs/cron"
)
2016-02-20 23:58:09 +03:00
var c = cron . New ( )
2019-07-16 03:13:03 +03:00
// Prevent duplicate running tasks.
var taskStatusTable = sync . NewStatusTable ( )
2016-11-25 11:19:24 +03:00
// NewContext begins cron tasks
2019-12-15 12:51:28 +03:00
// Each cron task is run within the shutdown context as a running server
// AtShutdown the cron server is stopped
2016-02-20 23:58:09 +03:00
func NewContext ( ) {
2020-05-17 02:31:38 +03:00
initBasicTasks ( )
initExtendedTasks ( )
2019-10-14 09:10:42 +03:00
2020-05-17 02:31:38 +03:00
lock . Lock ( )
for _ , task := range tasks {
if task . IsEnabled ( ) && task . DoRunAtStart ( ) {
go task . Run ( )
}
2019-10-14 09:10:42 +03:00
}
2016-02-20 23:58:09 +03:00
c . Start ( )
2020-05-17 02:31:38 +03:00
started = true
lock . Unlock ( )
graceful . GetManager ( ) . RunAtShutdown ( context . Background ( ) , func ( ) {
c . Stop ( )
lock . Lock ( )
started = false
lock . Unlock ( )
} )
}
// TaskTableRow represents a task row in the tasks table
type TaskTableRow struct {
Name string
Spec string
Next time . Time
Prev time . Time
ExecTimes int64
2014-06-13 21:01:52 +04:00
}
2020-05-17 02:31:38 +03:00
// TaskTable represents a table of tasks
type TaskTable [ ] * TaskTableRow
2016-02-20 23:58:09 +03:00
// ListTasks returns all running cron tasks.
2020-05-17 02:31:38 +03:00
func ListTasks ( ) TaskTable {
entries := c . Entries ( )
eMap := map [ string ] * cron . Entry { }
for _ , e := range entries {
eMap [ e . Description ] = e
}
lock . Lock ( )
defer lock . Unlock ( )
tTable := make ( [ ] * TaskTableRow , 0 , len ( tasks ) )
for _ , task := range tasks {
spec := "-"
var (
next time . Time
prev time . Time
)
if e , ok := eMap [ task . Name ] ; ok {
spec = e . Spec
next = e . Next
prev = e . Prev
}
task . lock . Lock ( )
tTable = append ( tTable , & TaskTableRow {
Name : task . Name ,
Spec : spec ,
Next : next ,
Prev : prev ,
ExecTimes : task . ExecTimes ,
} )
task . lock . Unlock ( )
}
return tTable
2014-04-13 05:30:09 +04:00
}