/
enqueue.go
64 lines (56 loc) · 1.42 KB
/
enqueue.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
55
56
57
58
59
60
61
62
63
64
package workers
import (
"crypto/rand"
"encoding/json"
"fmt"
"io"
"time"
)
type EnqueueData struct {
Queue string `json:"queue,omitempty"`
Class string `json:"class"`
Args interface{} `json:"args"`
Jid string `json:"jid"`
EnqueuedAt float64 `json:"enqueued_at"`
EnqueueOptions
}
type EnqueueOptions struct {
RetryCount int `json:"retry_count,omitempty"`
Retry bool `json:"retry,omitempty"`
At float64 `json:"at,omitempty"`
}
func generateJid() string {
// Return 12 random bytes as 24 character hex
b := make([]byte, 12)
_, err := io.ReadFull(rand.Reader, b)
if err != nil {
return ""
}
return fmt.Sprintf("%x", b)
}
func Enqueue(queue, class string, args interface{}) error {
return EnqueueWithOptions(queue, class, args, EnqueueOptions{})
}
func EnqueueWithOptions(queue, class string, args interface{}, opts EnqueueOptions) error {
conn := Config.Pool.Get()
defer conn.Close()
data := EnqueueData{
Queue: queue,
Class: class,
Args: args,
Jid: generateJid(),
EnqueuedAt: float64(time.Now().UnixNano()) / 1000000000,
EnqueueOptions: opts,
}
bytes, err := json.Marshal(data)
if err != nil {
return err
}
_, err = conn.Do("sadd", Config.namespace+"queues", queue)
if err != nil {
return err
}
queue = Config.namespace + "queue:" + queue
_, err = conn.Do("rpush", queue, bytes)
return err
}