Adds rate limiting

examples/realtime-advanced/main.go

@@ -26,7 +26,8 @@ func StartWorkers() {
 func StartGin() {
 	gin.SetMode(gin.ReleaseMode)
 
-	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)

examples/realtime-advanced/routes.go

@@ -3,12 +3,24 @@ package main
 import (
 	"html"
 	"io"
+	"log"
 	"strings"
 	"time"
 
 	"github.com/gin-gonic/gin"
 )
 
+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)
+		}
+		c.AbortWithStatus(503)
+	}
+}
+
 func index(c *gin.Context) {
 	c.Redirect(301, "/room/hn")
 }

examples/realtime-advanced/stats.go

@@ -8,6 +8,7 @@ import (
 	"github.com/manucorporat/stats"
 )
 
+var ips = stats.New()
 var messages = stats.New()
 var users = stats.New()
 var mutexStats sync.RWMutex