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
65
66
67
68
|
package amqp
import (
"fmt"
"github.com/spiral/roadrunner/v2/plugins/jobs/structs"
"github.com/streadway/amqp"
)
const (
rrID string = "rr-id"
rrJob string = "rr-job"
rrAttempt string = "rr-attempt"
rrMaxAttempts string = "rr-max_attempts"
rrTimeout string = "rr-timeout"
rrDelay string = "rr-delay"
rrRetryDelay string = "rr-retry_delay"
)
// pack job metadata into headers
func pack(id string, attempt uint64, j *structs.Job) amqp.Table {
return amqp.Table{
rrID: id,
rrJob: j.Job,
rrAttempt: attempt,
rrMaxAttempts: j.Options.Attempts,
rrTimeout: j.Options.Timeout,
rrDelay: j.Options.Delay,
rrRetryDelay: j.Options.RetryDelay,
}
}
// unpack restores jobs.Options
func unpack(d amqp.Delivery) (id string, attempt int, j *structs.Job, err error) { //nolint:deadcode,unused
j = &structs.Job{Payload: string(d.Body), Options: &structs.Options{}}
if _, ok := d.Headers[rrID].(string); !ok {
return "", 0, nil, fmt.Errorf("missing header `%s`", rrID)
}
if _, ok := d.Headers[rrAttempt].(uint64); !ok {
return "", 0, nil, fmt.Errorf("missing header `%s`", rrAttempt)
}
if _, ok := d.Headers[rrJob].(string); !ok {
return "", 0, nil, fmt.Errorf("missing header `%s`", rrJob)
}
j.Job = d.Headers[rrJob].(string)
if _, ok := d.Headers[rrMaxAttempts].(uint64); ok {
j.Options.Attempts = d.Headers[rrMaxAttempts].(uint64)
}
if _, ok := d.Headers[rrTimeout].(uint64); ok {
j.Options.Timeout = d.Headers[rrTimeout].(uint64)
}
if _, ok := d.Headers[rrDelay].(uint64); ok {
j.Options.Delay = d.Headers[rrDelay].(uint64)
}
if _, ok := d.Headers[rrRetryDelay].(uint64); ok {
j.Options.RetryDelay = d.Headers[rrRetryDelay].(uint64)
}
return d.Headers[rrID].(string), int(d.Headers[rrAttempt].(uint64)), j, nil
}
|
