-
Notifications
You must be signed in to change notification settings - Fork 0
/
rabbitMQ-antipode.go
130 lines (109 loc) · 2.65 KB
/
rabbitMQ-antipode.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
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
129
130
package antipode
import (
"context"
"encoding/json"
"fmt"
"log"
amqp "github.com/rabbitmq/amqp091-go"
)
type RabbitMQ struct {
connection *amqp.Connection
queue string
}
// How can I close the connection?
// É má prática manter a conexão sempre aberta?
func CreateRabbitMQ(rabbit_host string, rabbit_port string, rabbit_user string, rabbit_password string, queue string) RabbitMQ {
conn, err := amqp.Dial("amqp://" + rabbit_user + ":" + rabbit_password + "@" + rabbit_host + ":" + rabbit_port + "/")
if err != nil {
fmt.Println(err)
return RabbitMQ{}
}
//defer conn.Close()
return RabbitMQ{conn, queue}
}
func (r RabbitMQ) write(ctx context.Context, _ string, key string, obj AntiObj) error {
jsonAntiObj, err := json.Marshal(obj)
if err != nil {
return err
}
channel, err := r.connection.Channel()
if err != nil {
return err
}
defer channel.Close()
queue, err := channel.QueueDeclare(
r.queue, // Queue name
false, // Durable
false, // Delete when unused
false, // Exclusive
false, // No-wait
nil, // Arguments
)
if err != nil {
return err
}
err = channel.PublishWithContext(ctx,
"", // exchange
queue.Name, // routing key
false, // mandatory
false, // immediate
amqp.Publishing{
ContentType: "application/json",
Body: jsonAntiObj,
})
if err != nil {
return err
}
return err
}
func (r RabbitMQ) read(ctx context.Context, _ string, key string) (AntiObj, error) {
channel, err := r.connection.Channel()
if err != nil {
return AntiObj{}, err
}
//defer channel.Close()
queue, err := channel.QueueDeclare(
r.queue, // Queue name
false, // Durable
false, // Delete when unused
false, // Exclusive
false, // No-wait
nil, // Arguments
)
if err != nil {
return AntiObj{}, err
}
// Consume one message from the queue
err = channel.Qos(1, 0, false)
if err != nil {
return AntiObj{}, err
}
msgs, err := channel.Consume(
queue.Name, // Queue
"", // Consumer tag
false, // Auto-ack
false, // Exclusive
false, // No local
false, // No wait
nil, // Args
)
if err != nil {
log.Fatalf("Failed to consume messages from queue: %v", err)
}
// Wait for the first message to arrive and send an acknowledgement
msg := <-msgs
channel.Close()
err = msg.Ack(true)
if err != nil {
return AntiObj{}, err
}
var antiObj AntiObj
err = json.Unmarshal(msg.Body, &antiObj)
if err != nil {
return AntiObj{}, fmt.Errorf("Failed to unmarshal JSON: %v", err)
}
return antiObj, err
}
func (r RabbitMQ) barrier(ctx context.Context, lineage []WriteIdentifier, datastoreID string) error {
return nil
}