modify mq
This commit is contained in:
parent
e044a590d4
commit
d82c5ed055
19
mq/export.go
19
mq/export.go
|
@ -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 注册消息处理函数
|
||||||
|
|
Loading…
Reference in New Issue