fix limit result

emanuelef · Oct 8, 2023 · 48a36c0 · 48a36c0
1 parent 3a242a3
commit 48a36c0
Showing 1 changed file with 16 additions and 3 deletions.
diff --git a/main.go b/main.go
@@ -290,17 +290,30 @@ func main() {
 	})
 
 	app.Get("/limits", func(c *fiber.Ctx) error {
-		randomIndex := rand.Intn(len(maps.Keys(ghStatClients)))
-		clientKey := c.Query("client", maps.Keys(ghStatClients)[randomIndex])
-		client, ok := ghStatClients[clientKey]
+		client, ok := ghStatClients["PAT"]
 		if !ok {
 			return c.Status(404).SendString("Resource not found")
 		}
 		result, err := client.GetCurrentLimits(ctx)
+
+		if err != nil {
+			log.Fatalf("Error getting limits %v", err)
+		}
+
+		client, ok = ghStatClients["PAT2"]
+		if !ok {
+			return c.Status(404).SendString("Resource not found")
+		}
+
+		tmpResult, err := client.GetCurrentLimits(ctx)
+
 		if err != nil {
 			log.Fatalf("Error getting limits %v", err)
 		}
 
+		result.Remaining += tmpResult.Remaining
+		result.Limit += tmpResult.Limit
+
 		return c.JSON(result)
 	})