mirror of
https://gitee.com/johng/gf
synced 2026-06-06 02:25:47 +08:00
13524a36bc
## 描述
本PR为`gmap`、`gset`和`gtree`容器引入了`NilChecker`机制，以解决Go语言中的`typed
nil`问题。该实现允许用户注册自定义的nil检查函数来确定值是否应被视为nil，这对于处理那些会被存储到容器中的`typed
nil`值特别有用。
## 情况描述
当前`gmap`等容器的泛型容器存在对`value`的`nil`值无法正确过滤的问题，例如以下例子中如果使用默认的`if any(value)
!=
nil`去判断就会得到错误的结果，原因是会出现带有类型的`(*Student)(nil)`直接和`nil`比较或者使用`any`强转都是不对的，使用反射可以解决但是性能太差了，所以换个思虑我们让用户自己决定如何判断`nil`就能解决这个问题
```golang
func main() {
type Student struct {
Name string
Age int
}
m1 := gmap.NewKVMap[int, *Student](true)
for i := 0; i < 10; i++ {
m1.GetOrSetFuncLock(i, func() *Student {
if i%2 == 0 {
return &Student{}
}
return nil
})
}
fmt.Println(m1.Size()) // 10
m2 := gmap.NewKVMap[int, *Student](true)
m2.RegisterNilChecker(func(student *Student) bool {
return student == nil
})
for i := 0; i < 10; i++ {
m2.GetOrSetFuncLock(i, func() *Student {
if i%2 == 0 {
return &Student{}
}
return nil
})
}
fmt.Println(m2.Size()) // 5
}
```
## 变更内容
- 在gmap、gset和gtree包中添加了`NilChecker`类型定义
- 扩展容器结构体，增加`nilChecker`字段来存储自定义nil检查函数
- 实现了`RegisterNilChecker`方法，允许用户注册自定义nil检查逻辑
- 添加了`isNil`内部方法，优先使用自定义nil检查函数或回退到默认的`any(v) == nil`检查
- 更新关键操作（AddIfNotExist、Set等）以利用nil检查机制
- 为所有三个容器类型添加了全面的测试用例以验证nilchecker功能
---------
Co-authored-by: github-actions[bot] <github-actions[bot]@users.noreply.github.com>
Co-authored-by: Copilot <175728472+Copilot@users.noreply.github.com>