Merge pull request #2 from Selly-Modules/update-retry-config

update retry config
This commit is contained in:
Sinh Luu 2022-08-10 17:18:17 +07:00 committed by GitHub
commit 21c2ce0592
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
2 changed files with 16 additions and 8 deletions

View File

@ -1,6 +1,10 @@
package queue
import "time"
import (
"time"
"github.com/hibiken/asynq"
)
// Config ...
type Config struct {
@ -14,6 +18,7 @@ type Config struct {
Priority ConfigPriority
TaskTimeout time.Duration
RetryDelayFunc asynq.RetryDelayFunc
}
// ConfigRedis ...

View File

@ -50,6 +50,13 @@ func initServer(redisConn asynq.RedisClientOpt, cfg Config) *asynq.ServeMux {
cfg.Priority.Low = 1
cfg.Priority.StrictMode = false
}
retryDelayFunc := cfg.RetryDelayFunc
if retryDelayFunc == nil {
// Default delay in 10s
retryDelayFunc = func(n int, e error, t *asynq.Task) time.Duration {
return 10 * time.Second
}
}
// Init server
server := asynq.NewServer(redisConn, asynq.Config{
@ -61,11 +68,7 @@ func initServer(redisConn asynq.RedisClientOpt, cfg Config) *asynq.ServeMux {
},
StrictPriority: cfg.Priority.StrictMode,
// TODO:
// This is default option, retry after 10s, will add to config later
RetryDelayFunc: func(n int, e error, t *asynq.Task) time.Duration {
return 10 * time.Second
},
RetryDelayFunc: retryDelayFunc,
})
// Init mux server