add rate limiter for api commands

danielgtaylor · Mar 16, 2023 · eb9d928 · eb9d928
1 parent 126ef04
commit eb9d928
Showing 1 changed file with 7 additions and 1 deletion.
diff --git a/cli/operation.go b/cli/operation.go
@@ -139,7 +139,13 @@ func (o Operation) command() *cobra.Command {
 
 			req, _ := http.NewRequest(o.Method, uri, body)
 			req.Header = headers
-			MakeRequestAndFormat(req)
+			rateLimiter := Throttle{
+				rate:     viper.GetInt("rsh-load-rate"),
+				requests: viper.GetInt("rsh-requests"),
+			}
+			rateLimiter.RateLimit(func() {
+				MakeRequestAndFormat(req)
+			})
 		},
 	}