亲宝软件园·资讯

展开

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

m0_62924778 人气:0
想了解Golang 利用反射对结构体优雅排序的操作方法的相关内容吗,m0_62924778在本文为您仔细讲解Golang反射结构体排序的相关知识和一些Code实例,欢迎阅读和指正,我们先划重点:Golang反射结构体排序,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)
	}
}

 运行结果

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

加载全部内容

相关教程
猜你喜欢
用户评论