Go slices.SortFunc 怎么按业务优先级排序结构体切片

如果结构体里的业务优先级是“紧急、普通、低”这类字符串，不要把字符串直接交给 slices.SortFunc 比较。更稳妥的做法是先用 priorityRank 映射成整数，再在同优先级时补一个确定的第二排序字段。这样比较函数每次都能返回负数、正数或 0，排序结果也更容易测试。

一、先把业务优先级变成可比较的整数

字符串优先级不能按字典序比较。例如“低”可能排在“普通”前面，和产品定义完全相反。把业务词映射到 Priority 后，PriorityScore 只承担排序语义，结构体仍保留原始的 Priority 用于展示。

type Task struct {
	ID             int
	Priority       string
	PriorityScore  int
	CreatedAt      int64
}

var priorityRank = map[string]int{
	"紧急": 0,
	"普通": 1,
	"低":   2,
}

func scoreOf(priority string) int {
	if score, ok := priorityRank[priority]; ok {
		return score
	}
	// 未知优先级放到末尾，避免被误判成最高级。
	return len(priorityRank)
}

这里选择从 0 开始且数值越小越靠前。未知值不应悄悄变成 0，否则一条拼写错误的任务会跑到“紧急”之前；生产代码还可以在入库时拒绝未知值，但排序层仍建议保留兜底。

二、比较函数按优先级再按字段收敛平局

slices.SortFunc 的第二个参数不是“是否小于”的布尔函数，而是返回整数的比较函数：小于返回负数，大于返回正数，相等或不可区分返回 0。先比 PriorityScore，再比 CreatedAt 和 ID，可以让同级任务也有确定顺序。

import (
	"cmp"
	"slices"
)

func sortTasks(tasks []Task) {
	slices.SortFunc(tasks, func(a, b Task) int {
		if n := cmp.Compare(a.PriorityScore, b.PriorityScore); n != 0 {
			return n
		}
		if n := cmp.Compare(a.CreatedAt, b.CreatedAt); n != 0 {
			return n
		}
		// 最后用 ID 收敛平局，完全相同才返回 0。
		return cmp.Compare(a.ID, b.ID)
	})
}

调用前要确保每个元素都填过 PriorityScore，或者在比较函数内部调用 scoreOf。不要写成“只要 a 优先级不是紧急就返回 1”这种分支，它可能同时判定 a 和 b，破坏严格弱序。SortFunc 还会直接改变传入的结构体切片；需要保留原始顺序时先复制一份。

三、需要稳定顺序时改用 SortStableFunc

如果产品要求“同优先级继续沿用接口返回顺序”，比较函数可以只关注优先级，然后改用 slices.SortStableFunc。普通的 slices.SortFunc 不保证相等元素的原始顺序，所以不能把“返回 0”误解成稳定排序承诺。

func sortTasksStable(tasks []Task) {
	slices.SortStableFunc(tasks, func(a, b Task) int {
		// 返回 0 表示同优先级；稳定排序会保留它们的输入顺序。
		return cmp.Compare(a.PriorityScore, b.PriorityScore)
	})
}

两种策略的取舍很清楚：需要可重复的全序，就补上时间和 ID；需要保留上游顺序，就使用稳定排序。不要在比较函数里读取当前时间、随机数或可变全局状态，否则同一对元素可能前后返回不同结果。

四、用 IsSortedFunc 做轻量检查

排序逻辑最好只定义一次，检查时复用同一个比较函数。测试数据至少覆盖紧急、普通、低、未知和同级任务；检查函数返回 false 时，优先看映射表和比较字段，而不是先怀疑 slices 包。

func taskCompare(a, b Task) int {
	if n := cmp.Compare(scoreOf(a.Priority), scoreOf(b.Priority)); n != 0 {
		return n
	}
	return cmp.Compare(a.ID, b.ID)
}

// 排序后复用同一规则，避免测试和生产各写一套比较逻辑。
sorted := slices.IsSortedFunc(tasks, taskCompare)
if !sorted {
	panic("任务排序规则未满足")
}

如果只是验证生产数据，可以把 panic 换成日志和指标；示例强调的是复用比较函数。排序前后还要留意切片是否与其他代码共享底层数组，因为原地排序会让所有持有该切片视图的调用方看到顺序变化。

相关问题

slices.SortFunc 能按降序排吗？

可以，把比较参数反过来，或对最终比较结果取反；但要统一处理溢出和相等情况，推荐继续使用 cmp.Compare(b, a)。

未知优先级应该排在哪里？

由业务决定，常见做法是统一放到末尾并记录异常。关键是所有未知值必须得到同一个明确分数。

为什么同级任务每次顺序不一样？

因为 SortFunc 不保证稳定。如果同级原始顺序有意义，改用 SortStableFunc；如果需要固定顺序，则增加时间或 ID 作为第二关键字。