开启 Go 泛型时代：第三方泛型库分享

大家好，我是煎鱼。

最近 Go1.18 即将发布，其中包含了一个杀手级的特性：泛型。有了泛型后，就能干很多奇奇怪怪的事情了。

先是配套第三方库的使用，这是我在初学 Go 时最经常听到的，就是想说：“如果有泛型，就能干很多很多的事，就像 PHP array 一样” 等真香语句。

好物分享第 1 篇，定位是纯介绍。

介绍

今天分享的是已开源的泛型工具库 github.com/samber/lo，Star 3k，是一个具有 Lodash 风格的 Go 泛型库：

Lodash 指的是一个一致性、模块化、高性能的 JavaScript 实用工具库，JS 的例子大致如下：

_.chunk(['a', 'b', 'c', 'd'], 2);
// => [['a', 'b'], ['c', 'd']]
 
_.chunk(['a', 'b', 'c', 'd'], 3);
// => [['a', 'b', 'c'], ['d']]

相当于是他 Go 版本的类 API 翻译。

使用

安装 Go1.18Beta 版本：

$ go install golang.org/dl/go1.18beta2@latest
$ go1.18beta2 download

安装第三方开源库：

$ go1.18beta2 github.com/samber/lo

就可以使用了。

下面我例举几个比较常用的，完整的 API 列表大家可以到项目中查看。

https://pkg.go.dev/github.com/samber/lo

类型转换

操作一种类型的 Map/Slice，并将其转化为另一种类型的 Map/Slice：

import "github.com/samber/lo"

lo.Map[int64, string]([]int64{1, 2, 3, 4}, func(x int64, _ int) string {
    return strconv.FormatInt(x, 10)
})

输出结果：

[]string{"1", "2", "3", "4"}

并发处理

并发起 goroutine 来处理所传入的数据，并在内部会处理好顺序问题。最终结果会以相同的顺序返回：

import (
 "fmt"
 lop "github.com/samber/lo/parallel"
 "strconv"
)

func main() {
 s := lop.Map[int64, string]([]int64{1, 2, 3, 4}, func(x int64, _ int) string {
  return strconv.FormatInt(x, 10)
 })
 fmt.Println(s)
}

输出结果：

[1 2 3 4]

包含和筛选

还记得平时写 Slice 什么的，需要去根据类型不同写很多个函数吗？现在可以直接一个函数干完了。

判断如果集合中是否存在特定元素，如下：

present := lo.Contains[int]([]int{0, 1, 2, 3, 4, 5}, 5)

输出结果：

true

筛选出集合中符合条件的元素，如下：

even := lo.Filter[int]([]int{1, 2, 3, 4}, func(x int, _ int) bool {
    return x%2 == 0
})

输出结果：

[]int{2, 4}

唯一值

平时在处理业务数据时，还要处理 Slice、Map 里的重复数据，分分钟就多了 10 行代码，还有可能写错。

现在直接如此使用：

uniqValues := lo.Uniq[int]([]int{1, 2, 2, 1})

输出结果：

[]int{1, 2}

可以看到集合中重复的 1 和 2 被去掉了，输出的顺序是按照输入的先后顺序决定。

分组和切割

像是对业务数据做分组时，我们常常要把用户和其行为动作等挂靠在一起。以前得根据不同类型自己写。

现在可以直接调用带闭包的分组方法：

import lo "github.com/samber/lo"

groups := lo.GroupBy[int, int]([]int{0, 1, 2, 3, 4, 5}, func(i int) int {
    return i%3
})

输出结果：

map[int][]int{0: []int{0, 3}, 1: []int{1, 4}, 2: []int{2, 5}}

集合其根据闭包所执行的值进行了分组。

也可以做一系列动作，例如切割集合：

lo.Chunk[int]([]int{0, 1, 2, 3, 4, 5}, 2)

lo.Chunk[int]([]int{0, 1, 2, 3, 4, 5, 6}, 2)

输出结果：

[][]int{{0, 1}, {2, 3}, {4, 5}}

[][]int{{0, 1}, {2, 3}, {4, 5}, {6}}

三元运算

虽然 Go 不会支持三元运算符，但能够通过配套方法实现类似效果：

result := lo.Ternary[string](true, "a", "b")

result := lo.Ternary[string](false, "a", "b")

输出结果：

a

b

总结

今天这篇好物分享中，我们针对最近热议的 Go 泛型配套开源工具库进行了简单的介绍和分享。

从即将发布的 1.18 进度条来看，感觉马上就可以用上了，甚至可以用来学习泛型的基本实操知识。

关注煎鱼，获取业内第一手消息和知识 👇

你好，我是煎鱼，出版过 Go 畅销书《Go 语言编程之旅》，再到获得 GOP（Go 领域最有观点专家）荣誉，点击蓝字查看我的出书之路。

日常分享高质量文章，输出 Go 面试、工作经验、架构设计，加微信拉读者交流群，和大家交流！