localhots
/
burlesque
1
0
Fork 0
Code Releases Activity

Pass done channel to counter

This commit is contained in:
Gregory Eremin 2014-09-24 19:34:39 +04:00
parent d74792a591
commit 959bc56bc3
2 changed files with 5 additions and 5 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

6
hub/hub.go
View File

@ -47,9 +47,9 @@ func (h *Hub) Pub(queue string, msg []byte) bool {
} }
func (h *Hub) Sub(s *Subscription) { func (h *Hub) Sub(s *Subscription) {
for _, q := range s.Queues { for _, queue := range s.Queues {
if msg, ok := h.storage.Get(q); ok { if msg, ok := h.storage.Get(queue, s.Done()); ok {
s.Send(Message{q, msg}) s.Send(Message{queue, msg})
return return
} }
} }

4
storage/storage.go
View File

@ -35,7 +35,7 @@ func New(path string) (s *Storage, err error) {
return return
} }
func (s *Storage) Get(queue string) (message []byte, ok bool) { func (s *Storage) Get(queue string, done <-chan struct{}) (message []byte, ok bool) {
if _, exist := s.counters[queue]; !exist { if _, exist := s.counters[queue]; !exist {
return return
} }
@ -46,7 +46,7 @@ func (s *Storage) Get(queue string) (message []byte, ok bool) {
var index uint var index uint
select { select {
case index = <-s.counters[queue].stream: case index = <-s.counters[queue].stream:
default: case <-done:
return return
} }