add boolean return value

Author: Bychin

kpq/keyed_priority_queue.go

@@ -145,7 +145,7 @@ func (pq *KeyedPriorityQueue[K, V]) Pop() (K, V, bool) {
 
 // BlockingPop removes and returns the highest priority key and value from the priority queue.
 // In case queue is empty, it blocks until next Push happens or ctx is closed.
-func (pq *KeyedPriorityQueue[K, V]) BlockingPop(ctx context.Context) (K, V) {
+func (pq *KeyedPriorityQueue[K, V]) BlockingPop(ctx context.Context) (K, V, bool) {
 	pq.mu.Lock()
 
 	if len(pq.pm) == 0 {
@@ -155,14 +155,15 @@ func (pq *KeyedPriorityQueue[K, V]) BlockingPop(ctx context.Context) (K, V) {
 		case <-ctx.Done():
 			var k K
 			var v V
-			return k, v
+			return k, v, false
 		case pair := <-pq.fastTrack:
-			return pair.k, pair.v
+			return pair.k, pair.v, true
 		}
 	}
 
 	defer pq.mu.Unlock()
-	return pq.pop()
+	k, v := pq.pop()
+	return k, v, true
 }
 
 func (pq *KeyedPriorityQueue[K, V]) pop() (K, V) {

kpq/keyed_priority_queue_test.go

@@ -1,6 +1,7 @@
 package kpq
 
 import (
+	"context"
 	"errors"
 	"fmt"
 	"testing"
@@ -314,7 +315,7 @@ func TestKeyedPriorityQueue_BlockingPop(t *testing.T) {
 
 		for _, tc := range testCases {
 			t.Run(fmt.Sprintf("%s_%d", tc.wantKey, tc.wantValue), func(t *testing.T) {
-				gotKey, gotValue := pq.BlockingPop()
+				gotKey, gotValue, _ := pq.BlockingPop(context.Background())
 
 				if gotKey != tc.wantKey {
 					t.Errorf("pq.BlockingPop(): got key %q; want %q", gotKey, tc.wantKey)
@@ -394,7 +395,7 @@ func TestKeyedPriorityQueue_BlockingPop(t *testing.T) {
 
 		for _, tc := range testCases {
 			t.Run(fmt.Sprintf("%s_%d", tc.wantKey, tc.wantValue), func(t *testing.T) {
-				gotKey, gotValue := pq.BlockingPop()
+				gotKey, gotValue, _ := pq.BlockingPop(context.Background())
 
 				if gotKey != tc.wantKey {
 					t.Errorf("pq.Pop(): got key %q; want %q", gotKey, tc.wantKey)