This commit is contained in:
gazebo
2019-02-22 14:08:52 +08:00
parent fb77048058
commit e132209cc5
2 changed files with 7 additions and 8 deletions

View File

@@ -39,9 +39,10 @@ func (l *Limiter) AccessAPI(apiName string) {
if bucket == nil && l.defaultConfig != nil {
l.Lock()
if bucket = l.limitConfig[apiName]; bucket == nil {
v := l.defaultConfig
bucket = ratelimit.NewBucketWithQuantum(time.Duration(v.TimeGapMilliSecond)*time.Millisecond, int64(v.MaxAccessCount), int64(v.MaxAccessCount))
l.limitConfig[apiName] = bucket
if v := l.defaultConfig; v != nil {
bucket = ratelimit.NewBucketWithQuantum(time.Duration(v.TimeGapMilliSecond)*time.Millisecond, int64(v.MaxAccessCount), int64(v.MaxAccessCount))
l.limitConfig[apiName] = bucket
}
}
l.Unlock()
}