使用标准库实现的可测试的定时任务函数

要创建可测试的定时任务函数，可以利用标准库实现。使用gocron库存在一些局限性，如缺乏停止机制，导致测试困难。为了解决这个问题，可以修改RunSchedule函数，使其返回一个停止函数。通过在测试中调用该停止函数，可以控制任务的执行并验证函数的行为，从而实现可测试的定时任务功能。

我需要按一定的时间间隔运行作业（点击休息端点，然后将消息发送到队列）。目前我有这个功能，它使用 gocron - https://github.com/go-co-op/gocron

但是目前没有停止该功能的机制，因此我无法干净地测试它。

func RunSchedule(cronExpression string, myFunction func()) error {
    scheduler := gocron.NewScheduler(time.UTC)
    _, err := scheduler.Cron(cronExpression).StartImmediately().Do(myFunction)
    if err != nil {
        return err
    }
    scheduler.StartBlocking()
    return nil
}

它按要求执行，并以所需的时间间隔运行参数化函数，但我确信必须有一个更清晰的解决方案 - 可能是标准库。

正确答案

您可以只返回 .stop 函数：

func runschedule(cronexpression string, myfunction func()) (stop func(), err error) {
    scheduler := gocron.newscheduler(time.utc)
    _, err = scheduler.cron(cronexpression).startimmediately().do(myfunction)
    if err != nil {
        return nil, err
    }
    go func() {
        scheduler.startblocking()
    }()

    return scheduler.stop, nil
}

然后在您的测试中您可以执行以下操作：

called := 0
stop, err := RunSchedule("some-schedule", func() {
    called++
})
time.Sleep(time.Second * 1) // whatever you need to do 
stop()
if called != 1 {
    t.Fail("called should be 1")
}

终于介绍完啦！小伙伴们，这篇关于《使用标准库实现的可测试的定时任务函数》的介绍应该让你收获多多了吧！欢迎大家收藏或分享给更多需要学习的朋友吧~golang学习网公众号也会发布Golang相关知识，快来关注吧！