Update the broker interface
This commit is contained in:
47
broker/options.go
Normal file
47
broker/options.go
Normal file
@@ -0,0 +1,47 @@
|
||||
package broker
|
||||
|
||||
type Options struct{}
|
||||
|
||||
type PublishOptions struct{}
|
||||
|
||||
type SubscribeOptions struct {
|
||||
// AutoAck defaults to true
|
||||
AutoAck bool
|
||||
// NumHandlers defaults to 1
|
||||
NumHandlers int
|
||||
}
|
||||
|
||||
type Option func(*Options)
|
||||
|
||||
type PublishOption func(*PublishOptions)
|
||||
|
||||
type SubscribeOption func(*SubscribeOptions)
|
||||
|
||||
// DisableAutoAck will disable auto acking of messages
|
||||
// after they have been handled.
|
||||
func DisableAutoAck() SubscribeOption {
|
||||
return func(o *SubscribeOptions) {
|
||||
o.AutoAck = false
|
||||
}
|
||||
}
|
||||
|
||||
// NumHandlers sets the number of concurrent handlers to create
|
||||
// for a subscriber.
|
||||
func NumHandlers(i int) SubscribeOption {
|
||||
return func(o *SubscribeOptions) {
|
||||
o.NumHandlers = i
|
||||
}
|
||||
}
|
||||
|
||||
func newSubscribeOptions(opts ...SubscribeOption) SubscribeOptions {
|
||||
opt := SubscribeOptions{
|
||||
AutoAck: true,
|
||||
NumHandlers: 1,
|
||||
}
|
||||
|
||||
for _, o := range opts {
|
||||
o(&opt)
|
||||
}
|
||||
|
||||
return opt
|
||||
}
|
Reference in New Issue
Block a user