From 01d15bccf3b1dcc51516b1f64d50950b31d7f8fb Mon Sep 17 00:00:00 2001 From: Runxi Yu Date: Fri, 6 Mar 2026 21:19:56 +0800 Subject: Urgh I made some wrong amends and I'm too tired to separate the commits out this time ancestor: Split out of reachability mergebase: Add merge base routines internal/commitquery: Add commit query context engine thingy internal/peel: Shared tag peeling errors: Shared object query errors internal/testgit: Add rooted repo helpers; remove raw path access objectstore/memory: Add in-memory object store objectid: Add Compare helper --- internal/commitquery/priority_queue.go | 68 ++++++++++++++++++++++++++++++++++ 1 file changed, 68 insertions(+) create mode 100644 internal/commitquery/priority_queue.go (limited to 'internal/commitquery/priority_queue.go') diff --git a/internal/commitquery/priority_queue.go b/internal/commitquery/priority_queue.go new file mode 100644 index 00000000..a7a4876e --- /dev/null +++ b/internal/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 { + ctx *Context + items []NodeIndex +} + +// NewPriorityQueue builds one empty priority queue over one query context. +func NewPriorityQueue(ctx *Context) *PriorityQueue { + queue := &PriorityQueue{ctx: ctx} + 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.ctx.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