Add function to return the number of keys currently active.

1 files changed, 7 insertions, 0 deletions

diff --git a/queue.go b/queue.go
index a1628a5..3d61ec3 100644
--- a/queue.go
+++ b/queue.go
@@ -274,6 +274,13 @@ func (q *Queue[K]) LenKey(key K) int {
 	return l
 }
 
+// LenKeys returns the current number of keys in the queue.
+func (q *Queue[K]) LenKeys() int {
+	q.partitionsMu.Lock()
+	defer q.partitionsMu.Unlock()
+	return len(q.partitions)
+}
+
 // Stop makes every blocking operation return immediately with ok set to false.
 func (q *Queue[K]) Stop() {
 	q.once.Do(q.init)