go-pkg/asynq
2
0
Fork 0
mirror of https://github.com/hibiken/asynq.git synced 2025-09-16 11:32:26 +08:00
Code Issues Projects Releases Wiki Activity

Define Schedule and RetryLater method for RDB

Browse Source
This commit is contained in:
Ken Hibino
2019-12-04 06:45:30 -08:00
parent 985018e1b5
commit 4531e90b9d
5 changed files with 67 additions and 16 deletions
Download Patch File Download Diff File Expand all files Collapse all files

14
internal/rdb/rdb.go
View File

@@ -173,8 +173,18 @@ func (r *RDB) Done(msg *TaskMessage) error {
return nil
}
// Schedule adds the task to the zset to be processd at the specified time.
func (r *RDB) Schedule(zset string, processAt time.Time, msg *TaskMessage) error {
// Schedule adds the task to the backlog queue to be processed in the future.
func (r *RDB) Schedule(msg *TaskMessage, processAt time.Time) error {
return r.schedule(Scheduled, processAt, msg)
}
// RetryLater adds the task to the backlog queue to be retried in the future.
func (r *RDB) RetryLater(msg *TaskMessage, processAt time.Time) error {
return r.schedule(Retry, processAt, msg)
}
// schedule adds the task to the zset to be processd at the specified time.
func (r *RDB) schedule(zset string, processAt time.Time, msg *TaskMessage) error {
bytes, err := json.Marshal(msg)
if err != nil {
return fmt.Errorf("could not marshal %+v to json: %v", msg, err)