admin
e6956aa001
- React frontend with route-level code splitting - Backend rebranded from Baihu to TaskPool - DB brand migration script and local compatibility
70 lines
1.6 KiB
Go
70 lines
1.6 KiB
Go
package executor
|
|
|
|
import (
|
|
"sync"
|
|
|
|
"github.com/engigu/taskpool/internal/logger"
|
|
"github.com/engigu/taskpool/internal/systime"
|
|
"github.com/robfig/cron/v3"
|
|
)
|
|
|
|
type SysCronManager struct {
|
|
cron *cron.Cron
|
|
}
|
|
|
|
var (
|
|
sysCronInstance *SysCronManager
|
|
sysCronOnce sync.Once
|
|
)
|
|
|
|
|
|
// InitSysCron 初始化系统的内部定时器
|
|
func InitSysCron(){
|
|
GetSysCron()
|
|
}
|
|
|
|
// GetSysCron 获取内部系统定时器服务单例
|
|
func GetSysCron() *SysCronManager {
|
|
sysCronOnce.Do(func() {
|
|
// 使用秒级精度，指定为东八区
|
|
c := cron.New(cron.WithSeconds(), cron.WithLocation(systime.CST))
|
|
c.Start()
|
|
sysCronInstance = &SysCronManager{
|
|
cron: c,
|
|
}
|
|
logger.Infof("[SysCron] 内部系统定时管理器已启动")
|
|
})
|
|
return sysCronInstance
|
|
}
|
|
|
|
// AddJob 添加内部系统任务，spec为cron表达式（支持 @every 30s 这种快捷方式）
|
|
func (s *SysCronManager) AddJob(spec string, cmd func()) (cron.EntryID, error) {
|
|
id, err := s.cron.AddFunc(spec, cmd)
|
|
if err != nil {
|
|
logger.Errorf("[SysCron] 无法添加系统任务: %s, err: %v", spec, err)
|
|
return 0, err
|
|
}
|
|
return id, nil
|
|
}
|
|
|
|
// AddJobWithRun 立即开启一个协程异步执行一次任务，随后将其加入到系统定时任务中
|
|
func (s *SysCronManager) AddJobWithRun(spec string, cmd func()) (cron.EntryID, error) {
|
|
// 立即异步执行一次
|
|
go func() {
|
|
defer func() {
|
|
if r := recover(); r != nil {
|
|
logger.Errorf("[SysCron] 立即执行任务时发生 panic: %v", r)
|
|
}
|
|
}()
|
|
cmd()
|
|
}()
|
|
|
|
// 然后加入定时器
|
|
return s.AddJob(spec, cmd)
|
|
}
|
|
|
|
// RemoveJob 动态移除指定的系统定时任务
|
|
func (s *SysCronManager) RemoveJob(id cron.EntryID) {
|
|
s.cron.Remove(id)
|
|
}
|