shanshanzhong
c582087c0f
feat: 添加版本和构建时间变量 fix: 修正短信队列类型注释错误 style: 清理未使用的代码和测试文件 docs: 更新安装文档中的下载链接 chore: 迁移数据库脚本添加日志和订阅配置
43 lines
846 B
Go
43 lines
846 B
Go
package logger
|
|
|
|
import (
|
|
"sync/atomic"
|
|
"time"
|
|
|
|
"github.com/perfect-panel/server/pkg/syncx"
|
|
"github.com/perfect-panel/server/pkg/timex"
|
|
)
|
|
|
|
type limitedExecutor struct {
|
|
threshold time.Duration
|
|
lastTime *syncx.AtomicDuration
|
|
discarded uint32
|
|
}
|
|
|
|
func newLimitedExecutor(milliseconds int) *limitedExecutor {
|
|
return &limitedExecutor{
|
|
threshold: time.Duration(milliseconds) * time.Millisecond,
|
|
lastTime: syncx.NewAtomicDuration(),
|
|
}
|
|
}
|
|
|
|
func (le *limitedExecutor) logOrDiscard(execute func()) {
|
|
if le == nil || le.threshold <= 0 {
|
|
execute()
|
|
return
|
|
}
|
|
|
|
now := timex.Now()
|
|
if now-le.lastTime.Load() <= le.threshold {
|
|
atomic.AddUint32(&le.discarded, 1)
|
|
} else {
|
|
le.lastTime.Set(now)
|
|
discarded := atomic.SwapUint32(&le.discarded, 0)
|
|
if discarded > 0 {
|
|
Errorf("Discarded %d error messages", discarded)
|
|
}
|
|
|
|
execute()
|
|
}
|
|
}
|