diff --git a/realtime-advanced/main.go b/realtime-advanced/main.go index 6aa781e..1f3c858 100644 --- a/realtime-advanced/main.go +++ b/realtime-advanced/main.go @@ -27,7 +27,7 @@ func StartGin() { gin.SetMode(gin.ReleaseMode) router := gin.New() - router.Use(rateLimit, gin.Recovery(), gin.Logger()) + router.Use(rateLimit, gin.Recovery()) router.LoadHTMLGlob("resources/*.templ.html") router.Static("/static", "resources/static") router.GET("/", index) diff --git a/realtime-advanced/routes.go b/realtime-advanced/routes.go index 8a7ac64..71c2ceb 100644 --- a/realtime-advanced/routes.go +++ b/realtime-advanced/routes.go @@ -13,9 +13,12 @@ import ( func rateLimit(c *gin.Context) { ip := c.ClientIP() value := ips.Add(ip, 1) - if value > 800 { - if int(value)%700 == 0 { - log.Printf("ip block: %s, count: %f\n", ip, value) + if int(value)%200 == 0 { + log.Printf("ip: %s, count: %f\n", ip, value) + } + if value > 300 { + if int(value)%200 == 0 { + log.Printf("ip blocked", ip, value) } c.AbortWithStatus(503) }