GoCron - 简单流畅的定时任务调度库

GoCron 是一个轻量级的 Golang 定时任务调度库，它提供了简单易用的 API 来安排周期性任务。下面我将详细介绍如何使用 GoCron 实现定时任务调度。

安装

首先安装 GoCron 库：

go get github.com/go-co-op/gocron

基本使用

1. 创建调度器

package main

import (
	"fmt"
	"time"
	
	"github.com/go-co-op/gocron"
)

func main() {
	// 创建调度器实例
	s := gocron.NewScheduler(time.UTC)
	
	// 添加任务
	_, err := s.Every(1).Seconds().Do(task)
	if err != nil {
		panic(err)
	}
	
	// 启动调度器
	s.StartBlocking()
}

func task() {
	fmt.Println("任务执行:", time.Now().Format("2006-01-02 15:04:05"))
}

2. 多种时间间隔设置

// 每5秒执行一次
s.Every(5).Seconds().Do(task)

// 每1分钟执行一次
s.Every(1).Minute().Do(task)

// 每2小时执行一次
s.Every(2).Hours().Do(task)

// 每天执行一次
s.Every(1).Day().Do(task)

// 每周一执行
s.Every(1).Monday().Do(task)

// 每月1号执行
s.Every(1).Month(1).Do(task)

3. 指定具体时间执行

// 每天10:30执行
s.Every(1).Day().At("10:30").Do(task)

// 每周一10:30执行
s.Every(1).Monday().At("10:30").Do(task)

// 每月1号10:30执行
s.Every(1).Month(1).At("10:30").Do(task)

高级功能

1. 带参数的任务

func greet(name string) {
	fmt.Printf("Hello, %s! Time: %s\n", name, time.Now().Format("15:04:05"))
}

func main() {
	s := gocron.NewScheduler(time.UTC)
	
	// 传递参数给任务
	_, err := s.Every(2).Seconds().Do(greet, "Alice")
	if err != nil {
		panic(err)
	}
	
	s.StartBlocking()
}

2. 任务标签与移除任务

func main() {
	s := gocron.NewScheduler(time.UTC)
	
	// 添加带标签的任务
	_, err := s.Every(1).Second().Tag("task1").Do(task)
	if err != nil {
		panic(err)
	}
	
	// 5秒后移除任务
	time.AfterFunc(5*time.Second, func() {
		err := s.RemoveByTag("task1")
		if err != nil {
			fmt.Println("移除任务失败:", err)
		} else {
			fmt.Println("任务已移除")
		}
	})
	
	s.StartBlocking()
}

3. 单例模式（防止任务重叠）

func main() {
	s := gocron.NewScheduler(time.UTC)
	
	// 设置单例模式 - 如果前一个任务还在运行，则跳过本次执行
	_, err := s.Every(1).Second().SingletonMode().Do(longRunningTask)
	if err != nil {
		panic(err)
	}
	
	s.StartBlocking()
}

func longRunningTask() {
	fmt.Println("长任务开始:", time.Now().Format("15:04:05"))
	time.Sleep(3 * time.Second)
	fmt.Println("长任务结束:", time.Now().Format("15:04:05"))
}

4. 并发限制

func main() {
	s := gocron.NewScheduler(time.UTC)
	
	// 设置最大并发任务数为2
	s.SetMaxConcurrentJobs(2, gocron.RescheduleMode)
	
	// 添加多个任务
	for i := 0; i < 5; i++ {
		_, err := s.Every(1).Second().Do(taskWithID, i)
		if err != nil {
			panic(err)
		}
	}
	
	s.StartBlocking()
}

func taskWithID(id int) {
	fmt.Printf("任务 %d 开始: %s\n", id, time.Now().Format("15:04:05"))
	time.Sleep(3 * time.Second)
	fmt.Printf("任务 %d 结束: %s\n", id, time.Now().Format("15:04:05"))
}

5. Cron 表达式支持

func main() {
	s := gocron.NewScheduler(time.UTC)
	
	// 使用标准cron表达式
	_, err := s.Cron("*/5 * * * *").Do(task) // 每5分钟执行一次
	if err != nil {
		panic(err)
	}
	
	s.StartBlocking()
}

实际应用示例

package main

import (
	"fmt"
	"time"
	
	"github.com/go-co-op/gocron"
)

func main() {
	// 创建调度器
	scheduler := gocron.NewScheduler(time.UTC)
	
	// 添加多个任务
	_, _ = scheduler.Every(5).Seconds().Do(monitorSystem)
	_, _ = scheduler.Every(1).Hour().Do(cleanTempFiles)
	_, _ = scheduler.Every(1).Day().At("02:00").Do(backupDatabase)
	
	// 启动调度器
	fmt.Println("定时任务调度器启动...")
	scheduler.StartBlocking()
}

func monitorSystem() {
	fmt.Printf("[%s] 监控系统状态...\n", time.Now().Format("2006-01-02 15:04:05"))
	// 实际监控逻辑...
}

func cleanTempFiles() {
	fmt.Printf("[%s] 清理临时文件...\n", time.Now().Format("2006-01-02 15:04:05"))
	// 实际清理逻辑...
}

func backupDatabase() {
	fmt.Printf("[%s] 数据库备份开始...\n", time.Now().Format("2006-01-02 15:04:05"))
	// 实际备份逻辑...
	time.Sleep(10 * time.Second) // 模拟耗时操作
	fmt.Printf("[%s] 数据库备份完成\n", time.Now().Format("2006-01-02 15:04:05"))
}

注意事项

1. 默认情况下，GoCron 使用 UTC 时区，可以通过 NewScheduler 指定本地时区
2. 长时间运行的任务可能会影响调度精度，考虑使用单例模式
3. 在生产环境中，建议添加适当的错误处理和恢复机制
4. 对于分布式系统，需要考虑分布式锁机制

GoCron 是一个简单但功能强大的定时任务调度库，适合大多数定时任务场景。对于更复杂的需求，可以考虑结合其他工具如消息队列或专门的作业调度系统。