From 4a7475ba385ec6516b4851a81685ce14d65d1dcd Mon Sep 17 00:00:00 2001
From: Pieter Noordhuis <pieter.noordhuis@databricks.com>
Date: Fri, 26 Mar 2021 10:20:53 +0100
Subject: [PATCH] Fix the HTTP request rate limiter

The rate limiter was initialized with a maximum of 1 request per second with a
burst size of `RateLimitPerSecond`. Because we only use the `limiter.Wait()`
API, we can fix the burst at 1.
---
 common/client.go | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/common/client.go b/common/client.go
index 1f113c06fc..9939ecd3f5 100644
--- a/common/client.go
+++ b/common/client.go
@@ -198,7 +198,7 @@ func (c *DatabricksClient) configureHTTPCLient() {
 	if c.RateLimitPerSecond == 0 {
 		c.RateLimitPerSecond = DefaultRateLimitPerSecond
 	}
-	c.rateLimiter = rate.NewLimiter(rate.Every(1*time.Second), c.RateLimitPerSecond)
+	c.rateLimiter = rate.NewLimiter(rate.Limit(c.RateLimitPerSecond), 1)
 	// Set up a retryable HTTP Client to handle cases where the service returns
 	// a transient error on initial creation
 	retryDelayDuration := 10 * time.Second