Add rate limitting

d1jk · May 13, 2015 · 964f6e3 · 964f6e3
1 parent a721061
commit 964f6e3
Show file tree

Hide file tree

Showing 2 changed files with 18 additions and 1 deletion.
diff --git a/realtime-advanced/limit.go b/realtime-advanced/limit.go
@@ -0,0 +1,15 @@
+package main
+
+import "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 > 1000 {
+ c.AbortWithStatus(401)
+ }
+}
diff --git a/realtime-advanced/main.go b/realtime-advanced/main.go
@@ -8,7 +8,9 @@ import (
 )
 
 func main() {
- router := gin.Default()
+ router := gin.New()
+ router.Use(ratelimit, gin.Recovery(), gin.Logger())
+
  router.LoadHTMLGlob("resources/*.templ.html")
  router.Static("/static", "resources/static")
  router.GET("/", index)