From 86324ff1ae353194c6f1af917b33a4a8c2554c0f Mon Sep 17 00:00:00 2001 From: AikoCute Date: Sun, 20 Nov 2022 10:10:03 +0700 Subject: [PATCH] Logs for GlobalDeviceLimit --- common/limiter/limiter.go | 6 ++++-- 1 file changed, 4 insertions(+), 2 deletions(-) diff --git a/common/limiter/limiter.go b/common/limiter/limiter.go index 99009dd..08c22c5 100644 --- a/common/limiter/limiter.go +++ b/common/limiter/limiter.go @@ -4,6 +4,7 @@ package limiter import ( "context" "fmt" + "log" "strings" "sync" "time" @@ -47,6 +48,7 @@ func New() *Limiter { func (l *Limiter) AddInboundLimiter(tag string, nodeSpeedLimit uint64, userList *[]api.UserInfo, globalDeviceLimit *GlobalDeviceLimitConfig) error { // global limit if globalDeviceLimit.Limit > 0 { + log.Printf("[%s] GlobalDeviceLimit limit: %d", tag, globalDeviceLimit.Limit) l.r = redis.NewClient(&redis.Options{ Addr: globalDeviceLimit.RedisAddr, Password: globalDeviceLimit.RedisPassword, @@ -100,7 +102,7 @@ func (l *Limiter) UpdateInboundLimiter(tag string, updatedUserList *[]api.UserIn } } } else { - return fmt.Errorf("no such inbound in limiter: %s", tag) + return newError("no such inbound in limiter: %s", tag).AtError() } return nil } @@ -135,7 +137,7 @@ func (l *Limiter) GetOnlineDevice(tag string) (*[]api.OnlineUser, error) { return true }) } else { - return nil, fmt.Errorf("no such inbound in limiter: %s", tag) + return nil, newError("no such inbound in limiter: %s", tag).AtError() } return &onlineUser, nil }