[Fix] Correctly include query parameters for APIs whose request objects contain the body as a field

databricks-sdk-java/src/main/java/com/databricks/sdk/core/ApiClient.java

@@ -151,7 +151,7 @@ private ApiClient(Builder builder) {
     bodyLogger = new BodyLogger(mapper, 1024, debugTruncateBytes);
   }
 
-  private static <I> void setQuery(Request in, I entity) {
+  public static <I> void setQuery(Request in, I entity) {
     if (entity == null) {
       return;
     }
@@ -198,65 +198,10 @@ protected <I, O> O withJavaType(
     }
   }
 
-  public <O> O HEAD(String path, Class<O> target, Map<String, String> headers) {
-    return HEAD(path, null, target, headers);
-  }
-
-  public <I, O> O HEAD(String path, I in, Class<O> target, Map<String, String> headers) {
-    try {
-      return execute(prepareRequest("HEAD", path, in, headers), target);
-    } catch (IOException e) {
-      throw new DatabricksException("IO error: " + e.getMessage(), e);
-    }
-  }
-
-  public <O> O GET(String path, Class<O> target, Map<String, String> headers) {
-    return GET(path, null, target, headers);
-  }
-
-  public <I, O> O GET(String path, I in, Class<O> target, Map<String, String> headers) {
-    try {
-      return execute(prepareRequest("GET", path, in, headers), target);
-    } catch (IOException e) {
-      throw new DatabricksException("IO error: " + e.getMessage(), e);
-    }
-  }
-
-  public <O> O POST(String path, Class<O> target, Map<String, String> headers) {
-    try {
-      return execute(prepareRequest("POST", path, null, headers), target);
-    } catch (IOException e) {
-      throw new DatabricksException("IO error: " + e.getMessage(), e);
-    }
-  }
-
-  public <I, O> O POST(String path, I in, Class<O> target, Map<String, String> headers) {
-    try {
-      return execute(prepareRequest("POST", path, in, headers), target);
-    } catch (IOException e) {
-      throw new DatabricksException("IO error: " + e.getMessage(), e);
-    }
-  }
-
-  public <I, O> O PUT(String path, I in, Class<O> target, Map<String, String> headers) {
-    try {
-      return execute(prepareRequest("PUT", path, in, headers), target);
-    } catch (IOException e) {
-      throw new DatabricksException("IO error: " + e.getMessage(), e);
-    }
-  }
-
-  public <I, O> O PATCH(String path, I in, Class<O> target, Map<String, String> headers) {
-    try {
-      return execute(prepareRequest("PATCH", path, in, headers), target);
-    } catch (IOException e) {
-      throw new DatabricksException("IO error: " + e.getMessage(), e);
-    }
-  }
-
-  public <I, O> O DELETE(String path, I in, Class<O> target, Map<String, String> headers) {
+  public <I, O> O execute(
+      String method, String path, I in, Class<O> target, Map<String, String> headers) {
     try {
-      return execute(prepareRequest("DELETE", path, in, headers), target);
+      return execute(prepareRequest(method, path, in, headers), target);
     } catch (IOException e) {
       throw new DatabricksException("IO error: " + e.getMessage(), e);
     }
@@ -294,7 +239,7 @@ private <I> Request prepareRequest(String method, String path, I in, Map<String,
    * @param target Expected pojo type
    * @return POJO of requested type
    */
-  private <T> T execute(Request in, Class<T> target) throws IOException {
+  public <T> T execute(Request in, Class<T> target) throws IOException {
     Response out = getResponse(in);
     if (target == Void.class) {
       return null;
@@ -533,7 +478,7 @@ public <T> void deserialize(Response response, T object) throws IOException {
     }
   }
 
-  private String serialize(Object body) throws JsonProcessingException {
+  public String serialize(Object body) throws JsonProcessingException {
     if (body == null) {
       return null;
     }

databricks-sdk-java/src/main/java/com/databricks/sdk/core/DatabricksConfig.java

@@ -586,10 +586,13 @@ private OpenIDConnectEndpoints fetchDefaultOidcEndpoints() throws IOException {
             .withHttpClient(getHttpClient())
             .withGetHostFunc(v -> getHost())
             .build();
-    return apiClient.GET(
-        "/oidc/.well-known/oauth-authorization-server",
-        OpenIDConnectEndpoints.class,
-        new HashMap<>());
+    try {
+      return apiClient.execute(
+          new Request("GET", "/oidc/.well-known/oauth-authorization-server"),
+          OpenIDConnectEndpoints.class);
+    } catch (IOException e) {
+      throw new DatabricksException("IO error: " + e.getMessage(), e);
+    }
   }
 
   @Override

databricks-sdk-java/src/main/java/com/databricks/sdk/core/oauth/RefreshableTokenSource.java

@@ -4,6 +4,7 @@
 import com.databricks.sdk.core.DatabricksException;
 import com.databricks.sdk.core.http.FormRequest;
 import com.databricks.sdk.core.http.HttpClient;
+import com.databricks.sdk.core.http.Request;
 import java.time.LocalDateTime;
 import java.time.temporal.ChronoUnit;
 import java.util.Base64;
@@ -61,12 +62,11 @@ protected static Token retrieveToken(
         break;
     }
     headers.put("Content-Type", "application/x-www-form-urlencoded");
+    Request req = new Request("POST", tokenUrl, FormRequest.wrapValuesInList(params));
+    req.withHeaders(headers);
     try {
       ApiClient apiClient = new ApiClient.Builder().withHttpClient(hc).build();
-
-      OAuthResponse resp =
-          apiClient.POST(
-              tokenUrl, FormRequest.wrapValuesInList(params), OAuthResponse.class, headers);
+      OAuthResponse resp = apiClient.execute(req, OAuthResponse.class);
       if (resp.getErrorCode() != null) {
         throw new IllegalArgumentException(resp.getErrorCode() + ": " + resp.getErrorSummary());
       }