modify mq

This commit is contained in:
sk 2024-12-20 10:43:12 +08:00
parent e044a590d4
commit d82c5ed055
1 changed files with 13 additions and 6 deletions

View File

@ -55,18 +55,18 @@ type RabbitMQData struct {
Data interface{} Data interface{}
} }
func (c *MessageMgr) Send(data interface{}, name ...string) error { func (c *MessageMgr) Send(data interface{}, name string, options ...broker.PublishOption) error {
if msg, ok := data.(*RabbitMQData); ok { if msg, ok := data.(*RabbitMQData); ok {
return Send(msg.MQName, msg.Data) return Send(msg.MQName, msg.Data, options...)
} }
if len(name) > 0 && name[0] != "" { if len(name) > 0 {
return Send(name[0], data) return Send(name, data, options...)
} }
key := c.getName(data) key := c.getName(data)
if key == "" { if key == "" {
key = "_null_" key = "_null_"
} }
return Send(key, data) return Send(key, data, options...)
} }
type RegisterHandlerParam struct { type RegisterHandlerParam struct {
@ -132,7 +132,14 @@ func RegisterMessage(param *RegisterMessageParam) {
// Write 发送消息 // Write 发送消息
// 默认队列名称规则队列前缀_消息结构体名称 // 默认队列名称规则队列前缀_消息结构体名称
func Write(data interface{}, name ...string) error { func Write(data interface{}, name ...string) error {
return MessageMgrSingle.Send(data, name...) if len(name) > 0 && name[0] != "" {
return MessageMgrSingle.Send(data, name[0])
}
return MessageMgrSingle.Send(data, "")
}
func WriteWithOptions(data interface{}, name string, opts ...broker.PublishOption) error {
return MessageMgrSingle.Send(data, name, opts...)
} }
// RegisterHandler 注册消息处理函数 // RegisterHandler 注册消息处理函数