-
Notifications
You must be signed in to change notification settings - Fork 0
/
rabbit.go
71 lines (57 loc) · 1.39 KB
/
rabbit.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
65
66
67
68
69
70
71
package bank
import (
"encoding/json"
"github.com/jonsch318/royalafg/pkg/config"
"github.com/jonsch318/royalafg/pkg/errors"
"github.com/spf13/viper"
"github.com/streadway/amqp"
)
type RabbitBankConnection struct {
conn *amqp.Connection
ch *amqp.Channel
}
func NewRabbitBankConnection(url string) (*RabbitBankConnection, error) {
conn, err := amqp.Dial(url)
if err != nil {
return nil, err
}
ch, err := conn.Channel()
if err != nil {
return nil, err
}
if err := ch.ExchangeDeclare(viper.GetString(config.RabbitExchange), "direct", true, false, false, false, nil); err != nil {
return nil, err
}
return &RabbitBankConnection{
conn: conn,
ch: ch,
}, nil
}
func (r *RabbitBankConnection) publishCommand(commandType string, body []byte) error {
if body == nil {
return &errors.BodyNullError{}
}
headers := make(map[string]interface{})
headers["CommandType"] = commandType
if err := r.ch.Publish(
viper.GetString(config.RabbitExchange),
viper.GetString(config.RabbitBankQueue),
false,
false,
amqp.Publishing{
Headers: headers,
ContentType: "application/json",
Body: body,
DeliveryMode: amqp.Transient,
}); err != nil {
return err
}
return nil
}
func (r *RabbitBankConnection) PublishCommand(command *Command) error {
buf, err := json.Marshal(command)
if err != nil {
return err
}
return r.publishCommand(command.CommandType, buf)
}