详解Go语言中的Slice链式操作

作者:cipher 时间:2024-05-22 10:29:05 

示例

首先模拟一个业务场景,有订单、产品、自定义订单三个结构体,订单中包含多个产品:

type Order struct {
Id       string
Products []Product
}

type Product struct {
Id    string
Price int
}

type CustomOrder struct {
Id string
}

初始化模拟数据:

var orders = []Order{
{
Id: "o1",
Products: []Product{
{
Id:    "p1",
Price: 1,
},
{
Id:    "p2",
Price: 2,
},
},
},
{
Id: "o2",
Products: []Product{
{
Id:    "p3",
Price: 3,
},
{
Id:    "p4",
Price: 4,
},
},
},
}

接下来对订单列表做各种操作:

// 过滤Id为o2的订单
func TestFilter(t *testing.T) {
res := Lists[Order](orders).Filter(func(o any) bool {
return o.(Order).Id == "o2"
}).Collect()
t.Log(res) // [{o2 [{p3 3} {p4 4}]}]
}

// 将订单列表映射为自定义订单列表
func TestMap(t *testing.T) {
res := Lists[CustomOrder](orders).Map(func(o any) any {
return CustomOrder{
Id: "custom-" + o.(Order).Id,
}
}).Collect()
t.Log(res) // [{custom-o1} {custom-o2}]
}

// 将每个订单里的产品展开,并映射为自定义订单
func TestFlatAndMap(t *testing.T) {
res := Lists[CustomOrder](orders).
Flat(func(o any) []any {
return Lists[any](o.(Order).Products).ToList()
}).
Map(func(p any) any {
return CustomOrder{
Id: "ProductId-" + p.(Product).Id,
}
}).Collect()
t.Log(res) // [{ProductId-p1} {ProductId-p2} {ProductId-p3} {ProductId-p4}]
}

// 找到所有订单产品中价格最贵的那个产品
func TestMax(t *testing.T) {
res, found := Lists[Product](orders).
Flat(func(o any) []any {
return Lists[any](o.(Order).Products).ToList()
}).
Max(func(i, j any) bool {
return i.(Product).Price > j.(Product).Price
})
t.Log(found, res) // true {p4 4}
}

原理

type List[T any] struct {
list []any
}

将 go 中的原生切片包装成 List[T] 结构体,特别说明其中的泛型 T 是最终结果的元素类型,并不是原始传入切片的类型。

这样设计是因为 go 只能在构造结构体时指定泛型,因此将 List[T] 的泛型指定为最终结果的元素类型,就可以在操作完成后调用 Collect() 方法,得到最终的 T 类型切片,方便后面的业务逻辑使用。

因为 go 不支持在接受者函数中定义泛型,因此所有操作函数的参数和返回值类型只能定义为any,然后在函数体内转换为业务结构体使用,例如上面的 i.(Product).Price

此后将每一种操作,例如Filter、Map、Flat等,都返回List[T] 结构体,就可以实现链式操作。

实现

type List[T any] struct {
list []any
}

func Lists[T any](items any) *List[T] {
rv := reflect.ValueOf(items)
if rv.Kind() != reflect.Slice {
panic(fmt.Sprintf("not supported type: %v, please use slice instead", rv.Kind()))
}
l := rv.Len()
s := make([]any, 0, l)
for i := 0; i < l; i++ {
s = append(s, rv.Index(i).Interface())
}
return &List[T]{
list: s,
}
}

func (s *List[T]) Filter(fn func(any) bool) *List[T] {
l := make([]any, 0)
for _, e := range s.list {
if fn(e) {
l = append(l, e)
}
}
s.list = l
return s
}

func (s *List[T]) Map(fn func(any) any) *List[T] {
l := make([]any, 0)
for _, element := range s.list {
l = append(l, fn(element))
}
return &List[T]{
list: l,
}
}

func (s *List[T]) Flat(fn func(any) []any) *List[T] {
l := make([]any, 0)
for _, element := range s.list {
l = append(l, fn(element)...)
}
return &List[T]{
list: l,
}
}

func (s *List[T]) Sort(fn func(i, j any) bool) *List[T] {
if len(s.list) <= 0 {
return s
}
sort.SliceStable(s.list, func(i, j int) bool {
return fn(s.list[i], s.list[j])
})
return s
}

func (s *List[T]) Max(fn func(i, j any) bool) (T, bool) {
return s.Sort(fn).FindFirst()
}

func (s *List[T]) FindFirst() (T, bool) {
if len(s.list) <= 0 {
var nonsense T
return nonsense, false
}
return s.list[0].(T), true
}

func (s *List[T]) ToList() []any {
return s.list
}

func (s *List[T]) Collect() []T {
t := make([]T, 0)
for _, a := range s.list {
t = append(t, a.(T))
}
return t
}

来源:https://juejin.cn/post/7219130999685709884

标签:Go,Slice,链式,操作
0
投稿

猜你喜欢

  • Django模型验证器介绍与源码分析

    2023-10-19 13:49:53
  • 详解Python 关联规则分析

    2023-09-03 11:54:18
  • Python实现随机取一个矩阵数组的某几行

    2021-10-04 16:45:52
  • Python自动录入ERP系统数据

    2022-03-09 06:05:41
  • Python中根据时间自动创建文件夹的代码实现

    2023-07-06 02:42:01
  • PhpStorm配置debug环境的详细过程

    2023-05-26 20:04:32
  • 对python pandas 画移动平均线的方法详解

    2023-11-16 04:21:33
  • Linux删除系统自带版本Python过程详解

    2023-10-21 03:12:13
  • python使用glob检索文件的操作

    2022-04-26 19:05:49
  • pyinstaller通过spec文件打包py程序的步骤

    2021-02-05 01:49:57
  • node.js回调函数之阻塞调用与非阻塞调用

    2024-05-05 09:21:26
  • JavaScript数组合并的8种常见方法小结

    2024-04-16 09:32:55
  • 利用pandas读取中文数据集的方法

    2021-11-23 12:33:48
  • Access数据库导入Mysql的方法之一

    2023-11-17 12:41:41
  • MySQL 存储过程的基本用法介绍

    2024-01-20 15:56:33
  • jQuery的ajax下载blob文件

    2024-04-19 10:18:47
  • 教你用Python写安卓游戏外挂

    2023-10-21 17:43:29
  • 浅谈PyTorch的可重复性问题(如何使实验结果可复现)

    2021-07-16 06:34:33
  • python虚拟机之描述器实现原理与源码分析

    2022-05-24 14:41:10
  • PHP For循环字母A-Z当超过26个字母时输出AA,AB,AC

    2023-10-07 08:09:10
  • asp之家 网络编程 m.aspxhome.com