Standard library implementation of gocron function (testable)

php editor Baicao introduces you to a function called gocron, which is a standard library implementation and has testable features. The gocron function is designed to provide an easy way to create and manage scheduled tasks. It can help developers easily write and schedule multiple scheduled tasks, and also provides flexible testing functions to facilitate developers to conduct unit testing and functional verification. Whether in a development environment or a production environment, the gocron function is a powerful and reliable tool that can be used to implement various complex timing task requirements.

Question content

I need to run a job at certain intervals (hit the rest endpoint and then send a message to the queue). Currently I have this function which uses gocron - https://github.com/go-co-op/gocron

But there is currently no mechanism to stop this feature, so I can't test it cleanly.

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
}

It does what is asked and runs the parameterized function at the required intervals, but I'm sure there must be a cleaner solution - probably the standard library.

Solution

You can just return the .stop function:

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
}

Then in your test you can do the following:

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")
}

The above is the detailed content of Standard library implementation of gocron function (testable). For more information, please follow other related articles on the PHP Chinese website!