Context

databricks · May 13, 2024 · 0df16da · 0df16da
1 parent b7c06af
commit 0df16da
Show file tree

Hide file tree

Showing 2 changed files with 1 addition and 2 deletions.
diff --git a/config/config.go b/config/config.go
@@ -220,7 +220,6 @@ func (c *Config) GetToken() (*oauth2.Token, error) {
 	if err != nil {
 		return nil, err
 	}
-	//ctx := c.refreshClient.InContextForOAuth2(c.refreshCtx)
 	if h, ok := c.credentialsProvider.(credentials.OAuthCredentialsProvider); ok {
 		return h.Token()
 	} else {

diff --git a/httpclient/oauth_token.go b/httpclient/oauth_token.go
@@ -35,7 +35,7 @@ func (c *ApiClient) GetOAuthToken(authDetails string, token *oauth2.Token) (*cre
 		WithUrlEncodedData(data),
 		WithResponseUnmarshal(&response),
 	}
-	err := c.Do(context.Background(), http.MethodPost, path, opts...)
+	err := c.Do(c.InContextForOAuth2(context.Background()), http.MethodPost, path, opts...)
 	if err != nil {
 		return nil, err
 	}