Limit added

models/DB.go

@@ -6,4 +6,5 @@ type ApiUsage struct {
 	Ip     string `json:"ip"`
 	Method string `json:"method"`
 	Date   string `json:"date"`
+	Cnt    int    `json:"cnt"`
 }

models/database.go

@@ -41,3 +41,9 @@ func WriteLog(sub string, token string, ip string, method string) {
 	input := ApiUsage{Sub: sub, Token: token, Ip: ip, Method: method, Date: date}
 	DB.Create(&input)
 }
+
+func Count(sub string) int {
+	apiUsage := ApiUsage{}
+	DB.Raw("SELECT count(1) AS cnt	FROM ( SELECT * FROM nodereal.api_usages WHERE date BETWEEN DATE_ADD(NOW(),INTERVAL -1 MONTH ) AND NOW() ) AS a	WHERE a.sub= ?", sub).Scan(&apiUsage)
+	return apiUsage.Cnt
+}

models/database_test.go

@@ -37,13 +37,14 @@ func TestDB(t *testing.T) {
 	// DB
 	ConnectDatabase()
 	log.Logger.Info.Println(DbStr)
-	Read()
-	Write()
-	Read()
-	Update()
-	Read()
-	Delete()
-	Read()
+	// Read()
+	// Write()
+	// Read()
+	// Update()
+	// Read()
+	// Delete()
+	// Read()
+	log.Logger.Info.Println("cnt", Count("9900cd1f-9cf1-44fb-857e-b189eb0fe249"))
 
 }
 
@@ -54,12 +55,6 @@ func Read() {
 	log.Logger.Info.Println(apiUsage)
 }
 
-func Check(token string) {
-	var apiUsage = ApiUsage{}
-	// DB.Where("token = ?", token).First(apiUsage)
-	DB.Exec("").First(apiUsage)
-}
-
 func Write() {
 	log.Logger.Info.Println("Write test")
 	input := ApiUsage{Token: "test", Method: "/asddf", Date: "2012-02-20 12:11:11"}

routes/routes.go

@@ -100,11 +100,16 @@ func AuthorizationMiddleware(c *gin.Context) {
 		return
 	}
 	// Usage check from DB
-	// 1. Read usage data ( * find by tokenString)
+	// 1. Read usage data ( sub )
+	requestsCnt := models.Count(claims.Sub)
 
-	// 2. Usage process
-
-	// 3. Write usage data
+	if requestsCnt >= 100000 {
+		log.Logger.Error.Println("Exceeded the limit.", "sub:", claims.Sub, "(", requestsCnt, ")")
+		c.JSON(http.StatusUnauthorized, gin.H{"status": http.StatusUnauthorized, "error": "You've already exceeded the limit. (100,000 Requests)"})
+		c.Abort()
+		return
+	}
+	// 2. Write usage data Log
 	models.WriteLog(claims.Sub, tokenString, c.ClientIP(), c.Request.URL.Path)
 
 	c.Next()