Mirrors
/
prologic-msgbus
mirror of https://git.mills.io/prologic/msgbus.git
6
0
Fork 0
Code Issues Projects Releases Wiki Activity

Fixed subscribers gauge

This commit is contained in:
James Mills 2018-05-08 00:18:34 -07:00
parent 6ce5fb7104
commit 8da5d41e65
No known key found for this signature in database
GPG Key ID: AC4C014F1440EBD6
1 changed files with 13 additions and 14 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

27
msgbus.go
View File

@ -352,13 +352,8 @@ func (mb *MessageBus) NotifyAll(message Message) {
// Subscribe ...
func (mb *MessageBus) Subscribe(id, topic string) chan Message {
defer func() {
if mb.metrics != nil {
mb.metrics.Gauge("bus", "subscribers").Inc()
}
}()
log.Debugf("[msgbus] Subscribe id=%s topic=%s", id, topic)
t, ok := mb.topics[topic]
if !ok {
t = &Topic{Name: topic, TTL: mb.ttl, Created: time.Now()}
@ -372,22 +367,22 @@ func (mb *MessageBus) Subscribe(id, topic string) chan Message {
}
if ls.Exists(id) {
// Already verified th listener exists
// Already verified the listener exists
ch, _ := ls.Get(id)
return ch
}
if mb.metrics != nil {
mb.metrics.Gauge("bus", "subscribers").Inc()
}
return ls.Add(id)
}
// Unsubscribe ...
func (mb *MessageBus) Unsubscribe(id, topic string) {
defer func() {
if mb.metrics != nil {
mb.metrics.Gauge("bus", "subscribers").Dec()
}
}()
log.Debugf("[msgbus] Unsubscribe id=%s topic=%s", id, topic)
t, ok := mb.topics[topic]
if !ok {
return
@ -399,8 +394,12 @@ func (mb *MessageBus) Unsubscribe(id, topic string) {
}
if ls.Exists(id) {
// Already verified th listener exists
// Already verified the listener exists
ls.Remove(id)
if mb.metrics != nil {
mb.metrics.Gauge("bus", "subscribers").Dec()
}
}
}