xiaoqidun/limit
1
0
Fork 0
mirror of https://github.com/xiaoqidun/limit.git synced 2025-10-11 14:30:23 +08:00
Code Issues Packages Projects Releases Wiki Activity

style(代码风格): 微调代码风格

Browse Source
This commit is contained in:
xiaoqidun
2025-09-23 22:42:19 +08:00
parent 6cc84c0e49
commit cb0ec0059c
1 changed files with 0 additions and 8 deletions
Download Patch File Download Diff File Expand all files Collapse all files

8
limit.go
View File

@@ -69,17 +69,14 @@ func NewWithConfig(config Config) *Limiter {
if config.Expiration == 0 { if config.Expiration == 0 {
config.Expiration = 30 * time.Minute config.Expiration = 30 * time.Minute
} }
// 确保分片数量是2的幂，以便进行高效的位运算 // 确保分片数量是2的幂，以便进行高效的位运算
if config.ShardCount <= 0 || (config.ShardCount&(config.ShardCount-1)) != 0 { if config.ShardCount <= 0 || (config.ShardCount&(config.ShardCount-1)) != 0 {
config.ShardCount = defaultShardCount config.ShardCount = defaultShardCount
} }
l := &Limiter{ l := &Limiter{
shards: make([]*shard, config.ShardCount), shards: make([]*shard, config.ShardCount),
config: config, config: config,
} }
// 初始化所有分片 // 初始化所有分片
for i := 0; i < config.ShardCount; i++ { for i := 0; i < config.ShardCount; i++ {
l.shards[i] = newShard(config.GCInterval, config.Expiration) l.shards[i] = newShard(config.GCInterval, config.Expiration)
@@ -166,7 +163,6 @@ func (s *shard) gc(interval, expiration time.Duration) {
return return
default: default:
} }
select { select {
case <-ticker.C: case <-ticker.C:
s.mutex.Lock() s.mutex.Lock()
@@ -239,20 +235,16 @@ func (s *shard) stop() {
s.stopOnce.Do(func() { s.stopOnce.Do(func() {
close(s.stopCh) close(s.stopCh)
}) })
// 等待 gc goroutine 完全退出 // 等待 gc goroutine 完全退出
s.waitGroup.Wait() s.waitGroup.Wait()
// 锁定并进行最终的资源清理 // 锁定并进行最终的资源清理
// 因为 gc 已经退出，所以此时只有 Get/Del 会竞争锁 // 因为 gc 已经退出，所以此时只有 Get/Del 会竞争锁
s.mutex.Lock() s.mutex.Lock()
defer s.mutex.Unlock() defer s.mutex.Unlock()
// 检查是否已被清理，防止重复操作 // 检查是否已被清理，防止重复操作
if s.limiter == nil { if s.limiter == nil {
return return
} }
// 将所有 session 对象放回对象池 // 将所有 session 对象放回对象池
for _, sess := range s.limiter { for _, sess := range s.limiter {
sess.limiter = nil sess.limiter = nil