0ct0pu5/ente
1
0
Fork 0
Code Issues Pull requests Projects Releases Packages Wiki Activity Actions

[server] Update alert copy & frequency (#1129)

Browse source 
## Description

## Tests
This commit is contained in:
Neeraj Gupta 2024-03-18 10:15:16 +05:30 • committed by GitHub
commit c7fd976ab7
No known key found for this signature in database
GPG key ID: B5690EEEBB952194
3 changed files with 5 additions and 3 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

2
server/cmd/museum/main.go
View file

@ -178,7 +178,7 @@ func main() {
authCache := cache.New(1*time.Minute, 15*time.Minute)
accessTokenCache := cache.New(1*time.Minute, 15*time.Minute)
discordController := discord.NewDiscordController(userRepo, hostName, environment)
rateLimiter := middleware.NewRateLimitMiddleware(discordController, 5000, 5*time.Second)
rateLimiter := middleware.NewRateLimitMiddleware(discordController, 1000, 1*time.Second)
defer rateLimiter.Stop()
emailNotificationCtrl := &email.EmailNotificationController{

2
server/pkg/controller/discord/discord.go
View file

@ -106,7 +106,7 @@ func (c *DiscordController) NotifyAccountDelete(userID int64, paymentProvider st
}
func (c *DiscordController) NotifyPotentialAbuse(message string) {
c.Notify(message)
c.Notify(fmt.Sprintf("%s: %s", c.HostName, message))
}
func (c *DiscordController) getTimeSinceSignUp(userID int64) string {

4
server/pkg/middleware/rate_limit.go
View file

@ -76,7 +76,9 @@ func rateLimiter(interval string) *limiter.Limiter {
func (r *RateLimitMiddleware) GlobalRateLimiter() gin.HandlerFunc {
return func(c *gin.Context) {
if !r.Increment() {
go r.discordCtrl.NotifyPotentialAbuse("Global rate limit breached")
if r.count%100 == 0 {
go r.discordCtrl.NotifyPotentialAbuse(fmt.Sprintf("Global ratelimit (%d) breached %d", r.limit, r.count))
}
c.AbortWithStatusJSON(http.StatusTooManyRequests, gin.H{"error": "Rate limit breached, try later"})
return
}