创建消费者
package service
import (
"encoding/json"
"log"
"os"
"strings"
"sync"
"time"
"yy-data-processing/common/config"
"yy-data-processing/common/database"
"yy-data-processing/model"
)
func Consumer() {
// 声明队列,没有则创建
// 队列名称、是否持久化、所有消费者与队列断开时是否自动删除队列、是否独享(不同连接的channel能否使用该队列)
_, err := database.RabbitChannel.QueueDeclare(config.Config.QueueName, true, false, false, false, nil)
if err != nil {
log.Printf("声明队列 %v 失败, error: %v", config.Config.QueueName, err)
panic(err)
}
err = database.RabbitChannel.Qos(
1, // prefetch count 服务器将在收到确认之前将那么多消息传递给消费者。
0, // prefetch size 服务器将尝试在收到消费者的确认之前至少将那么多字节的交付保持刷新到网络
false, // 当 global 为 true 时,这些 Qos 设置适用于同一连接上所有通道上的所有现有和未来消费者。当为 false 时,Channel.Qos 设置将应用于此频道上的所有现有和未来消费者
)
if err != nil {
log.Printf("rabbitmq设置Qos失败, error: %v", err)
}
// 队列名称、consumer、auto-ack、是否独享
// deliveries是一个管道,有消息到队列,就会消费,消费者的消息只需要从deliveries这个管道获取
deliveries, err := database.RabbitChannel.Consume(config.Config.QueueName, "", false, false, false, false, nil)
if err != nil {
log.Printf("从队列 %v 获取数据失败, error: %v", config.Config.QueueName, err)
} else {
log.Println("从消费队列获取任务成功")
}
// 阻塞住
for {
select {
case message := <-deliveries:
closed := database.CheckRabbitClosed(*database.RabbitChannel)
if closed == 1 { // channel 已关闭,重连一下
database.InitRabbitmq()
err = database.RabbitChannel.Qos(1, 0, false)
if err != nil {
log.Printf("rabbitmq重连后设置Qos失败, error: %v", err)
}
} else {
msgData := string(message.Body)
request := model.Request{}
err := json.Unmarshal([]byte(msgData), &request)
if err != nil {
log.Printf("解析rabbitmq数据 %v 失败, error: %v", msgData, err)
} else {
// TODO...
// 处理逻辑
// 处理完毕手动ACK
message.Ack(true)
}
}
}
}
}