Go微服务中用robfig/cron防止多调度及任务并发执行

针对你的两个核心问题（任务串行执行、防止重复创建cron实例），可以通过以下方式优化：

一、确保任务串行执行（前一次完成再触发下一次） #

你当前的代码把整个任务执行都放在互斥锁内，虽然能实现串行，但锁持有时间过长（等于任务执行时间），这不是最优写法。正确的做法是只在修改/检查运行状态时加锁，任务执行逻辑放在锁外：

var isProcessRunning bool
var mu sync.Mutex

func RunProcess() {
    // 加锁检查并标记运行状态
    mu.Lock()
    if isProcessRunning {
        fmt.Println("Already Running...")
        mu.Unlock()
        return
    }
    isProcessRunning = true
    mu.Unlock() // 立即释放锁，避免长时间持有

    // 执行实际任务（锁外执行）
    fmt.Println("Running...")
    time.Sleep(15 * time.Second) // 模拟工作

    // 任务完成后，加锁标记为未运行
    mu.Lock()
    isProcessRunning = false
    mu.Unlock()
}

这样既保证了同一时间只有一个任务实例在运行，又不会让互斥锁长时间占用，避免影响其他可能的锁操作。

二、防止多次调用InitCron创建重复cron实例 #

使用sync.Once可以确保InitCron中的初始化逻辑只执行一次，无论被调用多少次：

var (
    cronInstance *cron.Cron
    initOnce     sync.Once
)

func InitCron() {
    initOnce.Do(func() {
        // 仅执行一次的初始化逻辑
        cronInstance = cron.New(cron.WithSeconds())
        _, err := cronInstance.AddFunc("*/10 * * * * *", RunProcess)
        if err != nil {
            fmt.Println("Error adding cron job:", err)
            return
        }
        cronInstance.Start()
        fmt.Println("Cron scheduler started successfully")
    })

    // 移除原有的select{}，因为如果多次调用InitCron，第一次已经阻塞了，后续调用会卡住
    // 应该把阻塞逻辑放在main函数里
}

func main() {
    InitCron()
    // 阻塞主线程，保持服务运行
    select {}
}

关键说明： #

• sync.Once的Do方法会保证传入的函数只被执行一次，即使InitCron被多次调用，也不会重复创建cron实例和添加任务。
• 把阻塞用的select{}从InitCron移到main函数，避免多次调用InitCron时后续调用被阻塞。

完整优化后的代码 #

package main

import (
    "fmt"
    "sync"
    "time"

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

var (
    isProcessRunning bool
    mu               sync.Mutex
    cronInstance     *cron.Cron
    initOnce         sync.Once
)

func RunProcess() {
    mu.Lock()
    if isProcessRunning {
        fmt.Println("Already Running...")
        mu.Unlock()
        return
    }
    isProcessRunning = true
    mu.Unlock()

    fmt.Println("Running...")
    // 模拟工作
    time.Sleep(15 * time.Second)

    mu.Lock()
    isProcessRunning = false
    mu.Unlock()
}

func InitCron() {
    initOnce.Do(func() {
        cronInstance = cron.New(cron.WithSeconds())
        _, err := cronInstance.AddFunc("*/10 * * * * *", RunProcess)
        if err != nil {
            fmt.Println("Error adding cron job:", err)
            return
        }
        cronInstance.Start()
        fmt.Println("Cron scheduler initialized and started")
    })
}

func main() {
    InitCron()
    // 多次调用也不会重复初始化
    InitCron()
    InitCron()

    // 阻塞主线程
    select {}
}

额外建议 #

• 如果你需要优雅关闭cron任务（比如收到终止信号时），可以监听os.Signal，调用cronInstance.Stop()来停止调度器。
• 对于更复杂的任务管理，可以考虑把任务状态和cron实例封装到一个结构体中，符合Go的编程习惯，避免过多全局变量。

内容的提问来源于stack exchange，提问作者Harry