Better rate limiting

yuanyu90221 · May 13, 2015 · c2a089a · c2a089a
1 parent e8897d7
commit c2a089a
Show file tree

Hide file tree

Showing 2 changed files with 11 additions and 4 deletions.
diff --git a/realtime-advanced/limit.go b/realtime-advanced/limit.go
@@ -1,15 +1,22 @@
 package main
 
-import "github.com/gin-gonic/gin"
+import (
+ "log"
+
+ "github.com/gin-gonic/gin"
+)
 
 import "github.com/manucorporat/stats"
 
 var ips = stats.New()
 
 func ratelimit(c *gin.Context) {
  ip := c.ClientIP()
- value := ips.Add(ip, 1)
- if value > 400 {
+ value := uint64(ips.Add(ip, 1))
+ if value >= 400 {
+ if value%400 == 0 {
+ log.Printf("BlockedIP:%s Requests:%d\n", ip, value)
+ }
  c.AbortWithStatus(401)
  }
 }
diff --git a/realtime-advanced/main.go b/realtime-advanced/main.go
@@ -9,7 +9,7 @@ import (
 
 func main() {
  router := gin.New()
- router.Use(gin.Logger(), ratelimit)
+ router.Use(ratelimit, gin.Recovery(), gin.Logger())
 
  router.LoadHTMLGlob("resources/*.templ.html")
  router.Static("/static", "resources/static")