You can not select more than 25 topics
Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
63 lines
1.8 KiB
63 lines
1.8 KiB
package routing
|
|
|
|
import (
|
|
amqp "github.com/rabbitmq/amqp091-go"
|
|
"goskeleton/app/global/variable"
|
|
)
|
|
|
|
// 等 go 泛型稳定以后，生产者和消费者初始化参数的设置，本段代码就可以继续精简
|
|
// 目前 apply(*producer) 的参数只能固定为生产者或者消费者其中之一的具体类型
|
|
|
|
// 1.生产者初始化参数定义
|
|
|
|
// OptionsProd 定义动态设置参数接口
|
|
type OptionsProd interface {
|
|
apply(*producer)
|
|
}
|
|
|
|
// OptionFunc 以函数形式实现上面的接口
|
|
type OptionFunc func(*producer)
|
|
|
|
func (f OptionFunc) apply(prod *producer) {
|
|
f(prod)
|
|
}
|
|
|
|
// SetProdMsgDelayParams 开发者设置生产者初始化时的参数
|
|
func SetProdMsgDelayParams(enableMsgDelayPlugin bool) OptionsProd {
|
|
return OptionFunc(func(p *producer) {
|
|
p.enableDelayMsgPlugin = enableMsgDelayPlugin
|
|
p.exchangeType = "x-delayed-message"
|
|
p.args = amqp.Table{
|
|
"x-delayed-type": "direct",
|
|
}
|
|
p.exchangeName = variable.ConfigYml.GetString("RabbitMq.Routing.DelayedExchangeName")
|
|
// 延迟消息队列，交换机、消息全部设置为持久
|
|
p.durable = true
|
|
})
|
|
}
|
|
|
|
// 2.消费者端初始化参数定义
|
|
|
|
// OptionsConsumer 定义动态设置参数接口
|
|
type OptionsConsumer interface {
|
|
apply(*consumer)
|
|
}
|
|
|
|
// OptionsConsumerFunc 以函数形式实现上面的接口
|
|
type OptionsConsumerFunc func(*consumer)
|
|
|
|
func (f OptionsConsumerFunc) apply(cons *consumer) {
|
|
f(cons)
|
|
}
|
|
|
|
// SetConsMsgDelayParams 开发者设置消费者端初始化时的参数
|
|
func SetConsMsgDelayParams(enableDelayMsgPlugin bool) OptionsConsumer {
|
|
return OptionsConsumerFunc(func(c *consumer) {
|
|
c.enableDelayMsgPlugin = enableDelayMsgPlugin
|
|
c.exchangeType = "x-delayed-message"
|
|
c.exchangeName = variable.ConfigYml.GetString("RabbitMq.Routing.DelayedExchangeName")
|
|
// 延迟消息队列，交换机、消息全部设置为持久
|
|
c.durable = true
|
|
})
|
|
}
|