65 lines
1.8 KiB
Go
65 lines
1.8 KiB
Go
|
package mqproducer
|
|||
|
|
|
|||
|
|
import (
|
|||
|
|
"context"
|
|||
|
|
"encoding/json"
|
|||
|
|
"github.com/apache/rocketmq-client-go/v2"
|
|||
|
|
"github.com/apache/rocketmq-client-go/v2/primitive"
|
|||
|
|
"github.com/apache/rocketmq-client-go/v2/producer"
|
|||
|
|
"log"
|
|||
|
|
"sync"
|
|||
|
|
"toutoukan/model/usermodel/userOrder"
|
|||
|
|
)
|
|||
|
|
|
|||
|
|
var normalProducer rocketmq.Producer
|
|||
|
|
var producerInitOnce sync.Once // 用于保证生产者只初始化一次
|
|||
|
|
|
|||
|
|
// 发送普通消息
|
|||
|
|
func SendNormalMessage(OrderId string, GoodsID string, UserID int) error {
|
|||
|
|
var initErr error
|
|||
|
|
// 使用 sync.Once 保证生产者只初始化一次
|
|||
|
|
producerInitOnce.Do(func() {
|
|||
|
|
normalProducer, initErr = rocketmq.NewProducer(
|
|||
|
|
producer.WithNameServer([]string{"127.0.0.1:9876"}), // RocketMQ namesrv 地址
|
|||
|
|
producer.WithGroupName("order_producer_group"), // 生产者组名
|
|||
|
|
)
|
|||
|
|
if initErr != nil {
|
|||
|
|
log.Fatalf("初始化普通生产者失败: %v", initErr)
|
|||
|
|
return
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
if initErr = normalProducer.Start(); initErr != nil {
|
|||
|
|
log.Fatalf("启动普通生产者失败: %v", initErr)
|
|||
|
|
return
|
|||
|
|
}
|
|||
|
|
})
|
|||
|
|
// 若初始化过程有错误，直接返回
|
|||
|
|
if initErr != nil {
|
|||
|
|
log.Printf("生产者初始化失败: %v", initErr)
|
|||
|
|
return initErr
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
// 构造消息内容
|
|||
|
|
msgData := userOrder.OrderMessage{
|
|||
|
|
OrderID: OrderId,
|
|||
|
|
GoodsID: GoodsID,
|
|||
|
|
UserID: UserID,
|
|||
|
|
Quantity: 1,
|
|||
|
|
}
|
|||
|
|
msgBody, _ := json.Marshal(msgData)
|
|||
|
|
|
|||
|
|
// 创建消息（指定主题和内容）
|
|||
|
|
msg := primitive.NewMessage("killgoods", msgBody) // 主题需提前创建或开启自动创建
|
|||
|
|
|
|||
|
|
// 发送消息（同步发送，等待结果）
|
|||
|
|
result, err := normalProducer.SendSync(context.Background(), msg)
|
|||
|
|
if err != nil {
|
|||
|
|
log.Printf("普通消息发送失败: %v", err)
|
|||
|
|
return err
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
log.Printf("普通消息发送成功，消息ID: %s, 队列: %d, 偏移量: %d",
|
|||
|
|
result.MsgID, result.MessageQueue.QueueId, result.QueueOffset)
|
|||
|
|
return nil
|
|||
|
|
}
|