2021-10-18 07:16:42 +00:00
|
|
|
package queue
|
|
|
|
|
|
|
|
import (
|
|
|
|
"github.com/hibiken/asynq"
|
|
|
|
)
|
|
|
|
|
|
|
|
// RunTask ...
|
2021-11-30 15:21:26 +00:00
|
|
|
func (i Instance) RunTask(typename string, payload []byte, retryTimes int) (*asynq.TaskInfo, error) {
|
2021-10-18 07:16:42 +00:00
|
|
|
// Create task and options
|
|
|
|
task := asynq.NewTask(typename, payload)
|
|
|
|
options := make([]asynq.Option, 0)
|
|
|
|
|
|
|
|
// Retry times
|
|
|
|
if retryTimes < 0 {
|
|
|
|
retryTimes = 0
|
|
|
|
}
|
|
|
|
options = append(options, asynq.MaxRetry(retryTimes))
|
|
|
|
|
|
|
|
// Enqueue task
|
2021-11-30 15:21:26 +00:00
|
|
|
return i.Client.Enqueue(task, options...)
|
2021-10-18 07:16:42 +00:00
|
|
|
}
|
|
|
|
|
|
|
|
// ScheduledTask create new task and run at specific time
|
|
|
|
// cronSpec follow cron expression
|
|
|
|
// https://www.freeformatter.com/cron-expression-generator-quartz.html
|
2021-11-30 15:21:26 +00:00
|
|
|
func (i Instance) ScheduledTask(typename string, payload []byte, cronSpec string) (string, error) {
|
2021-10-18 07:16:42 +00:00
|
|
|
// Create task and options
|
|
|
|
task := asynq.NewTask(typename, payload)
|
|
|
|
|
|
|
|
// TODO: Support options later
|
|
|
|
// options := make([]asynq.Option, 0)
|
|
|
|
|
2021-11-30 15:21:26 +00:00
|
|
|
return i.Scheduler.Register(cronSpec, task)
|
2021-10-18 07:16:42 +00:00
|
|
|
}
|