e4fee05c96
Update the GO tutorials with PublishWithContext instead of Publish Publish is deprecated. As context, I used: `context.WithTimeout(context.Background(), 5*time.Second)` update the Golang Version to 1.7 Signed-off-by: Gabriele Santomaggio G.santomaggio@gmail.com
96 lines
1.9 KiB
Go
96 lines
1.9 KiB
Go
package main
|
|
|
|
import (
|
|
"context"
|
|
"log"
|
|
"strconv"
|
|
"time"
|
|
|
|
amqp "github.com/rabbitmq/amqp091-go"
|
|
)
|
|
|
|
func failOnError(err error, msg string) {
|
|
if err != nil {
|
|
log.Panicf("%s: %s", msg, err)
|
|
}
|
|
}
|
|
|
|
func fib(n int) int {
|
|
if n == 0 {
|
|
return 0
|
|
} else if n == 1 {
|
|
return 1
|
|
} else {
|
|
return fib(n-1) + fib(n-2)
|
|
}
|
|
}
|
|
|
|
func main() {
|
|
conn, err := amqp.Dial("amqp://guest:guest@localhost:5672/")
|
|
failOnError(err, "Failed to connect to RabbitMQ")
|
|
defer conn.Close()
|
|
|
|
ch, err := conn.Channel()
|
|
failOnError(err, "Failed to open a channel")
|
|
defer ch.Close()
|
|
|
|
q, err := ch.QueueDeclare(
|
|
"rpc_queue", // name
|
|
false, // durable
|
|
false, // delete when unused
|
|
false, // exclusive
|
|
false, // no-wait
|
|
nil, // arguments
|
|
)
|
|
failOnError(err, "Failed to declare a queue")
|
|
|
|
err = ch.Qos(
|
|
1, // prefetch count
|
|
0, // prefetch size
|
|
false, // global
|
|
)
|
|
failOnError(err, "Failed to set QoS")
|
|
|
|
msgs, err := ch.Consume(
|
|
q.Name, // queue
|
|
"", // consumer
|
|
false, // auto-ack
|
|
false, // exclusive
|
|
false, // no-local
|
|
false, // no-wait
|
|
nil, // args
|
|
)
|
|
failOnError(err, "Failed to register a consumer")
|
|
|
|
var forever chan struct{}
|
|
|
|
go func() {
|
|
ctx, cancel := context.WithTimeout(context.Background(), 5*time.Second)
|
|
defer cancel()
|
|
for d := range msgs {
|
|
n, err := strconv.Atoi(string(d.Body))
|
|
failOnError(err, "Failed to convert body to integer")
|
|
|
|
log.Printf(" [.] fib(%d)", n)
|
|
response := fib(n)
|
|
|
|
err = ch.PublishWithContext(ctx,
|
|
"", // exchange
|
|
d.ReplyTo, // routing key
|
|
false, // mandatory
|
|
false, // immediate
|
|
amqp.Publishing{
|
|
ContentType: "text/plain",
|
|
CorrelationId: d.CorrelationId,
|
|
Body: []byte(strconv.Itoa(response)),
|
|
})
|
|
failOnError(err, "Failed to publish a message")
|
|
|
|
d.Ack(false)
|
|
}
|
|
}()
|
|
|
|
log.Printf(" [*] Awaiting RPC requests")
|
|
<-forever
|
|
}
|