vbatts/pkg
1
0
Fork 0
Code Pull requests Releases Activity

Merge pull request #19498 from mountkin/refactor-jsonfilelog-reader

Browse source 
use pubsub instead of filenotify to follow json logs
This commit is contained in:
Alexander Morozov 2016-02-16 13:55:48 -08:00
commit 3fcb349dba
1 changed files with 9 additions and 3 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

12
pubsub/publisher.go
View file

@ -54,18 +54,24 @@ func (p *Publisher) SubscribeTopic(topic topicFunc) chan interface{} {
// Evict removes the specified subscriber from receiving any more messages. // Evict removes the specified subscriber from receiving any more messages.
func (p *Publisher) Evict(sub chan interface{}) { func (p *Publisher) Evict(sub chan interface{}) {
p.m.Lock() p.m.Lock()
delete(p.subscribers, sub) if _, ok := p.subscribers[sub]; ok {
close(sub) delete(p.subscribers, sub)
close(sub)
}
p.m.Unlock() p.m.Unlock()
} }
// Publish sends the data in v to all subscribers currently registered with the publisher. // Publish sends the data in v to all subscribers currently registered with the publisher.
func (p *Publisher) Publish(v interface{}) { func (p *Publisher) Publish(v interface{}) {
p.m.RLock() p.m.RLock()
if len(p.subscribers) == 0 {
p.m.RUnlock()
return
}
wg := new(sync.WaitGroup) wg := new(sync.WaitGroup)
for sub, topic := range p.subscribers { for sub, topic := range p.subscribers {
wg.Add(1) wg.Add(1)
go p.sendTopic(sub, topic, v, wg) go p.sendTopic(sub, topic, v, wg)
} }
wg.Wait() wg.Wait()