go 切片进行链式操作并支持泛型

背景：

由于团队不是专业级别的go开发人员，主开发还是java，用惯了java的lambda表达式特别是流式操作，所以在用go语言时，发现切片处理起来比较麻烦，看看能不能支持类似流式操作，我这边就研究了下这块业务，发现是可以实现的，同时要支持所有类型，所以必然要支持泛型才可以，背景介绍完，开始正题：

定义泛型的类型形参列表：

// SliceAble
//
//	@Description: 泛型，如果需要支持其他类型，请自行添加
//	@Author  zhaosy
//	@date  2024-07-12 17:07:22
type SliceAble interface {Person | string | int
}

定义切片类型


// Slice
// 定义新切片类型
type Slice[T SliceAble] []T// NewSlice [T SliceAble]
//
//	@Author  zhaosy
//	@Description: newSlice
//	@date  2024-07-13 13:22:23
func NewSlice[T SliceAble](arr []T) Slice[T] {//这里为了不影响原始切片数据，所以进行copy，如果不关心可以删除直接返回arr即可copyArr := make([]T, len(arr), len(arr))copy(copyArr, arr)return copyArr
}// Filter
//
//	@Author  zhaosy
//	@Description: 过滤，demo：
//	people := []lang.Person{
//		{"Alice", 45},
//		{"Elizabeth", 75},
//		{"Alice", 25},
//		{"Bob", 30},
//		{"Alice", 2},
//		{"Bob", 4},
//		{"Colin", 5},
//		{"Elizabeth", 34},
//	}
//
//	p := lang.NewSlice(people).Filter(func(i lang.Person) bool {
//		return i.Age > 30
//	})
//
//	fmt.Println(p) //[{Alice 45} {Elizabeth 75} {Elizabeth 34}]
//	@date  2024-07-13 11:48:46
func (s Slice[T]) Filter(f func(T) bool) Slice[T] {var temp Slice[T]for _, v := range s {b := f(v)if b {temp = append(temp, v)}}return temp
}// Sort
//
//	@Author  zhaosy
//	@Description: 排序：
//	排序规则根据函数放回的bool类型进行排序，demo：
//	people := []lang.Person{
//		{"Alice", 45},
//		{"Elizabeth", 75},
//		{"Alice", 25},
//		{"Bob", 30},
//		{"Alice", 2},
//		{"Bob", 4},
//		{"Colin", 5},
//		{"Elizabeth", 34},
//	}
//	lang.NewSlice(people).Sort(func(s lang.Slice[lang.Person], i, j int) bool {
//		return s[i].Age > s[j].Age
//	})
//	fmt.Println(people) //结果：[{Elizabeth 75} {Alice 45} {Elizabeth 34} {Bob 30} {Alice 25} {Colin 5} {Bob 4} {Alice 2}]
//
//	@date  2024-07-13 11:41:14
func (s Slice[T]) Sort(f func(s Slice[T], i, j int) bool) Slice[T] {sort.SliceStable(s, func(i, j int) bool {return f(s, i, j)})return s
}// Append
//
//	@Author  zhaosy
//	@Description: 追加元素；采用go 底层切片append方式
//	@date  2024-07-13 13:13:09
func (s Slice[T]) Append(newObj T) Slice[T] {s = append(s, newObj)return s
}// OnePointerObj
//
//	@Author  zhaosy
//	@Description: 返回单条数据实例，是指针类型，且判断是否为空
//	@date  2024-07-13 13:13:41
func (s Slice[T]) OnePointerObj() *T {if len(s) == 0 {return nil}return &s[0]
}// OnePointerObj
//
//	@Author  zhaosy
//	@Description: 返回json字符串，如果为空，直接返回空字符串
//	@date  2024-07-13 13:47:41
func (s Slice[T]) ToJsonStr() string {b, err := json.Marshal(s)if err != nil {fmt.Println(err)return ""}return string(b)
}

测试：

import ("fmt""testing"
)type Person struct {Name string `json:"name"`Age  int    `json:"age"`
}func TestSlice_All(t *testing.T) {people := []Person{{"张三", 45},{"李四", 2},{"王五", 25},{"小明", 15},{"小红", 2},{"小蓝", 7},{"小白", 5},{"小花", 34},}fmt.Println("原始数据：", people)//需求：根据年龄进行升序排序p := NewSlice(people).Sort(func(s Slice[Person], i, j int) bool {return s[i].Age < s[j].Age})fmt.Println("排序后：", p) //结果：[{李四 2} {小红 2} {小白 5} {小蓝 7} {小明 15} {王五 25} {小花 34} {张三 45}]//需求：过滤掉小于18岁的信息p = NewSlice(people).Filter(func(pe Person) bool {return pe.Age > 18})fmt.Println("过滤18岁以后的：", p) //结果：[{张三 45} {王五 25} {小花 34}]//需求：向切片最后一个追加一个元素p = NewSlice(people).Append(Person{Name: "李玲",Age:  23,})fmt.Println("追加元素后：", p) //结果：[{张三 45} {李四 2} {王五 25} {小明 15} {小红 2} {小蓝 7} {小白 5} {小花 34} {李玲 23}]//需求：获取切片第一个（这块目的是经过其他方法处理过后，直接获取第一个，比较简单直接）one := NewSlice(people).OnePointerObj()fmt.Println("第一个元素：", *one) //结果：{张三 45}//来一个比较常用的链式需求，需求：新增一个元素，并输出大于18岁且按照年龄进行升序排序的列表,最后以json方式输出jsonStr := NewSlice(people).Append(Person{"李华", 35}).Sort(func(s Slice[Person], i, j int) bool {return s[i].Age < s[j].Age}).Filter(func(pe Person) bool {return pe.Age > 18}).ToJsonStr()fmt.Println("结果：", jsonStr) //结果：[{"name":"王五","age":25},{"name":"小花","age":34},{"name":"李华","age":35},{"name":"张三","age":45}]//后期有其他需求可以继续添加，这样就比较简单了
}