/
listener.go
82 lines (66 loc) · 1.41 KB
/
listener.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
package nano
import (
"errors"
log "github.com/Sirupsen/logrus"
"github.com/streadway/amqp"
)
type rpcHandler func(string, []byte) (string, []byte)
func rpcListen(dbURI string, name string, handler rpcHandler) error {
log.Debug("[RPC] listenning")
conn, err := amqp.Dial(dbURI)
if err != nil {
return err
}
defer conn.Close()
ch, err := conn.Channel()
if err != nil {
return err
}
defer ch.Close()
q, err := ch.QueueDeclare(
name, // name
false, // durable
false, // delete when usused
false, // exclusive
false, // no-wait
nil, // arguments
)
if err != nil {
return err
}
log.Debugf("AMQP Queue created: %s\n", q.Name)
msgs, err := ch.Consume(
q.Name, // queue
"", // consumer
false, // auto-ack
false, // exclusive
false, // no-local
false, // no-wait
nil, // args
)
if err != nil {
return err
}
for d := range msgs {
go handleReq(ch, d, handler)
}
return errors.New("Connection Lost")
}
func handleReq(ch *amqp.Channel, d amqp.Delivery, handler rpcHandler) {
contentType, body := handler(d.ContentType, d.Body)
err := ch.Publish(
"", // exchange
d.ReplyTo, // routing key
false, // mandatory
false, // immediate
amqp.Publishing{
ContentType: contentType,
CorrelationId: d.CorrelationId,
Body: body,
})
if err != nil {
log.Error("Failed to publish a message: " + err.Error())
return
}
d.Ack(false)
}