More changes!

This commit is contained in:
Manu Mtz-Almeida 2015-05-14 20:52:45 +02:00
parent c3ecb91dbf
commit 3db1479acc
1 changed files with 5 additions and 3 deletions

View File

@ -13,9 +13,11 @@ import (
func rateLimit(c *gin.Context) { func rateLimit(c *gin.Context) {
ip := c.ClientIP() ip := c.ClientIP()
value := ips.Add(ip, 1) value := ips.Add(ip, 1)
fmt.Printf("ip: %s, count: %f\n", ip, value) if int(value)%50 == 0 {
if value >= 300 { fmt.Printf("ip: %s, count: %d\n", ip, int(value))
if int(value)%300 == 0 { }
if value >= 200 {
if int(value)%200 == 0 {
fmt.Println("ip blocked") fmt.Println("ip blocked")
} }
c.AbortWithStatus(503) c.AbortWithStatus(503)