2014-04-13 10:30:09 +09:00
|
|
|
// Copyright 2014 The Gogs Authors. All rights reserved.
|
2019-07-16 09:13:03 +09:00
|
|
|
// Copyright 2019 The Gitea Authors. All rights reserved.
|
2014-04-13 10:30:09 +09: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 18:51:28 +09:00
|
|
|
"context"
|
2014-06-14 02:01:52 +09:00
|
|
|
"time"
|
|
|
|
|
2019-12-15 18:51:28 +09:00
|
|
|
"code.gitea.io/gitea/modules/graceful"
|
2019-07-16 09:13:03 +09:00
|
|
|
"code.gitea.io/gitea/modules/sync"
|
|
|
|
|
|
|
|
"github.com/gogs/cron"
|
|
|
|
)
|
|
|
|
|
2016-02-21 05:58:09 +09:00
|
|
|
var c = cron.New()
|
|
|
|
|
2019-07-16 09:13:03 +09:00
|
|
|
// Prevent duplicate running tasks.
|
|
|
|
var taskStatusTable = sync.NewStatusTable()
|
|
|
|
|
2016-11-25 17:19:24 +09:00
|
|
|
// NewContext begins cron tasks
|
2019-12-15 18:51:28 +09:00
|
|
|
// Each cron task is run within the shutdown context as a running server
|
|
|
|
// AtShutdown the cron server is stopped
|
2016-02-21 05:58:09 +09:00
|
|
|
func NewContext() {
|
2020-05-17 08:31:38 +09:00
|
|
|
initBasicTasks()
|
|
|
|
initExtendedTasks()
|
2019-10-14 15:10:42 +09:00
|
|
|
|
2020-05-17 08:31:38 +09:00
|
|
|
lock.Lock()
|
|
|
|
for _, task := range tasks {
|
|
|
|
if task.IsEnabled() && task.DoRunAtStart() {
|
|
|
|
go task.Run()
|
|
|
|
}
|
2019-10-14 15:10:42 +09:00
|
|
|
}
|
|
|
|
|
2016-02-21 05:58:09 +09:00
|
|
|
c.Start()
|
2020-05-17 08:31:38 +09: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 {
|
2022-03-29 10:31:07 +09:00
|
|
|
Name string
|
|
|
|
Spec string
|
|
|
|
Next time.Time
|
|
|
|
Prev time.Time
|
|
|
|
Status string
|
|
|
|
LastMessage string
|
|
|
|
LastDoer string
|
|
|
|
ExecTimes int64
|
|
|
|
task *Task
|
|
|
|
}
|
|
|
|
|
|
|
|
func (t *TaskTableRow) FormatLastMessage(locale string) string {
|
|
|
|
if t.Status == "finished" {
|
|
|
|
return t.task.GetConfig().FormatMessage(locale, t.Name, t.Status, t.LastDoer)
|
|
|
|
}
|
|
|
|
|
|
|
|
return t.task.GetConfig().FormatMessage(locale, t.Name, t.Status, t.LastDoer, t.LastMessage)
|
2014-06-14 02:01:52 +09:00
|
|
|
}
|
|
|
|
|
2020-05-17 08:31:38 +09:00
|
|
|
// TaskTable represents a table of tasks
|
|
|
|
type TaskTable []*TaskTableRow
|
|
|
|
|
2016-02-21 05:58:09 +09:00
|
|
|
// ListTasks returns all running cron tasks.
|
2020-05-17 08:31:38 +09: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{
|
2022-03-29 10:31:07 +09:00
|
|
|
Name: task.Name,
|
|
|
|
Spec: spec,
|
|
|
|
Next: next,
|
|
|
|
Prev: prev,
|
|
|
|
ExecTimes: task.ExecTimes,
|
|
|
|
LastMessage: task.LastMessage,
|
|
|
|
Status: task.Status,
|
|
|
|
LastDoer: task.LastDoer,
|
|
|
|
task: task,
|
2020-05-17 08:31:38 +09:00
|
|
|
})
|
|
|
|
task.lock.Unlock()
|
|
|
|
}
|
|
|
|
|
|
|
|
return tTable
|
2014-04-13 10:30:09 +09:00
|
|
|
}
|