网络编程
位置：首页>> 网络编程>> Go语言>> Golang 利用反射对结构体优雅排序的操作方法

Golang 利用反射对结构体优雅排序的操作方法

作者：m0_62924778　　发布时间：2024-05-10 13:56:42　

标签：Golang,结构体,排序

最近开始实习，工作技术栈主要Python和Golang，目前的任务把Python模块重构为GO模块，然后出现了一个问题，就是要将一个结构体按结构体中各个字段进行排序，然后写入Redis,对于Pyhon来说for循环就能解决，但是对于Go语言来说，每一次排序都要写一个比较函数，写出来的代码太丑，非常长，代码结构是一致，只是比较字段不一样而已，个人无法接受啊，网上搜索也没搜索到合适解决方法，所以自己想了一个解决方法来优雅排序。

比较函数：


func reflectCmp(i, j interface{}, fieldName string) bool {  //按传入fieldName 排序
valI := reflect.ValueOf(i).FieldByName(fieldName).Interface()
valJ := reflect.ValueOf(j).FieldByName(fieldName).Interface()
switch s := valI.(type) {
case string:
return s < valJ.(string)
case float64:
return s < valJ.(float64)
case int:
return s < valJ.(int)
default:
fmt.Println("The type is unknown")
}
return true
}

将自己要比较的类型加入switch中即可。

假设我们有一个结构体为


type person struct {
Age int
Name string
}

那么如何来排序呢？


func main() {
pSlice := []person{{20,"a"},{15,"be"},{32,"cc"},{2,"aa"}}
var p person
typeOfA := reflect.TypeOf(p)

// 遍历结构体所有成员
for i := 0; i < typeOfA.NumField(); i++ {
// 获取每个成员的结构体字段类型
fieldType := typeOfA.Field(i)
sort.Slice(pSlice,func(i, j int) bool {
return reflectCmp(pSlice[i],pSlice[j],fieldType.Name)
})
fmt.Println(pSlice)
}
}

 运行结果

Golang 利用反射对结构体优雅排序的操作方法

利用反射，不管结构体有多少个字段，我们都能用一个for 排序即可优雅完成排序。

来源：https://blog.csdn.net/m0_62924778/article/details/120744879

0
投稿

猜你喜欢

手机版 网络编程 asp之家 www.aspxhome.com