Golang实现定时任务

这篇博客展示了如何在Go语言中利用第三方库`github.com/robfig/cron/v3`创建一个每5秒执行一次的任务。主要代码包括初始化cron实例,添加定时任务函数`timer`并启动调度器。`timer`函数仅打印一句日志,说明任务执行情况。
package main

import (
	"github.com/robfig/cron/v3"
	"log"
)

func main() {
	c := cron.New()
	_, _ = c.AddFunc("@every 5s", timer)
	c.Start()
	select {}
}

func timer() {
	log.Println("每5s执行一次")
}

实现一个**基于 Golang 的分布式定时任务调度系统**,我们需要解决以下几个核心问题: 1. **定时触发任务(类似 Cron)** 2. **支持多节点部署(分布式)** 3. **避免多个节点重复执行同一个任务(分布式锁)** 4. **任务的持久化与状态管理(使用数据库或 Redis)** 5. **可扩展性、高可用性** 下面是一个使用 `Golang + Redis(Redlock 分布式锁)+ Cron 表达式解析` 实现的简单但完整的分布式定时任务调度系统。 --- ### ✅ 核心组件说明 - `robfig/cron`: 用于解析和调度 cron 表达式 - `Redis`: 存储任务元数据并实现分布式锁(防重复执行) - 每个节点独立运行,通过抢锁决定谁执行任务 --- ## 🚀 完整代码实现 ```go // main.go package main import ( "context" "fmt" "log" "time" "github.com/go-redis/redis/v8" "github.com/robfig/cron/v3" ) var rdb *redis.Client var ctx = context.Background() const ( LockKeyPrefix = "scheduler:lock:" LockTimeout = 30 * time.Second // 锁超时时间 TaskExecutionTime = 10 * time.Second // 模拟任务执行时间 ) type Task struct { ID string Spec string // cron 表达式 Command string // 要执行的任务命令(比如调用 API、打印日志等) NodeID string // 当前节点 ID } func init() { // 初始化 Redis 客户端 rdb = redis.NewClient(&redis.Options{ Addr: "localhost:6379", Password: "", DB: 0, }) // 测试连接 if _, err := rdb.Ping(ctx).Result(); err != nil { log.Fatal("无法连接到 Redis:", err) } } // 尝试获取分布式锁 func tryAcquireLock(taskID string, nodeID string) bool { success, err := rdb.SetNX(ctx, LockKeyPrefix+taskID, nodeID, LockTimeout).Result() if err != nil { log.Printf("获取锁时出错: %v", err) return false } return success } // 释放分布式锁 func releaseLock(taskID string) { rdb.Del(ctx, LockKeyPrefix+taskID) } // 执行具体任务逻辑 func executeTask(task Task) { log.Printf("[开始] 节点 %s 正在执行任务: %s (ID: %s)", task.NodeID, task.Command, task.ID) time.Sleep(TaskExecutionTime) // 模拟任务耗时 log.Printf("[完成] 节点 %s 成功完成任务: %s", task.NodeID, task.Command) releaseLock(task.ID) } // 启动调度器 func startScheduler(nodeID string) { log.Printf("节点 %s 已启动调度器...", nodeID) c := cron.New(cron.WithSeconds()) // 支持秒级 cron tasks := []Task{ {ID: "task1", Spec: "*/10 * * * * *", Command: "发送邮件通知", NodeID: nodeID}, {ID: "task2", Spec: "*/15 * * * * *", Command: "同步用户数据", NodeID: nodeID}, {ID: "task3", Spec: "0 */1 * * * *", Command: "生成报表", NodeID: nodeID}, } for _, task := range tasks { _, err := c.AddFunc(task.Spec, func() { log.Printf("触发任务: %s (ID: %s)", task.Command, task.ID) if tryAcquireLock(task.ID, task.NodeID) { go executeTask(task) } else { log.Printf("任务 %s 已被其他节点锁定,跳过执行。", task.ID) } }) if err != nil { log.Printf("添加任务失败: %v", err) } } c.Start() defer c.Stop() // 阻塞主程序 select {} } func main() { nodeID := fmt.Sprintf("node-%d", time.Now().Unix()%1000) // 简单模拟不同节点 startScheduler(nodeID) } ``` --- ## 🔧 运行方式 1. 安装依赖: ```bash go mod init scheduler go get github.com/go-redis/redis/v8 go get github.com/robfig/cron/v3 ``` 2. 启动多个实例(模拟分布式环境): ```bash # 终端1 go run main.go # 终端2 go run main.go ``` 3. 观察输出:你会发现相同任务只由其中一个节点执行(因为抢到了锁) --- ## 💡 原理解释 | 组件 | 作用 | |------|------| | `cron/v3` | 解析 `* * * * * *` 格式的定时规则,并在对应时间触发函数 | | `Redis SetNX` | 实现分布式互斥锁,确保同一时刻只有一个节点能执行任务 | | `LockKeyPrefix + taskID` | 每个任务对应唯一锁 key | | `LockTimeout` | 防止死锁(如节点崩溃未释放锁) | > ⚠️ 注意:生产环境中建议使用更健壮的分布式锁机制,例如 [Redlock 算法](https://redis.io/docs/manual/patterns/distributed-locks/) 或使用 etcd/ZooKeeper。 --- ## ✅ 特性总结 - ✅ 支持标准 Cron 表达式(含秒) - ✅ 多节点部署不重复执行 - ✅ 使用 Redis 实现轻量级锁 - ✅ 易于扩展为 Web API 管理任务 - ✅ 可持久化任务到数据库(后续可加) --- ## 🛠️ 可改进方向(进阶) 1. **任务持久化**:将任务存储在 MySQL/PostgreSQL 中,支持动态增删改查。 2. **Web UI / API 接口**:提供 HTTP 接口来管理任务。 3. **任务失败重试机制** 4. **日志记录与监控告警** 5. **使用 etcd 替代 Redis 实现 leader election** 6. **任务分片与负载均衡** --- ###
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值