Delete key from partitions map if it is empty.

1 files changed, 3 insertions, 0 deletions

diff --git a/queue.go b/queue.go
index 3d61ec3..04eaef7 100644
--- a/queue.go
+++ b/queue.go
@@ -244,6 +244,9 @@ func (q *Queue[K]) Pop() (item Item[K], ok bool) {
 		q.snoozeQ.Push(key, time.Now().Add(q.interval()).UnixMicro())
 		q.snoozeQNotEmpty.Signal()
 	} else {
+		q.partitionsMu.Lock()
+		delete(q.partitions, key)
+		q.partitionsMu.Unlock()
 		q.inactiveKeys[key] = time.Now().Add(q.interval())
 	}
 	q.snoozeQMu.Unlock()