OceanSlim/grain
1
0
Fork 0
mirror of https://github.com/0ceanSlim/grain.git synced 2024-11-21 16:17:13 +00:00
Code Issues Actions Packages Projects Releases Wiki Activity

refactor limit functions

Browse Source
This commit is contained in:
0ceanSlim 2024-09-05 11:09:03 -04:00
parent da46e23ae0
commit b22fbe6aa8
3 changed files with 6 additions and 6 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
config/rateLimiter.go
View File

@ -32,7 +32,7 @@ type RateLimiter struct {
var rateLimiterInstance *RateLimiter var rateLimiterInstance *RateLimiter
var rateOnce sync.Once var rateOnce sync.Once
func SetupRateLimiter(cfg *config.ServerConfig) { func SetRateLimit(cfg *config.ServerConfig) {
rateLimiter := NewRateLimiter( rateLimiter := NewRateLimiter(
rate.Limit(cfg.RateLimit.WsLimit), rate.Limit(cfg.RateLimit.WsLimit),
cfg.RateLimit.WsBurst, cfg.RateLimit.WsBurst,

6
config/setSizeLimit.go
View File

@ -12,13 +12,13 @@ type SizeLimiter struct {
mu sync.RWMutex mu sync.RWMutex
} }
func SetupSizeLimiter(cfg *config.ServerConfig) { func SetSizeLimit(cfg *config.ServerConfig) {
sizeLimiter := NewSizeLimiter(cfg.RateLimit.MaxEventSize) sizeLimiter := NewSizeLimiter(cfg.RateLimit.MaxEventSize)
for _, kindSizeLimit := range cfg.RateLimit.KindSizeLimits { for _, kindSizeLimit := range cfg.RateLimit.KindSizeLimits {
sizeLimiter.AddKindSizeLimit(kindSizeLimit.Kind, kindSizeLimit.MaxSize) sizeLimiter.AddKindSizeLimit(kindSizeLimit.Kind, kindSizeLimit.MaxSize)
} }
SetSizeLimit(sizeLimiter) SizeLimit(sizeLimiter)
} }
func NewSizeLimiter(globalMaxSize int) *SizeLimiter { func NewSizeLimiter(globalMaxSize int) *SizeLimiter {
@ -35,7 +35,7 @@ func GetSizeLimiter() *SizeLimiter {
return sizeLimiterInstance return sizeLimiterInstance
} }
func SetSizeLimit(sl *SizeLimiter) { func SizeLimit(sl *SizeLimiter) {
sizeOnce.Do(func() { sizeOnce.Do(func() {
sizeLimiterInstance = sl sizeLimiterInstance = sl
}) })

4
main.go
View File

@ -46,8 +46,8 @@ func main() {
log.Fatal("Error initializing database: ", err) log.Fatal("Error initializing database: ", err)
} }
config.SetupRateLimiter(cfg) config.SetRateLimit(cfg)
config.SetupSizeLimiter(cfg) config.SetSizeLimit(cfg)
config.ClearTemporaryBans() config.ClearTemporaryBans()