Pass done channel to counter

2014-09-24 19:34:39 +04:00
parent d74792a591
commit 959bc56bc3
2 changed files with 5 additions and 5 deletions
@@ -47,9 +47,9 @@ func (h *Hub) Pub(queue string, msg []byte) bool {
 }

 func (h *Hub) Sub(s *Subscription) {
-	for _, q := range s.Queues {
-		if msg, ok := h.storage.Get(q); ok {
-			s.Send(Message{q, msg})
+	for _, queue := range s.Queues {
+		if msg, ok := h.storage.Get(queue, s.Done()); ok {
+			s.Send(Message{queue, msg})
 			return
 		}
 	}