mirror of
https://gitee.com/milvus-io/milvus.git
synced 2024-12-05 13:28:49 +08:00
782f942b6f
Signed-off-by: Congqi Xia <congqi.xia@zilliz.com>
86 lines
2.4 KiB
Go
86 lines
2.4 KiB
Go
// Licensed to the LF AI & Data foundation under one
|
|
// or more contributor license agreements. See the NOTICE file
|
|
// distributed with this work for additional information
|
|
// regarding copyright ownership. The ASF licenses this file
|
|
// to you under the Apache License, Version 2.0 (the
|
|
// "License"); you may not use this file except in compliance
|
|
// with the License. You may obtain a copy of the License at
|
|
//
|
|
// http://www.apache.org/licenses/LICENSE-2.0
|
|
//
|
|
// Unless required by applicable law or agreed to in writing, software
|
|
// distributed under the License is distributed on an "AS IS" BASIS,
|
|
// WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
|
|
// See the License for the specific language governing permissions and
|
|
// limitations under the License.
|
|
|
|
package log
|
|
|
|
import (
|
|
"sync/atomic"
|
|
|
|
"github.com/uber/jaeger-client-go/utils"
|
|
"go.uber.org/zap"
|
|
)
|
|
|
|
// MLogger is a wrapper type of zap.Logger.
|
|
type MLogger struct {
|
|
*zap.Logger
|
|
rl atomic.Value // *utils.ReconfigurableRateLimiter
|
|
}
|
|
|
|
// With encapsulates zap.Logger With method to return MLogger instance.
|
|
func (l *MLogger) With(fields ...zap.Field) *MLogger {
|
|
nl := &MLogger{
|
|
Logger: l.Logger.With(fields...),
|
|
}
|
|
return nl
|
|
}
|
|
|
|
// WithRateGroup uses named RateLimiter for this logger.
|
|
func (l *MLogger) WithRateGroup(groupName string, creditPerSecond, maxBalance float64) *MLogger {
|
|
rl := utils.NewRateLimiter(creditPerSecond, maxBalance)
|
|
actual, loaded := _namedRateLimiters.LoadOrStore(groupName, rl)
|
|
if loaded {
|
|
rl.Update(creditPerSecond, maxBalance)
|
|
rl = actual.(*utils.ReconfigurableRateLimiter)
|
|
}
|
|
l.rl.Store(rl)
|
|
return l
|
|
}
|
|
|
|
func (l *MLogger) r() utils.RateLimiter {
|
|
val := l.rl.Load()
|
|
if l.rl.Load() == nil {
|
|
return R()
|
|
}
|
|
return val.(*utils.ReconfigurableRateLimiter)
|
|
}
|
|
|
|
// RatedDebug calls log.Debug with RateLimiter.
|
|
func (l *MLogger) RatedDebug(cost float64, msg string, fields ...zap.Field) bool {
|
|
if l.r().CheckCredit(cost) {
|
|
l.Debug(msg, fields...)
|
|
return true
|
|
}
|
|
return false
|
|
}
|
|
|
|
// RatedInfo calls log.Info with RateLimiter.
|
|
func (l *MLogger) RatedInfo(cost float64, msg string, fields ...zap.Field) bool {
|
|
if l.r().CheckCredit(cost) {
|
|
l.Info(msg, fields...)
|
|
return true
|
|
}
|
|
return false
|
|
}
|
|
|
|
// RatedWarn calls log.Warn with RateLimiter.
|
|
func (l *MLogger) RatedWarn(cost float64, msg string, fields ...zap.Field) bool {
|
|
if l.r().CheckCredit(cost) {
|
|
l.Warn(msg, fields...)
|
|
return true
|
|
}
|
|
return false
|
|
}
|