Initial commit: TaskPool React panel

- React frontend with route-level code splitting - Backend rebranded from Baihu to TaskPool - DB brand migration script and local compatibility
2026-07-26 08:43:52 +08:00
commit e6956aa001
397 changed files with 73621 additions and 0 deletions
@@ -0,0 +1,239 @@
+package executor
+
+import (
+	"fmt"
+	"math/rand"
+	"strings"
+	"sync"
+	"time"
+
+	"github.com/engigu/taskpool/internal/systime"
+
+	"github.com/robfig/cron/v3"
+)
+
+// 东八区时区（默认）
+var defaultLocation = systime.CST
+
+// CronManager 统一的任务调度管理器
+type CronManager struct {
+	cron      *cron.Cron
+	scheduler *Scheduler
+	entryMap  map[string]cron.EntryID // task ID -> cron entry ID
+	mu        sync.RWMutex
+	logger    SchedulerLogger
+	OnTrigger func(task CronTask) *ExecutionRequest // 任务触发时的请求构造工厂
+}
+
+// NewCronManager 创建一个新的计划任务管理器
+func NewCronManager(scheduler *Scheduler) *CronManager {
+	// 使用秒级精度的 cron parser
+	c := cron.New(cron.WithSeconds(), cron.WithLocation(defaultLocation))
+
+	m := &CronManager{
+		cron:      c,
+		scheduler: scheduler,
+		entryMap:  make(map[string]cron.EntryID),
+		logger:    &DefaultLogger{},
+	}
+
+	if scheduler != nil && scheduler.logger != nil {
+		m.logger = scheduler.logger
+	}
+
+	return m
+}
+
+// SetLogger 设置自定义日志实现
+func (m *CronManager) SetLogger(logger SchedulerLogger) {
+	m.mu.Lock()
+	defer m.mu.Unlock()
+	m.logger = logger
+}
+
+// SetScheduler 更新关联的调度器实例
+func (m *CronManager) SetScheduler(scheduler *Scheduler) {
+	m.mu.Lock()
+	defer m.mu.Unlock()
+	m.scheduler = scheduler
+}
+
+// Start 启动调度器
+func (m *CronManager) Start() {
+	m.cron.Start()
+	m.logger.Infof("[CronManager] 调度管理服务已启动")
+}
+
+// Stop 停止调度器
+func (m *CronManager) Stop() {
+	ctx := m.cron.Stop()
+	<-ctx.Done()
+	m.logger.Infof("[CronManager] 调度管理服务已停止")
+}
+
+// AddTask 添加或更新计划任务
+func (m *CronManager) AddTask(task CronTask) error {
+	m.mu.Lock()
+	defer m.mu.Unlock()
+
+	taskID := task.GetID()
+
+	// 如果已存在，先移除旧的
+	if entryID, exists := m.entryMap[taskID]; exists {
+		m.cron.Remove(entryID)
+		delete(m.entryMap, taskID)
+	}
+
+	// 准备任务执行函数
+	cmd := task.GetCommand()
+	name := task.GetName()
+	timeout := task.GetTimeout()
+	workDir := task.GetWorkDir()
+	envs := task.GetEnvs()
+	languages := task.GetLanguages()
+	useMise := task.UseMise()
+	secrets := task.GetSecrets()
+
+	schedule := strings.TrimSpace(task.GetSchedule())
+	entryID, err := m.cron.AddFunc(schedule, func() {
+		defer func() {
+			if r := recover(); r != nil {
+				m.logger.Errorf("[CronManager] 任务 #%s 执行过程中发生 Panic: %v", taskID, r)
+			}
+		}()
+
+		// 构造执行请求的 Builder
+		reqBuilder := func() *ExecutionRequest {
+			if m.OnTrigger != nil {
+				return m.OnTrigger(task)
+			}
+			return &ExecutionRequest{
+				TaskID:      taskID,
+				Name:        name,
+				Command:     cmd,
+				PreCommand:  task.GetPreCommand(),
+				PostCommand: task.GetPostCommand(),
+				Type:        TaskTypeCron,
+				Timeout:     timeout,
+				WorkDir:     workDir,
+				Envs: func() []string {
+					if vars := task.GetEnvVars(); len(vars) > 0 {
+						return vars
+					}
+					return ParseEnvVars(envs)
+				}(),
+				Secrets:   secrets,
+				Languages: languages,
+				UseMise:   useMise,
+			}
+		}
+
+		randomRange := task.GetRandomRange()
+		if randomRange > 0 && m.scheduler != nil {
+			// 生成 0 到 randomRange 之间的随机秒数
+			delaySeconds := rand.Intn(randomRange)
+			delay := time.Duration(delaySeconds) * time.Second
+			m.logger.Infof("[CronManager] 任务 %s (#%s) 将随机延迟 %v (范围: %ds) 后入队", name, taskID, delay, randomRange)
+
+			// 使用调度器的延时投递功能，不阻塞当前 Cron 协程
+			m.scheduler.EnqueueDelayed(delay, reqBuilder)
+		} else {
+			m.logger.Infof("[CronManager] 触发计划任务: %s (#%s)", name, taskID)
+			if m.scheduler != nil {
+				m.scheduler.EnqueueOrExecute(reqBuilder())
+			}
+		}
+
+		// 触发下次运行时间更新事件
+		m.triggerNextRunEvent(taskID, &ExecutionRequest{TaskID: taskID})
+	})
+
+	if err != nil {
+		m.logger.Errorf("[CronManager] 添加任务失败 #%s: %v", taskID, err)
+		return err
+	}
+
+	m.entryMap[taskID] = entryID
+	m.logger.Infof("[CronManager] 已添加调度: %s (#%s) [%s]", name, taskID, task.GetSchedule())
+
+	// 初始触发一次下次运行时间通知
+	go func() {
+		req := &ExecutionRequest{
+			TaskID:  taskID,
+			Name:    name,
+			Type:    TaskTypeCron,
+			UseMise: task.UseMise(),
+		}
+		m.triggerNextRunEvent(taskID, req)
+	}()
+
+	return nil
+}
+
+// RemoveTask 移除计划任务
+func (m *CronManager) RemoveTask(taskID string) {
+	m.mu.Lock()
+	defer m.mu.Unlock()
+
+	if entryID, exists := m.entryMap[taskID]; exists {
+		m.cron.Remove(entryID)
+		delete(m.entryMap, taskID)
+		m.logger.Infof("[CronManager] 任务已移除 #%s", taskID)
+	}
+}
+
+// triggerNextRunEvent 触发下次运行时间更新事件
+func (m *CronManager) triggerNextRunEvent(taskID string, req *ExecutionRequest) {
+	m.mu.RLock()
+	entryID, exists := m.entryMap[taskID]
+	m.mu.RUnlock()
+
+	if !exists {
+		return
+	}
+
+	entry := m.cron.Entry(entryID)
+	if !entry.Next.IsZero() && m.scheduler != nil && m.scheduler.handler != nil {
+		m.scheduler.handler.OnCronNextRun(req, entry.Next)
+	}
+}
+
+// ValidateCron 校验 Cron 表达式
+func (m *CronManager) ValidateCron(expression string) error {
+	expression = strings.TrimSpace(expression)
+	if expression == "" {
+		return fmt.Errorf("cron 表达式不能为空")
+	}
+
+	// 如果不是以 @ 开头的描述符，检查位数
+	if !strings.HasPrefix(expression, "@") {
+		fields := strings.Fields(expression)
+		if len(fields) != 6 {
+			return fmt.Errorf("cron 表达式必须为 6 位 (秒 分 时 日 月 周)")
+		}
+	}
+
+	parser := cron.NewParser(cron.Second | cron.Minute | cron.Hour | cron.Dom | cron.Month | cron.Dow | cron.Descriptor)
+	_, err := parser.Parse(expression)
+	return err
+}
+
+// GetEntry 获取任务详情
+func (m *CronManager) GetEntry(taskID string) (cron.Entry, bool) {
+	m.mu.RLock()
+	defer m.mu.RUnlock()
+
+	entryID, exists := m.entryMap[taskID]
+	if !exists {
+		return cron.Entry{}, false
+	}
+
+	return m.cron.Entry(entryID), true
+}
+
+// GetScheduledCount 获取已调度任务总数
+func (m *CronManager) GetScheduledCount() int {
+	m.mu.RLock()
+	defer m.mu.RUnlock()
+	return len(m.entryMap)
+}