tcjlj/flvx
1
0
Fork 0
mirror of https://github.com/Sagit-chu/flvx.git synced 2026-10-11 19:56:37 +08:00
Code Issues Packages Projects Releases Wiki Activity

fix(limiter): fix traffic limiter ScopeClient behavior to allow per-user limits

Browse Source
This commit is contained in:
sagit
2026-02-09 03:12:33 +00:00
parent 218084b542
commit d06e02998b
1 changed files with 6 additions and 0 deletions
Download Patch File Download Diff File Expand all files Collapse all files
go-gost/x/limiter/traffic/traffic.go
+6
View File
@@ -136,6 +136,9 @@ func (l *trafficLimiter) In(ctx context.Context, key string, opts ...limiter.Opt
return nil return nil
case limiter.ScopeClient: case limiter.ScopeClient:
if lim, ok := l.inLimits.Get(key); ok && lim != nil {
return lim.(traffic.Limiter)
}
return nil return nil
case limiter.ScopeConn: case limiter.ScopeConn:
@@ -215,6 +218,9 @@ func (l *trafficLimiter) Out(ctx context.Context, key string, opts ...limiter.Op
return nil return nil
case limiter.ScopeClient: case limiter.ScopeClient:
if lim, ok := l.outLimits.Get(key); ok && lim != nil {
return lim.(traffic.Limiter)
}
return nil return nil
case limiter.ScopeConn: case limiter.ScopeConn: