More aggressive rate limit

yuanyu90221 · May 14, 2015 · 237c042 · 237c042
1 parent 74967c0
commit 237c042
Show file tree

Hide file tree

Showing 2 changed files with 7 additions and 4 deletions.
diff --git 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
@@ -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)
  }