gin/examples/realtime-advanced/limit.go

23 lines
341 B
Go

package main
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 := uint64(ips.Add(ip, 1))
if value >= 400 {
if value%400 == 0 {
log.Printf("BlockedIP:%s Requests:%d\n", ip, value)
}
c.AbortWithStatus(401)
}
}