go-underscore：Golang 实用集合功能增强工具库

go-underscore 是一个受 JavaScript 的 underscore.js 和 lodash 库启发的 Golang 工具库，它为 Go 的原生集合类型（如切片和映射）提供了许多实用的函数式编程方法。

主要功能

go-underscore 提供了以下主要功能：

1. 集合操作（过滤、映射、归约等）
2. 实用函数（查找、包含、分组等）
3. 链式调用支持
4. 并发安全操作

安装

go get github.com/ahl5esoft/golang-underscore

基本用法示例

1. 过滤操作

package main

import (
	"fmt"
	u "github.com/ahl5esoft/golang-underscore"
)

func main() {
	nums := []int{1, 2, 3, 4, 5, 6, 7, 8, 9, 10}
	
	// 过滤出偶数
	var evenNumbers []int
	u.Chain(nums).Filter(func(num int, _ int) bool {
		return num%2 == 0
	}).Value(&evenNumbers)
	
	fmt.Println(evenNumbers) // 输出: [2 4 6 8 10]
}

2. 映射操作

package main

import (
	"fmt"
	u "github.com/ahl5esoft/golang-underscore"
)

func main() {
	nums := []int{1, 2, 3, 4, 5}
	
	// 将每个数字平方
	var squares []int
	u.Chain(nums).Map(func(num int, _ int) int {
		return num * num
	}).Value(&squares)
	
	fmt.Println(squares) // 输出: [1 4 9 16 25]
}

3. 归约操作

package main

import (
	"fmt"
	u "github.com/ahl5esoft/golang-underscore"
)

func main() {
	nums := []int{1, 2, 3, 4, 5}
	
	// 计算总和
	var sum int
	u.Chain(nums).Reduce(func(acc, num int, _ int) int {
		return acc + num
	}, 0, &sum)
	
	fmt.Println(sum) // 输出: 15
}

4. 查找操作

package main

import (
	"fmt"
	u "github.com/ahl5esoft/golang-underscore"
)

func main() {
	users := []struct {
		ID   int
		Name string
	}{
		{1, "Alice"},
		{2, "Bob"},
		{3, "Charlie"},
	}
	
	// 查找ID为2的用户
	var user struct {
		ID   int
		Name string
	}
	found := u.Chain(users).Find(func(u struct {
		ID   int
		Name string
	}, _ int) bool {
		return u.ID == 2
	}).Value(&user)
	
	if found {
		fmt.Println(user.Name) // 输出: Bob
	}
}

5. 分组操作

package main

import (
	"fmt"
	u "github.com/ahl5esoft/golang-underscore"
)

func main() {
	users := []struct {
		Name string
		Age  int
	}{
		{"Alice", 25},
		{"Bob", 30},
		{"Charlie", 25},
		{"David", 30},
	}
	
	// 按年龄分组
	var grouped map[int][]string
	u.Chain(users).Group(func(u struct {
		Name string
		Age  int
	}, _ int) int {
		return u.Age
	}, &grouped)
	
	fmt.Println(grouped) // 输出: map[25:[Alice Charlie] 30:[Bob David]]
}

6. 链式调用

package main

import (
	"fmt"
	u "github.com/ahl5esoft/golang-underscore"
)

func main() {
	nums := []int{1, 2, 3, 4, 5, 6, 7, 8, 9, 10}
	
	// 链式操作：过滤偶数 -> 平方 -> 取前3个
	var result []int
	u.Chain(nums).Filter(func(num int, _ int) bool {
		return num%2 == 0
	}).Map(func(num int, _ int) int {
		return num * num
	}).Take(3).Value(&result)
	
	fmt.Println(result) // 输出: [4 16 36]
}

性能考虑

虽然 go-underscore 提供了便利的函数式操作，但在性能敏感的场景中，直接使用原生循环可能更高效。go-underscore 更适合在业务逻辑复杂、需要清晰表达意图的场景中使用。

总结

go-underscore 为 Golang 开发者提供了一套实用的集合操作工具，特别适合需要函数式编程风格的场景。它简化了常见的集合操作，使代码更加简洁易读。虽然性能上可能不如原生循环，但在大多数业务场景中，这种差异是可以接受的。

如果你经常需要处理复杂的集合操作，go-underscore 可以显著提高你的开发效率和代码可读性。