golang对自定义类型进行排序的解决方法
前言
Go语言支持我们自定义类型，我们大家在实际项目中，常常需要根据一个结构体类型的某个字段进行排序。之前遇到这个问题不知道如何解决，后来在网上搜索了相关问题，找到了一些好的解决方案，此处参考下，做个总结吧。
由于golang的sort包本身就提供了相应的功能，我们就没必要重复的造个轮子了，来看看如何利用sort包来实现吧。
sort包浅谈
golang中也实现了排序算法的包sort包，sort包在内部实现了四种基本的排序算法：插入排序（insertionSort）、归并排序（symMerge）、堆排序（heapSort）和快速排序（quickSort）；sort包会依据实际数据自动选择最优的排序算法。
所以我们写代码时只需要考虑实现sort.Interface这个类型就可以了。
粗略的看看sort包
funcSort(dataInterface){
//Switchtoheapsortifdepthof2*ceil(lg(n+1))isreached.
n:=data.Len()
maxDepth:=0
fori:=n;i>0;i>>=1{
maxDepth++
}
maxDepth*=2
quickSort(data,0,n,maxDepth)
}
typeInterfaceinterface{
//Lenisthenumberofelementsinthecollection.
Len()int
//Lessreportswhethertheelementwith
//indexishouldsortbeforetheelementwithindexj.
Less(i,jint)bool
//Swapswapstheelementswithindexesiandj.
Swap(i,jint)
}
//内部实现的四种排序算法
//插入排序
funcinsertionSort(dataInterface,a,bint)
//堆排序
funcheapSort(dataInterface,a,bint)
//快速排序
funcquickSort(dataInterface,a,b,maxDepthint)
//归并排序
funcsymMerge(dataInterface,a,m,bint)
所以要调用sort.Sort()来实现自定义类型排序，只需要我们的类型实现Interface接口类型中的三个方法即可。
先看看sort包本身对于[]int类型如何排序
//首先定义了一个[]int类型的别名IntSlice
typeIntSlice[]int
//获取此slice的长度
func(pIntSlice)Len()int{returnlen(p)}
//比较两个元素大小升序
func(pIntSlice)Less(i,jint)bool{returnp[i]
照葫芦画瓢我们来对自定义的结构体类型进行降序排序
packagemain
import(
"fmt"
"sort"
)
typePersonstruct{
Namestring
Ageint
}
typePersons[]Person
//获取此slice的长度
func(pPersons)Len()int{returnlen(p)}
//根据元素的年龄降序排序（此处按照自己的业务逻辑写）
func(pPersons)Less(i,jint)bool{
returnp[i].Age>p[j].Age
}
//交换数据
func(pPersons)Swap(i,jint){p[i],p[j]=p[j],p[i]}
funcmain(){
persons:=Persons{
{
Name:"test1",
Age:20,
},
{
Name:"test2",
Age:22,
},
{
Name:"test3",
Age:21,
},
}
fmt.Println("排序前")
for_,person:=rangepersons{
fmt.Println(person.Name,":",person.Age)
}
sort.Sort(persons)
fmt.Println("排序后")
for_,person:=rangepersons{
fmt.Println(person.Name,":",person.Age)
}
}
其实，一般Len()和Swap()基本不做改变，只有涉及到元素比较的Less()方法会有所改变。
当我们对某一个结构体中多个字段进行排序时怎么办，难道每排序一个就写下这三个方法么，当然不是。我们可以利用嵌套结构体来解决这个问题。因为嵌套结构体可以继承父结构体的所有属性和方法
比如我想对上面Person的Name字段和Age对要排序，我们可以利用嵌套结构体来改进一下。
packagemain
import(
"fmt"
"sort"
)
typePersonstruct{
Namestring
Ageint
}
typePersons[]Person
//Len()方法和Swap()方法不用变化
//获取此slice的长度
func(pPersons)Len()int{returnlen(p)}
//交换数据
func(pPersons)Swap(i,jint){p[i],p[j]=p[j],p[i]}
//嵌套结构体将继承Person的所有属性和方法
//所以相当于SortByName也实现了Len()和Swap()方法
typeSortByNamestruct{Persons}
//根据元素的姓名长度降序排序（此处按照自己的业务逻辑写）
func(pSortByName)Less(i,jint)bool{
returnlen(p.Persons[i].Name)>len(p.Persons[j].Name)
}
typeSortByAgestruct{Persons}
//根据元素的年龄降序排序（此处按照自己的业务逻辑写）
func(pSortByAge)Less(i,jint)bool{
returnp.Persons[i].Age>p.Persons[j].Age
}
funcmain(){
persons:=Persons{
{
Name:"test123",
Age:20,
},
{
Name:"test1",
Age:22,
},
{
Name:"test12",
Age:21,
},
}
fmt.Println("排序前")
for_,person:=rangepersons{
fmt.Println(person.Name,":",person.Age)
}
sort.Sort(SortByName{persons})
fmt.Println("排序后")
for_,person:=rangepersons{
fmt.Println(person.Name,":",person.Age)
}
}
总结
以上就是这篇文章的全部内容了，希望本文的内容对大家的学习或者工作具有一定的参考学习价值，如果有疑问大家可以留言交流，谢谢大家对毛票票的支持。