//PublishMessage creates a new task to a queue
func (mq *MessageQueue) PublishMessage(message Message, routingKey string, corrID string, replyTo string) {
bodyJSON, err := json.Marshal(message)
util.FailOnError(err, "Failed to parse message to JSON")
err = mq.channel.Publish(
exchangeName, // exchange
routingKey, // routing key
false, // mandatory
false, // immediate
amqp.Publishing{
ContentType: "application/json",
CorrelationId: corrID,
ReplyTo: replyTo,
Body: bodyJSON,
})
util.FailOnError(err, "Failed to publish a message")
}
//NewQueue creates a new message queue and bind to the exchange
func (mq *MessageQueue) NewQueue(name string, routingKey string) {
q, err := mq.channel.QueueDeclare(
name, // name
false, // durable
false, // delete when usused
true, // exclusive
false, // noWait
nil, // arguments
)
util.FailOnError(err, "Failed to create a queue: "+name)
err = mq.channel.QueueBind(
q.Name, // queue name
routingKey, // routing key
exchangeName, // exchange
false,
nil)
util.FailOnError(err, "Failed to bind a queue")
}
//NewMQ creates a new message queue object
func NewMQ() *MessageQueue {
conn, err := amqp.Dial("amqp://*****:*****@localhost:5672/")
util.FailOnError(err, "Failed to connect to RabbitMQ")
ch, err := conn.Channel()
util.FailOnError(err, "Failed to open a channel")
err = ch.ExchangeDeclare(
exchangeName, // name
"direct", // type
true, // durable
false, // auto-deleted
false, // internal
false, // no-wait
nil, // arguments
)
util.FailOnError(err, "Failed to declare an exchange")
return &MessageQueue{
channel: ch,
}
}
//GetMessages sent to the queue
func (mq *MessageQueue) GetMessages(queueName string) <-chan amqp.Delivery {
msgs, err := mq.channel.Consume(
queueName, // queue
"", // consumer
true, // auto-ack
false, // exclusive
false, // no-local
false, // no-wait
nil, // args
)
util.FailOnError(err, "Failed to register a consumer")
return msgs
}
//LoadJSON load a Trip with a JSON string
func (t *Trip) LoadJSON(tripJSON string) {
err := json.Unmarshal([]byte(tripJSON), &t)
util.FailOnError(err, "Failed to load Trip from JSON")
}
//ToJSON returns a JSON string with the Trip
func (t *Trip) ToJSON() string {
tripJSON, err := json.Marshal(t)
util.FailOnError(err, "Failed to convert Trip Struct to JSON")
return string(tripJSON)
}
