golang 框架如何实现分布式定时任务？-Golang-PHP中文网

golang 框架如何实现分布式定时任务？

王林

发布： 2024-08-08 11:21:03

原创

844人浏览过

在 go 中实现分布式定时任务，可以使用框架 cron、viper、celery 和 quartz。cron 基于 cron 表达式定义任务，viper 可实现动态加载表达式。celery 是一款任务队列，支持定时任务。quartz 是一款企业级调度框架，具有强大功能。实战案例使用 cron 框架，每天凌晨 3 点执行发送邮件任务。viper 可用于从配置文件中动态加载 cron 表达式，简化任务更新。

golang 框架如何实现分布式定时任务？

Go 框架中的分布式定时任务实现

在分布式系统中，安排和执行定时任务至关重要，尤其是对于需要在特定时间点或定期执行的任务。本文将介绍 Go 中实现分布式定时任务的常用框架及实战案例。

常用的 Go 定时任务框架

立即学习“go语言免费学习笔记（深入）”；

cron：一个简单易用的框架，基于 cron 表达式定义定时任务，使用 Goroutine 执行任务。
Viper：一个用于读取配置和环境变量的框架，可以结合内置的 Watcher 实现动态加载 cron 表达式。
Celery：一个分布式的任务队列，支持多种语言和任务类型，包括定时任务。
Quartz：一个企业级的定时任务调度框架，具有强大的功能和可扩展性。

实战案例

假设我们想要构建一个每天凌晨 3 点发送邮件提醒的定时任务。使用 cron 框架，可以这样实现：

蓝色的商业服务企业网站UI模板

基于Bootstrap框架实现的通用的商业公司网站设计模板。适用于商业咨询公司、互联网商务公司网站。响应式的布局兼容任何大小的设备。

189

查看详情

import (
    "fmt"
    "time"

    "github.com/robfig/cron/v3"
)

func main() {
    // 创建一个 cron 实例
    c := cron.New()

    // 定义一个每天凌晨 3 点执行的任务
    _, err := c.AddFunc("@daily 03:00", func() {
        // 编写执行任务的代码，例如发送邮件
        fmt.Println("任务执行成功！")
    })
    if err != nil {
        // 处理错误
    }

    // 启动 cron 服务
    c.Start()

    // 保持 main 函数运行，直到收到终止信号（例如 Ctrl+C）
    ch := make(chan struct{})
    <-ch
}

登录后复制

运行此程序，cron 将在每天凌晨 3 点执行任务。要停止 cron，只需向 ch 通道发送一个信号即可。

动态加载 cron 表达式

使用 Viper，可以动态加载 cron 表达式，例如从配置文件中读取：

import (
    "fmt"
    "time"

    "github.com/spf13/viper"
    "github.com/robfig/cron/v3"
)

func main() {
    // 从配置文件中读取 cron 表达式
    viper.SetConfigFile("config.yaml")
    viper.SetConfigType("yaml")
    viper.ReadInConfig()
    cronExpr := viper.GetString("cron.expression")

    // 创建一个 cron 实例
    c := cron.New()

    // 定义一个使用动态 cron 表达式的任务
    _, err := c.AddFunc(cronExpr, func() {
        fmt.Println("任务执行成功！")
    })
    if err != nil {
        // 处理错误
    }

    // 启动 cron 服务
    c.Start()

    // 保持 main 函数运行，直到收到终止信号（例如 Ctrl+C）
    ch := make(chan struct{})
    <-ch
}

登录后复制

在 config.yaml 文件中，可以设置 cron 表达式：