feat: 实现数据库备份和恢复并发调度 (#84)

server/pkg/runner/priority_queue_test.go

@@ -0,0 +1,67 @@
+package runner
+
+import (
+	"github.com/stretchr/testify/require"
+	"testing"
+)
+
+func TestChangePriority(t *testing.T) {
+	q := NewPriorityQueue[*priorityElement](100,
+		func(src *priorityElement, dst *priorityElement) bool {
+			return src.Priority < dst.Priority
+		})
+	e1 := &priorityElement{
+		Data:     10,
+		Priority: 200,
+	}
+	_ = q.Enqueue(e1)
+	e2 := &priorityElement{
+		Data:     10,
+		Priority: 100,
+	}
+	_ = q.Enqueue(e2)
+	//e1.Priority = 10
+	val, _ := q.Dequeue()
+	println(val)
+}
+
+type priorityElement struct {
+	Data     any
+	Priority int
+}
+
+func TestPriorityQueue_Remove(t *testing.T) {
+	q := NewPriorityQueue[*priorityElement](100,
+		func(src *priorityElement, dst *priorityElement) bool {
+			return src.Priority < dst.Priority
+		})
+
+	for i := 8; i > 0; i-- {
+		q.Enqueue(&priorityElement{Priority: i})
+	}
+	requirePriorities(t, q)
+
+	q.Remove(8)
+	requirePriorities(t, q)
+	q.Remove(7)
+	requirePriorities(t, q)
+
+	q.Remove(2)
+	requirePriorities(t, q)
+
+	q.Remove(1)
+	requirePriorities(t, q)
+
+	q.Remove(0)
+	requirePriorities(t, q)
+}
+
+func requirePriorities(t *testing.T, q *PriorityQueue[*priorityElement]) {
+	ps := make([]int, 0, q.Len())
+	for _, val := range q.data[1:] {
+		ps = append(ps, val.Priority)
+	}
+	for i := q.Len(); i >= 2; i-- {
+		require.False(t, q.less(q.data[i], q.data[i/2]), ps)
+	}
+}