Go 定时任务方法封装

package main
import (
  "fmt"
  "time"
)
func main() {
  // 创建一个定时任务
  task := NewTask(5*time.Second, func() {
    fmt.Println("定时任务执行")
  })
  // 启动定时任务
  task.Start()
  // 等待一段时间
  time.Sleep(10 * time.Second)
  // 停止定时任务
  task.Stop()
}
type Task struct {
  interval time.Duration
  fn       func()
  stopChan chan struct{}
}
func NewTask(interval time.Duration, fn func()) *Task {
  return &Task{
    interval: interval,
    fn:       fn,
    stopChan: make(chan struct{}),
  }
}
func (t *Task) Start() {
  go func() {
    ticker := time.NewTicker(t.interval)
    defer ticker.Stop()
    for {
      select {
      case <-ticker.C:
        t.fn()
      case <-t.stopChan:
        return
      }
    }
  }()
}
func (t *Task) Stop() {
  close(t.stopChan)
}