From 040b572d95e4ca27e1ada6113c405b8a1eb4a669 Mon Sep 17 00:00:00 2001 From: Runxi Yu Date: Wed, 11 Mar 2026 20:41:32 +0800 Subject: commitquery: Merge from ancestor and mergebases --- commitquery/priority_queue.go | 68 +++++++++++++++++++++++++++++++++++++++++++ 1 file changed, 68 insertions(+) create mode 100644 commitquery/priority_queue.go (limited to 'commitquery/priority_queue.go') diff --git a/commitquery/priority_queue.go b/commitquery/priority_queue.go new file mode 100644 index 00000000..0ca57f7d --- /dev/null +++ b/commitquery/priority_queue.go @@ -0,0 +1,68 @@ +package commitquery + +import "container/heap" + +// priorityQueue orders internal nodes using one query context's comparator. +type priorityQueue struct { + query *Query + items []nodeIndex +} + +// newPriorityQueue builds one empty priority queue over one query context. +func newPriorityQueue(query *Query) *priorityQueue { + queue := &priorityQueue{query: query} + heap.Init(queue) + + return queue +} + +// Len reports the number of queued items. +func (queue *priorityQueue) Len() int { + return len(queue.items) +} + +// Less reports whether one heap slot sorts ahead of another. +func (queue *priorityQueue) Less(left, right int) bool { + return queue.query.compare(queue.items[left], queue.items[right]) > 0 +} + +// Swap exchanges two heap slots. +func (queue *priorityQueue) Swap(left, right int) { + queue.items[left], queue.items[right] = queue.items[right], queue.items[left] +} + +// Push appends one heap element. +func (queue *priorityQueue) Push(item any) { + idx, ok := item.(nodeIndex) + if !ok { + panic("commitquery: heap push item is not a nodeIndex") + } + + queue.items = append(queue.items, idx) +} + +// Pop removes one heap element. +func (queue *priorityQueue) Pop() any { + last := len(queue.items) - 1 + item := queue.items[last] + queue.items = queue.items[:last] + + return item +} + +// PushNode inserts one internal node. +func (queue *priorityQueue) PushNode(idx nodeIndex) { + heap.Push(queue, idx) +} + +// PopNode removes the highest-priority internal node. +func (queue *priorityQueue) PopNode() nodeIndex { + item := heap.Pop(queue) + + idx, ok := item.(nodeIndex) + if !ok { + panic("commitquery: heap pop item is not a nodeIndex") + } + + return idx +} -- cgit v1.3.1-10-gc9f91