forked from jrallison/go-workers
-
Notifications
You must be signed in to change notification settings - Fork 2
/
enqueue_option.go
54 lines (43 loc) · 1.13 KB
/
enqueue_option.go
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
package workers
import "time"
const (
DEFAULT_MAX_RETRIES = 25
)
type EnqueueOptions interface {
Apply(param *EnqueueParam)
}
type EnqueueOptFunc func(param *EnqueueParam)
func (e EnqueueOptFunc) Apply(param *EnqueueParam) {
e(param)
}
func WithIn(in time.Duration) EnqueueOptFunc {
return (EnqueueOptFunc)(func(param *EnqueueParam) {
param.At = nowToSecondsWithNanoPrecision() + durationToSecondsWithNanoPrecision(in)
})
}
func WithAt(at time.Time) EnqueueOptFunc {
return (EnqueueOptFunc)(func(param *EnqueueParam) {
param.At = timeToSecondsWithNanoPrecision(at)
})
}
func WithRetry() EnqueueOptFunc {
return (EnqueueOptFunc)(func(param *EnqueueParam) {
param.MaxRetries = DEFAULT_MAX_RETRIES
})
}
func WithMaxRetries(maxRetries int) EnqueueOptFunc {
return (EnqueueOptFunc)(func(param *EnqueueParam) {
param.MaxRetries = maxRetries
})
}
func WithJid(jid string) EnqueueOptFunc {
return (EnqueueOptFunc)(func(param *EnqueueParam) {
param.Jid = jid
})
}
func defaultEnqueueOpt(param *EnqueueParam) {
param.At = nowToSecondsWithNanoPrecision()
param.MaxRetries = 0
param.RetryCount = 0
param.Jid = generateJid()
}