Golang并发安全泛型集合(Set)设计与实现
并发安全泛型集合（core/pkg/set）设计与实现
1. 简介
set包提供CSet[T comparable]：在 Go 泛型下实现的线程安全集合（元素不重复）。底层使用map[T]struct{}作为紧凑存储（空 struct 不占值内存），对外通过sync.RWMutex隔离并发读写。
VSS项目应用：流名去重、会话 key 占用标记、邀请/拉流防并发击穿、设备侧状态去重等的场景。
2. 设计要点
2.1 为何不导出底层 map 类型
底层setMap[T]不导出，仅CSet[T]暴露 API，避免绕过锁直接改map的误用。
2.2Range：快照遍历与死锁规避
若在RWMutex读锁持有期间执行用户回调f，而f内部再次调用同一CSet的Add/Remove（写锁），会典型地读锁未释放又等写锁，在标准RWMutex上容易死锁。
因此Range实现为：
- 短临界区：在锁内仅复制当前键列表（
snapshotKeys）； - 锁外枚举：对快照逐元素调用
f。
2.3Values与快照
Values()与snapshotKeys一致：返回当前时刻所有元素的一份新切片；顺序未定义（与map遍历顺序一致）。适合一次性打印或调试，不适合依赖稳定排序（若需要排序请调用方slices.Sort。
2.4 容量提示New(hint uint)
New将hint传给make(map, hint)，仅为减少扩容次数；不是集合大小上限。高并发写热点集合时可适当增大 hint。
2.5nil接收者
对(*CSet[T])(nil)调用Add/Remove/Clear/Range等为空操作（不 panic）；Contains为 false，IsEmpty为 true，Size为 0，Values为 nil，与常见「防御性」用法一致。
2.6 批量Add/Remove
Add(elements ...T)：多参数等价于依次加入，单次持锁完成，减少锁次数。Remove(elements ...T)：批量删除。
3. API 一览
| 方法 | 说明 |
|---|---|
New[T](hint uint) *CSet[T] | 初始化构造 |
Add(elements ...T) | 并入集合 |
Remove(elements ...T) | 删除 |
Clear() | 清空 |
Contains(T) bool | 是否包含 |
IsEmpty() bool | 是否为空 |
Size() int | 元素个数 |
Range(func(T) bool) | 快照遍历，false提前结束 |
Values() []T | 快照切片 |
4. 在项目中的典型用途（VSS）
ServiceContext中示例：
PubStreamExistsState：记录已存在推流名，避免重复占流；InviteRequestState：同一streamName防击穿；FetchDeviceVideoState/TalkSipSendStatus：设备或会话的进行中标记。
这类场景共同特点：高并发读Contains、短路径写Add/Remove，RWMutex较互斥锁更友好。
5. 与其它容器选型
| 需求 | 更适合 |
|---|---|
| 键值对、按 key 取 value | xmap/sync.Map/ 业务 map+锁 |
| 只关心成员与否、去重 | CSet |
| 读多写少、key 为 string | 可考虑sync.Map |
6. 并发与性能提示
- 读多写少时：
Contains/Size/IsEmpty走读锁，可并行。 Range/Values会与写操作抢占锁复制快照；快照长度大时复制成本与O(n)内存分配相关，避免在超大集合上高频调用。- 基准测试：
go test ./core/pkg/set/... -bench=. -benchmem
7. 总结
core/pkg/set用小 API 面 + 不导出底层 map + 快照式Range，在保持并发安全的同时，
规避绕过锁与Range回调重入死锁两类常见问题；
配合New与批量 Add/Remove，适合在信令与媒体网关中做轻量级集合状态。