71 lines
1.4 KiB
Go
71 lines
1.4 KiB
Go
package main
|
|
|
|
import (
|
|
"bytes"
|
|
"fmt"
|
|
"github.com/streadway/amqp"
|
|
"log"
|
|
"time"
|
|
)
|
|
|
|
func failOnError(err error, msg string) {
|
|
if err != nil {
|
|
log.Fatalf("%s: %s", msg, err)
|
|
panic(fmt.Sprintf("%s: %s", msg, err))
|
|
}
|
|
}
|
|
|
|
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(
|
|
"task_queue", // name
|
|
true, // durable
|
|
false, // delete when unused
|
|
false, // exclusive
|
|
false, // no-wait
|
|
nil, // arguments
|
|
)
|
|
failOnError(err, "Failed to declare a queue")
|
|
|
|
err = ch.Qos(
|
|
3, // 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")
|
|
|
|
forever := make(chan bool)
|
|
|
|
go func() {
|
|
for d := range msgs {
|
|
log.Printf("Received a message: %s", d.Body)
|
|
d.Ack(false)
|
|
dot_count := bytes.Count(d.Body, []byte("."))
|
|
t := time.Duration(dot_count)
|
|
time.Sleep(t * time.Second)
|
|
log.Printf("Done")
|
|
}
|
|
}()
|
|
|
|
log.Printf(" [*] Waiting for messages. To exit press CTRL+C")
|
|
<-forever
|
|
}
|