From 7fe9683f288834f589188f88afa6942bba5a1247 Mon Sep 17 00:00:00 2001
From: V2Ray <admin@v2ray.com>
Date: Sat, 10 Oct 2015 21:29:37 +0200
Subject: [PATCH] Cleanup queue in time

---
 common/collect/timed_queue.go | 25 +++++++++++++------------
 1 file changed, 13 insertions(+), 12 deletions(-)

diff --git a/common/collect/timed_queue.go b/common/collect/timed_queue.go
index 1595723d6..7a3cfcefa 100644
--- a/common/collect/timed_queue.go
+++ b/common/collect/timed_queue.go
@@ -71,8 +71,7 @@ func (queue *TimedQueue) RemovedEntries() <-chan interface{} {
 }
 
 func (queue *TimedQueue) cleanup(tick <-chan time.Time) {
-	for {
-		now := <-tick
+	for now := range tick {
 		queue.access.RLock()
 		queueLen := queue.queue.Len()
 		queue.access.RUnlock()
@@ -80,16 +79,18 @@ func (queue *TimedQueue) cleanup(tick <-chan time.Time) {
 			continue
 		}
 		nowSec := now.UTC().Unix()
-		queue.access.RLock()
-		firstEntryTime := queue.queue[0].timeSec
-		queue.access.RUnlock()
-		if firstEntryTime > nowSec {
-			continue
-		}
-		queue.access.Lock()
-		firstEntryValue := heap.Pop(&queue.queue).(*timedQueueEntry).value
-		queue.access.Unlock()
+		for {
+			queue.access.RLock()
+			entry := queue.queue[0]
+			queue.access.RUnlock()
+			if entry.timeSec > nowSec {
+				break
+			}
+			queue.access.Lock()
+			heap.Pop(&queue.queue)
+			queue.access.Unlock()
 
-		queue.removed <- firstEntryValue
+			queue.removed <- entry.value
+		}
 	}
 }