Go bitset 怎样枚举所有置位的索引

如果要在 Go 中枚举 github.com/bits-and-blooms/bitset 的所有置位索引，Go 1.23 及以上优先写 for i := range b.EachSet()；需要兼容更早版本时，用 NextSet 循环逐个取出索引。若要把结果交给后续流水线，再把每个 uint 追加到 []uint。官方地址：https://pkg.go.dev/github.com/bits-and-blooms/bitset。

先把枚举任务分成三种 API 角色

把 bitset 当成一组“索引是否启用”的集合即可：调用 Set 打开某个索引，枚举时只关心那些值为真的位置。工作流的触发场景通常是把这些位置导出、转换成任务列表，或交给下一段过滤逻辑。此时先决定输出是流式消费，还是必须收集为一个切片。

安装依赖并确认 EachSet 的版本权限

模块中先添加依赖：

# 把 bitset 加入当前 Go module
go get github.com/bits-and-blooms/bitset

EachSet 的文档签名是 iter.Seq[uint]，官方 README 说明 Go 1.23 或更高版本可以直接用 range 遍历。项目需要兼容更早 Go 版本时，不要强行照搬这段写法，而是切换到下一节的 NextSet 方案。

让流水线阶段输出全部置位索引

在 Go 1.23 及以上，最短的收集方式如下。EachSet 只枚举已置位的索引，未置位的空洞不会出现在结果里。

package main

import (
    "fmt"

    "github.com/bits-and-blooms/bitset"
)

func main() {
    var b bitset.BitSet
    // 只设置三个离散索引，观察枚举结果不会补齐中间空洞。
    b.Set(2).Set(7).Set(11)

    indexes := make([]uint, 0, b.Count())
    // EachSet 产生所有置位索引，range 逐个交给后续阶段。
    for index := range b.EachSet() {
        indexes = append(indexes, index)
    }

    fmt.Println(indexes) // [2 7 11]
}

这里的容量提示使用 Count()，只是减少切片扩容机会；它不改变枚举结果。若下游可以边读边处理，也可以在 range 体内直接调用处理函数，不必创建 indexes。

用 NextSet 兼容旧版本或显式控制游标

NextSet(start) 返回从给定位置开始找到的下一个置位索引和布尔值。布尔值为 false 时表示后面没有可枚举的置位索引，不能继续使用上一次的索引。

func collectSetBits(b *bitset.BitSet) []uint {
    // 用 Count 预留容量，但仍以 NextSet 的 found 结果作为结束条件。
    indexes := make([]uint, 0, b.Count())
    for cursor := uint(0); ; {
        index, found := b.NextSet(cursor)
        if !found {
            break
        }
        indexes = append(indexes, index)
        // 加一后继续查找，避免反复拿到当前置位索引。
        cursor = index + 1
    }
    return indexes
}

这个写法适合需要把游标记录到日志、按区间分段，或项目仍然支持旧 Go 版本的场景。若索引已经接近 uint 的最大值，业务代码还应自行约束输入范围，避免 index + 1 溢出后回到较小位置。

给枚举结果加上数量与范围门禁

批处理任务中，最常见的误判是把 Len() 当成置位数量。官方文档中，Count() 表示置位 bit 的数量；Len() 表示比最高置位索引大一的位置。因此只设置索引 11 时，Count() 是 1，而 Len() 是 12。空 BitSet 则没有置位索引，二者都应按空集合语义处理。

失败处理：版本、类型与并发边界

如果编译器不认识 EachSet 的迭代写法，先确认项目的 Go 版本，再改用 NextSet；不要为了通过编译把索引强制转换成可能溢出的有符号类型。需要批量读取时可以查看 NextSetMany 的文档签名，但仍应以返回的索引数量为准。

另外，bitset 官方说明同一个 BitSet 默认不适合被多个 goroutine 无同步地访问。枚举期间如果另一个 goroutine 还在 Set 或 Clear，问题不是“少枚举了一个索引”这么简单，而是整个对象访问协议不成立。可以用单一拥有者传递 BitSet，或用互斥锁保护访问。

常见问题

EachSet 会返回所有从 0 到 Len 的数字吗？

不会。它只返回已经置位的索引，中间没有置位的位置会被跳过。

为什么 Count 和 Len 的结果经常不一样？

因为一个统计置位数量，一个描述最高置位索引之后的范围长度。稀疏 bitset 中二者差距会很大。

枚举时可以同时修改 BitSet 吗？

不要把枚举和无同步修改放在多个 goroutine 中。先确定对象所有权，必要时串行访问或使用互斥锁。