go实现组内排序取前n

有个业务涉及到了需要取数据分组后的前两条，内容大概是要获取每个用户热度最高的前两个视频，本来是用sql的 row_number实现的，结果觉得逻辑略显复杂，而且也不想让数据库处理复杂的查询，于是打算把数据查出来后，通过代码进行排序和取前二的值（可能这种做法不太可行，但没办法，数据量不大），组内排序代码如下:借鉴了希尔排序做法

//分组排序取前n
//结构为 map[组id]map[组内每个值的id]排序的值   n为取前n
//返回：map[组id][]组内id
func aaa(arr map[int32]map[int32]int32, n int) (res map[int32][]int32) {
	type kv struct {
		id  int32
		val int32
	}
	res = make(map[int32][]int32)
	for groupId, groupVal := range arr {
		kvArr := make([]kv, 0, len(groupVal))
		for k, v := range groupVal {
			kvArr = append(kvArr, kv{id: k, val: v})
		}
		for k, v := range kvArr {
			value := v
			var j int32
			for j = int32(k - 1); j >= 0 && value.val > kvArr[j].val; j-- {
				kvArr[j+1] = kvArr[j]
			}
			kvArr[j+1] = value
		}
		l := len(kvArr)
		for kn := 0; kn < n; kn++ {
			if l <= kn {
				continue
			}
			res[groupId] = append(res[groupId], kvArr[kn].id)
		}
	}
	return res
}