This commit is contained in:
Gregory Eremin 2014-09-10 16:53:41 +04:00
parent 69b94c077f
commit d913f52efa

View File

@ -23,6 +23,7 @@ func (h *Hub) Pub(queue string, msg []byte) bool {
if s.Queue == queue { if s.Queue == queue {
select { select {
case <-s.Done(): case <-s.Done():
// FIXME: Cleanup needed
continue continue
default: default:
} }
@ -42,6 +43,7 @@ func (h *Hub) Sub(s *Subscription) {
if msg, ok := h.storage.Get(s.Queue); ok { if msg, ok := h.storage.Get(s.Queue); ok {
s.Send(msg) s.Send(msg)
} else { } else {
// FIXME: Race condition
h.subscribers = append(h.subscribers, s) h.subscribers = append(h.subscribers, s)
} }
} }