Golang任务调度实现方法详解

type Task struct {
    ID       string
    Run      func()           // 任务函数
    Time     time.Time        // 执行时间
    Interval time.Duration    // 周期间隔（0表示只执行一次）
    Cancel   chan bool        // 取消信号
}

func (s *Scheduler) start() {
    ticker := time.NewTicker(100 * time.Millisecond)
    defer ticker.Stop()
for {
    select {
    case task := <-s.add:
        s.tasks[task.ID] = task
        s.queue = append(s.queue, task)
        sortTasksByTime(s.queue) // 按时间排序

    case id := <-s.remove:
        delete(s.tasks, id)
        s.queue = filterQueue(s.queue, id)

    case <-ticker.C:
        now := time.Now()
        var i int
        for i < len(s.queue) {
            task := s.queue[i]
            if task.Time.After(now) {
                break
            }
            go func(t *Task) {
                t.Run()
                if t.Interval > 0 {
                    t.Time = t.Time.Add(t.Interval)
                    s.add <- t
                }
            }(task)
            delete(s.tasks, task.ID)
            s.queue = append(s.queue[:i], s.queue[i+1:]...)
        }

    case <-s.stop:
        return
    }
}
}

func main() {
    sched := NewScheduler()
// 一次性任务：5秒后执行
sched.add <- &Task{
    ID:   "once",
    Run:  func() { fmt.Println("One-time task executed") },
    Time: time.Now().Add(5 * time.Second),
}

// 周期任务：每2秒打印一次
sched.add <- &Task{
    ID:       "periodic",
    Run:      func() { fmt.Println("Tick", time.Now()) },
    Time:     time.Now().Add(2 * time.Second),
    Interval: 2 * time.Second,
}

// 运行一段时间后退出
time.Sleep(10 * time.Second)
}