diff options
Diffstat (limited to 'plugins/broadcast')
| -rw-r--r-- | plugins/broadcast/plugin.go | 19 |
1 files changed, 15 insertions, 4 deletions
diff --git a/plugins/broadcast/plugin.go b/plugins/broadcast/plugin.go index 04a4fb80..6ddef806 100644 --- a/plugins/broadcast/plugin.go +++ b/plugins/broadcast/plugin.go @@ -116,7 +116,7 @@ func (p *Plugin) PublishAsync(m *pubsub.Message) { }() } -func (p *Plugin) GetDriver(key string) (pubsub.SubReader, error) { +func (p *Plugin) GetDriver(key string) (pubsub.SubReader, error) { //nolint:gocognit const op = errors.Op("broadcast_plugin_get_driver") // choose a driver @@ -164,17 +164,28 @@ func (p *Plugin) GetDriver(key string) (pubsub.SubReader, error) { return nil, errors.E(op, err) } - // save the pubsub under a config key - // + // if section already exists, return new connection + if _, ok := p.publishers[configKey]; ok { + return ps, nil + } + + // if not - initialize a connection p.publishers[configKey] = ps return ps, nil + + // then try global if local does not exist case p.cfgPlugin.Has(redis): ps, err := p.constructors[redis].PSConstruct(configKey) if err != nil { return nil, errors.E(op, err) } - // save the pubsub + // if section already exists, return new connection + if _, ok := p.publishers[configKey]; ok { + return ps, nil + } + + // if not - initialize a connection p.publishers[configKey] = ps return ps, nil } |