Added queue.Shift()

Author: limingxinleo

priority_queue.go

@@ -18,6 +18,13 @@ func NewPriorityQueue[T any]() *PriorityQueue[T] {
 	return &PriorityQueue[T]{}
 }
 
+func (q *PriorityQueue[T]) Shift() T {
+	l := len(q.elements)
+	x := q.elements[0]
+	q.elements = q.elements[1:l]
+	return x.Element
+}
+
 func (q *PriorityQueue[T]) Insert(element T, priority int64) bool {
 	q.priority--
 

priority_queue_test.go

@@ -23,3 +23,16 @@ func TestPriorityQueue_SamePriority(t *testing.T) {
 
 	assert.Equal(t, []string{"Swoole", "Swow", "Hyperf"}, queue.ToArray())
 }
+
+func TestPriorityQueue_Shift(t *testing.T) {
+	queue := &PriorityQueue[string]{}
+	queue.Insert("Hyperf", 1)
+	queue.Insert("Swoole", 0)
+	queue.Insert("Swow", 2)
+
+	queue.Sort()
+	assert.Equal(t, "Swow", queue.Shift())
+	assert.Equal(t, "Hyperf", queue.Shift())
+	assert.Equal(t, "Swoole", queue.Shift())
+	assert.Equal(t, 0, queue.Len())
+}