shithub: furgit

ref: b46eba214daa9a6ede179ed543033b0f3485ec2e
dir: /commitquery/priority_queue.go/

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
}