diff --git a/.codegen/impl.java.tmpl b/.codegen/impl.java.tmpl index 9cf5b1354..e9a3d5541 100644 --- a/.codegen/impl.java.tmpl +++ b/.codegen/impl.java.tmpl @@ -4,6 +4,7 @@ package com.databricks.sdk.service.{{.Package.Name}}; import java.io.IOException; import java.util.Collection; import java.util.Map; +import java.util.HashMap; import com.databricks.sdk.core.ApiClient; import com.databricks.sdk.core.DatabricksException; @@ -26,18 +27,26 @@ class {{.PascalName}}Impl implements {{.PascalName}}Service { String path = {{if .PathParts -}} String.format("{{range .PathParts}}{{.Prefix}}{{if or .Field .IsAccountId}}%s{{end}}{{ end }}"{{ range .PathParts }}{{if .Field}}, request.get{{.Field.PascalName}}(){{ else if .IsAccountId }}, apiClient.configuredAccountID(){{end}}{{ end }}) {{- else}}"{{.Path}}"{{end}}; + {{ template "headers" . }} {{if .Response -}} - {{- if .Response.ArrayValue -}} return apiClient.getCollection(path, null, {{template "type" .Response.ArrayValue}}.class); - {{- else if .Response.MapValue -}} return apiClient.getStringMap(path, request); + {{- if .Response.ArrayValue -}} return apiClient.getCollection(path, null, {{template "type" .Response.ArrayValue}}.class, headers); + {{- else if .Response.MapValue -}} return apiClient.getStringMap(path, request, headers); {{- else -}} return apiClient.{{.Verb}}(path{{if .Request}}, request{{end}}{{if .Response -}}, {{if .Response.ArrayValue }}Collection {{- else if .Response.MapValue }}Map {{- else}}{{template "type" .Response}} - {{- end -}}{{else}}, Void{{end}}.class);{{end}} + {{- end -}}{{else}}, Void{{end}}.class, headers);{{end}} {{- else -}}apiClient.{{.Verb}}(path{{if .Request}}, request{{end}}{{if .Response -}}, {{if .Response.ArrayValue }}Collection {{- else if .Response.MapValue }}Map {{- else}}{{template "type" .Response}} - {{- end -}}{{else}}, Void{{end}}.class); + {{- end -}}{{else}}, Void{{end}}.class, headers); {{end}} } {{end}} -} \ No newline at end of file +} + +{{ define "headers" -}} + Map headers = new HashMap<>(); + {{- range $key, $value := .FixedRequestHeaders }} + headers.put("{{$key}}", "{{$value}}"); + {{- end -}} +{{- end }} diff --git a/databricks-sdk-java/src/main/java/com/databricks/sdk/core/ApiClient.java b/databricks-sdk-java/src/main/java/com/databricks/sdk/core/ApiClient.java index 5bd02d015..b5dff7e3e 100644 --- a/databricks-sdk-java/src/main/java/com/databricks/sdk/core/ApiClient.java +++ b/databricks-sdk-java/src/main/java/com/databricks/sdk/core/ApiClient.java @@ -13,9 +13,7 @@ import com.fasterxml.jackson.databind.ObjectMapper; import com.fasterxml.jackson.databind.SerializationFeature; import java.io.IOException; -import java.util.Collection; -import java.util.Map; -import java.util.Random; +import java.util.*; import org.slf4j.Logger; import org.slf4j.LoggerFactory; @@ -85,30 +83,46 @@ private ObjectMapper makeObjectMapper() { return mapper; } - private Request withQuery(Request in, I entity) { + private static void setQuery(Request in, I entity) { if (entity == null) { - return in; + return; } for (GrpcTranscodingQueryParamsSerializer.HeaderEntry e : GrpcTranscodingQueryParamsSerializer.serialize(entity)) { in.withQueryParam(e.getKey(), e.getValue()); } - return in; } - public Collection getCollection(String path, I in, Class element) { + private static void setHeaders(Request in, Map headers) { + if (headers == null) { + return; + } + for (Map.Entry e : headers.entrySet()) { + in.withHeader(e.getKey(), e.getValue()); + } + } + + public Collection getCollection( + String path, I in, Class element, Map headers) { return withJavaType( - path, in, mapper.getTypeFactory().constructCollectionType(Collection.class, element)); + path, + in, + mapper.getTypeFactory().constructCollectionType(Collection.class, element), + headers); } - public Map getStringMap(String path, I in) { + public Map getStringMap(String path, I in, Map headers) { return withJavaType( - path, in, mapper.getTypeFactory().constructMapType(Map.class, String.class, String.class)); + path, + in, + mapper.getTypeFactory().constructMapType(Map.class, String.class, String.class), + headers); } - protected O withJavaType(String path, I in, JavaType javaType) { + protected O withJavaType( + String path, I in, JavaType javaType, Map headers) { try { - Request request = withQuery(new Request("GET", path), in); + Request request = prepareRequest("GET", path, in, headers); Response response = getResponse(request); return deserialize(response.getBody(), javaType); } catch (IOException e) { @@ -116,50 +130,58 @@ protected O withJavaType(String path, I in, JavaType javaType) { } } - public O GET(String path, Class target) { - return GET(path, null, target); + public O GET(String path, Class target, Map headers) { + return GET(path, null, target, headers); } - public O GET(String path, I in, Class target) { + public O GET(String path, I in, Class target, Map headers) { try { - return execute(withQuery(new Request("GET", path), in), target); + return execute(prepareRequest("GET", path, in, headers), target); } catch (IOException e) { throw new DatabricksException("IO error: " + e.getMessage(), e); } } - public O POST(String path, I in, Class target) { + public O POST(String path, I in, Class target, Map headers) { try { - return execute(new Request("POST", path, serialize(in)), target); + return execute(prepareRequest("POST", path, in, headers), target); } catch (IOException e) { throw new DatabricksException("IO error: " + e.getMessage(), e); } } - public O PUT(String path, I in, Class target) { + public O PUT(String path, I in, Class target, Map headers) { try { - return execute(new Request("PUT", path, serialize(in)), target); + return execute(prepareRequest("PUT", path, in, headers), target); } catch (IOException e) { throw new DatabricksException("IO error: " + e.getMessage(), e); } } - public O PATCH(String path, I in, Class target) { + public O PATCH(String path, I in, Class target, Map headers) { try { - return execute(new Request("PATCH", path, serialize(in)), target); + return execute(prepareRequest("PATCH", path, in, headers), target); } catch (IOException e) { throw new DatabricksException("IO error: " + e.getMessage(), e); } } - public O DELETE(String path, I in, Class target) { + public O DELETE(String path, I in, Class target, Map headers) { try { - return execute(withQuery(new Request("DELETE", path), in), target); + return execute(prepareRequest("DELETE", path, in, headers), target); } catch (IOException e) { throw new DatabricksException("IO error: " + e.getMessage(), e); } } + private Request prepareRequest( + String method, String path, I in, Map headers) { + Request req = new Request(method, path); + setQuery(req, in); + setHeaders(req, headers); + return req; + } + /** * Executes HTTP request with retries and converts it to proper POJO * @@ -177,7 +199,6 @@ private T execute(Request in, Class target) throws IOException { private Response getResponse(Request in) { in.withUrl(config.getHost() + in.getUrl()); - in.withHeader("Accept", "application/json"); return executeInner(in); } diff --git a/databricks-sdk-java/src/main/java/com/databricks/sdk/service/billing/BillableUsageImpl.java b/databricks-sdk-java/src/main/java/com/databricks/sdk/service/billing/BillableUsageImpl.java index 1b7836d98..bf243b185 100755 --- a/databricks-sdk-java/src/main/java/com/databricks/sdk/service/billing/BillableUsageImpl.java +++ b/databricks-sdk-java/src/main/java/com/databricks/sdk/service/billing/BillableUsageImpl.java @@ -3,6 +3,8 @@ import com.databricks.sdk.core.ApiClient; import com.databricks.sdk.support.Generated; +import java.util.HashMap; +import java.util.Map; /** Package-local implementation of BillableUsage */ @Generated @@ -17,6 +19,7 @@ public BillableUsageImpl(ApiClient apiClient) { public void download(DownloadRequest request) { String path = String.format("/api/2.0/accounts/%s/usage/download", apiClient.configuredAccountID()); - apiClient.GET(path, request, Void.class); + Map headers = new HashMap<>(); + apiClient.GET(path, request, Void.class, headers); } } diff --git a/databricks-sdk-java/src/main/java/com/databricks/sdk/service/billing/BudgetsImpl.java b/databricks-sdk-java/src/main/java/com/databricks/sdk/service/billing/BudgetsImpl.java index 082da2f4e..874f31f81 100755 --- a/databricks-sdk-java/src/main/java/com/databricks/sdk/service/billing/BudgetsImpl.java +++ b/databricks-sdk-java/src/main/java/com/databricks/sdk/service/billing/BudgetsImpl.java @@ -3,6 +3,8 @@ import com.databricks.sdk.core.ApiClient; import com.databricks.sdk.support.Generated; +import java.util.HashMap; +import java.util.Map; /** Package-local implementation of Budgets */ @Generated @@ -16,7 +18,10 @@ public BudgetsImpl(ApiClient apiClient) { @Override public WrappedBudgetWithStatus create(WrappedBudget request) { String path = String.format("/api/2.0/accounts/%s/budget", apiClient.configuredAccountID()); - return apiClient.POST(path, request, WrappedBudgetWithStatus.class); + Map headers = new HashMap<>(); + headers.put("Accept", "application/json"); + headers.put("Content-Type", "application/json"); + return apiClient.POST(path, request, WrappedBudgetWithStatus.class, headers); } @Override @@ -25,7 +30,9 @@ public void delete(DeleteBudgetRequest request) { String.format( "/api/2.0/accounts/%s/budget/%s", apiClient.configuredAccountID(), request.getBudgetId()); - apiClient.DELETE(path, request, Void.class); + Map headers = new HashMap<>(); + headers.put("Accept", "application/json"); + apiClient.DELETE(path, request, Void.class, headers); } @Override @@ -34,13 +41,17 @@ public WrappedBudgetWithStatus get(GetBudgetRequest request) { String.format( "/api/2.0/accounts/%s/budget/%s", apiClient.configuredAccountID(), request.getBudgetId()); - return apiClient.GET(path, request, WrappedBudgetWithStatus.class); + Map headers = new HashMap<>(); + headers.put("Accept", "application/json"); + return apiClient.GET(path, request, WrappedBudgetWithStatus.class, headers); } @Override public BudgetList list() { String path = String.format("/api/2.0/accounts/%s/budget", apiClient.configuredAccountID()); - return apiClient.GET(path, BudgetList.class); + Map headers = new HashMap<>(); + headers.put("Accept", "application/json"); + return apiClient.GET(path, BudgetList.class, headers); } @Override @@ -49,6 +60,9 @@ public void update(WrappedBudget request) { String.format( "/api/2.0/accounts/%s/budget/%s", apiClient.configuredAccountID(), request.getBudgetId()); - apiClient.PATCH(path, request, Void.class); + Map headers = new HashMap<>(); + headers.put("Accept", "application/json"); + headers.put("Content-Type", "application/json"); + apiClient.PATCH(path, request, Void.class, headers); } } diff --git a/databricks-sdk-java/src/main/java/com/databricks/sdk/service/billing/LogDeliveryImpl.java b/databricks-sdk-java/src/main/java/com/databricks/sdk/service/billing/LogDeliveryImpl.java index 7a674333d..bd0ed9b90 100755 --- a/databricks-sdk-java/src/main/java/com/databricks/sdk/service/billing/LogDeliveryImpl.java +++ b/databricks-sdk-java/src/main/java/com/databricks/sdk/service/billing/LogDeliveryImpl.java @@ -3,6 +3,8 @@ import com.databricks.sdk.core.ApiClient; import com.databricks.sdk.support.Generated; +import java.util.HashMap; +import java.util.Map; /** Package-local implementation of LogDelivery */ @Generated @@ -17,7 +19,10 @@ public LogDeliveryImpl(ApiClient apiClient) { public WrappedLogDeliveryConfiguration create(WrappedCreateLogDeliveryConfiguration request) { String path = String.format("/api/2.0/accounts/%s/log-delivery", apiClient.configuredAccountID()); - return apiClient.POST(path, request, WrappedLogDeliveryConfiguration.class); + Map headers = new HashMap<>(); + headers.put("Accept", "application/json"); + headers.put("Content-Type", "application/json"); + return apiClient.POST(path, request, WrappedLogDeliveryConfiguration.class, headers); } @Override @@ -26,14 +31,18 @@ public WrappedLogDeliveryConfiguration get(GetLogDeliveryRequest request) { String.format( "/api/2.0/accounts/%s/log-delivery/%s", apiClient.configuredAccountID(), request.getLogDeliveryConfigurationId()); - return apiClient.GET(path, request, WrappedLogDeliveryConfiguration.class); + Map headers = new HashMap<>(); + headers.put("Accept", "application/json"); + return apiClient.GET(path, request, WrappedLogDeliveryConfiguration.class, headers); } @Override public WrappedLogDeliveryConfigurations list(ListLogDeliveryRequest request) { String path = String.format("/api/2.0/accounts/%s/log-delivery", apiClient.configuredAccountID()); - return apiClient.GET(path, request, WrappedLogDeliveryConfigurations.class); + Map headers = new HashMap<>(); + headers.put("Accept", "application/json"); + return apiClient.GET(path, request, WrappedLogDeliveryConfigurations.class, headers); } @Override @@ -42,6 +51,9 @@ public void patchStatus(UpdateLogDeliveryConfigurationStatusRequest request) { String.format( "/api/2.0/accounts/%s/log-delivery/%s", apiClient.configuredAccountID(), request.getLogDeliveryConfigurationId()); - apiClient.PATCH(path, request, Void.class); + Map headers = new HashMap<>(); + headers.put("Accept", "application/json"); + headers.put("Content-Type", "application/json"); + apiClient.PATCH(path, request, Void.class, headers); } } diff --git a/databricks-sdk-java/src/main/java/com/databricks/sdk/service/catalog/AccountMetastoreAssignmentsImpl.java b/databricks-sdk-java/src/main/java/com/databricks/sdk/service/catalog/AccountMetastoreAssignmentsImpl.java index 4ce89f57d..9674b15a5 100755 --- a/databricks-sdk-java/src/main/java/com/databricks/sdk/service/catalog/AccountMetastoreAssignmentsImpl.java +++ b/databricks-sdk-java/src/main/java/com/databricks/sdk/service/catalog/AccountMetastoreAssignmentsImpl.java @@ -4,6 +4,8 @@ import com.databricks.sdk.core.ApiClient; import com.databricks.sdk.support.Generated; import java.util.Collection; +import java.util.HashMap; +import java.util.Map; /** Package-local implementation of AccountMetastoreAssignments */ @Generated @@ -20,7 +22,10 @@ public void create(AccountsCreateMetastoreAssignment request) { String.format( "/api/2.0/accounts/%s/workspaces/%s/metastores/%s", apiClient.configuredAccountID(), request.getWorkspaceId(), request.getMetastoreId()); - apiClient.POST(path, request, Void.class); + Map headers = new HashMap<>(); + headers.put("Accept", "application/json"); + headers.put("Content-Type", "application/json"); + apiClient.POST(path, request, Void.class, headers); } @Override @@ -29,7 +34,9 @@ public void delete(DeleteAccountMetastoreAssignmentRequest request) { String.format( "/api/2.0/accounts/%s/workspaces/%s/metastores/%s", apiClient.configuredAccountID(), request.getWorkspaceId(), request.getMetastoreId()); - apiClient.DELETE(path, request, Void.class); + Map headers = new HashMap<>(); + headers.put("Accept", "application/json"); + apiClient.DELETE(path, request, Void.class, headers); } @Override @@ -38,7 +45,9 @@ public AccountsMetastoreAssignment get(GetAccountMetastoreAssignmentRequest requ String.format( "/api/2.0/accounts/%s/workspaces/%s/metastore", apiClient.configuredAccountID(), request.getWorkspaceId()); - return apiClient.GET(path, request, AccountsMetastoreAssignment.class); + Map headers = new HashMap<>(); + headers.put("Accept", "application/json"); + return apiClient.GET(path, request, AccountsMetastoreAssignment.class, headers); } @Override @@ -47,7 +56,9 @@ public Collection list(ListAccountMetastoreAssignmentsReque String.format( "/api/2.0/accounts/%s/metastores/%s/workspaces", apiClient.configuredAccountID(), request.getMetastoreId()); - return apiClient.getCollection(path, null, MetastoreAssignment.class); + Map headers = new HashMap<>(); + headers.put("Accept", "application/json"); + return apiClient.getCollection(path, null, MetastoreAssignment.class, headers); } @Override @@ -56,6 +67,9 @@ public void update(AccountsUpdateMetastoreAssignment request) { String.format( "/api/2.0/accounts/%s/workspaces/%s/metastores/%s", apiClient.configuredAccountID(), request.getWorkspaceId(), request.getMetastoreId()); - apiClient.PUT(path, request, Void.class); + Map headers = new HashMap<>(); + headers.put("Accept", "application/json"); + headers.put("Content-Type", "application/json"); + apiClient.PUT(path, request, Void.class, headers); } } diff --git a/databricks-sdk-java/src/main/java/com/databricks/sdk/service/catalog/AccountMetastoresImpl.java b/databricks-sdk-java/src/main/java/com/databricks/sdk/service/catalog/AccountMetastoresImpl.java index f2b0961f7..cddea3a03 100755 --- a/databricks-sdk-java/src/main/java/com/databricks/sdk/service/catalog/AccountMetastoresImpl.java +++ b/databricks-sdk-java/src/main/java/com/databricks/sdk/service/catalog/AccountMetastoresImpl.java @@ -3,6 +3,8 @@ import com.databricks.sdk.core.ApiClient; import com.databricks.sdk.support.Generated; +import java.util.HashMap; +import java.util.Map; /** Package-local implementation of AccountMetastores */ @Generated @@ -16,7 +18,10 @@ public AccountMetastoresImpl(ApiClient apiClient) { @Override public AccountsMetastoreInfo create(AccountsCreateMetastore request) { String path = String.format("/api/2.0/accounts/%s/metastores", apiClient.configuredAccountID()); - return apiClient.POST(path, request, AccountsMetastoreInfo.class); + Map headers = new HashMap<>(); + headers.put("Accept", "application/json"); + headers.put("Content-Type", "application/json"); + return apiClient.POST(path, request, AccountsMetastoreInfo.class, headers); } @Override @@ -25,7 +30,9 @@ public void delete(DeleteAccountMetastoreRequest request) { String.format( "/api/2.0/accounts/%s/metastores/%s", apiClient.configuredAccountID(), request.getMetastoreId()); - apiClient.DELETE(path, request, Void.class); + Map headers = new HashMap<>(); + headers.put("Accept", "application/json"); + apiClient.DELETE(path, request, Void.class, headers); } @Override @@ -34,13 +41,17 @@ public AccountsMetastoreInfo get(GetAccountMetastoreRequest request) { String.format( "/api/2.0/accounts/%s/metastores/%s", apiClient.configuredAccountID(), request.getMetastoreId()); - return apiClient.GET(path, request, AccountsMetastoreInfo.class); + Map headers = new HashMap<>(); + headers.put("Accept", "application/json"); + return apiClient.GET(path, request, AccountsMetastoreInfo.class, headers); } @Override public ListMetastoresResponse list() { String path = String.format("/api/2.0/accounts/%s/metastores", apiClient.configuredAccountID()); - return apiClient.GET(path, ListMetastoresResponse.class); + Map headers = new HashMap<>(); + headers.put("Accept", "application/json"); + return apiClient.GET(path, ListMetastoresResponse.class, headers); } @Override @@ -49,6 +60,9 @@ public AccountsMetastoreInfo update(AccountsUpdateMetastore request) { String.format( "/api/2.0/accounts/%s/metastores/%s", apiClient.configuredAccountID(), request.getMetastoreId()); - return apiClient.PUT(path, request, AccountsMetastoreInfo.class); + Map headers = new HashMap<>(); + headers.put("Accept", "application/json"); + headers.put("Content-Type", "application/json"); + return apiClient.PUT(path, request, AccountsMetastoreInfo.class, headers); } } diff --git a/databricks-sdk-java/src/main/java/com/databricks/sdk/service/catalog/AccountStorageCredentialsImpl.java b/databricks-sdk-java/src/main/java/com/databricks/sdk/service/catalog/AccountStorageCredentialsImpl.java index 11271c6b5..51aba9700 100755 --- a/databricks-sdk-java/src/main/java/com/databricks/sdk/service/catalog/AccountStorageCredentialsImpl.java +++ b/databricks-sdk-java/src/main/java/com/databricks/sdk/service/catalog/AccountStorageCredentialsImpl.java @@ -3,6 +3,8 @@ import com.databricks.sdk.core.ApiClient; import com.databricks.sdk.support.Generated; +import java.util.HashMap; +import java.util.Map; /** Package-local implementation of AccountStorageCredentials */ @Generated @@ -19,7 +21,10 @@ public AccountsStorageCredentialInfo create(AccountsCreateStorageCredential requ String.format( "/api/2.0/accounts/%s/metastores/%s/storage-credentials", apiClient.configuredAccountID(), request.getMetastoreId()); - return apiClient.POST(path, request, AccountsStorageCredentialInfo.class); + Map headers = new HashMap<>(); + headers.put("Accept", "application/json"); + headers.put("Content-Type", "application/json"); + return apiClient.POST(path, request, AccountsStorageCredentialInfo.class, headers); } @Override @@ -28,7 +33,9 @@ public void delete(DeleteAccountStorageCredentialRequest request) { String.format( "/api/2.0/accounts/%s/metastores/%s/storage-credentials/", apiClient.configuredAccountID(), request.getMetastoreId()); - apiClient.DELETE(path, request, Void.class); + Map headers = new HashMap<>(); + headers.put("Accept", "application/json"); + apiClient.DELETE(path, request, Void.class, headers); } @Override @@ -37,7 +44,9 @@ public AccountsStorageCredentialInfo get(GetAccountStorageCredentialRequest requ String.format( "/api/2.0/accounts/%s/metastores/%s/storage-credentials/", apiClient.configuredAccountID(), request.getMetastoreId()); - return apiClient.GET(path, request, AccountsStorageCredentialInfo.class); + Map headers = new HashMap<>(); + headers.put("Accept", "application/json"); + return apiClient.GET(path, request, AccountsStorageCredentialInfo.class, headers); } @Override @@ -46,7 +55,9 @@ public ListStorageCredentialsResponse list(ListAccountStorageCredentialsRequest String.format( "/api/2.0/accounts/%s/metastores/%s/storage-credentials", apiClient.configuredAccountID(), request.getMetastoreId()); - return apiClient.GET(path, request, ListStorageCredentialsResponse.class); + Map headers = new HashMap<>(); + headers.put("Accept", "application/json"); + return apiClient.GET(path, request, ListStorageCredentialsResponse.class, headers); } @Override @@ -55,6 +66,9 @@ public AccountsStorageCredentialInfo update(AccountsUpdateStorageCredential requ String.format( "/api/2.0/accounts/%s/metastores/%s/storage-credentials/", apiClient.configuredAccountID(), request.getMetastoreId()); - return apiClient.PUT(path, request, AccountsStorageCredentialInfo.class); + Map headers = new HashMap<>(); + headers.put("Accept", "application/json"); + headers.put("Content-Type", "application/json"); + return apiClient.PUT(path, request, AccountsStorageCredentialInfo.class, headers); } } diff --git a/databricks-sdk-java/src/main/java/com/databricks/sdk/service/catalog/ArtifactAllowlistsImpl.java b/databricks-sdk-java/src/main/java/com/databricks/sdk/service/catalog/ArtifactAllowlistsImpl.java index b32cd2537..5a7a55677 100755 --- a/databricks-sdk-java/src/main/java/com/databricks/sdk/service/catalog/ArtifactAllowlistsImpl.java +++ b/databricks-sdk-java/src/main/java/com/databricks/sdk/service/catalog/ArtifactAllowlistsImpl.java @@ -3,6 +3,8 @@ import com.databricks.sdk.core.ApiClient; import com.databricks.sdk.support.Generated; +import java.util.HashMap; +import java.util.Map; /** Package-local implementation of ArtifactAllowlists */ @Generated @@ -17,13 +19,18 @@ public ArtifactAllowlistsImpl(ApiClient apiClient) { public ArtifactAllowlistInfo get(GetArtifactAllowlistRequest request) { String path = String.format("/api/2.1/unity-catalog/artifact-allowlists/%s", request.getArtifactType()); - return apiClient.GET(path, request, ArtifactAllowlistInfo.class); + Map headers = new HashMap<>(); + headers.put("Accept", "application/json"); + return apiClient.GET(path, request, ArtifactAllowlistInfo.class, headers); } @Override public ArtifactAllowlistInfo update(SetArtifactAllowlist request) { String path = String.format("/api/2.1/unity-catalog/artifact-allowlists/%s", request.getArtifactType()); - return apiClient.PUT(path, request, ArtifactAllowlistInfo.class); + Map headers = new HashMap<>(); + headers.put("Accept", "application/json"); + headers.put("Content-Type", "application/json"); + return apiClient.PUT(path, request, ArtifactAllowlistInfo.class, headers); } } diff --git a/databricks-sdk-java/src/main/java/com/databricks/sdk/service/catalog/CatalogsImpl.java b/databricks-sdk-java/src/main/java/com/databricks/sdk/service/catalog/CatalogsImpl.java index 8a8ad220b..d2b78e313 100755 --- a/databricks-sdk-java/src/main/java/com/databricks/sdk/service/catalog/CatalogsImpl.java +++ b/databricks-sdk-java/src/main/java/com/databricks/sdk/service/catalog/CatalogsImpl.java @@ -3,6 +3,8 @@ import com.databricks.sdk.core.ApiClient; import com.databricks.sdk.support.Generated; +import java.util.HashMap; +import java.util.Map; /** Package-local implementation of Catalogs */ @Generated @@ -16,30 +18,42 @@ public CatalogsImpl(ApiClient apiClient) { @Override public CatalogInfo create(CreateCatalog request) { String path = "/api/2.1/unity-catalog/catalogs"; - return apiClient.POST(path, request, CatalogInfo.class); + Map headers = new HashMap<>(); + headers.put("Accept", "application/json"); + headers.put("Content-Type", "application/json"); + return apiClient.POST(path, request, CatalogInfo.class, headers); } @Override public void delete(DeleteCatalogRequest request) { String path = String.format("/api/2.1/unity-catalog/catalogs/%s", request.getName()); - apiClient.DELETE(path, request, Void.class); + Map headers = new HashMap<>(); + headers.put("Accept", "application/json"); + apiClient.DELETE(path, request, Void.class, headers); } @Override public CatalogInfo get(GetCatalogRequest request) { String path = String.format("/api/2.1/unity-catalog/catalogs/%s", request.getName()); - return apiClient.GET(path, request, CatalogInfo.class); + Map headers = new HashMap<>(); + headers.put("Accept", "application/json"); + return apiClient.GET(path, request, CatalogInfo.class, headers); } @Override public ListCatalogsResponse list() { String path = "/api/2.1/unity-catalog/catalogs"; - return apiClient.GET(path, ListCatalogsResponse.class); + Map headers = new HashMap<>(); + headers.put("Accept", "application/json"); + return apiClient.GET(path, ListCatalogsResponse.class, headers); } @Override public CatalogInfo update(UpdateCatalog request) { String path = String.format("/api/2.1/unity-catalog/catalogs/%s", request.getName()); - return apiClient.PATCH(path, request, CatalogInfo.class); + Map headers = new HashMap<>(); + headers.put("Accept", "application/json"); + headers.put("Content-Type", "application/json"); + return apiClient.PATCH(path, request, CatalogInfo.class, headers); } } diff --git a/databricks-sdk-java/src/main/java/com/databricks/sdk/service/catalog/ConnectionsImpl.java b/databricks-sdk-java/src/main/java/com/databricks/sdk/service/catalog/ConnectionsImpl.java index b3c849ddb..05032a1dd 100755 --- a/databricks-sdk-java/src/main/java/com/databricks/sdk/service/catalog/ConnectionsImpl.java +++ b/databricks-sdk-java/src/main/java/com/databricks/sdk/service/catalog/ConnectionsImpl.java @@ -3,6 +3,8 @@ import com.databricks.sdk.core.ApiClient; import com.databricks.sdk.support.Generated; +import java.util.HashMap; +import java.util.Map; /** Package-local implementation of Connections */ @Generated @@ -16,30 +18,42 @@ public ConnectionsImpl(ApiClient apiClient) { @Override public ConnectionInfo create(CreateConnection request) { String path = "/api/2.1/unity-catalog/connections"; - return apiClient.POST(path, request, ConnectionInfo.class); + Map headers = new HashMap<>(); + headers.put("Accept", "application/json"); + headers.put("Content-Type", "application/json"); + return apiClient.POST(path, request, ConnectionInfo.class, headers); } @Override public void delete(DeleteConnectionRequest request) { String path = String.format("/api/2.1/unity-catalog/connections/%s", request.getNameArg()); - apiClient.DELETE(path, request, Void.class); + Map headers = new HashMap<>(); + headers.put("Accept", "application/json"); + apiClient.DELETE(path, request, Void.class, headers); } @Override public ConnectionInfo get(GetConnectionRequest request) { String path = String.format("/api/2.1/unity-catalog/connections/%s", request.getNameArg()); - return apiClient.GET(path, request, ConnectionInfo.class); + Map headers = new HashMap<>(); + headers.put("Accept", "application/json"); + return apiClient.GET(path, request, ConnectionInfo.class, headers); } @Override public ListConnectionsResponse list() { String path = "/api/2.1/unity-catalog/connections"; - return apiClient.GET(path, ListConnectionsResponse.class); + Map headers = new HashMap<>(); + headers.put("Accept", "application/json"); + return apiClient.GET(path, ListConnectionsResponse.class, headers); } @Override public ConnectionInfo update(UpdateConnection request) { String path = String.format("/api/2.1/unity-catalog/connections/%s", request.getNameArg()); - return apiClient.PATCH(path, request, ConnectionInfo.class); + Map headers = new HashMap<>(); + headers.put("Accept", "application/json"); + headers.put("Content-Type", "application/json"); + return apiClient.PATCH(path, request, ConnectionInfo.class, headers); } } diff --git a/databricks-sdk-java/src/main/java/com/databricks/sdk/service/catalog/CreateMetastoreAssignmentsResponseItem.java b/databricks-sdk-java/src/main/java/com/databricks/sdk/service/catalog/CreateMetastoreAssignmentsResponseItem.java deleted file mode 100755 index 63275bcfb..000000000 --- a/databricks-sdk-java/src/main/java/com/databricks/sdk/service/catalog/CreateMetastoreAssignmentsResponseItem.java +++ /dev/null @@ -1,75 +0,0 @@ -// Code generated from OpenAPI specs by Databricks SDK Generator. DO NOT EDIT. - -package com.databricks.sdk.service.catalog; - -import com.databricks.sdk.support.Generated; -import com.databricks.sdk.support.ToStringer; -import com.fasterxml.jackson.annotation.JsonProperty; -import java.util.Objects; - -@Generated -public class CreateMetastoreAssignmentsResponseItem { - /** A human-readable message describing the outcome of the creation */ - @JsonProperty("message") - private String message; - - /** */ - @JsonProperty("metastore_assignment") - private MetastoreAssignment metastoreAssignment; - - /** The returned HTTP status code for an individual creation in the batch */ - @JsonProperty("status_code") - private Long statusCode; - - public CreateMetastoreAssignmentsResponseItem setMessage(String message) { - this.message = message; - return this; - } - - public String getMessage() { - return message; - } - - public CreateMetastoreAssignmentsResponseItem setMetastoreAssignment( - MetastoreAssignment metastoreAssignment) { - this.metastoreAssignment = metastoreAssignment; - return this; - } - - public MetastoreAssignment getMetastoreAssignment() { - return metastoreAssignment; - } - - public CreateMetastoreAssignmentsResponseItem setStatusCode(Long statusCode) { - this.statusCode = statusCode; - return this; - } - - public Long getStatusCode() { - return statusCode; - } - - @Override - public boolean equals(Object o) { - if (this == o) return true; - if (o == null || getClass() != o.getClass()) return false; - CreateMetastoreAssignmentsResponseItem that = (CreateMetastoreAssignmentsResponseItem) o; - return Objects.equals(message, that.message) - && Objects.equals(metastoreAssignment, that.metastoreAssignment) - && Objects.equals(statusCode, that.statusCode); - } - - @Override - public int hashCode() { - return Objects.hash(message, metastoreAssignment, statusCode); - } - - @Override - public String toString() { - return new ToStringer(CreateMetastoreAssignmentsResponseItem.class) - .add("message", message) - .add("metastoreAssignment", metastoreAssignment) - .add("statusCode", statusCode) - .toString(); - } -} diff --git a/databricks-sdk-java/src/main/java/com/databricks/sdk/service/catalog/EffectiveAutoMaintenanceFlag.java b/databricks-sdk-java/src/main/java/com/databricks/sdk/service/catalog/EffectiveAutoMaintenanceFlag.java deleted file mode 100755 index 5dcb8fb7c..000000000 --- a/databricks-sdk-java/src/main/java/com/databricks/sdk/service/catalog/EffectiveAutoMaintenanceFlag.java +++ /dev/null @@ -1,81 +0,0 @@ -// Code generated from OpenAPI specs by Databricks SDK Generator. DO NOT EDIT. - -package com.databricks.sdk.service.catalog; - -import com.databricks.sdk.support.Generated; -import com.databricks.sdk.support.ToStringer; -import com.fasterxml.jackson.annotation.JsonProperty; -import java.util.Objects; - -@Generated -public class EffectiveAutoMaintenanceFlag { - /** - * The name of the object from which the flag was inherited. If there was no inheritance, this - * field is left blank. - */ - @JsonProperty("inherited_from_name") - private String inheritedFromName; - - /** - * The type of the object from which the flag was inherited. If there was no inheritance, this - * field is left blank. - */ - @JsonProperty("inherited_from_type") - private EffectiveAutoMaintenanceFlagInheritedFromType inheritedFromType; - - /** Whether auto maintenance should be enabled for this object and objects under it. */ - @JsonProperty("value") - private EnableAutoMaintenance value; - - public EffectiveAutoMaintenanceFlag setInheritedFromName(String inheritedFromName) { - this.inheritedFromName = inheritedFromName; - return this; - } - - public String getInheritedFromName() { - return inheritedFromName; - } - - public EffectiveAutoMaintenanceFlag setInheritedFromType( - EffectiveAutoMaintenanceFlagInheritedFromType inheritedFromType) { - this.inheritedFromType = inheritedFromType; - return this; - } - - public EffectiveAutoMaintenanceFlagInheritedFromType getInheritedFromType() { - return inheritedFromType; - } - - public EffectiveAutoMaintenanceFlag setValue(EnableAutoMaintenance value) { - this.value = value; - return this; - } - - public EnableAutoMaintenance getValue() { - return value; - } - - @Override - public boolean equals(Object o) { - if (this == o) return true; - if (o == null || getClass() != o.getClass()) return false; - EffectiveAutoMaintenanceFlag that = (EffectiveAutoMaintenanceFlag) o; - return Objects.equals(inheritedFromName, that.inheritedFromName) - && Objects.equals(inheritedFromType, that.inheritedFromType) - && Objects.equals(value, that.value); - } - - @Override - public int hashCode() { - return Objects.hash(inheritedFromName, inheritedFromType, value); - } - - @Override - public String toString() { - return new ToStringer(EffectiveAutoMaintenanceFlag.class) - .add("inheritedFromName", inheritedFromName) - .add("inheritedFromType", inheritedFromType) - .add("value", value) - .toString(); - } -} diff --git a/databricks-sdk-java/src/main/java/com/databricks/sdk/service/catalog/EffectiveAutoMaintenanceFlagInheritedFromType.java b/databricks-sdk-java/src/main/java/com/databricks/sdk/service/catalog/EffectiveAutoMaintenanceFlagInheritedFromType.java deleted file mode 100755 index 8fc3ee579..000000000 --- a/databricks-sdk-java/src/main/java/com/databricks/sdk/service/catalog/EffectiveAutoMaintenanceFlagInheritedFromType.java +++ /dev/null @@ -1,15 +0,0 @@ -// Code generated from OpenAPI specs by Databricks SDK Generator. DO NOT EDIT. - -package com.databricks.sdk.service.catalog; - -import com.databricks.sdk.support.Generated; - -/** - * The type of the object from which the flag was inherited. If there was no inheritance, this field - * is left blank. - */ -@Generated -public enum EffectiveAutoMaintenanceFlagInheritedFromType { - CATALOG, - SCHEMA, -} diff --git a/databricks-sdk-java/src/main/java/com/databricks/sdk/service/catalog/EnableAutoMaintenance.java b/databricks-sdk-java/src/main/java/com/databricks/sdk/service/catalog/EnableAutoMaintenance.java deleted file mode 100755 index 4a59df6fb..000000000 --- a/databricks-sdk-java/src/main/java/com/databricks/sdk/service/catalog/EnableAutoMaintenance.java +++ /dev/null @@ -1,13 +0,0 @@ -// Code generated from OpenAPI specs by Databricks SDK Generator. DO NOT EDIT. - -package com.databricks.sdk.service.catalog; - -import com.databricks.sdk.support.Generated; - -/** Whether auto maintenance should be enabled for this object and objects under it. */ -@Generated -public enum EnableAutoMaintenance { - DISABLE, - ENABLE, - INHERIT, -} diff --git a/databricks-sdk-java/src/main/java/com/databricks/sdk/service/catalog/ExternalLocationsImpl.java b/databricks-sdk-java/src/main/java/com/databricks/sdk/service/catalog/ExternalLocationsImpl.java index b85819244..f4a68dea1 100755 --- a/databricks-sdk-java/src/main/java/com/databricks/sdk/service/catalog/ExternalLocationsImpl.java +++ b/databricks-sdk-java/src/main/java/com/databricks/sdk/service/catalog/ExternalLocationsImpl.java @@ -3,6 +3,8 @@ import com.databricks.sdk.core.ApiClient; import com.databricks.sdk.support.Generated; +import java.util.HashMap; +import java.util.Map; /** Package-local implementation of ExternalLocations */ @Generated @@ -16,30 +18,42 @@ public ExternalLocationsImpl(ApiClient apiClient) { @Override public ExternalLocationInfo create(CreateExternalLocation request) { String path = "/api/2.1/unity-catalog/external-locations"; - return apiClient.POST(path, request, ExternalLocationInfo.class); + Map headers = new HashMap<>(); + headers.put("Accept", "application/json"); + headers.put("Content-Type", "application/json"); + return apiClient.POST(path, request, ExternalLocationInfo.class, headers); } @Override public void delete(DeleteExternalLocationRequest request) { String path = String.format("/api/2.1/unity-catalog/external-locations/%s", request.getName()); - apiClient.DELETE(path, request, Void.class); + Map headers = new HashMap<>(); + headers.put("Accept", "application/json"); + apiClient.DELETE(path, request, Void.class, headers); } @Override public ExternalLocationInfo get(GetExternalLocationRequest request) { String path = String.format("/api/2.1/unity-catalog/external-locations/%s", request.getName()); - return apiClient.GET(path, request, ExternalLocationInfo.class); + Map headers = new HashMap<>(); + headers.put("Accept", "application/json"); + return apiClient.GET(path, request, ExternalLocationInfo.class, headers); } @Override public ListExternalLocationsResponse list() { String path = "/api/2.1/unity-catalog/external-locations"; - return apiClient.GET(path, ListExternalLocationsResponse.class); + Map headers = new HashMap<>(); + headers.put("Accept", "application/json"); + return apiClient.GET(path, ListExternalLocationsResponse.class, headers); } @Override public ExternalLocationInfo update(UpdateExternalLocation request) { String path = String.format("/api/2.1/unity-catalog/external-locations/%s", request.getName()); - return apiClient.PATCH(path, request, ExternalLocationInfo.class); + Map headers = new HashMap<>(); + headers.put("Accept", "application/json"); + headers.put("Content-Type", "application/json"); + return apiClient.PATCH(path, request, ExternalLocationInfo.class, headers); } } diff --git a/databricks-sdk-java/src/main/java/com/databricks/sdk/service/catalog/FunctionsImpl.java b/databricks-sdk-java/src/main/java/com/databricks/sdk/service/catalog/FunctionsImpl.java index 6e05e5ce2..dc4fc9518 100755 --- a/databricks-sdk-java/src/main/java/com/databricks/sdk/service/catalog/FunctionsImpl.java +++ b/databricks-sdk-java/src/main/java/com/databricks/sdk/service/catalog/FunctionsImpl.java @@ -3,6 +3,8 @@ import com.databricks.sdk.core.ApiClient; import com.databricks.sdk.support.Generated; +import java.util.HashMap; +import java.util.Map; /** Package-local implementation of Functions */ @Generated @@ -16,30 +18,42 @@ public FunctionsImpl(ApiClient apiClient) { @Override public FunctionInfo create(CreateFunction request) { String path = "/api/2.1/unity-catalog/functions"; - return apiClient.POST(path, request, FunctionInfo.class); + Map headers = new HashMap<>(); + headers.put("Accept", "application/json"); + headers.put("Content-Type", "application/json"); + return apiClient.POST(path, request, FunctionInfo.class, headers); } @Override public void delete(DeleteFunctionRequest request) { String path = String.format("/api/2.1/unity-catalog/functions/%s", request.getName()); - apiClient.DELETE(path, request, Void.class); + Map headers = new HashMap<>(); + headers.put("Accept", "application/json"); + apiClient.DELETE(path, request, Void.class, headers); } @Override public FunctionInfo get(GetFunctionRequest request) { String path = String.format("/api/2.1/unity-catalog/functions/%s", request.getName()); - return apiClient.GET(path, request, FunctionInfo.class); + Map headers = new HashMap<>(); + headers.put("Accept", "application/json"); + return apiClient.GET(path, request, FunctionInfo.class, headers); } @Override public ListFunctionsResponse list(ListFunctionsRequest request) { String path = "/api/2.1/unity-catalog/functions"; - return apiClient.GET(path, request, ListFunctionsResponse.class); + Map headers = new HashMap<>(); + headers.put("Accept", "application/json"); + return apiClient.GET(path, request, ListFunctionsResponse.class, headers); } @Override public FunctionInfo update(UpdateFunction request) { String path = String.format("/api/2.1/unity-catalog/functions/%s", request.getName()); - return apiClient.PATCH(path, request, FunctionInfo.class); + Map headers = new HashMap<>(); + headers.put("Accept", "application/json"); + headers.put("Content-Type", "application/json"); + return apiClient.PATCH(path, request, FunctionInfo.class, headers); } } diff --git a/databricks-sdk-java/src/main/java/com/databricks/sdk/service/catalog/GrantsImpl.java b/databricks-sdk-java/src/main/java/com/databricks/sdk/service/catalog/GrantsImpl.java index 1851ea58c..a4f83e394 100755 --- a/databricks-sdk-java/src/main/java/com/databricks/sdk/service/catalog/GrantsImpl.java +++ b/databricks-sdk-java/src/main/java/com/databricks/sdk/service/catalog/GrantsImpl.java @@ -3,6 +3,8 @@ import com.databricks.sdk.core.ApiClient; import com.databricks.sdk.support.Generated; +import java.util.HashMap; +import java.util.Map; /** Package-local implementation of Grants */ @Generated @@ -19,7 +21,9 @@ public PermissionsList get(GetGrantRequest request) { String.format( "/api/2.1/unity-catalog/permissions/%s/%s", request.getSecurableType(), request.getFullName()); - return apiClient.GET(path, request, PermissionsList.class); + Map headers = new HashMap<>(); + headers.put("Accept", "application/json"); + return apiClient.GET(path, request, PermissionsList.class, headers); } @Override @@ -28,7 +32,9 @@ public EffectivePermissionsList getEffective(GetEffectiveRequest request) { String.format( "/api/2.1/unity-catalog/effective-permissions/%s/%s", request.getSecurableType(), request.getFullName()); - return apiClient.GET(path, request, EffectivePermissionsList.class); + Map headers = new HashMap<>(); + headers.put("Accept", "application/json"); + return apiClient.GET(path, request, EffectivePermissionsList.class, headers); } @Override @@ -37,6 +43,9 @@ public PermissionsList update(UpdatePermissions request) { String.format( "/api/2.1/unity-catalog/permissions/%s/%s", request.getSecurableType(), request.getFullName()); - return apiClient.PATCH(path, request, PermissionsList.class); + Map headers = new HashMap<>(); + headers.put("Accept", "application/json"); + headers.put("Content-Type", "application/json"); + return apiClient.PATCH(path, request, PermissionsList.class, headers); } } diff --git a/databricks-sdk-java/src/main/java/com/databricks/sdk/service/catalog/MetastoresImpl.java b/databricks-sdk-java/src/main/java/com/databricks/sdk/service/catalog/MetastoresImpl.java index eb0e4eb4a..b533753bc 100755 --- a/databricks-sdk-java/src/main/java/com/databricks/sdk/service/catalog/MetastoresImpl.java +++ b/databricks-sdk-java/src/main/java/com/databricks/sdk/service/catalog/MetastoresImpl.java @@ -3,6 +3,8 @@ import com.databricks.sdk.core.ApiClient; import com.databricks.sdk.support.Generated; +import java.util.HashMap; +import java.util.Map; /** Package-local implementation of Metastores */ @Generated @@ -17,69 +19,96 @@ public MetastoresImpl(ApiClient apiClient) { public void assign(CreateMetastoreAssignment request) { String path = String.format("/api/2.1/unity-catalog/workspaces/%s/metastore", request.getWorkspaceId()); - apiClient.PUT(path, request, Void.class); + Map headers = new HashMap<>(); + headers.put("Accept", "application/json"); + headers.put("Content-Type", "application/json"); + apiClient.PUT(path, request, Void.class, headers); } @Override public MetastoreInfo create(CreateMetastore request) { String path = "/api/2.1/unity-catalog/metastores"; - return apiClient.POST(path, request, MetastoreInfo.class); + Map headers = new HashMap<>(); + headers.put("Accept", "application/json"); + headers.put("Content-Type", "application/json"); + return apiClient.POST(path, request, MetastoreInfo.class, headers); } @Override public MetastoreAssignment current() { String path = "/api/2.1/unity-catalog/current-metastore-assignment"; - return apiClient.GET(path, MetastoreAssignment.class); + Map headers = new HashMap<>(); + headers.put("Accept", "application/json"); + return apiClient.GET(path, MetastoreAssignment.class, headers); } @Override public void delete(DeleteMetastoreRequest request) { String path = String.format("/api/2.1/unity-catalog/metastores/%s", request.getId()); - apiClient.DELETE(path, request, Void.class); + Map headers = new HashMap<>(); + headers.put("Accept", "application/json"); + apiClient.DELETE(path, request, Void.class, headers); } @Override public UpdatePredictiveOptimizationResponse enableOptimization( UpdatePredictiveOptimization request) { String path = "/api/2.0/predictive-optimization/service"; - return apiClient.PATCH(path, request, UpdatePredictiveOptimizationResponse.class); + Map headers = new HashMap<>(); + headers.put("Accept", "application/json"); + headers.put("Content-Type", "application/json"); + return apiClient.PATCH(path, request, UpdatePredictiveOptimizationResponse.class, headers); } @Override public MetastoreInfo get(GetMetastoreRequest request) { String path = String.format("/api/2.1/unity-catalog/metastores/%s", request.getId()); - return apiClient.GET(path, request, MetastoreInfo.class); + Map headers = new HashMap<>(); + headers.put("Accept", "application/json"); + return apiClient.GET(path, request, MetastoreInfo.class, headers); } @Override public ListMetastoresResponse list() { String path = "/api/2.1/unity-catalog/metastores"; - return apiClient.GET(path, ListMetastoresResponse.class); + Map headers = new HashMap<>(); + headers.put("Accept", "application/json"); + return apiClient.GET(path, ListMetastoresResponse.class, headers); } @Override public GetMetastoreSummaryResponse summary() { String path = "/api/2.1/unity-catalog/metastore_summary"; - return apiClient.GET(path, GetMetastoreSummaryResponse.class); + Map headers = new HashMap<>(); + headers.put("Accept", "application/json"); + return apiClient.GET(path, GetMetastoreSummaryResponse.class, headers); } @Override public void unassign(UnassignRequest request) { String path = String.format("/api/2.1/unity-catalog/workspaces/%s/metastore", request.getWorkspaceId()); - apiClient.DELETE(path, request, Void.class); + Map headers = new HashMap<>(); + headers.put("Accept", "application/json"); + apiClient.DELETE(path, request, Void.class, headers); } @Override public MetastoreInfo update(UpdateMetastore request) { String path = String.format("/api/2.1/unity-catalog/metastores/%s", request.getId()); - return apiClient.PATCH(path, request, MetastoreInfo.class); + Map headers = new HashMap<>(); + headers.put("Accept", "application/json"); + headers.put("Content-Type", "application/json"); + return apiClient.PATCH(path, request, MetastoreInfo.class, headers); } @Override public void updateAssignment(UpdateMetastoreAssignment request) { String path = String.format("/api/2.1/unity-catalog/workspaces/%s/metastore", request.getWorkspaceId()); - apiClient.PATCH(path, request, Void.class); + Map headers = new HashMap<>(); + headers.put("Accept", "application/json"); + headers.put("Content-Type", "application/json"); + apiClient.PATCH(path, request, Void.class, headers); } } diff --git a/databricks-sdk-java/src/main/java/com/databricks/sdk/service/catalog/SchemasImpl.java b/databricks-sdk-java/src/main/java/com/databricks/sdk/service/catalog/SchemasImpl.java index bf5cc4b1c..491966ccb 100755 --- a/databricks-sdk-java/src/main/java/com/databricks/sdk/service/catalog/SchemasImpl.java +++ b/databricks-sdk-java/src/main/java/com/databricks/sdk/service/catalog/SchemasImpl.java @@ -3,6 +3,8 @@ import com.databricks.sdk.core.ApiClient; import com.databricks.sdk.support.Generated; +import java.util.HashMap; +import java.util.Map; /** Package-local implementation of Schemas */ @Generated @@ -16,30 +18,42 @@ public SchemasImpl(ApiClient apiClient) { @Override public SchemaInfo create(CreateSchema request) { String path = "/api/2.1/unity-catalog/schemas"; - return apiClient.POST(path, request, SchemaInfo.class); + Map headers = new HashMap<>(); + headers.put("Accept", "application/json"); + headers.put("Content-Type", "application/json"); + return apiClient.POST(path, request, SchemaInfo.class, headers); } @Override public void delete(DeleteSchemaRequest request) { String path = String.format("/api/2.1/unity-catalog/schemas/%s", request.getFullName()); - apiClient.DELETE(path, request, Void.class); + Map headers = new HashMap<>(); + headers.put("Accept", "application/json"); + apiClient.DELETE(path, request, Void.class, headers); } @Override public SchemaInfo get(GetSchemaRequest request) { String path = String.format("/api/2.1/unity-catalog/schemas/%s", request.getFullName()); - return apiClient.GET(path, request, SchemaInfo.class); + Map headers = new HashMap<>(); + headers.put("Accept", "application/json"); + return apiClient.GET(path, request, SchemaInfo.class, headers); } @Override public ListSchemasResponse list(ListSchemasRequest request) { String path = "/api/2.1/unity-catalog/schemas"; - return apiClient.GET(path, request, ListSchemasResponse.class); + Map headers = new HashMap<>(); + headers.put("Accept", "application/json"); + return apiClient.GET(path, request, ListSchemasResponse.class, headers); } @Override public SchemaInfo update(UpdateSchema request) { String path = String.format("/api/2.1/unity-catalog/schemas/%s", request.getFullName()); - return apiClient.PATCH(path, request, SchemaInfo.class); + Map headers = new HashMap<>(); + headers.put("Accept", "application/json"); + headers.put("Content-Type", "application/json"); + return apiClient.PATCH(path, request, SchemaInfo.class, headers); } } diff --git a/databricks-sdk-java/src/main/java/com/databricks/sdk/service/catalog/SecurableTagsImpl.java b/databricks-sdk-java/src/main/java/com/databricks/sdk/service/catalog/SecurableTagsImpl.java index 107152684..fd697d51e 100755 --- a/databricks-sdk-java/src/main/java/com/databricks/sdk/service/catalog/SecurableTagsImpl.java +++ b/databricks-sdk-java/src/main/java/com/databricks/sdk/service/catalog/SecurableTagsImpl.java @@ -3,6 +3,8 @@ import com.databricks.sdk.core.ApiClient; import com.databricks.sdk.support.Generated; +import java.util.HashMap; +import java.util.Map; /** Package-local implementation of SecurableTags */ @Generated @@ -19,7 +21,9 @@ public TagSecurableAssignmentsList list(ListSecurableTagsRequest request) { String.format( "/api/2.1/unity-catalog/securable-tags/%s/%s", request.getSecurableType(), request.getFullName()); - return apiClient.GET(path, request, TagSecurableAssignmentsList.class); + Map headers = new HashMap<>(); + headers.put("Accept", "application/json"); + return apiClient.GET(path, request, TagSecurableAssignmentsList.class, headers); } @Override @@ -28,6 +32,9 @@ public TagSecurableAssignmentsList update(UpdateTags request) { String.format( "/api/2.1/unity-catalog/securable-tags/%s/%s", request.getSecurableType(), request.getFullName()); - return apiClient.PATCH(path, request, TagSecurableAssignmentsList.class); + Map headers = new HashMap<>(); + headers.put("Accept", "application/json"); + headers.put("Content-Type", "application/json"); + return apiClient.PATCH(path, request, TagSecurableAssignmentsList.class, headers); } } diff --git a/databricks-sdk-java/src/main/java/com/databricks/sdk/service/catalog/StorageCredentialsImpl.java b/databricks-sdk-java/src/main/java/com/databricks/sdk/service/catalog/StorageCredentialsImpl.java index cfc1033c2..10a3743b1 100755 --- a/databricks-sdk-java/src/main/java/com/databricks/sdk/service/catalog/StorageCredentialsImpl.java +++ b/databricks-sdk-java/src/main/java/com/databricks/sdk/service/catalog/StorageCredentialsImpl.java @@ -3,6 +3,8 @@ import com.databricks.sdk.core.ApiClient; import com.databricks.sdk.support.Generated; +import java.util.HashMap; +import java.util.Map; /** Package-local implementation of StorageCredentials */ @Generated @@ -16,36 +18,51 @@ public StorageCredentialsImpl(ApiClient apiClient) { @Override public StorageCredentialInfo create(CreateStorageCredential request) { String path = "/api/2.1/unity-catalog/storage-credentials"; - return apiClient.POST(path, request, StorageCredentialInfo.class); + Map headers = new HashMap<>(); + headers.put("Accept", "application/json"); + headers.put("Content-Type", "application/json"); + return apiClient.POST(path, request, StorageCredentialInfo.class, headers); } @Override public void delete(DeleteStorageCredentialRequest request) { String path = String.format("/api/2.1/unity-catalog/storage-credentials/%s", request.getName()); - apiClient.DELETE(path, request, Void.class); + Map headers = new HashMap<>(); + headers.put("Accept", "application/json"); + apiClient.DELETE(path, request, Void.class, headers); } @Override public StorageCredentialInfo get(GetStorageCredentialRequest request) { String path = String.format("/api/2.1/unity-catalog/storage-credentials/%s", request.getName()); - return apiClient.GET(path, request, StorageCredentialInfo.class); + Map headers = new HashMap<>(); + headers.put("Accept", "application/json"); + return apiClient.GET(path, request, StorageCredentialInfo.class, headers); } @Override public ListStorageCredentialsResponse list() { String path = "/api/2.1/unity-catalog/storage-credentials"; - return apiClient.GET(path, ListStorageCredentialsResponse.class); + Map headers = new HashMap<>(); + headers.put("Accept", "application/json"); + return apiClient.GET(path, ListStorageCredentialsResponse.class, headers); } @Override public StorageCredentialInfo update(UpdateStorageCredential request) { String path = String.format("/api/2.1/unity-catalog/storage-credentials/%s", request.getName()); - return apiClient.PATCH(path, request, StorageCredentialInfo.class); + Map headers = new HashMap<>(); + headers.put("Accept", "application/json"); + headers.put("Content-Type", "application/json"); + return apiClient.PATCH(path, request, StorageCredentialInfo.class, headers); } @Override public ValidateStorageCredentialResponse validate(ValidateStorageCredential request) { String path = "/api/2.1/unity-catalog/validate-storage-credentials"; - return apiClient.POST(path, request, ValidateStorageCredentialResponse.class); + Map headers = new HashMap<>(); + headers.put("Accept", "application/json"); + headers.put("Content-Type", "application/json"); + return apiClient.POST(path, request, ValidateStorageCredentialResponse.class, headers); } } diff --git a/databricks-sdk-java/src/main/java/com/databricks/sdk/service/catalog/SubentityTagsImpl.java b/databricks-sdk-java/src/main/java/com/databricks/sdk/service/catalog/SubentityTagsImpl.java index 3c569fef8..1bcaceeb6 100755 --- a/databricks-sdk-java/src/main/java/com/databricks/sdk/service/catalog/SubentityTagsImpl.java +++ b/databricks-sdk-java/src/main/java/com/databricks/sdk/service/catalog/SubentityTagsImpl.java @@ -3,6 +3,8 @@ import com.databricks.sdk.core.ApiClient; import com.databricks.sdk.support.Generated; +import java.util.HashMap; +import java.util.Map; /** Package-local implementation of SubentityTags */ @Generated @@ -19,7 +21,9 @@ public TagSubentityAssignmentsList list(ListSubentityTagsRequest request) { String.format( "/api/2.1/unity-catalog/subentity-tags/%s/%s/%s", request.getSecurableType(), request.getFullName(), request.getSubentityName()); - return apiClient.GET(path, request, TagSubentityAssignmentsList.class); + Map headers = new HashMap<>(); + headers.put("Accept", "application/json"); + return apiClient.GET(path, request, TagSubentityAssignmentsList.class, headers); } @Override @@ -28,6 +32,9 @@ public TagSubentityAssignmentsList update(UpdateTags request) { String.format( "/api/2.1/unity-catalog/subentity-tags/%s/%s/%s", request.getSecurableType(), request.getFullName(), request.getSubentityName()); - return apiClient.PATCH(path, request, TagSubentityAssignmentsList.class); + Map headers = new HashMap<>(); + headers.put("Accept", "application/json"); + headers.put("Content-Type", "application/json"); + return apiClient.PATCH(path, request, TagSubentityAssignmentsList.class, headers); } } diff --git a/databricks-sdk-java/src/main/java/com/databricks/sdk/service/catalog/SystemSchemasImpl.java b/databricks-sdk-java/src/main/java/com/databricks/sdk/service/catalog/SystemSchemasImpl.java index 9068f99d6..fdc106cef 100755 --- a/databricks-sdk-java/src/main/java/com/databricks/sdk/service/catalog/SystemSchemasImpl.java +++ b/databricks-sdk-java/src/main/java/com/databricks/sdk/service/catalog/SystemSchemasImpl.java @@ -3,6 +3,8 @@ import com.databricks.sdk.core.ApiClient; import com.databricks.sdk.support.Generated; +import java.util.HashMap; +import java.util.Map; /** Package-local implementation of SystemSchemas */ @Generated @@ -19,7 +21,9 @@ public void disable(DisableRequest request) { String.format( "/api/2.1/unity-catalog/metastores/%s/systemschemas/%s", request.getMetastoreId(), request.getSchemaName()); - apiClient.DELETE(path, request, Void.class); + Map headers = new HashMap<>(); + headers.put("Accept", "application/json"); + apiClient.DELETE(path, request, Void.class, headers); } @Override @@ -28,7 +32,9 @@ public void enable(EnableRequest request) { String.format( "/api/2.1/unity-catalog/metastores/%s/systemschemas/%s", request.getMetastoreId(), request.getSchemaName()); - apiClient.PUT(path, request, Void.class); + Map headers = new HashMap<>(); + headers.put("Accept", "application/json"); + apiClient.PUT(path, request, Void.class, headers); } @Override @@ -36,6 +42,8 @@ public ListSystemSchemasResponse list(ListSystemSchemasRequest request) { String path = String.format( "/api/2.1/unity-catalog/metastores/%s/systemschemas", request.getMetastoreId()); - return apiClient.GET(path, request, ListSystemSchemasResponse.class); + Map headers = new HashMap<>(); + headers.put("Accept", "application/json"); + return apiClient.GET(path, request, ListSystemSchemasResponse.class, headers); } } diff --git a/databricks-sdk-java/src/main/java/com/databricks/sdk/service/catalog/TableConstraintsImpl.java b/databricks-sdk-java/src/main/java/com/databricks/sdk/service/catalog/TableConstraintsImpl.java index 8517783b7..2c210d3a4 100755 --- a/databricks-sdk-java/src/main/java/com/databricks/sdk/service/catalog/TableConstraintsImpl.java +++ b/databricks-sdk-java/src/main/java/com/databricks/sdk/service/catalog/TableConstraintsImpl.java @@ -3,6 +3,8 @@ import com.databricks.sdk.core.ApiClient; import com.databricks.sdk.support.Generated; +import java.util.HashMap; +import java.util.Map; /** Package-local implementation of TableConstraints */ @Generated @@ -16,12 +18,17 @@ public TableConstraintsImpl(ApiClient apiClient) { @Override public TableConstraint create(CreateTableConstraint request) { String path = "/api/2.1/unity-catalog/constraints"; - return apiClient.POST(path, request, TableConstraint.class); + Map headers = new HashMap<>(); + headers.put("Accept", "application/json"); + headers.put("Content-Type", "application/json"); + return apiClient.POST(path, request, TableConstraint.class, headers); } @Override public void delete(DeleteTableConstraintRequest request) { String path = String.format("/api/2.1/unity-catalog/constraints/%s", request.getFullName()); - apiClient.DELETE(path, request, Void.class); + Map headers = new HashMap<>(); + headers.put("Accept", "application/json"); + apiClient.DELETE(path, request, Void.class, headers); } } diff --git a/databricks-sdk-java/src/main/java/com/databricks/sdk/service/catalog/TablesImpl.java b/databricks-sdk-java/src/main/java/com/databricks/sdk/service/catalog/TablesImpl.java index 9201366a6..53b043863 100755 --- a/databricks-sdk-java/src/main/java/com/databricks/sdk/service/catalog/TablesImpl.java +++ b/databricks-sdk-java/src/main/java/com/databricks/sdk/service/catalog/TablesImpl.java @@ -3,6 +3,8 @@ import com.databricks.sdk.core.ApiClient; import com.databricks.sdk.support.Generated; +import java.util.HashMap; +import java.util.Map; /** Package-local implementation of Tables */ @Generated @@ -16,30 +18,41 @@ public TablesImpl(ApiClient apiClient) { @Override public void delete(DeleteTableRequest request) { String path = String.format("/api/2.1/unity-catalog/tables/%s", request.getFullName()); - apiClient.DELETE(path, request, Void.class); + Map headers = new HashMap<>(); + headers.put("Accept", "application/json"); + apiClient.DELETE(path, request, Void.class, headers); } @Override public TableInfo get(GetTableRequest request) { String path = String.format("/api/2.1/unity-catalog/tables/%s", request.getFullName()); - return apiClient.GET(path, request, TableInfo.class); + Map headers = new HashMap<>(); + headers.put("Accept", "application/json"); + return apiClient.GET(path, request, TableInfo.class, headers); } @Override public ListTablesResponse list(ListTablesRequest request) { String path = "/api/2.1/unity-catalog/tables"; - return apiClient.GET(path, request, ListTablesResponse.class); + Map headers = new HashMap<>(); + headers.put("Accept", "application/json"); + return apiClient.GET(path, request, ListTablesResponse.class, headers); } @Override public ListTableSummariesResponse listSummaries(ListSummariesRequest request) { String path = "/api/2.1/unity-catalog/table-summaries"; - return apiClient.GET(path, request, ListTableSummariesResponse.class); + Map headers = new HashMap<>(); + headers.put("Accept", "application/json"); + return apiClient.GET(path, request, ListTableSummariesResponse.class, headers); } @Override public void update(UpdateTableRequest request) { String path = String.format("/api/2.1/unity-catalog/tables/%s", request.getFullName()); - apiClient.PATCH(path, request, Void.class); + Map headers = new HashMap<>(); + headers.put("Accept", "application/json"); + headers.put("Content-Type", "application/json"); + apiClient.PATCH(path, request, Void.class, headers); } } diff --git a/databricks-sdk-java/src/main/java/com/databricks/sdk/service/catalog/UpdateAutoMaintenance.java b/databricks-sdk-java/src/main/java/com/databricks/sdk/service/catalog/UpdateAutoMaintenance.java deleted file mode 100755 index e918118ca..000000000 --- a/databricks-sdk-java/src/main/java/com/databricks/sdk/service/catalog/UpdateAutoMaintenance.java +++ /dev/null @@ -1,58 +0,0 @@ -// Code generated from OpenAPI specs by Databricks SDK Generator. DO NOT EDIT. - -package com.databricks.sdk.service.catalog; - -import com.databricks.sdk.support.Generated; -import com.databricks.sdk.support.ToStringer; -import com.fasterxml.jackson.annotation.JsonProperty; -import java.util.Objects; - -@Generated -public class UpdateAutoMaintenance { - /** Whether to enable auto maintenance on the metastore. */ - @JsonProperty("enable") - private Boolean enable; - - /** Unique identifier of metastore. */ - @JsonProperty("metastore_id") - private String metastoreId; - - public UpdateAutoMaintenance setEnable(Boolean enable) { - this.enable = enable; - return this; - } - - public Boolean getEnable() { - return enable; - } - - public UpdateAutoMaintenance setMetastoreId(String metastoreId) { - this.metastoreId = metastoreId; - return this; - } - - public String getMetastoreId() { - return metastoreId; - } - - @Override - public boolean equals(Object o) { - if (this == o) return true; - if (o == null || getClass() != o.getClass()) return false; - UpdateAutoMaintenance that = (UpdateAutoMaintenance) o; - return Objects.equals(enable, that.enable) && Objects.equals(metastoreId, that.metastoreId); - } - - @Override - public int hashCode() { - return Objects.hash(enable, metastoreId); - } - - @Override - public String toString() { - return new ToStringer(UpdateAutoMaintenance.class) - .add("enable", enable) - .add("metastoreId", metastoreId) - .toString(); - } -} diff --git a/databricks-sdk-java/src/main/java/com/databricks/sdk/service/catalog/UpdateAutoMaintenanceResponse.java b/databricks-sdk-java/src/main/java/com/databricks/sdk/service/catalog/UpdateAutoMaintenanceResponse.java deleted file mode 100755 index 68a097f79..000000000 --- a/databricks-sdk-java/src/main/java/com/databricks/sdk/service/catalog/UpdateAutoMaintenanceResponse.java +++ /dev/null @@ -1,77 +0,0 @@ -// Code generated from OpenAPI specs by Databricks SDK Generator. DO NOT EDIT. - -package com.databricks.sdk.service.catalog; - -import com.databricks.sdk.support.Generated; -import com.databricks.sdk.support.ToStringer; -import com.fasterxml.jackson.annotation.JsonProperty; -import java.util.Objects; - -@Generated -public class UpdateAutoMaintenanceResponse { - /** Whether auto maintenance is enabled on the metastore. */ - @JsonProperty("state") - private Boolean state; - - /** - * Id of the auto maintenance service principal. This will be the user used to run maintenance - * tasks. - */ - @JsonProperty("user_id") - private Long userId; - - /** Name of the auto maintenance service principal. */ - @JsonProperty("username") - private String username; - - public UpdateAutoMaintenanceResponse setState(Boolean state) { - this.state = state; - return this; - } - - public Boolean getState() { - return state; - } - - public UpdateAutoMaintenanceResponse setUserId(Long userId) { - this.userId = userId; - return this; - } - - public Long getUserId() { - return userId; - } - - public UpdateAutoMaintenanceResponse setUsername(String username) { - this.username = username; - return this; - } - - public String getUsername() { - return username; - } - - @Override - public boolean equals(Object o) { - if (this == o) return true; - if (o == null || getClass() != o.getClass()) return false; - UpdateAutoMaintenanceResponse that = (UpdateAutoMaintenanceResponse) o; - return Objects.equals(state, that.state) - && Objects.equals(userId, that.userId) - && Objects.equals(username, that.username); - } - - @Override - public int hashCode() { - return Objects.hash(state, userId, username); - } - - @Override - public String toString() { - return new ToStringer(UpdateAutoMaintenanceResponse.class) - .add("state", state) - .add("userId", userId) - .add("username", username) - .toString(); - } -} diff --git a/databricks-sdk-java/src/main/java/com/databricks/sdk/service/catalog/VolumesImpl.java b/databricks-sdk-java/src/main/java/com/databricks/sdk/service/catalog/VolumesImpl.java index 1f3028162..26382aefc 100755 --- a/databricks-sdk-java/src/main/java/com/databricks/sdk/service/catalog/VolumesImpl.java +++ b/databricks-sdk-java/src/main/java/com/databricks/sdk/service/catalog/VolumesImpl.java @@ -3,6 +3,8 @@ import com.databricks.sdk.core.ApiClient; import com.databricks.sdk.support.Generated; +import java.util.HashMap; +import java.util.Map; /** Package-local implementation of Volumes */ @Generated @@ -16,30 +18,41 @@ public VolumesImpl(ApiClient apiClient) { @Override public VolumeInfo create(CreateVolumeRequestContent request) { String path = "/api/2.1/unity-catalog/volumes"; - return apiClient.POST(path, request, VolumeInfo.class); + Map headers = new HashMap<>(); + headers.put("Accept", "application/json"); + headers.put("Content-Type", "application/json"); + return apiClient.POST(path, request, VolumeInfo.class, headers); } @Override public void delete(DeleteVolumeRequest request) { String path = String.format("/api/2.1/unity-catalog/volumes/%s", request.getFullNameArg()); - apiClient.DELETE(path, request, Void.class); + Map headers = new HashMap<>(); + apiClient.DELETE(path, request, Void.class, headers); } @Override public ListVolumesResponseContent list(ListVolumesRequest request) { String path = "/api/2.1/unity-catalog/volumes"; - return apiClient.GET(path, request, ListVolumesResponseContent.class); + Map headers = new HashMap<>(); + headers.put("Accept", "application/json"); + return apiClient.GET(path, request, ListVolumesResponseContent.class, headers); } @Override public VolumeInfo read(ReadVolumeRequest request) { String path = String.format("/api/2.1/unity-catalog/volumes/%s", request.getFullNameArg()); - return apiClient.GET(path, request, VolumeInfo.class); + Map headers = new HashMap<>(); + headers.put("Accept", "application/json"); + return apiClient.GET(path, request, VolumeInfo.class, headers); } @Override public VolumeInfo update(UpdateVolumeRequestContent request) { String path = String.format("/api/2.1/unity-catalog/volumes/%s", request.getFullNameArg()); - return apiClient.PATCH(path, request, VolumeInfo.class); + Map headers = new HashMap<>(); + headers.put("Accept", "application/json"); + headers.put("Content-Type", "application/json"); + return apiClient.PATCH(path, request, VolumeInfo.class, headers); } } diff --git a/databricks-sdk-java/src/main/java/com/databricks/sdk/service/catalog/WorkspaceBindingsImpl.java b/databricks-sdk-java/src/main/java/com/databricks/sdk/service/catalog/WorkspaceBindingsImpl.java index bd463e292..1c0721e03 100755 --- a/databricks-sdk-java/src/main/java/com/databricks/sdk/service/catalog/WorkspaceBindingsImpl.java +++ b/databricks-sdk-java/src/main/java/com/databricks/sdk/service/catalog/WorkspaceBindingsImpl.java @@ -3,6 +3,8 @@ import com.databricks.sdk.core.ApiClient; import com.databricks.sdk.support.Generated; +import java.util.HashMap; +import java.util.Map; /** Package-local implementation of WorkspaceBindings */ @Generated @@ -17,13 +19,18 @@ public WorkspaceBindingsImpl(ApiClient apiClient) { public CurrentWorkspaceBindings get(GetWorkspaceBindingRequest request) { String path = String.format("/api/2.1/unity-catalog/workspace-bindings/catalogs/%s", request.getName()); - return apiClient.GET(path, request, CurrentWorkspaceBindings.class); + Map headers = new HashMap<>(); + headers.put("Accept", "application/json"); + return apiClient.GET(path, request, CurrentWorkspaceBindings.class, headers); } @Override public CurrentWorkspaceBindings update(UpdateWorkspaceBindings request) { String path = String.format("/api/2.1/unity-catalog/workspace-bindings/catalogs/%s", request.getName()); - return apiClient.PATCH(path, request, CurrentWorkspaceBindings.class); + Map headers = new HashMap<>(); + headers.put("Accept", "application/json"); + headers.put("Content-Type", "application/json"); + return apiClient.PATCH(path, request, CurrentWorkspaceBindings.class, headers); } } diff --git a/databricks-sdk-java/src/main/java/com/databricks/sdk/service/compute/ClusterPoliciesImpl.java b/databricks-sdk-java/src/main/java/com/databricks/sdk/service/compute/ClusterPoliciesImpl.java index c3293c4f9..6abd470c5 100755 --- a/databricks-sdk-java/src/main/java/com/databricks/sdk/service/compute/ClusterPoliciesImpl.java +++ b/databricks-sdk-java/src/main/java/com/databricks/sdk/service/compute/ClusterPoliciesImpl.java @@ -3,6 +3,8 @@ import com.databricks.sdk.core.ApiClient; import com.databricks.sdk.support.Generated; +import java.util.HashMap; +import java.util.Map; /** Package-local implementation of ClusterPolicies */ @Generated @@ -16,25 +18,36 @@ public ClusterPoliciesImpl(ApiClient apiClient) { @Override public CreatePolicyResponse create(CreatePolicy request) { String path = "/api/2.0/policies/clusters/create"; - return apiClient.POST(path, request, CreatePolicyResponse.class); + Map headers = new HashMap<>(); + headers.put("Accept", "application/json"); + headers.put("Content-Type", "application/json"); + return apiClient.POST(path, request, CreatePolicyResponse.class, headers); } @Override public void delete(DeletePolicy request) { String path = "/api/2.0/policies/clusters/delete"; - apiClient.POST(path, request, Void.class); + Map headers = new HashMap<>(); + headers.put("Accept", "application/json"); + headers.put("Content-Type", "application/json"); + apiClient.POST(path, request, Void.class, headers); } @Override public void edit(EditPolicy request) { String path = "/api/2.0/policies/clusters/edit"; - apiClient.POST(path, request, Void.class); + Map headers = new HashMap<>(); + headers.put("Accept", "application/json"); + headers.put("Content-Type", "application/json"); + apiClient.POST(path, request, Void.class, headers); } @Override public Policy get(GetClusterPolicyRequest request) { String path = "/api/2.0/policies/clusters/get"; - return apiClient.GET(path, request, Policy.class); + Map headers = new HashMap<>(); + headers.put("Accept", "application/json"); + return apiClient.GET(path, request, Policy.class, headers); } @Override @@ -44,7 +57,9 @@ public GetClusterPolicyPermissionLevelsResponse getClusterPolicyPermissionLevels String.format( "/api/2.0/permissions/cluster-policies/%s/permissionLevels", request.getClusterPolicyId()); - return apiClient.GET(path, request, GetClusterPolicyPermissionLevelsResponse.class); + Map headers = new HashMap<>(); + headers.put("Accept", "application/json"); + return apiClient.GET(path, request, GetClusterPolicyPermissionLevelsResponse.class, headers); } @Override @@ -52,13 +67,17 @@ public ClusterPolicyPermissions getClusterPolicyPermissions( GetClusterPolicyPermissionsRequest request) { String path = String.format("/api/2.0/permissions/cluster-policies/%s", request.getClusterPolicyId()); - return apiClient.GET(path, request, ClusterPolicyPermissions.class); + Map headers = new HashMap<>(); + headers.put("Accept", "application/json"); + return apiClient.GET(path, request, ClusterPolicyPermissions.class, headers); } @Override public ListPoliciesResponse list(ListClusterPoliciesRequest request) { String path = "/api/2.0/policies/clusters/list"; - return apiClient.GET(path, request, ListPoliciesResponse.class); + Map headers = new HashMap<>(); + headers.put("Accept", "application/json"); + return apiClient.GET(path, request, ListPoliciesResponse.class, headers); } @Override @@ -66,7 +85,10 @@ public ClusterPolicyPermissions setClusterPolicyPermissions( ClusterPolicyPermissionsRequest request) { String path = String.format("/api/2.0/permissions/cluster-policies/%s", request.getClusterPolicyId()); - return apiClient.PUT(path, request, ClusterPolicyPermissions.class); + Map headers = new HashMap<>(); + headers.put("Accept", "application/json"); + headers.put("Content-Type", "application/json"); + return apiClient.PUT(path, request, ClusterPolicyPermissions.class, headers); } @Override @@ -74,6 +96,9 @@ public ClusterPolicyPermissions updateClusterPolicyPermissions( ClusterPolicyPermissionsRequest request) { String path = String.format("/api/2.0/permissions/cluster-policies/%s", request.getClusterPolicyId()); - return apiClient.PATCH(path, request, ClusterPolicyPermissions.class); + Map headers = new HashMap<>(); + headers.put("Accept", "application/json"); + headers.put("Content-Type", "application/json"); + return apiClient.PATCH(path, request, ClusterPolicyPermissions.class, headers); } } diff --git a/databricks-sdk-java/src/main/java/com/databricks/sdk/service/compute/ClustersImpl.java b/databricks-sdk-java/src/main/java/com/databricks/sdk/service/compute/ClustersImpl.java index 90caed689..b4dbf7987 100755 --- a/databricks-sdk-java/src/main/java/com/databricks/sdk/service/compute/ClustersImpl.java +++ b/databricks-sdk-java/src/main/java/com/databricks/sdk/service/compute/ClustersImpl.java @@ -3,6 +3,8 @@ import com.databricks.sdk.core.ApiClient; import com.databricks.sdk.support.Generated; +import java.util.HashMap; +import java.util.Map; /** Package-local implementation of Clusters */ @Generated @@ -16,37 +18,54 @@ public ClustersImpl(ApiClient apiClient) { @Override public void changeOwner(ChangeClusterOwner request) { String path = "/api/2.0/clusters/change-owner"; - apiClient.POST(path, request, Void.class); + Map headers = new HashMap<>(); + headers.put("Accept", "application/json"); + headers.put("Content-Type", "application/json"); + apiClient.POST(path, request, Void.class, headers); } @Override public CreateClusterResponse create(CreateCluster request) { String path = "/api/2.0/clusters/create"; - return apiClient.POST(path, request, CreateClusterResponse.class); + Map headers = new HashMap<>(); + headers.put("Accept", "application/json"); + headers.put("Content-Type", "application/json"); + return apiClient.POST(path, request, CreateClusterResponse.class, headers); } @Override public void delete(DeleteCluster request) { String path = "/api/2.0/clusters/delete"; - apiClient.POST(path, request, Void.class); + Map headers = new HashMap<>(); + headers.put("Accept", "application/json"); + headers.put("Content-Type", "application/json"); + apiClient.POST(path, request, Void.class, headers); } @Override public void edit(EditCluster request) { String path = "/api/2.0/clusters/edit"; - apiClient.POST(path, request, Void.class); + Map headers = new HashMap<>(); + headers.put("Accept", "application/json"); + headers.put("Content-Type", "application/json"); + apiClient.POST(path, request, Void.class, headers); } @Override public GetEventsResponse events(GetEvents request) { String path = "/api/2.0/clusters/events"; - return apiClient.POST(path, request, GetEventsResponse.class); + Map headers = new HashMap<>(); + headers.put("Accept", "application/json"); + headers.put("Content-Type", "application/json"); + return apiClient.POST(path, request, GetEventsResponse.class, headers); } @Override public ClusterDetails get(GetClusterRequest request) { String path = "/api/2.0/clusters/get"; - return apiClient.GET(path, request, ClusterDetails.class); + Map headers = new HashMap<>(); + headers.put("Accept", "application/json"); + return apiClient.GET(path, request, ClusterDetails.class, headers); } @Override @@ -54,84 +73,120 @@ public GetClusterPermissionLevelsResponse getClusterPermissionLevels( GetClusterPermissionLevelsRequest request) { String path = String.format("/api/2.0/permissions/clusters/%s/permissionLevels", request.getClusterId()); - return apiClient.GET(path, request, GetClusterPermissionLevelsResponse.class); + Map headers = new HashMap<>(); + headers.put("Accept", "application/json"); + return apiClient.GET(path, request, GetClusterPermissionLevelsResponse.class, headers); } @Override public ClusterPermissions getClusterPermissions(GetClusterPermissionsRequest request) { String path = String.format("/api/2.0/permissions/clusters/%s", request.getClusterId()); - return apiClient.GET(path, request, ClusterPermissions.class); + Map headers = new HashMap<>(); + headers.put("Accept", "application/json"); + return apiClient.GET(path, request, ClusterPermissions.class, headers); } @Override public ListClustersResponse list(ListClustersRequest request) { String path = "/api/2.0/clusters/list"; - return apiClient.GET(path, request, ListClustersResponse.class); + Map headers = new HashMap<>(); + headers.put("Accept", "application/json"); + return apiClient.GET(path, request, ListClustersResponse.class, headers); } @Override public ListNodeTypesResponse listNodeTypes() { String path = "/api/2.0/clusters/list-node-types"; - return apiClient.GET(path, ListNodeTypesResponse.class); + Map headers = new HashMap<>(); + headers.put("Accept", "application/json"); + return apiClient.GET(path, ListNodeTypesResponse.class, headers); } @Override public ListAvailableZonesResponse listZones() { String path = "/api/2.0/clusters/list-zones"; - return apiClient.GET(path, ListAvailableZonesResponse.class); + Map headers = new HashMap<>(); + headers.put("Accept", "application/json"); + return apiClient.GET(path, ListAvailableZonesResponse.class, headers); } @Override public void permanentDelete(PermanentDeleteCluster request) { String path = "/api/2.0/clusters/permanent-delete"; - apiClient.POST(path, request, Void.class); + Map headers = new HashMap<>(); + headers.put("Accept", "application/json"); + headers.put("Content-Type", "application/json"); + apiClient.POST(path, request, Void.class, headers); } @Override public void pin(PinCluster request) { String path = "/api/2.0/clusters/pin"; - apiClient.POST(path, request, Void.class); + Map headers = new HashMap<>(); + headers.put("Accept", "application/json"); + headers.put("Content-Type", "application/json"); + apiClient.POST(path, request, Void.class, headers); } @Override public void resize(ResizeCluster request) { String path = "/api/2.0/clusters/resize"; - apiClient.POST(path, request, Void.class); + Map headers = new HashMap<>(); + headers.put("Accept", "application/json"); + headers.put("Content-Type", "application/json"); + apiClient.POST(path, request, Void.class, headers); } @Override public void restart(RestartCluster request) { String path = "/api/2.0/clusters/restart"; - apiClient.POST(path, request, Void.class); + Map headers = new HashMap<>(); + headers.put("Accept", "application/json"); + headers.put("Content-Type", "application/json"); + apiClient.POST(path, request, Void.class, headers); } @Override public ClusterPermissions setClusterPermissions(ClusterPermissionsRequest request) { String path = String.format("/api/2.0/permissions/clusters/%s", request.getClusterId()); - return apiClient.PUT(path, request, ClusterPermissions.class); + Map headers = new HashMap<>(); + headers.put("Accept", "application/json"); + headers.put("Content-Type", "application/json"); + return apiClient.PUT(path, request, ClusterPermissions.class, headers); } @Override public GetSparkVersionsResponse sparkVersions() { String path = "/api/2.0/clusters/spark-versions"; - return apiClient.GET(path, GetSparkVersionsResponse.class); + Map headers = new HashMap<>(); + headers.put("Accept", "application/json"); + return apiClient.GET(path, GetSparkVersionsResponse.class, headers); } @Override public void start(StartCluster request) { String path = "/api/2.0/clusters/start"; - apiClient.POST(path, request, Void.class); + Map headers = new HashMap<>(); + headers.put("Accept", "application/json"); + headers.put("Content-Type", "application/json"); + apiClient.POST(path, request, Void.class, headers); } @Override public void unpin(UnpinCluster request) { String path = "/api/2.0/clusters/unpin"; - apiClient.POST(path, request, Void.class); + Map headers = new HashMap<>(); + headers.put("Accept", "application/json"); + headers.put("Content-Type", "application/json"); + apiClient.POST(path, request, Void.class, headers); } @Override public ClusterPermissions updateClusterPermissions(ClusterPermissionsRequest request) { String path = String.format("/api/2.0/permissions/clusters/%s", request.getClusterId()); - return apiClient.PATCH(path, request, ClusterPermissions.class); + Map headers = new HashMap<>(); + headers.put("Accept", "application/json"); + headers.put("Content-Type", "application/json"); + return apiClient.PATCH(path, request, ClusterPermissions.class, headers); } } diff --git a/databricks-sdk-java/src/main/java/com/databricks/sdk/service/compute/CommandExecutionImpl.java b/databricks-sdk-java/src/main/java/com/databricks/sdk/service/compute/CommandExecutionImpl.java index 804011ce4..1a53931e9 100755 --- a/databricks-sdk-java/src/main/java/com/databricks/sdk/service/compute/CommandExecutionImpl.java +++ b/databricks-sdk-java/src/main/java/com/databricks/sdk/service/compute/CommandExecutionImpl.java @@ -3,6 +3,8 @@ import com.databricks.sdk.core.ApiClient; import com.databricks.sdk.support.Generated; +import java.util.HashMap; +import java.util.Map; /** Package-local implementation of CommandExecution */ @Generated @@ -16,36 +18,52 @@ public CommandExecutionImpl(ApiClient apiClient) { @Override public void cancel(CancelCommand request) { String path = "/api/1.2/commands/cancel"; - apiClient.POST(path, request, Void.class); + Map headers = new HashMap<>(); + headers.put("Accept", "application/json"); + headers.put("Content-Type", "application/json"); + apiClient.POST(path, request, Void.class, headers); } @Override public CommandStatusResponse commandStatus(CommandStatusRequest request) { String path = "/api/1.2/commands/status"; - return apiClient.GET(path, request, CommandStatusResponse.class); + Map headers = new HashMap<>(); + headers.put("Accept", "application/json"); + return apiClient.GET(path, request, CommandStatusResponse.class, headers); } @Override public ContextStatusResponse contextStatus(ContextStatusRequest request) { String path = "/api/1.2/contexts/status"; - return apiClient.GET(path, request, ContextStatusResponse.class); + Map headers = new HashMap<>(); + headers.put("Accept", "application/json"); + return apiClient.GET(path, request, ContextStatusResponse.class, headers); } @Override public Created create(CreateContext request) { String path = "/api/1.2/contexts/create"; - return apiClient.POST(path, request, Created.class); + Map headers = new HashMap<>(); + headers.put("Accept", "application/json"); + headers.put("Content-Type", "application/json"); + return apiClient.POST(path, request, Created.class, headers); } @Override public void destroy(DestroyContext request) { String path = "/api/1.2/contexts/destroy"; - apiClient.POST(path, request, Void.class); + Map headers = new HashMap<>(); + headers.put("Accept", "application/json"); + headers.put("Content-Type", "application/json"); + apiClient.POST(path, request, Void.class, headers); } @Override public Created execute(Command request) { String path = "/api/1.2/commands/execute"; - return apiClient.POST(path, request, Created.class); + Map headers = new HashMap<>(); + headers.put("Accept", "application/json"); + headers.put("Content-Type", "application/json"); + return apiClient.POST(path, request, Created.class, headers); } } diff --git a/databricks-sdk-java/src/main/java/com/databricks/sdk/service/compute/GlobalInitScriptsImpl.java b/databricks-sdk-java/src/main/java/com/databricks/sdk/service/compute/GlobalInitScriptsImpl.java index b6846695d..fcb1bac9c 100755 --- a/databricks-sdk-java/src/main/java/com/databricks/sdk/service/compute/GlobalInitScriptsImpl.java +++ b/databricks-sdk-java/src/main/java/com/databricks/sdk/service/compute/GlobalInitScriptsImpl.java @@ -3,6 +3,8 @@ import com.databricks.sdk.core.ApiClient; import com.databricks.sdk.support.Generated; +import java.util.HashMap; +import java.util.Map; /** Package-local implementation of GlobalInitScripts */ @Generated @@ -16,30 +18,40 @@ public GlobalInitScriptsImpl(ApiClient apiClient) { @Override public CreateResponse create(GlobalInitScriptCreateRequest request) { String path = "/api/2.0/global-init-scripts"; - return apiClient.POST(path, request, CreateResponse.class); + Map headers = new HashMap<>(); + headers.put("Accept", "application/json"); + headers.put("Content-Type", "application/json"); + return apiClient.POST(path, request, CreateResponse.class, headers); } @Override public void delete(DeleteGlobalInitScriptRequest request) { String path = String.format("/api/2.0/global-init-scripts/%s", request.getScriptId()); - apiClient.DELETE(path, request, Void.class); + Map headers = new HashMap<>(); + apiClient.DELETE(path, request, Void.class, headers); } @Override public GlobalInitScriptDetailsWithContent get(GetGlobalInitScriptRequest request) { String path = String.format("/api/2.0/global-init-scripts/%s", request.getScriptId()); - return apiClient.GET(path, request, GlobalInitScriptDetailsWithContent.class); + Map headers = new HashMap<>(); + headers.put("Accept", "application/json"); + return apiClient.GET(path, request, GlobalInitScriptDetailsWithContent.class, headers); } @Override public ListGlobalInitScriptsResponse list() { String path = "/api/2.0/global-init-scripts"; - return apiClient.GET(path, ListGlobalInitScriptsResponse.class); + Map headers = new HashMap<>(); + headers.put("Accept", "application/json"); + return apiClient.GET(path, ListGlobalInitScriptsResponse.class, headers); } @Override public void update(GlobalInitScriptUpdateRequest request) { String path = String.format("/api/2.0/global-init-scripts/%s", request.getScriptId()); - apiClient.PATCH(path, request, Void.class); + Map headers = new HashMap<>(); + headers.put("Content-Type", "application/json"); + apiClient.PATCH(path, request, Void.class, headers); } } diff --git a/databricks-sdk-java/src/main/java/com/databricks/sdk/service/compute/InstancePoolsImpl.java b/databricks-sdk-java/src/main/java/com/databricks/sdk/service/compute/InstancePoolsImpl.java index 2df3a704b..3efdb23d1 100755 --- a/databricks-sdk-java/src/main/java/com/databricks/sdk/service/compute/InstancePoolsImpl.java +++ b/databricks-sdk-java/src/main/java/com/databricks/sdk/service/compute/InstancePoolsImpl.java @@ -3,6 +3,8 @@ import com.databricks.sdk.core.ApiClient; import com.databricks.sdk.support.Generated; +import java.util.HashMap; +import java.util.Map; /** Package-local implementation of InstancePools */ @Generated @@ -16,25 +18,36 @@ public InstancePoolsImpl(ApiClient apiClient) { @Override public CreateInstancePoolResponse create(CreateInstancePool request) { String path = "/api/2.0/instance-pools/create"; - return apiClient.POST(path, request, CreateInstancePoolResponse.class); + Map headers = new HashMap<>(); + headers.put("Accept", "application/json"); + headers.put("Content-Type", "application/json"); + return apiClient.POST(path, request, CreateInstancePoolResponse.class, headers); } @Override public void delete(DeleteInstancePool request) { String path = "/api/2.0/instance-pools/delete"; - apiClient.POST(path, request, Void.class); + Map headers = new HashMap<>(); + headers.put("Accept", "application/json"); + headers.put("Content-Type", "application/json"); + apiClient.POST(path, request, Void.class, headers); } @Override public void edit(EditInstancePool request) { String path = "/api/2.0/instance-pools/edit"; - apiClient.POST(path, request, Void.class); + Map headers = new HashMap<>(); + headers.put("Accept", "application/json"); + headers.put("Content-Type", "application/json"); + apiClient.POST(path, request, Void.class, headers); } @Override public GetInstancePool get(GetInstancePoolRequest request) { String path = "/api/2.0/instance-pools/get"; - return apiClient.GET(path, request, GetInstancePool.class); + Map headers = new HashMap<>(); + headers.put("Accept", "application/json"); + return apiClient.GET(path, request, GetInstancePool.class, headers); } @Override @@ -43,7 +56,9 @@ public GetInstancePoolPermissionLevelsResponse getInstancePoolPermissionLevels( String path = String.format( "/api/2.0/permissions/instance-pools/%s/permissionLevels", request.getInstancePoolId()); - return apiClient.GET(path, request, GetInstancePoolPermissionLevelsResponse.class); + Map headers = new HashMap<>(); + headers.put("Accept", "application/json"); + return apiClient.GET(path, request, GetInstancePoolPermissionLevelsResponse.class, headers); } @Override @@ -51,13 +66,17 @@ public InstancePoolPermissions getInstancePoolPermissions( GetInstancePoolPermissionsRequest request) { String path = String.format("/api/2.0/permissions/instance-pools/%s", request.getInstancePoolId()); - return apiClient.GET(path, request, InstancePoolPermissions.class); + Map headers = new HashMap<>(); + headers.put("Accept", "application/json"); + return apiClient.GET(path, request, InstancePoolPermissions.class, headers); } @Override public ListInstancePools list() { String path = "/api/2.0/instance-pools/list"; - return apiClient.GET(path, ListInstancePools.class); + Map headers = new HashMap<>(); + headers.put("Accept", "application/json"); + return apiClient.GET(path, ListInstancePools.class, headers); } @Override @@ -65,7 +84,10 @@ public InstancePoolPermissions setInstancePoolPermissions( InstancePoolPermissionsRequest request) { String path = String.format("/api/2.0/permissions/instance-pools/%s", request.getInstancePoolId()); - return apiClient.PUT(path, request, InstancePoolPermissions.class); + Map headers = new HashMap<>(); + headers.put("Accept", "application/json"); + headers.put("Content-Type", "application/json"); + return apiClient.PUT(path, request, InstancePoolPermissions.class, headers); } @Override @@ -73,6 +95,9 @@ public InstancePoolPermissions updateInstancePoolPermissions( InstancePoolPermissionsRequest request) { String path = String.format("/api/2.0/permissions/instance-pools/%s", request.getInstancePoolId()); - return apiClient.PATCH(path, request, InstancePoolPermissions.class); + Map headers = new HashMap<>(); + headers.put("Accept", "application/json"); + headers.put("Content-Type", "application/json"); + return apiClient.PATCH(path, request, InstancePoolPermissions.class, headers); } } diff --git a/databricks-sdk-java/src/main/java/com/databricks/sdk/service/compute/InstanceProfilesImpl.java b/databricks-sdk-java/src/main/java/com/databricks/sdk/service/compute/InstanceProfilesImpl.java index 7c4cb66a0..cd90cac4d 100755 --- a/databricks-sdk-java/src/main/java/com/databricks/sdk/service/compute/InstanceProfilesImpl.java +++ b/databricks-sdk-java/src/main/java/com/databricks/sdk/service/compute/InstanceProfilesImpl.java @@ -3,6 +3,8 @@ import com.databricks.sdk.core.ApiClient; import com.databricks.sdk.support.Generated; +import java.util.HashMap; +import java.util.Map; /** Package-local implementation of InstanceProfiles */ @Generated @@ -16,24 +18,35 @@ public InstanceProfilesImpl(ApiClient apiClient) { @Override public void add(AddInstanceProfile request) { String path = "/api/2.0/instance-profiles/add"; - apiClient.POST(path, request, Void.class); + Map headers = new HashMap<>(); + headers.put("Accept", "application/json"); + headers.put("Content-Type", "application/json"); + apiClient.POST(path, request, Void.class, headers); } @Override public void edit(InstanceProfile request) { String path = "/api/2.0/instance-profiles/edit"; - apiClient.POST(path, request, Void.class); + Map headers = new HashMap<>(); + headers.put("Accept", "application/json"); + headers.put("Content-Type", "application/json"); + apiClient.POST(path, request, Void.class, headers); } @Override public ListInstanceProfilesResponse list() { String path = "/api/2.0/instance-profiles/list"; - return apiClient.GET(path, ListInstanceProfilesResponse.class); + Map headers = new HashMap<>(); + headers.put("Accept", "application/json"); + return apiClient.GET(path, ListInstanceProfilesResponse.class, headers); } @Override public void remove(RemoveInstanceProfile request) { String path = "/api/2.0/instance-profiles/remove"; - apiClient.POST(path, request, Void.class); + Map headers = new HashMap<>(); + headers.put("Accept", "application/json"); + headers.put("Content-Type", "application/json"); + apiClient.POST(path, request, Void.class, headers); } } diff --git a/databricks-sdk-java/src/main/java/com/databricks/sdk/service/compute/LibrariesImpl.java b/databricks-sdk-java/src/main/java/com/databricks/sdk/service/compute/LibrariesImpl.java index 3edbb964c..de279ca31 100755 --- a/databricks-sdk-java/src/main/java/com/databricks/sdk/service/compute/LibrariesImpl.java +++ b/databricks-sdk-java/src/main/java/com/databricks/sdk/service/compute/LibrariesImpl.java @@ -3,6 +3,8 @@ import com.databricks.sdk.core.ApiClient; import com.databricks.sdk.support.Generated; +import java.util.HashMap; +import java.util.Map; /** Package-local implementation of Libraries */ @Generated @@ -16,24 +18,34 @@ public LibrariesImpl(ApiClient apiClient) { @Override public ListAllClusterLibraryStatusesResponse allClusterStatuses() { String path = "/api/2.0/libraries/all-cluster-statuses"; - return apiClient.GET(path, ListAllClusterLibraryStatusesResponse.class); + Map headers = new HashMap<>(); + headers.put("Accept", "application/json"); + return apiClient.GET(path, ListAllClusterLibraryStatusesResponse.class, headers); } @Override public ClusterLibraryStatuses clusterStatus(ClusterStatusRequest request) { String path = "/api/2.0/libraries/cluster-status"; - return apiClient.GET(path, request, ClusterLibraryStatuses.class); + Map headers = new HashMap<>(); + headers.put("Accept", "application/json"); + return apiClient.GET(path, request, ClusterLibraryStatuses.class, headers); } @Override public void install(InstallLibraries request) { String path = "/api/2.0/libraries/install"; - apiClient.POST(path, request, Void.class); + Map headers = new HashMap<>(); + headers.put("Accept", "application/json"); + headers.put("Content-Type", "application/json"); + apiClient.POST(path, request, Void.class, headers); } @Override public void uninstall(UninstallLibraries request) { String path = "/api/2.0/libraries/uninstall"; - apiClient.POST(path, request, Void.class); + Map headers = new HashMap<>(); + headers.put("Accept", "application/json"); + headers.put("Content-Type", "application/json"); + apiClient.POST(path, request, Void.class, headers); } } diff --git a/databricks-sdk-java/src/main/java/com/databricks/sdk/service/compute/PolicyFamiliesImpl.java b/databricks-sdk-java/src/main/java/com/databricks/sdk/service/compute/PolicyFamiliesImpl.java index 8219ff1e3..9902a2134 100755 --- a/databricks-sdk-java/src/main/java/com/databricks/sdk/service/compute/PolicyFamiliesImpl.java +++ b/databricks-sdk-java/src/main/java/com/databricks/sdk/service/compute/PolicyFamiliesImpl.java @@ -3,6 +3,8 @@ import com.databricks.sdk.core.ApiClient; import com.databricks.sdk.support.Generated; +import java.util.HashMap; +import java.util.Map; /** Package-local implementation of PolicyFamilies */ @Generated @@ -16,12 +18,16 @@ public PolicyFamiliesImpl(ApiClient apiClient) { @Override public PolicyFamily get(GetPolicyFamilyRequest request) { String path = String.format("/api/2.0/policy-families/%s", request.getPolicyFamilyId()); - return apiClient.GET(path, request, PolicyFamily.class); + Map headers = new HashMap<>(); + headers.put("Accept", "application/json"); + return apiClient.GET(path, request, PolicyFamily.class, headers); } @Override public ListPolicyFamiliesResponse list(ListPolicyFamiliesRequest request) { String path = "/api/2.0/policy-families"; - return apiClient.GET(path, request, ListPolicyFamiliesResponse.class); + Map headers = new HashMap<>(); + headers.put("Accept", "application/json"); + return apiClient.GET(path, request, ListPolicyFamiliesResponse.class, headers); } } diff --git a/databricks-sdk-java/src/main/java/com/databricks/sdk/service/files/DbfsImpl.java b/databricks-sdk-java/src/main/java/com/databricks/sdk/service/files/DbfsImpl.java index 4f4032944..6a1efb994 100755 --- a/databricks-sdk-java/src/main/java/com/databricks/sdk/service/files/DbfsImpl.java +++ b/databricks-sdk-java/src/main/java/com/databricks/sdk/service/files/DbfsImpl.java @@ -3,6 +3,8 @@ import com.databricks.sdk.core.ApiClient; import com.databricks.sdk.support.Generated; +import java.util.HashMap; +import java.util.Map; /** Package-local implementation of Dbfs */ @Generated @@ -16,60 +18,87 @@ public DbfsImpl(ApiClient apiClient) { @Override public void addBlock(AddBlock request) { String path = "/api/2.0/dbfs/add-block"; - apiClient.POST(path, request, Void.class); + Map headers = new HashMap<>(); + headers.put("Accept", "application/json"); + headers.put("Content-Type", "application/json"); + apiClient.POST(path, request, Void.class, headers); } @Override public void close(Close request) { String path = "/api/2.0/dbfs/close"; - apiClient.POST(path, request, Void.class); + Map headers = new HashMap<>(); + headers.put("Accept", "application/json"); + headers.put("Content-Type", "application/json"); + apiClient.POST(path, request, Void.class, headers); } @Override public CreateResponse create(Create request) { String path = "/api/2.0/dbfs/create"; - return apiClient.POST(path, request, CreateResponse.class); + Map headers = new HashMap<>(); + headers.put("Accept", "application/json"); + headers.put("Content-Type", "application/json"); + return apiClient.POST(path, request, CreateResponse.class, headers); } @Override public void delete(Delete request) { String path = "/api/2.0/dbfs/delete"; - apiClient.POST(path, request, Void.class); + Map headers = new HashMap<>(); + headers.put("Accept", "application/json"); + headers.put("Content-Type", "application/json"); + apiClient.POST(path, request, Void.class, headers); } @Override public FileInfo getStatus(GetStatusRequest request) { String path = "/api/2.0/dbfs/get-status"; - return apiClient.GET(path, request, FileInfo.class); + Map headers = new HashMap<>(); + headers.put("Accept", "application/json"); + return apiClient.GET(path, request, FileInfo.class, headers); } @Override public ListStatusResponse list(ListDbfsRequest request) { String path = "/api/2.0/dbfs/list"; - return apiClient.GET(path, request, ListStatusResponse.class); + Map headers = new HashMap<>(); + headers.put("Accept", "application/json"); + return apiClient.GET(path, request, ListStatusResponse.class, headers); } @Override public void mkdirs(MkDirs request) { String path = "/api/2.0/dbfs/mkdirs"; - apiClient.POST(path, request, Void.class); + Map headers = new HashMap<>(); + headers.put("Accept", "application/json"); + headers.put("Content-Type", "application/json"); + apiClient.POST(path, request, Void.class, headers); } @Override public void move(Move request) { String path = "/api/2.0/dbfs/move"; - apiClient.POST(path, request, Void.class); + Map headers = new HashMap<>(); + headers.put("Accept", "application/json"); + headers.put("Content-Type", "application/json"); + apiClient.POST(path, request, Void.class, headers); } @Override public void put(Put request) { String path = "/api/2.0/dbfs/put"; - apiClient.POST(path, request, Void.class); + Map headers = new HashMap<>(); + headers.put("Accept", "application/json"); + headers.put("Content-Type", "application/json"); + apiClient.POST(path, request, Void.class, headers); } @Override public ReadResponse read(ReadDbfsRequest request) { String path = "/api/2.0/dbfs/read"; - return apiClient.GET(path, request, ReadResponse.class); + Map headers = new HashMap<>(); + headers.put("Accept", "application/json"); + return apiClient.GET(path, request, ReadResponse.class, headers); } } diff --git a/databricks-sdk-java/src/main/java/com/databricks/sdk/service/iam/AccountAccessControlImpl.java b/databricks-sdk-java/src/main/java/com/databricks/sdk/service/iam/AccountAccessControlImpl.java index 41c9803ed..43bc13198 100755 --- a/databricks-sdk-java/src/main/java/com/databricks/sdk/service/iam/AccountAccessControlImpl.java +++ b/databricks-sdk-java/src/main/java/com/databricks/sdk/service/iam/AccountAccessControlImpl.java @@ -3,6 +3,8 @@ import com.databricks.sdk.core.ApiClient; import com.databricks.sdk.support.Generated; +import java.util.HashMap; +import java.util.Map; /** Package-local implementation of AccountAccessControl */ @Generated @@ -20,7 +22,9 @@ public GetAssignableRolesForResourceResponse getAssignableRolesForResource( String.format( "/api/2.0/preview/accounts/%s/access-control/assignable-roles", apiClient.configuredAccountID()); - return apiClient.GET(path, request, GetAssignableRolesForResourceResponse.class); + Map headers = new HashMap<>(); + headers.put("Accept", "application/json"); + return apiClient.GET(path, request, GetAssignableRolesForResourceResponse.class, headers); } @Override @@ -29,7 +33,9 @@ public RuleSetResponse getRuleSet(GetRuleSetRequest request) { String.format( "/api/2.0/preview/accounts/%s/access-control/rule-sets", apiClient.configuredAccountID()); - return apiClient.GET(path, request, RuleSetResponse.class); + Map headers = new HashMap<>(); + headers.put("Accept", "application/json"); + return apiClient.GET(path, request, RuleSetResponse.class, headers); } @Override @@ -38,6 +44,9 @@ public RuleSetResponse updateRuleSet(UpdateRuleSetRequest request) { String.format( "/api/2.0/preview/accounts/%s/access-control/rule-sets", apiClient.configuredAccountID()); - return apiClient.PUT(path, request, RuleSetResponse.class); + Map headers = new HashMap<>(); + headers.put("Accept", "application/json"); + headers.put("Content-Type", "application/json"); + return apiClient.PUT(path, request, RuleSetResponse.class, headers); } } diff --git a/databricks-sdk-java/src/main/java/com/databricks/sdk/service/iam/AccountAccessControlProxyImpl.java b/databricks-sdk-java/src/main/java/com/databricks/sdk/service/iam/AccountAccessControlProxyImpl.java index 9f17b9e14..b3292da04 100755 --- a/databricks-sdk-java/src/main/java/com/databricks/sdk/service/iam/AccountAccessControlProxyImpl.java +++ b/databricks-sdk-java/src/main/java/com/databricks/sdk/service/iam/AccountAccessControlProxyImpl.java @@ -3,6 +3,8 @@ import com.databricks.sdk.core.ApiClient; import com.databricks.sdk.support.Generated; +import java.util.HashMap; +import java.util.Map; /** Package-local implementation of AccountAccessControlProxy */ @Generated @@ -17,18 +19,25 @@ public AccountAccessControlProxyImpl(ApiClient apiClient) { public GetAssignableRolesForResourceResponse getAssignableRolesForResource( GetAssignableRolesForResourceRequest request) { String path = "/api/2.0/preview/accounts/access-control/assignable-roles"; - return apiClient.GET(path, request, GetAssignableRolesForResourceResponse.class); + Map headers = new HashMap<>(); + headers.put("Accept", "application/json"); + return apiClient.GET(path, request, GetAssignableRolesForResourceResponse.class, headers); } @Override public RuleSetResponse getRuleSet(GetRuleSetRequest request) { String path = "/api/2.0/preview/accounts/access-control/rule-sets"; - return apiClient.GET(path, request, RuleSetResponse.class); + Map headers = new HashMap<>(); + headers.put("Accept", "application/json"); + return apiClient.GET(path, request, RuleSetResponse.class, headers); } @Override public RuleSetResponse updateRuleSet(UpdateRuleSetRequest request) { String path = "/api/2.0/preview/accounts/access-control/rule-sets"; - return apiClient.PUT(path, request, RuleSetResponse.class); + Map headers = new HashMap<>(); + headers.put("Accept", "application/json"); + headers.put("Content-Type", "application/json"); + return apiClient.PUT(path, request, RuleSetResponse.class, headers); } } diff --git a/databricks-sdk-java/src/main/java/com/databricks/sdk/service/iam/AccountGroupsImpl.java b/databricks-sdk-java/src/main/java/com/databricks/sdk/service/iam/AccountGroupsImpl.java index 5328fcf17..13017f65b 100755 --- a/databricks-sdk-java/src/main/java/com/databricks/sdk/service/iam/AccountGroupsImpl.java +++ b/databricks-sdk-java/src/main/java/com/databricks/sdk/service/iam/AccountGroupsImpl.java @@ -3,6 +3,8 @@ import com.databricks.sdk.core.ApiClient; import com.databricks.sdk.support.Generated; +import java.util.HashMap; +import java.util.Map; /** Package-local implementation of AccountGroups */ @Generated @@ -17,7 +19,10 @@ public AccountGroupsImpl(ApiClient apiClient) { public Group create(Group request) { String path = String.format("/api/2.0/accounts/%s/scim/v2/Groups", apiClient.configuredAccountID()); - return apiClient.POST(path, request, Group.class); + Map headers = new HashMap<>(); + headers.put("Accept", "application/json"); + headers.put("Content-Type", "application/json"); + return apiClient.POST(path, request, Group.class, headers); } @Override @@ -26,7 +31,8 @@ public void delete(DeleteAccountGroupRequest request) { String.format( "/api/2.0/accounts/%s/scim/v2/Groups/%s", apiClient.configuredAccountID(), request.getId()); - apiClient.DELETE(path, request, Void.class); + Map headers = new HashMap<>(); + apiClient.DELETE(path, request, Void.class, headers); } @Override @@ -35,14 +41,18 @@ public Group get(GetAccountGroupRequest request) { String.format( "/api/2.0/accounts/%s/scim/v2/Groups/%s", apiClient.configuredAccountID(), request.getId()); - return apiClient.GET(path, request, Group.class); + Map headers = new HashMap<>(); + headers.put("Accept", "application/json"); + return apiClient.GET(path, request, Group.class, headers); } @Override public ListGroupsResponse list(ListAccountGroupsRequest request) { String path = String.format("/api/2.0/accounts/%s/scim/v2/Groups", apiClient.configuredAccountID()); - return apiClient.GET(path, request, ListGroupsResponse.class); + Map headers = new HashMap<>(); + headers.put("Accept", "application/json"); + return apiClient.GET(path, request, ListGroupsResponse.class, headers); } @Override @@ -51,7 +61,10 @@ public void patch(PartialUpdate request) { String.format( "/api/2.0/accounts/%s/scim/v2/Groups/%s", apiClient.configuredAccountID(), request.getId()); - apiClient.PATCH(path, request, Void.class); + Map headers = new HashMap<>(); + headers.put("Accept", "application/json"); + headers.put("Content-Type", "application/json"); + apiClient.PATCH(path, request, Void.class, headers); } @Override @@ -60,6 +73,9 @@ public void update(Group request) { String.format( "/api/2.0/accounts/%s/scim/v2/Groups/%s", apiClient.configuredAccountID(), request.getId()); - apiClient.PUT(path, request, Void.class); + Map headers = new HashMap<>(); + headers.put("Accept", "application/json"); + headers.put("Content-Type", "application/json"); + apiClient.PUT(path, request, Void.class, headers); } } diff --git a/databricks-sdk-java/src/main/java/com/databricks/sdk/service/iam/AccountServicePrincipalsImpl.java b/databricks-sdk-java/src/main/java/com/databricks/sdk/service/iam/AccountServicePrincipalsImpl.java index d416c2c77..af1163981 100755 --- a/databricks-sdk-java/src/main/java/com/databricks/sdk/service/iam/AccountServicePrincipalsImpl.java +++ b/databricks-sdk-java/src/main/java/com/databricks/sdk/service/iam/AccountServicePrincipalsImpl.java @@ -3,6 +3,8 @@ import com.databricks.sdk.core.ApiClient; import com.databricks.sdk.support.Generated; +import java.util.HashMap; +import java.util.Map; /** Package-local implementation of AccountServicePrincipals */ @Generated @@ -18,7 +20,10 @@ public ServicePrincipal create(ServicePrincipal request) { String path = String.format( "/api/2.0/accounts/%s/scim/v2/ServicePrincipals", apiClient.configuredAccountID()); - return apiClient.POST(path, request, ServicePrincipal.class); + Map headers = new HashMap<>(); + headers.put("Accept", "application/json"); + headers.put("Content-Type", "application/json"); + return apiClient.POST(path, request, ServicePrincipal.class, headers); } @Override @@ -27,7 +32,8 @@ public void delete(DeleteAccountServicePrincipalRequest request) { String.format( "/api/2.0/accounts/%s/scim/v2/ServicePrincipals/%s", apiClient.configuredAccountID(), request.getId()); - apiClient.DELETE(path, request, Void.class); + Map headers = new HashMap<>(); + apiClient.DELETE(path, request, Void.class, headers); } @Override @@ -36,7 +42,9 @@ public ServicePrincipal get(GetAccountServicePrincipalRequest request) { String.format( "/api/2.0/accounts/%s/scim/v2/ServicePrincipals/%s", apiClient.configuredAccountID(), request.getId()); - return apiClient.GET(path, request, ServicePrincipal.class); + Map headers = new HashMap<>(); + headers.put("Accept", "application/json"); + return apiClient.GET(path, request, ServicePrincipal.class, headers); } @Override @@ -44,7 +52,9 @@ public ListServicePrincipalResponse list(ListAccountServicePrincipalsRequest req String path = String.format( "/api/2.0/accounts/%s/scim/v2/ServicePrincipals", apiClient.configuredAccountID()); - return apiClient.GET(path, request, ListServicePrincipalResponse.class); + Map headers = new HashMap<>(); + headers.put("Accept", "application/json"); + return apiClient.GET(path, request, ListServicePrincipalResponse.class, headers); } @Override @@ -53,7 +63,10 @@ public void patch(PartialUpdate request) { String.format( "/api/2.0/accounts/%s/scim/v2/ServicePrincipals/%s", apiClient.configuredAccountID(), request.getId()); - apiClient.PATCH(path, request, Void.class); + Map headers = new HashMap<>(); + headers.put("Accept", "application/json"); + headers.put("Content-Type", "application/json"); + apiClient.PATCH(path, request, Void.class, headers); } @Override @@ -62,6 +75,9 @@ public void update(ServicePrincipal request) { String.format( "/api/2.0/accounts/%s/scim/v2/ServicePrincipals/%s", apiClient.configuredAccountID(), request.getId()); - apiClient.PUT(path, request, Void.class); + Map headers = new HashMap<>(); + headers.put("Accept", "application/json"); + headers.put("Content-Type", "application/json"); + apiClient.PUT(path, request, Void.class, headers); } } diff --git a/databricks-sdk-java/src/main/java/com/databricks/sdk/service/iam/AccountUsersImpl.java b/databricks-sdk-java/src/main/java/com/databricks/sdk/service/iam/AccountUsersImpl.java index 2bff4396d..f80624783 100755 --- a/databricks-sdk-java/src/main/java/com/databricks/sdk/service/iam/AccountUsersImpl.java +++ b/databricks-sdk-java/src/main/java/com/databricks/sdk/service/iam/AccountUsersImpl.java @@ -3,6 +3,8 @@ import com.databricks.sdk.core.ApiClient; import com.databricks.sdk.support.Generated; +import java.util.HashMap; +import java.util.Map; /** Package-local implementation of AccountUsers */ @Generated @@ -17,7 +19,10 @@ public AccountUsersImpl(ApiClient apiClient) { public User create(User request) { String path = String.format("/api/2.0/accounts/%s/scim/v2/Users", apiClient.configuredAccountID()); - return apiClient.POST(path, request, User.class); + Map headers = new HashMap<>(); + headers.put("Accept", "application/json"); + headers.put("Content-Type", "application/json"); + return apiClient.POST(path, request, User.class, headers); } @Override @@ -26,7 +31,8 @@ public void delete(DeleteAccountUserRequest request) { String.format( "/api/2.0/accounts/%s/scim/v2/Users/%s", apiClient.configuredAccountID(), request.getId()); - apiClient.DELETE(path, request, Void.class); + Map headers = new HashMap<>(); + apiClient.DELETE(path, request, Void.class, headers); } @Override @@ -35,14 +41,18 @@ public User get(GetAccountUserRequest request) { String.format( "/api/2.0/accounts/%s/scim/v2/Users/%s", apiClient.configuredAccountID(), request.getId()); - return apiClient.GET(path, request, User.class); + Map headers = new HashMap<>(); + headers.put("Accept", "application/json"); + return apiClient.GET(path, request, User.class, headers); } @Override public ListUsersResponse list(ListAccountUsersRequest request) { String path = String.format("/api/2.0/accounts/%s/scim/v2/Users", apiClient.configuredAccountID()); - return apiClient.GET(path, request, ListUsersResponse.class); + Map headers = new HashMap<>(); + headers.put("Accept", "application/json"); + return apiClient.GET(path, request, ListUsersResponse.class, headers); } @Override @@ -51,7 +61,10 @@ public void patch(PartialUpdate request) { String.format( "/api/2.0/accounts/%s/scim/v2/Users/%s", apiClient.configuredAccountID(), request.getId()); - apiClient.PATCH(path, request, Void.class); + Map headers = new HashMap<>(); + headers.put("Accept", "application/json"); + headers.put("Content-Type", "application/json"); + apiClient.PATCH(path, request, Void.class, headers); } @Override @@ -60,6 +73,9 @@ public void update(User request) { String.format( "/api/2.0/accounts/%s/scim/v2/Users/%s", apiClient.configuredAccountID(), request.getId()); - apiClient.PUT(path, request, Void.class); + Map headers = new HashMap<>(); + headers.put("Accept", "application/json"); + headers.put("Content-Type", "application/json"); + apiClient.PUT(path, request, Void.class, headers); } } diff --git a/databricks-sdk-java/src/main/java/com/databricks/sdk/service/iam/CurrentUserImpl.java b/databricks-sdk-java/src/main/java/com/databricks/sdk/service/iam/CurrentUserImpl.java index 481814412..d33b0f8b1 100755 --- a/databricks-sdk-java/src/main/java/com/databricks/sdk/service/iam/CurrentUserImpl.java +++ b/databricks-sdk-java/src/main/java/com/databricks/sdk/service/iam/CurrentUserImpl.java @@ -3,6 +3,8 @@ import com.databricks.sdk.core.ApiClient; import com.databricks.sdk.support.Generated; +import java.util.HashMap; +import java.util.Map; /** Package-local implementation of CurrentUser */ @Generated @@ -16,6 +18,8 @@ public CurrentUserImpl(ApiClient apiClient) { @Override public User me() { String path = "/api/2.0/preview/scim/v2/Me"; - return apiClient.GET(path, User.class); + Map headers = new HashMap<>(); + headers.put("Accept", "application/json"); + return apiClient.GET(path, User.class, headers); } } diff --git a/databricks-sdk-java/src/main/java/com/databricks/sdk/service/iam/GroupsImpl.java b/databricks-sdk-java/src/main/java/com/databricks/sdk/service/iam/GroupsImpl.java index 1ac900afb..9b26ca969 100755 --- a/databricks-sdk-java/src/main/java/com/databricks/sdk/service/iam/GroupsImpl.java +++ b/databricks-sdk-java/src/main/java/com/databricks/sdk/service/iam/GroupsImpl.java @@ -3,6 +3,8 @@ import com.databricks.sdk.core.ApiClient; import com.databricks.sdk.support.Generated; +import java.util.HashMap; +import java.util.Map; /** Package-local implementation of Groups */ @Generated @@ -16,36 +18,50 @@ public GroupsImpl(ApiClient apiClient) { @Override public Group create(Group request) { String path = "/api/2.0/preview/scim/v2/Groups"; - return apiClient.POST(path, request, Group.class); + Map headers = new HashMap<>(); + headers.put("Accept", "application/json"); + headers.put("Content-Type", "application/json"); + return apiClient.POST(path, request, Group.class, headers); } @Override public void delete(DeleteGroupRequest request) { String path = String.format("/api/2.0/preview/scim/v2/Groups/%s", request.getId()); - apiClient.DELETE(path, request, Void.class); + Map headers = new HashMap<>(); + apiClient.DELETE(path, request, Void.class, headers); } @Override public Group get(GetGroupRequest request) { String path = String.format("/api/2.0/preview/scim/v2/Groups/%s", request.getId()); - return apiClient.GET(path, request, Group.class); + Map headers = new HashMap<>(); + headers.put("Accept", "application/json"); + return apiClient.GET(path, request, Group.class, headers); } @Override public ListGroupsResponse list(ListGroupsRequest request) { String path = "/api/2.0/preview/scim/v2/Groups"; - return apiClient.GET(path, request, ListGroupsResponse.class); + Map headers = new HashMap<>(); + headers.put("Accept", "application/json"); + return apiClient.GET(path, request, ListGroupsResponse.class, headers); } @Override public void patch(PartialUpdate request) { String path = String.format("/api/2.0/preview/scim/v2/Groups/%s", request.getId()); - apiClient.PATCH(path, request, Void.class); + Map headers = new HashMap<>(); + headers.put("Accept", "application/json"); + headers.put("Content-Type", "application/json"); + apiClient.PATCH(path, request, Void.class, headers); } @Override public void update(Group request) { String path = String.format("/api/2.0/preview/scim/v2/Groups/%s", request.getId()); - apiClient.PUT(path, request, Void.class); + Map headers = new HashMap<>(); + headers.put("Accept", "application/json"); + headers.put("Content-Type", "application/json"); + apiClient.PUT(path, request, Void.class, headers); } } diff --git a/databricks-sdk-java/src/main/java/com/databricks/sdk/service/iam/PermissionsImpl.java b/databricks-sdk-java/src/main/java/com/databricks/sdk/service/iam/PermissionsImpl.java index b0ae26b2b..cd11548dd 100755 --- a/databricks-sdk-java/src/main/java/com/databricks/sdk/service/iam/PermissionsImpl.java +++ b/databricks-sdk-java/src/main/java/com/databricks/sdk/service/iam/PermissionsImpl.java @@ -3,6 +3,8 @@ import com.databricks.sdk.core.ApiClient; import com.databricks.sdk.support.Generated; +import java.util.HashMap; +import java.util.Map; /** Package-local implementation of Permissions */ @Generated @@ -19,7 +21,9 @@ public ObjectPermissions get(GetPermissionRequest request) { String.format( "/api/2.0/permissions/%s/%s", request.getRequestObjectType(), request.getRequestObjectId()); - return apiClient.GET(path, request, ObjectPermissions.class); + Map headers = new HashMap<>(); + headers.put("Accept", "application/json"); + return apiClient.GET(path, request, ObjectPermissions.class, headers); } @Override @@ -28,7 +32,9 @@ public GetPermissionLevelsResponse getPermissionLevels(GetPermissionLevelsReques String.format( "/api/2.0/permissions/%s/%s/permissionLevels", request.getRequestObjectType(), request.getRequestObjectId()); - return apiClient.GET(path, request, GetPermissionLevelsResponse.class); + Map headers = new HashMap<>(); + headers.put("Accept", "application/json"); + return apiClient.GET(path, request, GetPermissionLevelsResponse.class, headers); } @Override @@ -37,7 +43,10 @@ public ObjectPermissions set(PermissionsRequest request) { String.format( "/api/2.0/permissions/%s/%s", request.getRequestObjectType(), request.getRequestObjectId()); - return apiClient.PUT(path, request, ObjectPermissions.class); + Map headers = new HashMap<>(); + headers.put("Accept", "application/json"); + headers.put("Content-Type", "application/json"); + return apiClient.PUT(path, request, ObjectPermissions.class, headers); } @Override @@ -46,6 +55,9 @@ public ObjectPermissions update(PermissionsRequest request) { String.format( "/api/2.0/permissions/%s/%s", request.getRequestObjectType(), request.getRequestObjectId()); - return apiClient.PATCH(path, request, ObjectPermissions.class); + Map headers = new HashMap<>(); + headers.put("Accept", "application/json"); + headers.put("Content-Type", "application/json"); + return apiClient.PATCH(path, request, ObjectPermissions.class, headers); } } diff --git a/databricks-sdk-java/src/main/java/com/databricks/sdk/service/iam/ServicePrincipalsImpl.java b/databricks-sdk-java/src/main/java/com/databricks/sdk/service/iam/ServicePrincipalsImpl.java index c9186a089..a67321a33 100755 --- a/databricks-sdk-java/src/main/java/com/databricks/sdk/service/iam/ServicePrincipalsImpl.java +++ b/databricks-sdk-java/src/main/java/com/databricks/sdk/service/iam/ServicePrincipalsImpl.java @@ -3,6 +3,8 @@ import com.databricks.sdk.core.ApiClient; import com.databricks.sdk.support.Generated; +import java.util.HashMap; +import java.util.Map; /** Package-local implementation of ServicePrincipals */ @Generated @@ -16,36 +18,50 @@ public ServicePrincipalsImpl(ApiClient apiClient) { @Override public ServicePrincipal create(ServicePrincipal request) { String path = "/api/2.0/preview/scim/v2/ServicePrincipals"; - return apiClient.POST(path, request, ServicePrincipal.class); + Map headers = new HashMap<>(); + headers.put("Accept", "application/json"); + headers.put("Content-Type", "application/json"); + return apiClient.POST(path, request, ServicePrincipal.class, headers); } @Override public void delete(DeleteServicePrincipalRequest request) { String path = String.format("/api/2.0/preview/scim/v2/ServicePrincipals/%s", request.getId()); - apiClient.DELETE(path, request, Void.class); + Map headers = new HashMap<>(); + apiClient.DELETE(path, request, Void.class, headers); } @Override public ServicePrincipal get(GetServicePrincipalRequest request) { String path = String.format("/api/2.0/preview/scim/v2/ServicePrincipals/%s", request.getId()); - return apiClient.GET(path, request, ServicePrincipal.class); + Map headers = new HashMap<>(); + headers.put("Accept", "application/json"); + return apiClient.GET(path, request, ServicePrincipal.class, headers); } @Override public ListServicePrincipalResponse list(ListServicePrincipalsRequest request) { String path = "/api/2.0/preview/scim/v2/ServicePrincipals"; - return apiClient.GET(path, request, ListServicePrincipalResponse.class); + Map headers = new HashMap<>(); + headers.put("Accept", "application/json"); + return apiClient.GET(path, request, ListServicePrincipalResponse.class, headers); } @Override public void patch(PartialUpdate request) { String path = String.format("/api/2.0/preview/scim/v2/ServicePrincipals/%s", request.getId()); - apiClient.PATCH(path, request, Void.class); + Map headers = new HashMap<>(); + headers.put("Accept", "application/json"); + headers.put("Content-Type", "application/json"); + apiClient.PATCH(path, request, Void.class, headers); } @Override public void update(ServicePrincipal request) { String path = String.format("/api/2.0/preview/scim/v2/ServicePrincipals/%s", request.getId()); - apiClient.PUT(path, request, Void.class); + Map headers = new HashMap<>(); + headers.put("Accept", "application/json"); + headers.put("Content-Type", "application/json"); + apiClient.PUT(path, request, Void.class, headers); } } diff --git a/databricks-sdk-java/src/main/java/com/databricks/sdk/service/iam/UsersImpl.java b/databricks-sdk-java/src/main/java/com/databricks/sdk/service/iam/UsersImpl.java index 358cbec46..74e2d5915 100755 --- a/databricks-sdk-java/src/main/java/com/databricks/sdk/service/iam/UsersImpl.java +++ b/databricks-sdk-java/src/main/java/com/databricks/sdk/service/iam/UsersImpl.java @@ -3,6 +3,8 @@ import com.databricks.sdk.core.ApiClient; import com.databricks.sdk.support.Generated; +import java.util.HashMap; +import java.util.Map; /** Package-local implementation of Users */ @Generated @@ -16,60 +18,84 @@ public UsersImpl(ApiClient apiClient) { @Override public User create(User request) { String path = "/api/2.0/preview/scim/v2/Users"; - return apiClient.POST(path, request, User.class); + Map headers = new HashMap<>(); + headers.put("Accept", "application/json"); + headers.put("Content-Type", "application/json"); + return apiClient.POST(path, request, User.class, headers); } @Override public void delete(DeleteUserRequest request) { String path = String.format("/api/2.0/preview/scim/v2/Users/%s", request.getId()); - apiClient.DELETE(path, request, Void.class); + Map headers = new HashMap<>(); + apiClient.DELETE(path, request, Void.class, headers); } @Override public User get(GetUserRequest request) { String path = String.format("/api/2.0/preview/scim/v2/Users/%s", request.getId()); - return apiClient.GET(path, request, User.class); + Map headers = new HashMap<>(); + headers.put("Accept", "application/json"); + return apiClient.GET(path, request, User.class, headers); } @Override public GetPasswordPermissionLevelsResponse getPasswordPermissionLevels() { String path = "/api/2.0/permissions/authorization/passwords/permissionLevels"; - return apiClient.GET(path, GetPasswordPermissionLevelsResponse.class); + Map headers = new HashMap<>(); + headers.put("Accept", "application/json"); + return apiClient.GET(path, GetPasswordPermissionLevelsResponse.class, headers); } @Override public PasswordPermissions getPasswordPermissions() { String path = "/api/2.0/permissions/authorization/passwords"; - return apiClient.GET(path, PasswordPermissions.class); + Map headers = new HashMap<>(); + headers.put("Accept", "application/json"); + return apiClient.GET(path, PasswordPermissions.class, headers); } @Override public ListUsersResponse list(ListUsersRequest request) { String path = "/api/2.0/preview/scim/v2/Users"; - return apiClient.GET(path, request, ListUsersResponse.class); + Map headers = new HashMap<>(); + headers.put("Accept", "application/json"); + return apiClient.GET(path, request, ListUsersResponse.class, headers); } @Override public void patch(PartialUpdate request) { String path = String.format("/api/2.0/preview/scim/v2/Users/%s", request.getId()); - apiClient.PATCH(path, request, Void.class); + Map headers = new HashMap<>(); + headers.put("Accept", "application/json"); + headers.put("Content-Type", "application/json"); + apiClient.PATCH(path, request, Void.class, headers); } @Override public PasswordPermissions setPasswordPermissions(PasswordPermissionsRequest request) { String path = "/api/2.0/permissions/authorization/passwords"; - return apiClient.PUT(path, request, PasswordPermissions.class); + Map headers = new HashMap<>(); + headers.put("Accept", "application/json"); + headers.put("Content-Type", "application/json"); + return apiClient.PUT(path, request, PasswordPermissions.class, headers); } @Override public void update(User request) { String path = String.format("/api/2.0/preview/scim/v2/Users/%s", request.getId()); - apiClient.PUT(path, request, Void.class); + Map headers = new HashMap<>(); + headers.put("Accept", "application/json"); + headers.put("Content-Type", "application/json"); + apiClient.PUT(path, request, Void.class, headers); } @Override public PasswordPermissions updatePasswordPermissions(PasswordPermissionsRequest request) { String path = "/api/2.0/permissions/authorization/passwords"; - return apiClient.PATCH(path, request, PasswordPermissions.class); + Map headers = new HashMap<>(); + headers.put("Accept", "application/json"); + headers.put("Content-Type", "application/json"); + return apiClient.PATCH(path, request, PasswordPermissions.class, headers); } } diff --git a/databricks-sdk-java/src/main/java/com/databricks/sdk/service/iam/WorkspaceAssignmentImpl.java b/databricks-sdk-java/src/main/java/com/databricks/sdk/service/iam/WorkspaceAssignmentImpl.java index e68463959..77f4c5b9a 100755 --- a/databricks-sdk-java/src/main/java/com/databricks/sdk/service/iam/WorkspaceAssignmentImpl.java +++ b/databricks-sdk-java/src/main/java/com/databricks/sdk/service/iam/WorkspaceAssignmentImpl.java @@ -3,6 +3,8 @@ import com.databricks.sdk.core.ApiClient; import com.databricks.sdk.support.Generated; +import java.util.HashMap; +import java.util.Map; /** Package-local implementation of WorkspaceAssignment */ @Generated @@ -19,7 +21,9 @@ public void delete(DeleteWorkspaceAssignmentRequest request) { String.format( "/api/2.0/accounts/%s/workspaces/%s/permissionassignments/principals/%s", apiClient.configuredAccountID(), request.getWorkspaceId(), request.getPrincipalId()); - apiClient.DELETE(path, request, Void.class); + Map headers = new HashMap<>(); + headers.put("Accept", "application/json"); + apiClient.DELETE(path, request, Void.class, headers); } @Override @@ -28,7 +32,9 @@ public WorkspacePermissions get(GetWorkspaceAssignmentRequest request) { String.format( "/api/2.0/accounts/%s/workspaces/%s/permissionassignments/permissions", apiClient.configuredAccountID(), request.getWorkspaceId()); - return apiClient.GET(path, request, WorkspacePermissions.class); + Map headers = new HashMap<>(); + headers.put("Accept", "application/json"); + return apiClient.GET(path, request, WorkspacePermissions.class, headers); } @Override @@ -37,7 +43,9 @@ public PermissionAssignments list(ListWorkspaceAssignmentRequest request) { String.format( "/api/2.0/accounts/%s/workspaces/%s/permissionassignments", apiClient.configuredAccountID(), request.getWorkspaceId()); - return apiClient.GET(path, request, PermissionAssignments.class); + Map headers = new HashMap<>(); + headers.put("Accept", "application/json"); + return apiClient.GET(path, request, PermissionAssignments.class, headers); } @Override @@ -46,6 +54,9 @@ public void update(UpdateWorkspaceAssignments request) { String.format( "/api/2.0/accounts/%s/workspaces/%s/permissionassignments/principals/%s", apiClient.configuredAccountID(), request.getWorkspaceId(), request.getPrincipalId()); - apiClient.PUT(path, request, Void.class); + Map headers = new HashMap<>(); + headers.put("Accept", "application/json"); + headers.put("Content-Type", "application/json"); + apiClient.PUT(path, request, Void.class, headers); } } diff --git a/databricks-sdk-java/src/main/java/com/databricks/sdk/service/jobs/JobsImpl.java b/databricks-sdk-java/src/main/java/com/databricks/sdk/service/jobs/JobsImpl.java index ff132cc96..b2f3fe6cf 100755 --- a/databricks-sdk-java/src/main/java/com/databricks/sdk/service/jobs/JobsImpl.java +++ b/databricks-sdk-java/src/main/java/com/databricks/sdk/service/jobs/JobsImpl.java @@ -3,6 +3,8 @@ import com.databricks.sdk.core.ApiClient; import com.databricks.sdk.support.Generated; +import java.util.HashMap; +import java.util.Map; /** Package-local implementation of Jobs */ @Generated @@ -16,43 +18,62 @@ public JobsImpl(ApiClient apiClient) { @Override public void cancelAllRuns(CancelAllRuns request) { String path = "/api/2.1/jobs/runs/cancel-all"; - apiClient.POST(path, request, Void.class); + Map headers = new HashMap<>(); + headers.put("Accept", "application/json"); + headers.put("Content-Type", "application/json"); + apiClient.POST(path, request, Void.class, headers); } @Override public void cancelRun(CancelRun request) { String path = "/api/2.1/jobs/runs/cancel"; - apiClient.POST(path, request, Void.class); + Map headers = new HashMap<>(); + headers.put("Accept", "application/json"); + headers.put("Content-Type", "application/json"); + apiClient.POST(path, request, Void.class, headers); } @Override public CreateResponse create(CreateJob request) { String path = "/api/2.1/jobs/create"; - return apiClient.POST(path, request, CreateResponse.class); + Map headers = new HashMap<>(); + headers.put("Accept", "application/json"); + headers.put("Content-Type", "application/json"); + return apiClient.POST(path, request, CreateResponse.class, headers); } @Override public void delete(DeleteJob request) { String path = "/api/2.1/jobs/delete"; - apiClient.POST(path, request, Void.class); + Map headers = new HashMap<>(); + headers.put("Accept", "application/json"); + headers.put("Content-Type", "application/json"); + apiClient.POST(path, request, Void.class, headers); } @Override public void deleteRun(DeleteRun request) { String path = "/api/2.1/jobs/runs/delete"; - apiClient.POST(path, request, Void.class); + Map headers = new HashMap<>(); + headers.put("Accept", "application/json"); + headers.put("Content-Type", "application/json"); + apiClient.POST(path, request, Void.class, headers); } @Override public ExportRunOutput exportRun(ExportRunRequest request) { String path = "/api/2.1/jobs/runs/export"; - return apiClient.GET(path, request, ExportRunOutput.class); + Map headers = new HashMap<>(); + headers.put("Accept", "application/json"); + return apiClient.GET(path, request, ExportRunOutput.class, headers); } @Override public Job get(GetJobRequest request) { String path = "/api/2.1/jobs/get"; - return apiClient.GET(path, request, Job.class); + Map headers = new HashMap<>(); + headers.put("Accept", "application/json"); + return apiClient.GET(path, request, Job.class, headers); } @Override @@ -60,78 +81,111 @@ public GetJobPermissionLevelsResponse getJobPermissionLevels( GetJobPermissionLevelsRequest request) { String path = String.format("/api/2.0/permissions/jobs/%s/permissionLevels", request.getJobId()); - return apiClient.GET(path, request, GetJobPermissionLevelsResponse.class); + Map headers = new HashMap<>(); + headers.put("Accept", "application/json"); + return apiClient.GET(path, request, GetJobPermissionLevelsResponse.class, headers); } @Override public JobPermissions getJobPermissions(GetJobPermissionsRequest request) { String path = String.format("/api/2.0/permissions/jobs/%s", request.getJobId()); - return apiClient.GET(path, request, JobPermissions.class); + Map headers = new HashMap<>(); + headers.put("Accept", "application/json"); + return apiClient.GET(path, request, JobPermissions.class, headers); } @Override public Run getRun(GetRunRequest request) { String path = "/api/2.1/jobs/runs/get"; - return apiClient.GET(path, request, Run.class); + Map headers = new HashMap<>(); + headers.put("Accept", "application/json"); + return apiClient.GET(path, request, Run.class, headers); } @Override public RunOutput getRunOutput(GetRunOutputRequest request) { String path = "/api/2.1/jobs/runs/get-output"; - return apiClient.GET(path, request, RunOutput.class); + Map headers = new HashMap<>(); + headers.put("Accept", "application/json"); + return apiClient.GET(path, request, RunOutput.class, headers); } @Override public ListJobsResponse list(ListJobsRequest request) { String path = "/api/2.1/jobs/list"; - return apiClient.GET(path, request, ListJobsResponse.class); + Map headers = new HashMap<>(); + headers.put("Accept", "application/json"); + return apiClient.GET(path, request, ListJobsResponse.class, headers); } @Override public ListRunsResponse listRuns(ListRunsRequest request) { String path = "/api/2.1/jobs/runs/list"; - return apiClient.GET(path, request, ListRunsResponse.class); + Map headers = new HashMap<>(); + headers.put("Accept", "application/json"); + return apiClient.GET(path, request, ListRunsResponse.class, headers); } @Override public RepairRunResponse repairRun(RepairRun request) { String path = "/api/2.1/jobs/runs/repair"; - return apiClient.POST(path, request, RepairRunResponse.class); + Map headers = new HashMap<>(); + headers.put("Accept", "application/json"); + headers.put("Content-Type", "application/json"); + return apiClient.POST(path, request, RepairRunResponse.class, headers); } @Override public void reset(ResetJob request) { String path = "/api/2.1/jobs/reset"; - apiClient.POST(path, request, Void.class); + Map headers = new HashMap<>(); + headers.put("Accept", "application/json"); + headers.put("Content-Type", "application/json"); + apiClient.POST(path, request, Void.class, headers); } @Override public RunNowResponse runNow(RunNow request) { String path = "/api/2.1/jobs/run-now"; - return apiClient.POST(path, request, RunNowResponse.class); + Map headers = new HashMap<>(); + headers.put("Accept", "application/json"); + headers.put("Content-Type", "application/json"); + return apiClient.POST(path, request, RunNowResponse.class, headers); } @Override public JobPermissions setJobPermissions(JobPermissionsRequest request) { String path = String.format("/api/2.0/permissions/jobs/%s", request.getJobId()); - return apiClient.PUT(path, request, JobPermissions.class); + Map headers = new HashMap<>(); + headers.put("Accept", "application/json"); + headers.put("Content-Type", "application/json"); + return apiClient.PUT(path, request, JobPermissions.class, headers); } @Override public SubmitRunResponse submit(SubmitRun request) { String path = "/api/2.1/jobs/runs/submit"; - return apiClient.POST(path, request, SubmitRunResponse.class); + Map headers = new HashMap<>(); + headers.put("Accept", "application/json"); + headers.put("Content-Type", "application/json"); + return apiClient.POST(path, request, SubmitRunResponse.class, headers); } @Override public void update(UpdateJob request) { String path = "/api/2.1/jobs/update"; - apiClient.POST(path, request, Void.class); + Map headers = new HashMap<>(); + headers.put("Accept", "application/json"); + headers.put("Content-Type", "application/json"); + apiClient.POST(path, request, Void.class, headers); } @Override public JobPermissions updateJobPermissions(JobPermissionsRequest request) { String path = String.format("/api/2.0/permissions/jobs/%s", request.getJobId()); - return apiClient.PATCH(path, request, JobPermissions.class); + Map headers = new HashMap<>(); + headers.put("Accept", "application/json"); + headers.put("Content-Type", "application/json"); + return apiClient.PATCH(path, request, JobPermissions.class, headers); } } diff --git a/databricks-sdk-java/src/main/java/com/databricks/sdk/service/ml/ExperimentsImpl.java b/databricks-sdk-java/src/main/java/com/databricks/sdk/service/ml/ExperimentsImpl.java index 1d712d0e4..d30d19482 100755 --- a/databricks-sdk-java/src/main/java/com/databricks/sdk/service/ml/ExperimentsImpl.java +++ b/databricks-sdk-java/src/main/java/com/databricks/sdk/service/ml/ExperimentsImpl.java @@ -3,6 +3,8 @@ import com.databricks.sdk.core.ApiClient; import com.databricks.sdk.support.Generated; +import java.util.HashMap; +import java.util.Map; /** Package-local implementation of Experiments */ @Generated @@ -16,49 +18,71 @@ public ExperimentsImpl(ApiClient apiClient) { @Override public CreateExperimentResponse createExperiment(CreateExperiment request) { String path = "/api/2.0/mlflow/experiments/create"; - return apiClient.POST(path, request, CreateExperimentResponse.class); + Map headers = new HashMap<>(); + headers.put("Accept", "application/json"); + headers.put("Content-Type", "application/json"); + return apiClient.POST(path, request, CreateExperimentResponse.class, headers); } @Override public CreateRunResponse createRun(CreateRun request) { String path = "/api/2.0/mlflow/runs/create"; - return apiClient.POST(path, request, CreateRunResponse.class); + Map headers = new HashMap<>(); + headers.put("Accept", "application/json"); + headers.put("Content-Type", "application/json"); + return apiClient.POST(path, request, CreateRunResponse.class, headers); } @Override public void deleteExperiment(DeleteExperiment request) { String path = "/api/2.0/mlflow/experiments/delete"; - apiClient.POST(path, request, Void.class); + Map headers = new HashMap<>(); + headers.put("Accept", "application/json"); + headers.put("Content-Type", "application/json"); + apiClient.POST(path, request, Void.class, headers); } @Override public void deleteRun(DeleteRun request) { String path = "/api/2.0/mlflow/runs/delete"; - apiClient.POST(path, request, Void.class); + Map headers = new HashMap<>(); + headers.put("Accept", "application/json"); + headers.put("Content-Type", "application/json"); + apiClient.POST(path, request, Void.class, headers); } @Override public DeleteRunsResponse deleteRuns(DeleteRuns request) { String path = "/api/2.0/mlflow/databricks/runs/delete-runs"; - return apiClient.POST(path, request, DeleteRunsResponse.class); + Map headers = new HashMap<>(); + headers.put("Accept", "application/json"); + headers.put("Content-Type", "application/json"); + return apiClient.POST(path, request, DeleteRunsResponse.class, headers); } @Override public void deleteTag(DeleteTag request) { String path = "/api/2.0/mlflow/runs/delete-tag"; - apiClient.POST(path, request, Void.class); + Map headers = new HashMap<>(); + headers.put("Accept", "application/json"); + headers.put("Content-Type", "application/json"); + apiClient.POST(path, request, Void.class, headers); } @Override public GetExperimentByNameResponse getByName(GetByNameRequest request) { String path = "/api/2.0/mlflow/experiments/get-by-name"; - return apiClient.GET(path, request, GetExperimentByNameResponse.class); + Map headers = new HashMap<>(); + headers.put("Accept", "application/json"); + return apiClient.GET(path, request, GetExperimentByNameResponse.class, headers); } @Override public Experiment getExperiment(GetExperimentRequest request) { String path = "/api/2.0/mlflow/experiments/get"; - return apiClient.GET(path, request, Experiment.class); + Map headers = new HashMap<>(); + headers.put("Accept", "application/json"); + return apiClient.GET(path, request, Experiment.class, headers); } @Override @@ -67,132 +91,192 @@ public GetExperimentPermissionLevelsResponse getExperimentPermissionLevels( String path = String.format( "/api/2.0/permissions/experiments/%s/permissionLevels", request.getExperimentId()); - return apiClient.GET(path, request, GetExperimentPermissionLevelsResponse.class); + Map headers = new HashMap<>(); + headers.put("Accept", "application/json"); + return apiClient.GET(path, request, GetExperimentPermissionLevelsResponse.class, headers); } @Override public ExperimentPermissions getExperimentPermissions(GetExperimentPermissionsRequest request) { String path = String.format("/api/2.0/permissions/experiments/%s", request.getExperimentId()); - return apiClient.GET(path, request, ExperimentPermissions.class); + Map headers = new HashMap<>(); + headers.put("Accept", "application/json"); + return apiClient.GET(path, request, ExperimentPermissions.class, headers); } @Override public GetMetricHistoryResponse getHistory(GetHistoryRequest request) { String path = "/api/2.0/mlflow/metrics/get-history"; - return apiClient.GET(path, request, GetMetricHistoryResponse.class); + Map headers = new HashMap<>(); + headers.put("Accept", "application/json"); + return apiClient.GET(path, request, GetMetricHistoryResponse.class, headers); } @Override public GetRunResponse getRun(GetRunRequest request) { String path = "/api/2.0/mlflow/runs/get"; - return apiClient.GET(path, request, GetRunResponse.class); + Map headers = new HashMap<>(); + headers.put("Accept", "application/json"); + return apiClient.GET(path, request, GetRunResponse.class, headers); } @Override public ListArtifactsResponse listArtifacts(ListArtifactsRequest request) { String path = "/api/2.0/mlflow/artifacts/list"; - return apiClient.GET(path, request, ListArtifactsResponse.class); + Map headers = new HashMap<>(); + headers.put("Accept", "application/json"); + return apiClient.GET(path, request, ListArtifactsResponse.class, headers); } @Override public ListExperimentsResponse listExperiments(ListExperimentsRequest request) { String path = "/api/2.0/mlflow/experiments/list"; - return apiClient.GET(path, request, ListExperimentsResponse.class); + Map headers = new HashMap<>(); + headers.put("Accept", "application/json"); + return apiClient.GET(path, request, ListExperimentsResponse.class, headers); } @Override public void logBatch(LogBatch request) { String path = "/api/2.0/mlflow/runs/log-batch"; - apiClient.POST(path, request, Void.class); + Map headers = new HashMap<>(); + headers.put("Accept", "application/json"); + headers.put("Content-Type", "application/json"); + apiClient.POST(path, request, Void.class, headers); } @Override public void logInputs(LogInputs request) { String path = "/api/2.0/mlflow/runs/log-inputs"; - apiClient.POST(path, request, Void.class); + Map headers = new HashMap<>(); + headers.put("Accept", "application/json"); + headers.put("Content-Type", "application/json"); + apiClient.POST(path, request, Void.class, headers); } @Override public void logMetric(LogMetric request) { String path = "/api/2.0/mlflow/runs/log-metric"; - apiClient.POST(path, request, Void.class); + Map headers = new HashMap<>(); + headers.put("Accept", "application/json"); + headers.put("Content-Type", "application/json"); + apiClient.POST(path, request, Void.class, headers); } @Override public void logModel(LogModel request) { String path = "/api/2.0/mlflow/runs/log-model"; - apiClient.POST(path, request, Void.class); + Map headers = new HashMap<>(); + headers.put("Accept", "application/json"); + headers.put("Content-Type", "application/json"); + apiClient.POST(path, request, Void.class, headers); } @Override public void logParam(LogParam request) { String path = "/api/2.0/mlflow/runs/log-parameter"; - apiClient.POST(path, request, Void.class); + Map headers = new HashMap<>(); + headers.put("Accept", "application/json"); + headers.put("Content-Type", "application/json"); + apiClient.POST(path, request, Void.class, headers); } @Override public void restoreExperiment(RestoreExperiment request) { String path = "/api/2.0/mlflow/experiments/restore"; - apiClient.POST(path, request, Void.class); + Map headers = new HashMap<>(); + headers.put("Accept", "application/json"); + headers.put("Content-Type", "application/json"); + apiClient.POST(path, request, Void.class, headers); } @Override public void restoreRun(RestoreRun request) { String path = "/api/2.0/mlflow/runs/restore"; - apiClient.POST(path, request, Void.class); + Map headers = new HashMap<>(); + headers.put("Accept", "application/json"); + headers.put("Content-Type", "application/json"); + apiClient.POST(path, request, Void.class, headers); } @Override public RestoreRunsResponse restoreRuns(RestoreRuns request) { String path = "/api/2.0/mlflow/databricks/runs/restore-runs"; - return apiClient.POST(path, request, RestoreRunsResponse.class); + Map headers = new HashMap<>(); + headers.put("Accept", "application/json"); + headers.put("Content-Type", "application/json"); + return apiClient.POST(path, request, RestoreRunsResponse.class, headers); } @Override public SearchExperimentsResponse searchExperiments(SearchExperiments request) { String path = "/api/2.0/mlflow/experiments/search"; - return apiClient.POST(path, request, SearchExperimentsResponse.class); + Map headers = new HashMap<>(); + headers.put("Accept", "application/json"); + headers.put("Content-Type", "application/json"); + return apiClient.POST(path, request, SearchExperimentsResponse.class, headers); } @Override public SearchRunsResponse searchRuns(SearchRuns request) { String path = "/api/2.0/mlflow/runs/search"; - return apiClient.POST(path, request, SearchRunsResponse.class); + Map headers = new HashMap<>(); + headers.put("Accept", "application/json"); + headers.put("Content-Type", "application/json"); + return apiClient.POST(path, request, SearchRunsResponse.class, headers); } @Override public ExperimentPermissions setExperimentPermissions(ExperimentPermissionsRequest request) { String path = String.format("/api/2.0/permissions/experiments/%s", request.getExperimentId()); - return apiClient.PUT(path, request, ExperimentPermissions.class); + Map headers = new HashMap<>(); + headers.put("Accept", "application/json"); + headers.put("Content-Type", "application/json"); + return apiClient.PUT(path, request, ExperimentPermissions.class, headers); } @Override public void setExperimentTag(SetExperimentTag request) { String path = "/api/2.0/mlflow/experiments/set-experiment-tag"; - apiClient.POST(path, request, Void.class); + Map headers = new HashMap<>(); + headers.put("Accept", "application/json"); + headers.put("Content-Type", "application/json"); + apiClient.POST(path, request, Void.class, headers); } @Override public void setTag(SetTag request) { String path = "/api/2.0/mlflow/runs/set-tag"; - apiClient.POST(path, request, Void.class); + Map headers = new HashMap<>(); + headers.put("Accept", "application/json"); + headers.put("Content-Type", "application/json"); + apiClient.POST(path, request, Void.class, headers); } @Override public void updateExperiment(UpdateExperiment request) { String path = "/api/2.0/mlflow/experiments/update"; - apiClient.POST(path, request, Void.class); + Map headers = new HashMap<>(); + headers.put("Accept", "application/json"); + headers.put("Content-Type", "application/json"); + apiClient.POST(path, request, Void.class, headers); } @Override public ExperimentPermissions updateExperimentPermissions(ExperimentPermissionsRequest request) { String path = String.format("/api/2.0/permissions/experiments/%s", request.getExperimentId()); - return apiClient.PATCH(path, request, ExperimentPermissions.class); + Map headers = new HashMap<>(); + headers.put("Accept", "application/json"); + headers.put("Content-Type", "application/json"); + return apiClient.PATCH(path, request, ExperimentPermissions.class, headers); } @Override public UpdateRunResponse updateRun(UpdateRun request) { String path = "/api/2.0/mlflow/runs/update"; - return apiClient.POST(path, request, UpdateRunResponse.class); + Map headers = new HashMap<>(); + headers.put("Accept", "application/json"); + headers.put("Content-Type", "application/json"); + return apiClient.POST(path, request, UpdateRunResponse.class, headers); } } diff --git a/databricks-sdk-java/src/main/java/com/databricks/sdk/service/ml/ModelRegistryImpl.java b/databricks-sdk-java/src/main/java/com/databricks/sdk/service/ml/ModelRegistryImpl.java index 868d74e7e..15d0b8160 100755 --- a/databricks-sdk-java/src/main/java/com/databricks/sdk/service/ml/ModelRegistryImpl.java +++ b/databricks-sdk-java/src/main/java/com/databricks/sdk/service/ml/ModelRegistryImpl.java @@ -3,6 +3,8 @@ import com.databricks.sdk.core.ApiClient; import com.databricks.sdk.support.Generated; +import java.util.HashMap; +import java.util.Map; /** Package-local implementation of ModelRegistry */ @Generated @@ -17,104 +19,145 @@ public ModelRegistryImpl(ApiClient apiClient) { public ApproveTransitionRequestResponse approveTransitionRequest( ApproveTransitionRequest request) { String path = "/api/2.0/mlflow/transition-requests/approve"; - return apiClient.POST(path, request, ApproveTransitionRequestResponse.class); + Map headers = new HashMap<>(); + headers.put("Accept", "application/json"); + headers.put("Content-Type", "application/json"); + return apiClient.POST(path, request, ApproveTransitionRequestResponse.class, headers); } @Override public CreateCommentResponse createComment(CreateComment request) { String path = "/api/2.0/mlflow/comments/create"; - return apiClient.POST(path, request, CreateCommentResponse.class); + Map headers = new HashMap<>(); + headers.put("Accept", "application/json"); + headers.put("Content-Type", "application/json"); + return apiClient.POST(path, request, CreateCommentResponse.class, headers); } @Override public CreateModelResponse createModel(CreateModelRequest request) { String path = "/api/2.0/mlflow/registered-models/create"; - return apiClient.POST(path, request, CreateModelResponse.class); + Map headers = new HashMap<>(); + headers.put("Accept", "application/json"); + headers.put("Content-Type", "application/json"); + return apiClient.POST(path, request, CreateModelResponse.class, headers); } @Override public CreateModelVersionResponse createModelVersion(CreateModelVersionRequest request) { String path = "/api/2.0/mlflow/model-versions/create"; - return apiClient.POST(path, request, CreateModelVersionResponse.class); + Map headers = new HashMap<>(); + headers.put("Accept", "application/json"); + headers.put("Content-Type", "application/json"); + return apiClient.POST(path, request, CreateModelVersionResponse.class, headers); } @Override public CreateTransitionRequestResponse createTransitionRequest(CreateTransitionRequest request) { String path = "/api/2.0/mlflow/transition-requests/create"; - return apiClient.POST(path, request, CreateTransitionRequestResponse.class); + Map headers = new HashMap<>(); + headers.put("Accept", "application/json"); + headers.put("Content-Type", "application/json"); + return apiClient.POST(path, request, CreateTransitionRequestResponse.class, headers); } @Override public CreateWebhookResponse createWebhook(CreateRegistryWebhook request) { String path = "/api/2.0/mlflow/registry-webhooks/create"; - return apiClient.POST(path, request, CreateWebhookResponse.class); + Map headers = new HashMap<>(); + headers.put("Accept", "application/json"); + headers.put("Content-Type", "application/json"); + return apiClient.POST(path, request, CreateWebhookResponse.class, headers); } @Override public void deleteComment(DeleteCommentRequest request) { String path = "/api/2.0/mlflow/comments/delete"; - apiClient.DELETE(path, request, Void.class); + Map headers = new HashMap<>(); + headers.put("Accept", "application/json"); + apiClient.DELETE(path, request, Void.class, headers); } @Override public void deleteModel(DeleteModelRequest request) { String path = "/api/2.0/mlflow/registered-models/delete"; - apiClient.DELETE(path, request, Void.class); + Map headers = new HashMap<>(); + headers.put("Accept", "application/json"); + apiClient.DELETE(path, request, Void.class, headers); } @Override public void deleteModelTag(DeleteModelTagRequest request) { String path = "/api/2.0/mlflow/registered-models/delete-tag"; - apiClient.DELETE(path, request, Void.class); + Map headers = new HashMap<>(); + headers.put("Accept", "application/json"); + apiClient.DELETE(path, request, Void.class, headers); } @Override public void deleteModelVersion(DeleteModelVersionRequest request) { String path = "/api/2.0/mlflow/model-versions/delete"; - apiClient.DELETE(path, request, Void.class); + Map headers = new HashMap<>(); + headers.put("Accept", "application/json"); + apiClient.DELETE(path, request, Void.class, headers); } @Override public void deleteModelVersionTag(DeleteModelVersionTagRequest request) { String path = "/api/2.0/mlflow/model-versions/delete-tag"; - apiClient.DELETE(path, request, Void.class); + Map headers = new HashMap<>(); + headers.put("Accept", "application/json"); + apiClient.DELETE(path, request, Void.class, headers); } @Override public void deleteTransitionRequest(DeleteTransitionRequestRequest request) { String path = "/api/2.0/mlflow/transition-requests/delete"; - apiClient.DELETE(path, request, Void.class); + Map headers = new HashMap<>(); + headers.put("Accept", "application/json"); + apiClient.DELETE(path, request, Void.class, headers); } @Override public void deleteWebhook(DeleteWebhookRequest request) { String path = "/api/2.0/mlflow/registry-webhooks/delete"; - apiClient.DELETE(path, request, Void.class); + Map headers = new HashMap<>(); + headers.put("Accept", "application/json"); + apiClient.DELETE(path, request, Void.class, headers); } @Override public GetLatestVersionsResponse getLatestVersions(GetLatestVersionsRequest request) { String path = "/api/2.0/mlflow/registered-models/get-latest-versions"; - return apiClient.POST(path, request, GetLatestVersionsResponse.class); + Map headers = new HashMap<>(); + headers.put("Accept", "application/json"); + headers.put("Content-Type", "application/json"); + return apiClient.POST(path, request, GetLatestVersionsResponse.class, headers); } @Override public GetModelResponse getModel(GetModelRequest request) { String path = "/api/2.0/mlflow/databricks/registered-models/get"; - return apiClient.GET(path, request, GetModelResponse.class); + Map headers = new HashMap<>(); + headers.put("Accept", "application/json"); + return apiClient.GET(path, request, GetModelResponse.class, headers); } @Override public GetModelVersionResponse getModelVersion(GetModelVersionRequest request) { String path = "/api/2.0/mlflow/model-versions/get"; - return apiClient.GET(path, request, GetModelVersionResponse.class); + Map headers = new HashMap<>(); + headers.put("Accept", "application/json"); + return apiClient.GET(path, request, GetModelVersionResponse.class, headers); } @Override public GetModelVersionDownloadUriResponse getModelVersionDownloadUri( GetModelVersionDownloadUriRequest request) { String path = "/api/2.0/mlflow/model-versions/get-download-uri"; - return apiClient.GET(path, request, GetModelVersionDownloadUriResponse.class); + Map headers = new HashMap<>(); + headers.put("Accept", "application/json"); + return apiClient.GET(path, request, GetModelVersionDownloadUriResponse.class, headers); } @Override @@ -124,7 +167,9 @@ public GetRegisteredModelPermissionLevelsResponse getRegisteredModelPermissionLe String.format( "/api/2.0/permissions/registered-models/%s/permissionLevels", request.getRegisteredModelId()); - return apiClient.GET(path, request, GetRegisteredModelPermissionLevelsResponse.class); + Map headers = new HashMap<>(); + headers.put("Accept", "application/json"); + return apiClient.GET(path, request, GetRegisteredModelPermissionLevelsResponse.class, headers); } @Override @@ -132,62 +177,86 @@ public RegisteredModelPermissions getRegisteredModelPermissions( GetRegisteredModelPermissionsRequest request) { String path = String.format("/api/2.0/permissions/registered-models/%s", request.getRegisteredModelId()); - return apiClient.GET(path, request, RegisteredModelPermissions.class); + Map headers = new HashMap<>(); + headers.put("Accept", "application/json"); + return apiClient.GET(path, request, RegisteredModelPermissions.class, headers); } @Override public ListModelsResponse listModels(ListModelsRequest request) { String path = "/api/2.0/mlflow/registered-models/list"; - return apiClient.GET(path, request, ListModelsResponse.class); + Map headers = new HashMap<>(); + headers.put("Accept", "application/json"); + return apiClient.GET(path, request, ListModelsResponse.class, headers); } @Override public ListTransitionRequestsResponse listTransitionRequests( ListTransitionRequestsRequest request) { String path = "/api/2.0/mlflow/transition-requests/list"; - return apiClient.GET(path, request, ListTransitionRequestsResponse.class); + Map headers = new HashMap<>(); + headers.put("Accept", "application/json"); + return apiClient.GET(path, request, ListTransitionRequestsResponse.class, headers); } @Override public ListRegistryWebhooks listWebhooks(ListWebhooksRequest request) { String path = "/api/2.0/mlflow/registry-webhooks/list"; - return apiClient.GET(path, request, ListRegistryWebhooks.class); + Map headers = new HashMap<>(); + headers.put("Accept", "application/json"); + return apiClient.GET(path, request, ListRegistryWebhooks.class, headers); } @Override public RejectTransitionRequestResponse rejectTransitionRequest(RejectTransitionRequest request) { String path = "/api/2.0/mlflow/transition-requests/reject"; - return apiClient.POST(path, request, RejectTransitionRequestResponse.class); + Map headers = new HashMap<>(); + headers.put("Accept", "application/json"); + headers.put("Content-Type", "application/json"); + return apiClient.POST(path, request, RejectTransitionRequestResponse.class, headers); } @Override public RenameModelResponse renameModel(RenameModelRequest request) { String path = "/api/2.0/mlflow/registered-models/rename"; - return apiClient.POST(path, request, RenameModelResponse.class); + Map headers = new HashMap<>(); + headers.put("Accept", "application/json"); + headers.put("Content-Type", "application/json"); + return apiClient.POST(path, request, RenameModelResponse.class, headers); } @Override public SearchModelVersionsResponse searchModelVersions(SearchModelVersionsRequest request) { String path = "/api/2.0/mlflow/model-versions/search"; - return apiClient.GET(path, request, SearchModelVersionsResponse.class); + Map headers = new HashMap<>(); + headers.put("Accept", "application/json"); + return apiClient.GET(path, request, SearchModelVersionsResponse.class, headers); } @Override public SearchModelsResponse searchModels(SearchModelsRequest request) { String path = "/api/2.0/mlflow/registered-models/search"; - return apiClient.GET(path, request, SearchModelsResponse.class); + Map headers = new HashMap<>(); + headers.put("Accept", "application/json"); + return apiClient.GET(path, request, SearchModelsResponse.class, headers); } @Override public void setModelTag(SetModelTagRequest request) { String path = "/api/2.0/mlflow/registered-models/set-tag"; - apiClient.POST(path, request, Void.class); + Map headers = new HashMap<>(); + headers.put("Accept", "application/json"); + headers.put("Content-Type", "application/json"); + apiClient.POST(path, request, Void.class, headers); } @Override public void setModelVersionTag(SetModelVersionTagRequest request) { String path = "/api/2.0/mlflow/model-versions/set-tag"; - apiClient.POST(path, request, Void.class); + Map headers = new HashMap<>(); + headers.put("Accept", "application/json"); + headers.put("Content-Type", "application/json"); + apiClient.POST(path, request, Void.class, headers); } @Override @@ -195,37 +264,55 @@ public RegisteredModelPermissions setRegisteredModelPermissions( RegisteredModelPermissionsRequest request) { String path = String.format("/api/2.0/permissions/registered-models/%s", request.getRegisteredModelId()); - return apiClient.PUT(path, request, RegisteredModelPermissions.class); + Map headers = new HashMap<>(); + headers.put("Accept", "application/json"); + headers.put("Content-Type", "application/json"); + return apiClient.PUT(path, request, RegisteredModelPermissions.class, headers); } @Override public TestRegistryWebhookResponse testRegistryWebhook(TestRegistryWebhookRequest request) { String path = "/api/2.0/mlflow/registry-webhooks/test"; - return apiClient.POST(path, request, TestRegistryWebhookResponse.class); + Map headers = new HashMap<>(); + headers.put("Accept", "application/json"); + headers.put("Content-Type", "application/json"); + return apiClient.POST(path, request, TestRegistryWebhookResponse.class, headers); } @Override public TransitionStageResponse transitionStage(TransitionModelVersionStageDatabricks request) { String path = "/api/2.0/mlflow/databricks/model-versions/transition-stage"; - return apiClient.POST(path, request, TransitionStageResponse.class); + Map headers = new HashMap<>(); + headers.put("Accept", "application/json"); + headers.put("Content-Type", "application/json"); + return apiClient.POST(path, request, TransitionStageResponse.class, headers); } @Override public UpdateCommentResponse updateComment(UpdateComment request) { String path = "/api/2.0/mlflow/comments/update"; - return apiClient.PATCH(path, request, UpdateCommentResponse.class); + Map headers = new HashMap<>(); + headers.put("Accept", "application/json"); + headers.put("Content-Type", "application/json"); + return apiClient.PATCH(path, request, UpdateCommentResponse.class, headers); } @Override public void updateModel(UpdateModelRequest request) { String path = "/api/2.0/mlflow/registered-models/update"; - apiClient.PATCH(path, request, Void.class); + Map headers = new HashMap<>(); + headers.put("Accept", "application/json"); + headers.put("Content-Type", "application/json"); + apiClient.PATCH(path, request, Void.class, headers); } @Override public void updateModelVersion(UpdateModelVersionRequest request) { String path = "/api/2.0/mlflow/model-versions/update"; - apiClient.PATCH(path, request, Void.class); + Map headers = new HashMap<>(); + headers.put("Accept", "application/json"); + headers.put("Content-Type", "application/json"); + apiClient.PATCH(path, request, Void.class, headers); } @Override @@ -233,12 +320,18 @@ public RegisteredModelPermissions updateRegisteredModelPermissions( RegisteredModelPermissionsRequest request) { String path = String.format("/api/2.0/permissions/registered-models/%s", request.getRegisteredModelId()); - return apiClient.PATCH(path, request, RegisteredModelPermissions.class); + Map headers = new HashMap<>(); + headers.put("Accept", "application/json"); + headers.put("Content-Type", "application/json"); + return apiClient.PATCH(path, request, RegisteredModelPermissions.class, headers); } @Override public void updateWebhook(UpdateRegistryWebhook request) { String path = "/api/2.0/mlflow/registry-webhooks/update"; - apiClient.PATCH(path, request, Void.class); + Map headers = new HashMap<>(); + headers.put("Accept", "application/json"); + headers.put("Content-Type", "application/json"); + apiClient.PATCH(path, request, Void.class, headers); } } diff --git a/databricks-sdk-java/src/main/java/com/databricks/sdk/service/oauth2/CustomAppIntegrationImpl.java b/databricks-sdk-java/src/main/java/com/databricks/sdk/service/oauth2/CustomAppIntegrationImpl.java index d15e5f85f..6f662e741 100755 --- a/databricks-sdk-java/src/main/java/com/databricks/sdk/service/oauth2/CustomAppIntegrationImpl.java +++ b/databricks-sdk-java/src/main/java/com/databricks/sdk/service/oauth2/CustomAppIntegrationImpl.java @@ -3,6 +3,8 @@ import com.databricks.sdk.core.ApiClient; import com.databricks.sdk.support.Generated; +import java.util.HashMap; +import java.util.Map; /** Package-local implementation of CustomAppIntegration */ @Generated @@ -18,7 +20,10 @@ public CreateCustomAppIntegrationOutput create(CreateCustomAppIntegration reques String path = String.format( "/api/2.0/accounts/%s/oauth2/custom-app-integrations", apiClient.configuredAccountID()); - return apiClient.POST(path, request, CreateCustomAppIntegrationOutput.class); + Map headers = new HashMap<>(); + headers.put("Accept", "application/json"); + headers.put("Content-Type", "application/json"); + return apiClient.POST(path, request, CreateCustomAppIntegrationOutput.class, headers); } @Override @@ -27,7 +32,9 @@ public void delete(DeleteCustomAppIntegrationRequest request) { String.format( "/api/2.0/accounts/%s/oauth2/custom-app-integrations/%s", apiClient.configuredAccountID(), request.getIntegrationId()); - apiClient.DELETE(path, request, Void.class); + Map headers = new HashMap<>(); + headers.put("Accept", "application/json"); + apiClient.DELETE(path, request, Void.class, headers); } @Override @@ -36,7 +43,9 @@ public GetCustomAppIntegrationOutput get(GetCustomAppIntegrationRequest request) String.format( "/api/2.0/accounts/%s/oauth2/custom-app-integrations/%s", apiClient.configuredAccountID(), request.getIntegrationId()); - return apiClient.GET(path, request, GetCustomAppIntegrationOutput.class); + Map headers = new HashMap<>(); + headers.put("Accept", "application/json"); + return apiClient.GET(path, request, GetCustomAppIntegrationOutput.class, headers); } @Override @@ -44,7 +53,9 @@ public GetCustomAppIntegrationsOutput list() { String path = String.format( "/api/2.0/accounts/%s/oauth2/custom-app-integrations", apiClient.configuredAccountID()); - return apiClient.GET(path, GetCustomAppIntegrationsOutput.class); + Map headers = new HashMap<>(); + headers.put("Accept", "application/json"); + return apiClient.GET(path, GetCustomAppIntegrationsOutput.class, headers); } @Override @@ -53,6 +64,9 @@ public void update(UpdateCustomAppIntegration request) { String.format( "/api/2.0/accounts/%s/oauth2/custom-app-integrations/%s", apiClient.configuredAccountID(), request.getIntegrationId()); - apiClient.PATCH(path, request, Void.class); + Map headers = new HashMap<>(); + headers.put("Accept", "application/json"); + headers.put("Content-Type", "application/json"); + apiClient.PATCH(path, request, Void.class, headers); } } diff --git a/databricks-sdk-java/src/main/java/com/databricks/sdk/service/oauth2/OAuthEnrollmentImpl.java b/databricks-sdk-java/src/main/java/com/databricks/sdk/service/oauth2/OAuthEnrollmentImpl.java index 2b8e93789..66a705da9 100755 --- a/databricks-sdk-java/src/main/java/com/databricks/sdk/service/oauth2/OAuthEnrollmentImpl.java +++ b/databricks-sdk-java/src/main/java/com/databricks/sdk/service/oauth2/OAuthEnrollmentImpl.java @@ -3,6 +3,8 @@ import com.databricks.sdk.core.ApiClient; import com.databricks.sdk.support.Generated; +import java.util.HashMap; +import java.util.Map; /** Package-local implementation of OAuthEnrollment */ @Generated @@ -17,13 +19,17 @@ public OAuthEnrollmentImpl(ApiClient apiClient) { public void create(CreateOAuthEnrollment request) { String path = String.format("/api/2.0/accounts/%s/oauth2/enrollment", apiClient.configuredAccountID()); - apiClient.POST(path, request, Void.class); + Map headers = new HashMap<>(); + headers.put("Content-Type", "application/json"); + apiClient.POST(path, request, Void.class, headers); } @Override public OAuthEnrollmentStatus get() { String path = String.format("/api/2.0/accounts/%s/oauth2/enrollment", apiClient.configuredAccountID()); - return apiClient.GET(path, OAuthEnrollmentStatus.class); + Map headers = new HashMap<>(); + headers.put("Accept", "application/json"); + return apiClient.GET(path, OAuthEnrollmentStatus.class, headers); } } diff --git a/databricks-sdk-java/src/main/java/com/databricks/sdk/service/oauth2/PublishedAppIntegrationImpl.java b/databricks-sdk-java/src/main/java/com/databricks/sdk/service/oauth2/PublishedAppIntegrationImpl.java index c86390344..583d6f4c6 100755 --- a/databricks-sdk-java/src/main/java/com/databricks/sdk/service/oauth2/PublishedAppIntegrationImpl.java +++ b/databricks-sdk-java/src/main/java/com/databricks/sdk/service/oauth2/PublishedAppIntegrationImpl.java @@ -3,6 +3,8 @@ import com.databricks.sdk.core.ApiClient; import com.databricks.sdk.support.Generated; +import java.util.HashMap; +import java.util.Map; /** Package-local implementation of PublishedAppIntegration */ @Generated @@ -19,7 +21,10 @@ public CreatePublishedAppIntegrationOutput create(CreatePublishedAppIntegration String.format( "/api/2.0/accounts/%s/oauth2/published-app-integrations", apiClient.configuredAccountID()); - return apiClient.POST(path, request, CreatePublishedAppIntegrationOutput.class); + Map headers = new HashMap<>(); + headers.put("Accept", "application/json"); + headers.put("Content-Type", "application/json"); + return apiClient.POST(path, request, CreatePublishedAppIntegrationOutput.class, headers); } @Override @@ -28,7 +33,9 @@ public void delete(DeletePublishedAppIntegrationRequest request) { String.format( "/api/2.0/accounts/%s/oauth2/published-app-integrations/%s", apiClient.configuredAccountID(), request.getIntegrationId()); - apiClient.DELETE(path, request, Void.class); + Map headers = new HashMap<>(); + headers.put("Accept", "application/json"); + apiClient.DELETE(path, request, Void.class, headers); } @Override @@ -37,7 +44,9 @@ public GetPublishedAppIntegrationOutput get(GetPublishedAppIntegrationRequest re String.format( "/api/2.0/accounts/%s/oauth2/published-app-integrations/%s", apiClient.configuredAccountID(), request.getIntegrationId()); - return apiClient.GET(path, request, GetPublishedAppIntegrationOutput.class); + Map headers = new HashMap<>(); + headers.put("Accept", "application/json"); + return apiClient.GET(path, request, GetPublishedAppIntegrationOutput.class, headers); } @Override @@ -46,7 +55,9 @@ public GetPublishedAppIntegrationsOutput list() { String.format( "/api/2.0/accounts/%s/oauth2/published-app-integrations", apiClient.configuredAccountID()); - return apiClient.GET(path, GetPublishedAppIntegrationsOutput.class); + Map headers = new HashMap<>(); + headers.put("Accept", "application/json"); + return apiClient.GET(path, GetPublishedAppIntegrationsOutput.class, headers); } @Override @@ -55,6 +66,9 @@ public void update(UpdatePublishedAppIntegration request) { String.format( "/api/2.0/accounts/%s/oauth2/published-app-integrations/%s", apiClient.configuredAccountID(), request.getIntegrationId()); - apiClient.PATCH(path, request, Void.class); + Map headers = new HashMap<>(); + headers.put("Accept", "application/json"); + headers.put("Content-Type", "application/json"); + apiClient.PATCH(path, request, Void.class, headers); } } diff --git a/databricks-sdk-java/src/main/java/com/databricks/sdk/service/oauth2/ServicePrincipalSecretsImpl.java b/databricks-sdk-java/src/main/java/com/databricks/sdk/service/oauth2/ServicePrincipalSecretsImpl.java index cc44a778d..57f8f9bb1 100755 --- a/databricks-sdk-java/src/main/java/com/databricks/sdk/service/oauth2/ServicePrincipalSecretsImpl.java +++ b/databricks-sdk-java/src/main/java/com/databricks/sdk/service/oauth2/ServicePrincipalSecretsImpl.java @@ -3,6 +3,8 @@ import com.databricks.sdk.core.ApiClient; import com.databricks.sdk.support.Generated; +import java.util.HashMap; +import java.util.Map; /** Package-local implementation of ServicePrincipalSecrets */ @Generated @@ -19,7 +21,9 @@ public CreateServicePrincipalSecretResponse create(CreateServicePrincipalSecretR String.format( "/api/2.0/accounts/%s/servicePrincipals/%s/credentials/secrets", apiClient.configuredAccountID(), request.getServicePrincipalId()); - return apiClient.POST(path, request, CreateServicePrincipalSecretResponse.class); + Map headers = new HashMap<>(); + headers.put("Accept", "application/json"); + return apiClient.POST(path, request, CreateServicePrincipalSecretResponse.class, headers); } @Override @@ -30,7 +34,8 @@ public void delete(DeleteServicePrincipalSecretRequest request) { apiClient.configuredAccountID(), request.getServicePrincipalId(), request.getSecretId()); - apiClient.DELETE(path, request, Void.class); + Map headers = new HashMap<>(); + apiClient.DELETE(path, request, Void.class, headers); } @Override @@ -39,6 +44,8 @@ public ListServicePrincipalSecretsResponse list(ListServicePrincipalSecretsReque String.format( "/api/2.0/accounts/%s/servicePrincipals/%s/credentials/secrets", apiClient.configuredAccountID(), request.getServicePrincipalId()); - return apiClient.GET(path, request, ListServicePrincipalSecretsResponse.class); + Map headers = new HashMap<>(); + headers.put("Accept", "application/json"); + return apiClient.GET(path, request, ListServicePrincipalSecretsResponse.class, headers); } } diff --git a/databricks-sdk-java/src/main/java/com/databricks/sdk/service/pipelines/PipelinesImpl.java b/databricks-sdk-java/src/main/java/com/databricks/sdk/service/pipelines/PipelinesImpl.java index 27cbb9a83..b60b63244 100755 --- a/databricks-sdk-java/src/main/java/com/databricks/sdk/service/pipelines/PipelinesImpl.java +++ b/databricks-sdk-java/src/main/java/com/databricks/sdk/service/pipelines/PipelinesImpl.java @@ -3,6 +3,8 @@ import com.databricks.sdk.core.ApiClient; import com.databricks.sdk.support.Generated; +import java.util.HashMap; +import java.util.Map; /** Package-local implementation of Pipelines */ @Generated @@ -16,19 +18,26 @@ public PipelinesImpl(ApiClient apiClient) { @Override public CreatePipelineResponse create(CreatePipeline request) { String path = "/api/2.0/pipelines"; - return apiClient.POST(path, request, CreatePipelineResponse.class); + Map headers = new HashMap<>(); + headers.put("Accept", "application/json"); + headers.put("Content-Type", "application/json"); + return apiClient.POST(path, request, CreatePipelineResponse.class, headers); } @Override public void delete(DeletePipelineRequest request) { String path = String.format("/api/2.0/pipelines/%s", request.getPipelineId()); - apiClient.DELETE(path, request, Void.class); + Map headers = new HashMap<>(); + headers.put("Accept", "application/json"); + apiClient.DELETE(path, request, Void.class, headers); } @Override public GetPipelineResponse get(GetPipelineRequest request) { String path = String.format("/api/2.0/pipelines/%s", request.getPipelineId()); - return apiClient.GET(path, request, GetPipelineResponse.class); + Map headers = new HashMap<>(); + headers.put("Accept", "application/json"); + return apiClient.GET(path, request, GetPipelineResponse.class, headers); } @Override @@ -37,13 +46,17 @@ public GetPipelinePermissionLevelsResponse getPipelinePermissionLevels( String path = String.format( "/api/2.0/permissions/pipelines/%s/permissionLevels", request.getPipelineId()); - return apiClient.GET(path, request, GetPipelinePermissionLevelsResponse.class); + Map headers = new HashMap<>(); + headers.put("Accept", "application/json"); + return apiClient.GET(path, request, GetPipelinePermissionLevelsResponse.class, headers); } @Override public PipelinePermissions getPipelinePermissions(GetPipelinePermissionsRequest request) { String path = String.format("/api/2.0/permissions/pipelines/%s", request.getPipelineId()); - return apiClient.GET(path, request, PipelinePermissions.class); + Map headers = new HashMap<>(); + headers.put("Accept", "application/json"); + return apiClient.GET(path, request, PipelinePermissions.class, headers); } @Override @@ -51,60 +64,84 @@ public GetUpdateResponse getUpdate(GetUpdateRequest request) { String path = String.format( "/api/2.0/pipelines/%s/updates/%s", request.getPipelineId(), request.getUpdateId()); - return apiClient.GET(path, request, GetUpdateResponse.class); + Map headers = new HashMap<>(); + headers.put("Accept", "application/json"); + return apiClient.GET(path, request, GetUpdateResponse.class, headers); } @Override public ListPipelineEventsResponse listPipelineEvents(ListPipelineEventsRequest request) { String path = String.format("/api/2.0/pipelines/%s/events", request.getPipelineId()); - return apiClient.GET(path, request, ListPipelineEventsResponse.class); + Map headers = new HashMap<>(); + headers.put("Accept", "application/json"); + return apiClient.GET(path, request, ListPipelineEventsResponse.class, headers); } @Override public ListPipelinesResponse listPipelines(ListPipelinesRequest request) { String path = "/api/2.0/pipelines"; - return apiClient.GET(path, request, ListPipelinesResponse.class); + Map headers = new HashMap<>(); + headers.put("Accept", "application/json"); + return apiClient.GET(path, request, ListPipelinesResponse.class, headers); } @Override public ListUpdatesResponse listUpdates(ListUpdatesRequest request) { String path = String.format("/api/2.0/pipelines/%s/updates", request.getPipelineId()); - return apiClient.GET(path, request, ListUpdatesResponse.class); + Map headers = new HashMap<>(); + headers.put("Accept", "application/json"); + return apiClient.GET(path, request, ListUpdatesResponse.class, headers); } @Override public void reset(ResetRequest request) { String path = String.format("/api/2.0/pipelines/%s/reset", request.getPipelineId()); - apiClient.POST(path, request, Void.class); + Map headers = new HashMap<>(); + headers.put("Accept", "application/json"); + apiClient.POST(path, request, Void.class, headers); } @Override public PipelinePermissions setPipelinePermissions(PipelinePermissionsRequest request) { String path = String.format("/api/2.0/permissions/pipelines/%s", request.getPipelineId()); - return apiClient.PUT(path, request, PipelinePermissions.class); + Map headers = new HashMap<>(); + headers.put("Accept", "application/json"); + headers.put("Content-Type", "application/json"); + return apiClient.PUT(path, request, PipelinePermissions.class, headers); } @Override public StartUpdateResponse startUpdate(StartUpdate request) { String path = String.format("/api/2.0/pipelines/%s/updates", request.getPipelineId()); - return apiClient.POST(path, request, StartUpdateResponse.class); + Map headers = new HashMap<>(); + headers.put("Accept", "application/json"); + headers.put("Content-Type", "application/json"); + return apiClient.POST(path, request, StartUpdateResponse.class, headers); } @Override public void stop(StopRequest request) { String path = String.format("/api/2.0/pipelines/%s/stop", request.getPipelineId()); - apiClient.POST(path, request, Void.class); + Map headers = new HashMap<>(); + headers.put("Accept", "application/json"); + apiClient.POST(path, request, Void.class, headers); } @Override public void update(EditPipeline request) { String path = String.format("/api/2.0/pipelines/%s", request.getPipelineId()); - apiClient.PUT(path, request, Void.class); + Map headers = new HashMap<>(); + headers.put("Accept", "application/json"); + headers.put("Content-Type", "application/json"); + apiClient.PUT(path, request, Void.class, headers); } @Override public PipelinePermissions updatePipelinePermissions(PipelinePermissionsRequest request) { String path = String.format("/api/2.0/permissions/pipelines/%s", request.getPipelineId()); - return apiClient.PATCH(path, request, PipelinePermissions.class); + Map headers = new HashMap<>(); + headers.put("Accept", "application/json"); + headers.put("Content-Type", "application/json"); + return apiClient.PATCH(path, request, PipelinePermissions.class, headers); } } diff --git a/databricks-sdk-java/src/main/java/com/databricks/sdk/service/provisioning/CredentialsImpl.java b/databricks-sdk-java/src/main/java/com/databricks/sdk/service/provisioning/CredentialsImpl.java index e6c7ee826..769f1755a 100755 --- a/databricks-sdk-java/src/main/java/com/databricks/sdk/service/provisioning/CredentialsImpl.java +++ b/databricks-sdk-java/src/main/java/com/databricks/sdk/service/provisioning/CredentialsImpl.java @@ -4,6 +4,8 @@ import com.databricks.sdk.core.ApiClient; import com.databricks.sdk.support.Generated; import java.util.Collection; +import java.util.HashMap; +import java.util.Map; /** Package-local implementation of Credentials */ @Generated @@ -18,7 +20,10 @@ public CredentialsImpl(ApiClient apiClient) { public Credential create(CreateCredentialRequest request) { String path = String.format("/api/2.0/accounts/%s/credentials", apiClient.configuredAccountID()); - return apiClient.POST(path, request, Credential.class); + Map headers = new HashMap<>(); + headers.put("Accept", "application/json"); + headers.put("Content-Type", "application/json"); + return apiClient.POST(path, request, Credential.class, headers); } @Override @@ -27,7 +32,9 @@ public void delete(DeleteCredentialRequest request) { String.format( "/api/2.0/accounts/%s/credentials/%s", apiClient.configuredAccountID(), request.getCredentialsId()); - apiClient.DELETE(path, request, Void.class); + Map headers = new HashMap<>(); + headers.put("Accept", "application/json"); + apiClient.DELETE(path, request, Void.class, headers); } @Override @@ -36,13 +43,17 @@ public Credential get(GetCredentialRequest request) { String.format( "/api/2.0/accounts/%s/credentials/%s", apiClient.configuredAccountID(), request.getCredentialsId()); - return apiClient.GET(path, request, Credential.class); + Map headers = new HashMap<>(); + headers.put("Accept", "application/json"); + return apiClient.GET(path, request, Credential.class, headers); } @Override public Collection list() { String path = String.format("/api/2.0/accounts/%s/credentials", apiClient.configuredAccountID()); - return apiClient.getCollection(path, null, Credential.class); + Map headers = new HashMap<>(); + headers.put("Accept", "application/json"); + return apiClient.getCollection(path, null, Credential.class, headers); } } diff --git a/databricks-sdk-java/src/main/java/com/databricks/sdk/service/provisioning/EncryptionKeysImpl.java b/databricks-sdk-java/src/main/java/com/databricks/sdk/service/provisioning/EncryptionKeysImpl.java index 10d82e220..ef8b82db1 100755 --- a/databricks-sdk-java/src/main/java/com/databricks/sdk/service/provisioning/EncryptionKeysImpl.java +++ b/databricks-sdk-java/src/main/java/com/databricks/sdk/service/provisioning/EncryptionKeysImpl.java @@ -4,6 +4,8 @@ import com.databricks.sdk.core.ApiClient; import com.databricks.sdk.support.Generated; import java.util.Collection; +import java.util.HashMap; +import java.util.Map; /** Package-local implementation of EncryptionKeys */ @Generated @@ -19,7 +21,10 @@ public CustomerManagedKey create(CreateCustomerManagedKeyRequest request) { String path = String.format( "/api/2.0/accounts/%s/customer-managed-keys", apiClient.configuredAccountID()); - return apiClient.POST(path, request, CustomerManagedKey.class); + Map headers = new HashMap<>(); + headers.put("Accept", "application/json"); + headers.put("Content-Type", "application/json"); + return apiClient.POST(path, request, CustomerManagedKey.class, headers); } @Override @@ -28,7 +33,9 @@ public void delete(DeleteEncryptionKeyRequest request) { String.format( "/api/2.0/accounts/%s/customer-managed-keys/%s", apiClient.configuredAccountID(), request.getCustomerManagedKeyId()); - apiClient.DELETE(path, request, Void.class); + Map headers = new HashMap<>(); + headers.put("Accept", "application/json"); + apiClient.DELETE(path, request, Void.class, headers); } @Override @@ -37,7 +44,9 @@ public CustomerManagedKey get(GetEncryptionKeyRequest request) { String.format( "/api/2.0/accounts/%s/customer-managed-keys/%s", apiClient.configuredAccountID(), request.getCustomerManagedKeyId()); - return apiClient.GET(path, request, CustomerManagedKey.class); + Map headers = new HashMap<>(); + headers.put("Accept", "application/json"); + return apiClient.GET(path, request, CustomerManagedKey.class, headers); } @Override @@ -45,6 +54,8 @@ public Collection list() { String path = String.format( "/api/2.0/accounts/%s/customer-managed-keys", apiClient.configuredAccountID()); - return apiClient.getCollection(path, null, CustomerManagedKey.class); + Map headers = new HashMap<>(); + headers.put("Accept", "application/json"); + return apiClient.getCollection(path, null, CustomerManagedKey.class, headers); } } diff --git a/databricks-sdk-java/src/main/java/com/databricks/sdk/service/provisioning/NetworksImpl.java b/databricks-sdk-java/src/main/java/com/databricks/sdk/service/provisioning/NetworksImpl.java index a45f53bb7..023e9060c 100755 --- a/databricks-sdk-java/src/main/java/com/databricks/sdk/service/provisioning/NetworksImpl.java +++ b/databricks-sdk-java/src/main/java/com/databricks/sdk/service/provisioning/NetworksImpl.java @@ -4,6 +4,8 @@ import com.databricks.sdk.core.ApiClient; import com.databricks.sdk.support.Generated; import java.util.Collection; +import java.util.HashMap; +import java.util.Map; /** Package-local implementation of Networks */ @Generated @@ -17,7 +19,10 @@ public NetworksImpl(ApiClient apiClient) { @Override public Network create(CreateNetworkRequest request) { String path = String.format("/api/2.0/accounts/%s/networks", apiClient.configuredAccountID()); - return apiClient.POST(path, request, Network.class); + Map headers = new HashMap<>(); + headers.put("Accept", "application/json"); + headers.put("Content-Type", "application/json"); + return apiClient.POST(path, request, Network.class, headers); } @Override @@ -26,7 +31,9 @@ public void delete(DeleteNetworkRequest request) { String.format( "/api/2.0/accounts/%s/networks/%s", apiClient.configuredAccountID(), request.getNetworkId()); - apiClient.DELETE(path, request, Void.class); + Map headers = new HashMap<>(); + headers.put("Accept", "application/json"); + apiClient.DELETE(path, request, Void.class, headers); } @Override @@ -35,12 +42,16 @@ public Network get(GetNetworkRequest request) { String.format( "/api/2.0/accounts/%s/networks/%s", apiClient.configuredAccountID(), request.getNetworkId()); - return apiClient.GET(path, request, Network.class); + Map headers = new HashMap<>(); + headers.put("Accept", "application/json"); + return apiClient.GET(path, request, Network.class, headers); } @Override public Collection list() { String path = String.format("/api/2.0/accounts/%s/networks", apiClient.configuredAccountID()); - return apiClient.getCollection(path, null, Network.class); + Map headers = new HashMap<>(); + headers.put("Accept", "application/json"); + return apiClient.getCollection(path, null, Network.class, headers); } } diff --git a/databricks-sdk-java/src/main/java/com/databricks/sdk/service/provisioning/PrivateAccessImpl.java b/databricks-sdk-java/src/main/java/com/databricks/sdk/service/provisioning/PrivateAccessImpl.java index 5e8399102..56ea36a4c 100755 --- a/databricks-sdk-java/src/main/java/com/databricks/sdk/service/provisioning/PrivateAccessImpl.java +++ b/databricks-sdk-java/src/main/java/com/databricks/sdk/service/provisioning/PrivateAccessImpl.java @@ -4,6 +4,8 @@ import com.databricks.sdk.core.ApiClient; import com.databricks.sdk.support.Generated; import java.util.Collection; +import java.util.HashMap; +import java.util.Map; /** Package-local implementation of PrivateAccess */ @Generated @@ -19,7 +21,10 @@ public PrivateAccessSettings create(UpsertPrivateAccessSettingsRequest request) String path = String.format( "/api/2.0/accounts/%s/private-access-settings", apiClient.configuredAccountID()); - return apiClient.POST(path, request, PrivateAccessSettings.class); + Map headers = new HashMap<>(); + headers.put("Accept", "application/json"); + headers.put("Content-Type", "application/json"); + return apiClient.POST(path, request, PrivateAccessSettings.class, headers); } @Override @@ -28,7 +33,9 @@ public void delete(DeletePrivateAccesRequest request) { String.format( "/api/2.0/accounts/%s/private-access-settings/%s", apiClient.configuredAccountID(), request.getPrivateAccessSettingsId()); - apiClient.DELETE(path, request, Void.class); + Map headers = new HashMap<>(); + headers.put("Accept", "application/json"); + apiClient.DELETE(path, request, Void.class, headers); } @Override @@ -37,7 +44,9 @@ public PrivateAccessSettings get(GetPrivateAccesRequest request) { String.format( "/api/2.0/accounts/%s/private-access-settings/%s", apiClient.configuredAccountID(), request.getPrivateAccessSettingsId()); - return apiClient.GET(path, request, PrivateAccessSettings.class); + Map headers = new HashMap<>(); + headers.put("Accept", "application/json"); + return apiClient.GET(path, request, PrivateAccessSettings.class, headers); } @Override @@ -45,7 +54,9 @@ public Collection list() { String path = String.format( "/api/2.0/accounts/%s/private-access-settings", apiClient.configuredAccountID()); - return apiClient.getCollection(path, null, PrivateAccessSettings.class); + Map headers = new HashMap<>(); + headers.put("Accept", "application/json"); + return apiClient.getCollection(path, null, PrivateAccessSettings.class, headers); } @Override @@ -54,6 +65,9 @@ public void replace(UpsertPrivateAccessSettingsRequest request) { String.format( "/api/2.0/accounts/%s/private-access-settings/%s", apiClient.configuredAccountID(), request.getPrivateAccessSettingsId()); - apiClient.PUT(path, request, Void.class); + Map headers = new HashMap<>(); + headers.put("Accept", "application/json"); + headers.put("Content-Type", "application/json"); + apiClient.PUT(path, request, Void.class, headers); } } diff --git a/databricks-sdk-java/src/main/java/com/databricks/sdk/service/provisioning/StorageImpl.java b/databricks-sdk-java/src/main/java/com/databricks/sdk/service/provisioning/StorageImpl.java index 1be4991a6..cf6b0a977 100755 --- a/databricks-sdk-java/src/main/java/com/databricks/sdk/service/provisioning/StorageImpl.java +++ b/databricks-sdk-java/src/main/java/com/databricks/sdk/service/provisioning/StorageImpl.java @@ -4,6 +4,8 @@ import com.databricks.sdk.core.ApiClient; import com.databricks.sdk.support.Generated; import java.util.Collection; +import java.util.HashMap; +import java.util.Map; /** Package-local implementation of Storage */ @Generated @@ -19,7 +21,10 @@ public StorageConfiguration create(CreateStorageConfigurationRequest request) { String path = String.format( "/api/2.0/accounts/%s/storage-configurations", apiClient.configuredAccountID()); - return apiClient.POST(path, request, StorageConfiguration.class); + Map headers = new HashMap<>(); + headers.put("Accept", "application/json"); + headers.put("Content-Type", "application/json"); + return apiClient.POST(path, request, StorageConfiguration.class, headers); } @Override @@ -28,7 +33,9 @@ public void delete(DeleteStorageRequest request) { String.format( "/api/2.0/accounts/%s/storage-configurations/%s", apiClient.configuredAccountID(), request.getStorageConfigurationId()); - apiClient.DELETE(path, request, Void.class); + Map headers = new HashMap<>(); + headers.put("Accept", "application/json"); + apiClient.DELETE(path, request, Void.class, headers); } @Override @@ -37,7 +44,9 @@ public StorageConfiguration get(GetStorageRequest request) { String.format( "/api/2.0/accounts/%s/storage-configurations/%s", apiClient.configuredAccountID(), request.getStorageConfigurationId()); - return apiClient.GET(path, request, StorageConfiguration.class); + Map headers = new HashMap<>(); + headers.put("Accept", "application/json"); + return apiClient.GET(path, request, StorageConfiguration.class, headers); } @Override @@ -45,6 +54,8 @@ public Collection list() { String path = String.format( "/api/2.0/accounts/%s/storage-configurations", apiClient.configuredAccountID()); - return apiClient.getCollection(path, null, StorageConfiguration.class); + Map headers = new HashMap<>(); + headers.put("Accept", "application/json"); + return apiClient.getCollection(path, null, StorageConfiguration.class, headers); } } diff --git a/databricks-sdk-java/src/main/java/com/databricks/sdk/service/provisioning/VpcEndpointsImpl.java b/databricks-sdk-java/src/main/java/com/databricks/sdk/service/provisioning/VpcEndpointsImpl.java index cec40f2db..065171b28 100755 --- a/databricks-sdk-java/src/main/java/com/databricks/sdk/service/provisioning/VpcEndpointsImpl.java +++ b/databricks-sdk-java/src/main/java/com/databricks/sdk/service/provisioning/VpcEndpointsImpl.java @@ -4,6 +4,8 @@ import com.databricks.sdk.core.ApiClient; import com.databricks.sdk.support.Generated; import java.util.Collection; +import java.util.HashMap; +import java.util.Map; /** Package-local implementation of VpcEndpoints */ @Generated @@ -18,7 +20,10 @@ public VpcEndpointsImpl(ApiClient apiClient) { public VpcEndpoint create(CreateVpcEndpointRequest request) { String path = String.format("/api/2.0/accounts/%s/vpc-endpoints", apiClient.configuredAccountID()); - return apiClient.POST(path, request, VpcEndpoint.class); + Map headers = new HashMap<>(); + headers.put("Accept", "application/json"); + headers.put("Content-Type", "application/json"); + return apiClient.POST(path, request, VpcEndpoint.class, headers); } @Override @@ -27,7 +32,9 @@ public void delete(DeleteVpcEndpointRequest request) { String.format( "/api/2.0/accounts/%s/vpc-endpoints/%s", apiClient.configuredAccountID(), request.getVpcEndpointId()); - apiClient.DELETE(path, request, Void.class); + Map headers = new HashMap<>(); + headers.put("Accept", "application/json"); + apiClient.DELETE(path, request, Void.class, headers); } @Override @@ -36,13 +43,17 @@ public VpcEndpoint get(GetVpcEndpointRequest request) { String.format( "/api/2.0/accounts/%s/vpc-endpoints/%s", apiClient.configuredAccountID(), request.getVpcEndpointId()); - return apiClient.GET(path, request, VpcEndpoint.class); + Map headers = new HashMap<>(); + headers.put("Accept", "application/json"); + return apiClient.GET(path, request, VpcEndpoint.class, headers); } @Override public Collection list() { String path = String.format("/api/2.0/accounts/%s/vpc-endpoints", apiClient.configuredAccountID()); - return apiClient.getCollection(path, null, VpcEndpoint.class); + Map headers = new HashMap<>(); + headers.put("Accept", "application/json"); + return apiClient.getCollection(path, null, VpcEndpoint.class, headers); } } diff --git a/databricks-sdk-java/src/main/java/com/databricks/sdk/service/provisioning/WorkspacesImpl.java b/databricks-sdk-java/src/main/java/com/databricks/sdk/service/provisioning/WorkspacesImpl.java index d2b8569a5..d798348cc 100755 --- a/databricks-sdk-java/src/main/java/com/databricks/sdk/service/provisioning/WorkspacesImpl.java +++ b/databricks-sdk-java/src/main/java/com/databricks/sdk/service/provisioning/WorkspacesImpl.java @@ -4,6 +4,8 @@ import com.databricks.sdk.core.ApiClient; import com.databricks.sdk.support.Generated; import java.util.Collection; +import java.util.HashMap; +import java.util.Map; /** Package-local implementation of Workspaces */ @Generated @@ -17,7 +19,10 @@ public WorkspacesImpl(ApiClient apiClient) { @Override public Workspace create(CreateWorkspaceRequest request) { String path = String.format("/api/2.0/accounts/%s/workspaces", apiClient.configuredAccountID()); - return apiClient.POST(path, request, Workspace.class); + Map headers = new HashMap<>(); + headers.put("Accept", "application/json"); + headers.put("Content-Type", "application/json"); + return apiClient.POST(path, request, Workspace.class, headers); } @Override @@ -26,7 +31,9 @@ public void delete(DeleteWorkspaceRequest request) { String.format( "/api/2.0/accounts/%s/workspaces/%s", apiClient.configuredAccountID(), request.getWorkspaceId()); - apiClient.DELETE(path, request, Void.class); + Map headers = new HashMap<>(); + headers.put("Accept", "application/json"); + apiClient.DELETE(path, request, Void.class, headers); } @Override @@ -35,13 +42,17 @@ public Workspace get(GetWorkspaceRequest request) { String.format( "/api/2.0/accounts/%s/workspaces/%s", apiClient.configuredAccountID(), request.getWorkspaceId()); - return apiClient.GET(path, request, Workspace.class); + Map headers = new HashMap<>(); + headers.put("Accept", "application/json"); + return apiClient.GET(path, request, Workspace.class, headers); } @Override public Collection list() { String path = String.format("/api/2.0/accounts/%s/workspaces", apiClient.configuredAccountID()); - return apiClient.getCollection(path, null, Workspace.class); + Map headers = new HashMap<>(); + headers.put("Accept", "application/json"); + return apiClient.getCollection(path, null, Workspace.class, headers); } @Override @@ -50,6 +61,9 @@ public void update(UpdateWorkspaceRequest request) { String.format( "/api/2.0/accounts/%s/workspaces/%s", apiClient.configuredAccountID(), request.getWorkspaceId()); - apiClient.PATCH(path, request, Void.class); + Map headers = new HashMap<>(); + headers.put("Accept", "application/json"); + headers.put("Content-Type", "application/json"); + apiClient.PATCH(path, request, Void.class, headers); } } diff --git a/databricks-sdk-java/src/main/java/com/databricks/sdk/service/serving/ServingEndpointsImpl.java b/databricks-sdk-java/src/main/java/com/databricks/sdk/service/serving/ServingEndpointsImpl.java index c43bbe266..b1b639902 100755 --- a/databricks-sdk-java/src/main/java/com/databricks/sdk/service/serving/ServingEndpointsImpl.java +++ b/databricks-sdk-java/src/main/java/com/databricks/sdk/service/serving/ServingEndpointsImpl.java @@ -3,6 +3,8 @@ import com.databricks.sdk.core.ApiClient; import com.databricks.sdk.support.Generated; +import java.util.HashMap; +import java.util.Map; /** Package-local implementation of ServingEndpoints */ @Generated @@ -19,31 +21,41 @@ public BuildLogsResponse buildLogs(BuildLogsRequest request) { String.format( "/api/2.0/serving-endpoints/%s/served-models/%s/build-logs", request.getName(), request.getServedModelName()); - return apiClient.GET(path, request, BuildLogsResponse.class); + Map headers = new HashMap<>(); + headers.put("Accept", "application/json"); + return apiClient.GET(path, request, BuildLogsResponse.class, headers); } @Override public ServingEndpointDetailed create(CreateServingEndpoint request) { String path = "/api/2.0/serving-endpoints"; - return apiClient.POST(path, request, ServingEndpointDetailed.class); + Map headers = new HashMap<>(); + headers.put("Accept", "application/json"); + headers.put("Content-Type", "application/json"); + return apiClient.POST(path, request, ServingEndpointDetailed.class, headers); } @Override public void delete(DeleteServingEndpointRequest request) { String path = String.format("/api/2.0/serving-endpoints/%s", request.getName()); - apiClient.DELETE(path, request, Void.class); + Map headers = new HashMap<>(); + headers.put("Accept", "application/json"); + apiClient.DELETE(path, request, Void.class, headers); } @Override public void exportMetrics(ExportMetricsRequest request) { String path = String.format("/api/2.0/serving-endpoints/%s/metrics", request.getName()); - apiClient.GET(path, request, Void.class); + Map headers = new HashMap<>(); + apiClient.GET(path, request, Void.class, headers); } @Override public ServingEndpointDetailed get(GetServingEndpointRequest request) { String path = String.format("/api/2.0/serving-endpoints/%s", request.getName()); - return apiClient.GET(path, request, ServingEndpointDetailed.class); + Map headers = new HashMap<>(); + headers.put("Accept", "application/json"); + return apiClient.GET(path, request, ServingEndpointDetailed.class, headers); } @Override @@ -53,7 +65,9 @@ public GetServingEndpointPermissionLevelsResponse getServingEndpointPermissionLe String.format( "/api/2.0/permissions/serving-endpoints/%s/permissionLevels", request.getServingEndpointId()); - return apiClient.GET(path, request, GetServingEndpointPermissionLevelsResponse.class); + Map headers = new HashMap<>(); + headers.put("Accept", "application/json"); + return apiClient.GET(path, request, GetServingEndpointPermissionLevelsResponse.class, headers); } @Override @@ -61,13 +75,17 @@ public ServingEndpointPermissions getServingEndpointPermissions( GetServingEndpointPermissionsRequest request) { String path = String.format("/api/2.0/permissions/serving-endpoints/%s", request.getServingEndpointId()); - return apiClient.GET(path, request, ServingEndpointPermissions.class); + Map headers = new HashMap<>(); + headers.put("Accept", "application/json"); + return apiClient.GET(path, request, ServingEndpointPermissions.class, headers); } @Override public ListEndpointsResponse list() { String path = "/api/2.0/serving-endpoints"; - return apiClient.GET(path, ListEndpointsResponse.class); + Map headers = new HashMap<>(); + headers.put("Accept", "application/json"); + return apiClient.GET(path, ListEndpointsResponse.class, headers); } @Override @@ -76,13 +94,17 @@ public ServerLogsResponse logs(LogsRequest request) { String.format( "/api/2.0/serving-endpoints/%s/served-models/%s/logs", request.getName(), request.getServedModelName()); - return apiClient.GET(path, request, ServerLogsResponse.class); + Map headers = new HashMap<>(); + headers.put("Accept", "application/json"); + return apiClient.GET(path, request, ServerLogsResponse.class, headers); } @Override public QueryEndpointResponse query(QueryRequest request) { String path = String.format("/serving-endpoints/%s/invocations", request.getName()); - return apiClient.POST(path, request, QueryEndpointResponse.class); + Map headers = new HashMap<>(); + headers.put("Accept", "application/json"); + return apiClient.POST(path, request, QueryEndpointResponse.class, headers); } @Override @@ -90,13 +112,19 @@ public ServingEndpointPermissions setServingEndpointPermissions( ServingEndpointPermissionsRequest request) { String path = String.format("/api/2.0/permissions/serving-endpoints/%s", request.getServingEndpointId()); - return apiClient.PUT(path, request, ServingEndpointPermissions.class); + Map headers = new HashMap<>(); + headers.put("Accept", "application/json"); + headers.put("Content-Type", "application/json"); + return apiClient.PUT(path, request, ServingEndpointPermissions.class, headers); } @Override public ServingEndpointDetailed updateConfig(EndpointCoreConfigInput request) { String path = String.format("/api/2.0/serving-endpoints/%s/config", request.getName()); - return apiClient.PUT(path, request, ServingEndpointDetailed.class); + Map headers = new HashMap<>(); + headers.put("Accept", "application/json"); + headers.put("Content-Type", "application/json"); + return apiClient.PUT(path, request, ServingEndpointDetailed.class, headers); } @Override @@ -104,6 +132,9 @@ public ServingEndpointPermissions updateServingEndpointPermissions( ServingEndpointPermissionsRequest request) { String path = String.format("/api/2.0/permissions/serving-endpoints/%s", request.getServingEndpointId()); - return apiClient.PATCH(path, request, ServingEndpointPermissions.class); + Map headers = new HashMap<>(); + headers.put("Accept", "application/json"); + headers.put("Content-Type", "application/json"); + return apiClient.PATCH(path, request, ServingEndpointPermissions.class, headers); } } diff --git a/databricks-sdk-java/src/main/java/com/databricks/sdk/service/settings/AccountIpAccessListsImpl.java b/databricks-sdk-java/src/main/java/com/databricks/sdk/service/settings/AccountIpAccessListsImpl.java index 413d23085..429d4c6f9 100755 --- a/databricks-sdk-java/src/main/java/com/databricks/sdk/service/settings/AccountIpAccessListsImpl.java +++ b/databricks-sdk-java/src/main/java/com/databricks/sdk/service/settings/AccountIpAccessListsImpl.java @@ -3,6 +3,8 @@ import com.databricks.sdk.core.ApiClient; import com.databricks.sdk.support.Generated; +import java.util.HashMap; +import java.util.Map; /** Package-local implementation of AccountIpAccessLists */ @Generated @@ -18,7 +20,10 @@ public CreateIpAccessListResponse create(CreateIpAccessList request) { String path = String.format( "/api/2.0/preview/accounts/%s/ip-access-lists", apiClient.configuredAccountID()); - return apiClient.POST(path, request, CreateIpAccessListResponse.class); + Map headers = new HashMap<>(); + headers.put("Accept", "application/json"); + headers.put("Content-Type", "application/json"); + return apiClient.POST(path, request, CreateIpAccessListResponse.class, headers); } @Override @@ -27,7 +32,8 @@ public void delete(DeleteAccountIpAccessListRequest request) { String.format( "/api/2.0/preview/accounts/%s/ip-access-lists/%s", apiClient.configuredAccountID(), request.getIpAccessListId()); - apiClient.DELETE(path, request, Void.class); + Map headers = new HashMap<>(); + apiClient.DELETE(path, request, Void.class, headers); } @Override @@ -36,7 +42,9 @@ public GetIpAccessListResponse get(GetAccountIpAccessListRequest request) { String.format( "/api/2.0/preview/accounts/%s/ip-access-lists/%s", apiClient.configuredAccountID(), request.getIpAccessListId()); - return apiClient.GET(path, request, GetIpAccessListResponse.class); + Map headers = new HashMap<>(); + headers.put("Accept", "application/json"); + return apiClient.GET(path, request, GetIpAccessListResponse.class, headers); } @Override @@ -44,7 +52,9 @@ public GetIpAccessListsResponse list() { String path = String.format( "/api/2.0/preview/accounts/%s/ip-access-lists", apiClient.configuredAccountID()); - return apiClient.GET(path, GetIpAccessListsResponse.class); + Map headers = new HashMap<>(); + headers.put("Accept", "application/json"); + return apiClient.GET(path, GetIpAccessListsResponse.class, headers); } @Override @@ -53,7 +63,10 @@ public void replace(ReplaceIpAccessList request) { String.format( "/api/2.0/preview/accounts/%s/ip-access-lists/%s", apiClient.configuredAccountID(), request.getIpAccessListId()); - apiClient.PUT(path, request, Void.class); + Map headers = new HashMap<>(); + headers.put("Accept", "application/json"); + headers.put("Content-Type", "application/json"); + apiClient.PUT(path, request, Void.class, headers); } @Override @@ -62,6 +75,9 @@ public void update(UpdateIpAccessList request) { String.format( "/api/2.0/preview/accounts/%s/ip-access-lists/%s", apiClient.configuredAccountID(), request.getIpAccessListId()); - apiClient.PATCH(path, request, Void.class); + Map headers = new HashMap<>(); + headers.put("Accept", "application/json"); + headers.put("Content-Type", "application/json"); + apiClient.PATCH(path, request, Void.class, headers); } } diff --git a/databricks-sdk-java/src/main/java/com/databricks/sdk/service/settings/AccountNetworkPolicyImpl.java b/databricks-sdk-java/src/main/java/com/databricks/sdk/service/settings/AccountNetworkPolicyImpl.java index e2b568588..4a8f39d83 100755 --- a/databricks-sdk-java/src/main/java/com/databricks/sdk/service/settings/AccountNetworkPolicyImpl.java +++ b/databricks-sdk-java/src/main/java/com/databricks/sdk/service/settings/AccountNetworkPolicyImpl.java @@ -3,6 +3,8 @@ import com.databricks.sdk.core.ApiClient; import com.databricks.sdk.support.Generated; +import java.util.HashMap; +import java.util.Map; /** Package-local implementation of AccountNetworkPolicy */ @Generated @@ -20,7 +22,9 @@ public DeleteAccountNetworkPolicyResponse deleteAccountNetworkPolicy( String.format( "/api/2.0/accounts/%s/settings/types/network_policy/names/default", apiClient.configuredAccountID()); - return apiClient.DELETE(path, request, DeleteAccountNetworkPolicyResponse.class); + Map headers = new HashMap<>(); + headers.put("Accept", "application/json"); + return apiClient.DELETE(path, request, DeleteAccountNetworkPolicyResponse.class, headers); } @Override @@ -30,7 +34,9 @@ public AccountNetworkPolicyMessage readAccountNetworkPolicy( String.format( "/api/2.0/accounts/%s/settings/types/network_policy/names/default", apiClient.configuredAccountID()); - return apiClient.GET(path, request, AccountNetworkPolicyMessage.class); + Map headers = new HashMap<>(); + headers.put("Accept", "application/json"); + return apiClient.GET(path, request, AccountNetworkPolicyMessage.class, headers); } @Override @@ -40,6 +46,9 @@ public AccountNetworkPolicyMessage updateAccountNetworkPolicy( String.format( "/api/2.0/accounts/%s/settings/types/network_policy/names/default", apiClient.configuredAccountID()); - return apiClient.PATCH(path, request, AccountNetworkPolicyMessage.class); + Map headers = new HashMap<>(); + headers.put("Accept", "application/json"); + headers.put("Content-Type", "application/json"); + return apiClient.PATCH(path, request, AccountNetworkPolicyMessage.class, headers); } } diff --git a/databricks-sdk-java/src/main/java/com/databricks/sdk/service/settings/AccountSettingsImpl.java b/databricks-sdk-java/src/main/java/com/databricks/sdk/service/settings/AccountSettingsImpl.java index cd2f5f42d..a30f7d135 100755 --- a/databricks-sdk-java/src/main/java/com/databricks/sdk/service/settings/AccountSettingsImpl.java +++ b/databricks-sdk-java/src/main/java/com/databricks/sdk/service/settings/AccountSettingsImpl.java @@ -3,6 +3,8 @@ import com.databricks.sdk.core.ApiClient; import com.databricks.sdk.support.Generated; +import java.util.HashMap; +import java.util.Map; /** Package-local implementation of AccountSettings */ @Generated @@ -20,7 +22,9 @@ public DeletePersonalComputeSettingResponse deletePersonalComputeSetting( String.format( "/api/2.0/accounts/%s/settings/types/dcp_acct_enable/names/default", apiClient.configuredAccountID()); - return apiClient.DELETE(path, request, DeletePersonalComputeSettingResponse.class); + Map headers = new HashMap<>(); + headers.put("Accept", "application/json"); + return apiClient.DELETE(path, request, DeletePersonalComputeSettingResponse.class, headers); } @Override @@ -30,7 +34,9 @@ public PersonalComputeSetting readPersonalComputeSetting( String.format( "/api/2.0/accounts/%s/settings/types/dcp_acct_enable/names/default", apiClient.configuredAccountID()); - return apiClient.GET(path, request, PersonalComputeSetting.class); + Map headers = new HashMap<>(); + headers.put("Accept", "application/json"); + return apiClient.GET(path, request, PersonalComputeSetting.class, headers); } @Override @@ -40,6 +46,9 @@ public PersonalComputeSetting updatePersonalComputeSetting( String.format( "/api/2.0/accounts/%s/settings/types/dcp_acct_enable/names/default", apiClient.configuredAccountID()); - return apiClient.PATCH(path, request, PersonalComputeSetting.class); + Map headers = new HashMap<>(); + headers.put("Accept", "application/json"); + headers.put("Content-Type", "application/json"); + return apiClient.PATCH(path, request, PersonalComputeSetting.class, headers); } } diff --git a/databricks-sdk-java/src/main/java/com/databricks/sdk/service/settings/IpAccessListsImpl.java b/databricks-sdk-java/src/main/java/com/databricks/sdk/service/settings/IpAccessListsImpl.java index 401d91781..95aa2bffa 100755 --- a/databricks-sdk-java/src/main/java/com/databricks/sdk/service/settings/IpAccessListsImpl.java +++ b/databricks-sdk-java/src/main/java/com/databricks/sdk/service/settings/IpAccessListsImpl.java @@ -3,6 +3,8 @@ import com.databricks.sdk.core.ApiClient; import com.databricks.sdk.support.Generated; +import java.util.HashMap; +import java.util.Map; /** Package-local implementation of IpAccessLists */ @Generated @@ -16,36 +18,50 @@ public IpAccessListsImpl(ApiClient apiClient) { @Override public CreateIpAccessListResponse create(CreateIpAccessList request) { String path = "/api/2.0/ip-access-lists"; - return apiClient.POST(path, request, CreateIpAccessListResponse.class); + Map headers = new HashMap<>(); + headers.put("Accept", "application/json"); + headers.put("Content-Type", "application/json"); + return apiClient.POST(path, request, CreateIpAccessListResponse.class, headers); } @Override public void delete(DeleteIpAccessListRequest request) { String path = String.format("/api/2.0/ip-access-lists/%s", request.getIpAccessListId()); - apiClient.DELETE(path, request, Void.class); + Map headers = new HashMap<>(); + apiClient.DELETE(path, request, Void.class, headers); } @Override public FetchIpAccessListResponse get(GetIpAccessListRequest request) { String path = String.format("/api/2.0/ip-access-lists/%s", request.getIpAccessListId()); - return apiClient.GET(path, request, FetchIpAccessListResponse.class); + Map headers = new HashMap<>(); + headers.put("Accept", "application/json"); + return apiClient.GET(path, request, FetchIpAccessListResponse.class, headers); } @Override public GetIpAccessListResponse list() { String path = "/api/2.0/ip-access-lists"; - return apiClient.GET(path, GetIpAccessListResponse.class); + Map headers = new HashMap<>(); + headers.put("Accept", "application/json"); + return apiClient.GET(path, GetIpAccessListResponse.class, headers); } @Override public void replace(ReplaceIpAccessList request) { String path = String.format("/api/2.0/ip-access-lists/%s", request.getIpAccessListId()); - apiClient.PUT(path, request, Void.class); + Map headers = new HashMap<>(); + headers.put("Accept", "application/json"); + headers.put("Content-Type", "application/json"); + apiClient.PUT(path, request, Void.class, headers); } @Override public void update(UpdateIpAccessList request) { String path = String.format("/api/2.0/ip-access-lists/%s", request.getIpAccessListId()); - apiClient.PATCH(path, request, Void.class); + Map headers = new HashMap<>(); + headers.put("Accept", "application/json"); + headers.put("Content-Type", "application/json"); + apiClient.PATCH(path, request, Void.class, headers); } } diff --git a/databricks-sdk-java/src/main/java/com/databricks/sdk/service/settings/TokenManagementImpl.java b/databricks-sdk-java/src/main/java/com/databricks/sdk/service/settings/TokenManagementImpl.java index 58ebec01a..693abe8fb 100755 --- a/databricks-sdk-java/src/main/java/com/databricks/sdk/service/settings/TokenManagementImpl.java +++ b/databricks-sdk-java/src/main/java/com/databricks/sdk/service/settings/TokenManagementImpl.java @@ -3,6 +3,8 @@ import com.databricks.sdk.core.ApiClient; import com.databricks.sdk.support.Generated; +import java.util.HashMap; +import java.util.Map; /** Package-local implementation of TokenManagement */ @Generated @@ -16,48 +18,66 @@ public TokenManagementImpl(ApiClient apiClient) { @Override public CreateOboTokenResponse createOboToken(CreateOboTokenRequest request) { String path = "/api/2.0/token-management/on-behalf-of/tokens"; - return apiClient.POST(path, request, CreateOboTokenResponse.class); + Map headers = new HashMap<>(); + headers.put("Accept", "application/json"); + headers.put("Content-Type", "application/json"); + return apiClient.POST(path, request, CreateOboTokenResponse.class, headers); } @Override public void delete(DeleteTokenManagementRequest request) { String path = String.format("/api/2.0/token-management/tokens/%s", request.getTokenId()); - apiClient.DELETE(path, request, Void.class); + Map headers = new HashMap<>(); + apiClient.DELETE(path, request, Void.class, headers); } @Override public TokenInfo get(GetTokenManagementRequest request) { String path = String.format("/api/2.0/token-management/tokens/%s", request.getTokenId()); - return apiClient.GET(path, request, TokenInfo.class); + Map headers = new HashMap<>(); + headers.put("Accept", "application/json"); + return apiClient.GET(path, request, TokenInfo.class, headers); } @Override public GetTokenPermissionLevelsResponse getTokenPermissionLevels() { String path = "/api/2.0/permissions/authorization/tokens/permissionLevels"; - return apiClient.GET(path, GetTokenPermissionLevelsResponse.class); + Map headers = new HashMap<>(); + headers.put("Accept", "application/json"); + return apiClient.GET(path, GetTokenPermissionLevelsResponse.class, headers); } @Override public TokenPermissions getTokenPermissions() { String path = "/api/2.0/permissions/authorization/tokens"; - return apiClient.GET(path, TokenPermissions.class); + Map headers = new HashMap<>(); + headers.put("Accept", "application/json"); + return apiClient.GET(path, TokenPermissions.class, headers); } @Override public ListTokensResponse list(ListTokenManagementRequest request) { String path = "/api/2.0/token-management/tokens"; - return apiClient.GET(path, request, ListTokensResponse.class); + Map headers = new HashMap<>(); + headers.put("Accept", "application/json"); + return apiClient.GET(path, request, ListTokensResponse.class, headers); } @Override public TokenPermissions setTokenPermissions(TokenPermissionsRequest request) { String path = "/api/2.0/permissions/authorization/tokens"; - return apiClient.PUT(path, request, TokenPermissions.class); + Map headers = new HashMap<>(); + headers.put("Accept", "application/json"); + headers.put("Content-Type", "application/json"); + return apiClient.PUT(path, request, TokenPermissions.class, headers); } @Override public TokenPermissions updateTokenPermissions(TokenPermissionsRequest request) { String path = "/api/2.0/permissions/authorization/tokens"; - return apiClient.PATCH(path, request, TokenPermissions.class); + Map headers = new HashMap<>(); + headers.put("Accept", "application/json"); + headers.put("Content-Type", "application/json"); + return apiClient.PATCH(path, request, TokenPermissions.class, headers); } } diff --git a/databricks-sdk-java/src/main/java/com/databricks/sdk/service/settings/TokensImpl.java b/databricks-sdk-java/src/main/java/com/databricks/sdk/service/settings/TokensImpl.java index 2709f91a9..44019a6a8 100755 --- a/databricks-sdk-java/src/main/java/com/databricks/sdk/service/settings/TokensImpl.java +++ b/databricks-sdk-java/src/main/java/com/databricks/sdk/service/settings/TokensImpl.java @@ -3,6 +3,8 @@ import com.databricks.sdk.core.ApiClient; import com.databricks.sdk.support.Generated; +import java.util.HashMap; +import java.util.Map; /** Package-local implementation of Tokens */ @Generated @@ -16,18 +18,26 @@ public TokensImpl(ApiClient apiClient) { @Override public CreateTokenResponse create(CreateTokenRequest request) { String path = "/api/2.0/token/create"; - return apiClient.POST(path, request, CreateTokenResponse.class); + Map headers = new HashMap<>(); + headers.put("Accept", "application/json"); + headers.put("Content-Type", "application/json"); + return apiClient.POST(path, request, CreateTokenResponse.class, headers); } @Override public void delete(RevokeTokenRequest request) { String path = "/api/2.0/token/delete"; - apiClient.POST(path, request, Void.class); + Map headers = new HashMap<>(); + headers.put("Accept", "application/json"); + headers.put("Content-Type", "application/json"); + apiClient.POST(path, request, Void.class, headers); } @Override public ListTokensResponse list() { String path = "/api/2.0/token/list"; - return apiClient.GET(path, ListTokensResponse.class); + Map headers = new HashMap<>(); + headers.put("Accept", "application/json"); + return apiClient.GET(path, ListTokensResponse.class, headers); } } diff --git a/databricks-sdk-java/src/main/java/com/databricks/sdk/service/settings/WorkspaceConfImpl.java b/databricks-sdk-java/src/main/java/com/databricks/sdk/service/settings/WorkspaceConfImpl.java index c863a3b54..6ae4e5384 100755 --- a/databricks-sdk-java/src/main/java/com/databricks/sdk/service/settings/WorkspaceConfImpl.java +++ b/databricks-sdk-java/src/main/java/com/databricks/sdk/service/settings/WorkspaceConfImpl.java @@ -3,6 +3,7 @@ import com.databricks.sdk.core.ApiClient; import com.databricks.sdk.support.Generated; +import java.util.HashMap; import java.util.Map; /** Package-local implementation of WorkspaceConf */ @@ -17,12 +18,16 @@ public WorkspaceConfImpl(ApiClient apiClient) { @Override public Map getStatus(GetStatusRequest request) { String path = "/api/2.0/workspace-conf"; - return apiClient.getStringMap(path, request); + Map headers = new HashMap<>(); + headers.put("Accept", "application/json"); + return apiClient.getStringMap(path, request, headers); } @Override public void setStatus(Map request) { String path = "/api/2.0/workspace-conf"; - apiClient.PATCH(path, request, Void.class); + Map headers = new HashMap<>(); + headers.put("Content-Type", "application/json"); + apiClient.PATCH(path, request, Void.class, headers); } } diff --git a/databricks-sdk-java/src/main/java/com/databricks/sdk/service/sharing/CleanRoomsImpl.java b/databricks-sdk-java/src/main/java/com/databricks/sdk/service/sharing/CleanRoomsImpl.java index d282108bc..1de3db59f 100755 --- a/databricks-sdk-java/src/main/java/com/databricks/sdk/service/sharing/CleanRoomsImpl.java +++ b/databricks-sdk-java/src/main/java/com/databricks/sdk/service/sharing/CleanRoomsImpl.java @@ -3,6 +3,8 @@ import com.databricks.sdk.core.ApiClient; import com.databricks.sdk.support.Generated; +import java.util.HashMap; +import java.util.Map; /** Package-local implementation of CleanRooms */ @Generated @@ -16,30 +18,42 @@ public CleanRoomsImpl(ApiClient apiClient) { @Override public CleanRoomInfo create(CreateCleanRoom request) { String path = "/api/2.1/unity-catalog/clean-rooms"; - return apiClient.POST(path, request, CleanRoomInfo.class); + Map headers = new HashMap<>(); + headers.put("Accept", "application/json"); + headers.put("Content-Type", "application/json"); + return apiClient.POST(path, request, CleanRoomInfo.class, headers); } @Override public void delete(DeleteCleanRoomRequest request) { String path = String.format("/api/2.1/unity-catalog/clean-rooms/%s", request.getNameArg()); - apiClient.DELETE(path, request, Void.class); + Map headers = new HashMap<>(); + headers.put("Accept", "application/json"); + apiClient.DELETE(path, request, Void.class, headers); } @Override public CleanRoomInfo get(GetCleanRoomRequest request) { String path = String.format("/api/2.1/unity-catalog/clean-rooms/%s", request.getNameArg()); - return apiClient.GET(path, request, CleanRoomInfo.class); + Map headers = new HashMap<>(); + headers.put("Accept", "application/json"); + return apiClient.GET(path, request, CleanRoomInfo.class, headers); } @Override public ListCleanRoomsResponse list() { String path = "/api/2.1/unity-catalog/clean-rooms"; - return apiClient.GET(path, ListCleanRoomsResponse.class); + Map headers = new HashMap<>(); + headers.put("Accept", "application/json"); + return apiClient.GET(path, ListCleanRoomsResponse.class, headers); } @Override public CleanRoomInfo update(UpdateCleanRoom request) { String path = String.format("/api/2.1/unity-catalog/clean-rooms/%s", request.getNameArg()); - return apiClient.PATCH(path, request, CleanRoomInfo.class); + Map headers = new HashMap<>(); + headers.put("Accept", "application/json"); + headers.put("Content-Type", "application/json"); + return apiClient.PATCH(path, request, CleanRoomInfo.class, headers); } } diff --git a/databricks-sdk-java/src/main/java/com/databricks/sdk/service/sharing/ProvidersImpl.java b/databricks-sdk-java/src/main/java/com/databricks/sdk/service/sharing/ProvidersImpl.java index cc579a396..ff677eac3 100755 --- a/databricks-sdk-java/src/main/java/com/databricks/sdk/service/sharing/ProvidersImpl.java +++ b/databricks-sdk-java/src/main/java/com/databricks/sdk/service/sharing/ProvidersImpl.java @@ -3,6 +3,8 @@ import com.databricks.sdk.core.ApiClient; import com.databricks.sdk.support.Generated; +import java.util.HashMap; +import java.util.Map; /** Package-local implementation of Providers */ @Generated @@ -16,36 +18,50 @@ public ProvidersImpl(ApiClient apiClient) { @Override public ProviderInfo create(CreateProvider request) { String path = "/api/2.1/unity-catalog/providers"; - return apiClient.POST(path, request, ProviderInfo.class); + Map headers = new HashMap<>(); + headers.put("Accept", "application/json"); + headers.put("Content-Type", "application/json"); + return apiClient.POST(path, request, ProviderInfo.class, headers); } @Override public void delete(DeleteProviderRequest request) { String path = String.format("/api/2.1/unity-catalog/providers/%s", request.getName()); - apiClient.DELETE(path, request, Void.class); + Map headers = new HashMap<>(); + headers.put("Accept", "application/json"); + apiClient.DELETE(path, request, Void.class, headers); } @Override public ProviderInfo get(GetProviderRequest request) { String path = String.format("/api/2.1/unity-catalog/providers/%s", request.getName()); - return apiClient.GET(path, request, ProviderInfo.class); + Map headers = new HashMap<>(); + headers.put("Accept", "application/json"); + return apiClient.GET(path, request, ProviderInfo.class, headers); } @Override public ListProvidersResponse list(ListProvidersRequest request) { String path = "/api/2.1/unity-catalog/providers"; - return apiClient.GET(path, request, ListProvidersResponse.class); + Map headers = new HashMap<>(); + headers.put("Accept", "application/json"); + return apiClient.GET(path, request, ListProvidersResponse.class, headers); } @Override public ListProviderSharesResponse listShares(ListSharesRequest request) { String path = String.format("/api/2.1/unity-catalog/providers/%s/shares", request.getName()); - return apiClient.GET(path, request, ListProviderSharesResponse.class); + Map headers = new HashMap<>(); + headers.put("Accept", "application/json"); + return apiClient.GET(path, request, ListProviderSharesResponse.class, headers); } @Override public ProviderInfo update(UpdateProvider request) { String path = String.format("/api/2.1/unity-catalog/providers/%s", request.getName()); - return apiClient.PATCH(path, request, ProviderInfo.class); + Map headers = new HashMap<>(); + headers.put("Accept", "application/json"); + headers.put("Content-Type", "application/json"); + return apiClient.PATCH(path, request, ProviderInfo.class, headers); } } diff --git a/databricks-sdk-java/src/main/java/com/databricks/sdk/service/sharing/RecipientActivationImpl.java b/databricks-sdk-java/src/main/java/com/databricks/sdk/service/sharing/RecipientActivationImpl.java index d8d1f85e8..49bd076a0 100755 --- a/databricks-sdk-java/src/main/java/com/databricks/sdk/service/sharing/RecipientActivationImpl.java +++ b/databricks-sdk-java/src/main/java/com/databricks/sdk/service/sharing/RecipientActivationImpl.java @@ -3,6 +3,8 @@ import com.databricks.sdk.core.ApiClient; import com.databricks.sdk.support.Generated; +import java.util.HashMap; +import java.util.Map; /** Package-local implementation of RecipientActivation */ @Generated @@ -19,7 +21,9 @@ public void getActivationUrlInfo(GetActivationUrlInfoRequest request) { String.format( "/api/2.1/unity-catalog/public/data_sharing_activation_info/%s", request.getActivationUrl()); - apiClient.GET(path, request, Void.class); + Map headers = new HashMap<>(); + headers.put("Accept", "application/json"); + apiClient.GET(path, request, Void.class, headers); } @Override @@ -27,6 +31,8 @@ public RetrieveTokenResponse retrieveToken(RetrieveTokenRequest request) { String path = String.format( "/api/2.1/unity-catalog/public/data_sharing_activation/%s", request.getActivationUrl()); - return apiClient.GET(path, request, RetrieveTokenResponse.class); + Map headers = new HashMap<>(); + headers.put("Accept", "application/json"); + return apiClient.GET(path, request, RetrieveTokenResponse.class, headers); } } diff --git a/databricks-sdk-java/src/main/java/com/databricks/sdk/service/sharing/RecipientsImpl.java b/databricks-sdk-java/src/main/java/com/databricks/sdk/service/sharing/RecipientsImpl.java index 860a4008a..9a603c716 100755 --- a/databricks-sdk-java/src/main/java/com/databricks/sdk/service/sharing/RecipientsImpl.java +++ b/databricks-sdk-java/src/main/java/com/databricks/sdk/service/sharing/RecipientsImpl.java @@ -3,6 +3,8 @@ import com.databricks.sdk.core.ApiClient; import com.databricks.sdk.support.Generated; +import java.util.HashMap; +import java.util.Map; /** Package-local implementation of Recipients */ @Generated @@ -16,44 +18,61 @@ public RecipientsImpl(ApiClient apiClient) { @Override public RecipientInfo create(CreateRecipient request) { String path = "/api/2.1/unity-catalog/recipients"; - return apiClient.POST(path, request, RecipientInfo.class); + Map headers = new HashMap<>(); + headers.put("Accept", "application/json"); + headers.put("Content-Type", "application/json"); + return apiClient.POST(path, request, RecipientInfo.class, headers); } @Override public void delete(DeleteRecipientRequest request) { String path = String.format("/api/2.1/unity-catalog/recipients/%s", request.getName()); - apiClient.DELETE(path, request, Void.class); + Map headers = new HashMap<>(); + headers.put("Accept", "application/json"); + apiClient.DELETE(path, request, Void.class, headers); } @Override public RecipientInfo get(GetRecipientRequest request) { String path = String.format("/api/2.1/unity-catalog/recipients/%s", request.getName()); - return apiClient.GET(path, request, RecipientInfo.class); + Map headers = new HashMap<>(); + headers.put("Accept", "application/json"); + return apiClient.GET(path, request, RecipientInfo.class, headers); } @Override public ListRecipientsResponse list(ListRecipientsRequest request) { String path = "/api/2.1/unity-catalog/recipients"; - return apiClient.GET(path, request, ListRecipientsResponse.class); + Map headers = new HashMap<>(); + headers.put("Accept", "application/json"); + return apiClient.GET(path, request, ListRecipientsResponse.class, headers); } @Override public RecipientInfo rotateToken(RotateRecipientToken request) { String path = String.format("/api/2.1/unity-catalog/recipients/%s/rotate-token", request.getName()); - return apiClient.POST(path, request, RecipientInfo.class); + Map headers = new HashMap<>(); + headers.put("Accept", "application/json"); + headers.put("Content-Type", "application/json"); + return apiClient.POST(path, request, RecipientInfo.class, headers); } @Override public GetRecipientSharePermissionsResponse sharePermissions(SharePermissionsRequest request) { String path = String.format("/api/2.1/unity-catalog/recipients/%s/share-permissions", request.getName()); - return apiClient.GET(path, request, GetRecipientSharePermissionsResponse.class); + Map headers = new HashMap<>(); + headers.put("Accept", "application/json"); + return apiClient.GET(path, request, GetRecipientSharePermissionsResponse.class, headers); } @Override public void update(UpdateRecipient request) { String path = String.format("/api/2.1/unity-catalog/recipients/%s", request.getName()); - apiClient.PATCH(path, request, Void.class); + Map headers = new HashMap<>(); + headers.put("Accept", "application/json"); + headers.put("Content-Type", "application/json"); + apiClient.PATCH(path, request, Void.class, headers); } } diff --git a/databricks-sdk-java/src/main/java/com/databricks/sdk/service/sharing/SharesImpl.java b/databricks-sdk-java/src/main/java/com/databricks/sdk/service/sharing/SharesImpl.java index ae6cb89d5..b95646c84 100755 --- a/databricks-sdk-java/src/main/java/com/databricks/sdk/service/sharing/SharesImpl.java +++ b/databricks-sdk-java/src/main/java/com/databricks/sdk/service/sharing/SharesImpl.java @@ -3,6 +3,8 @@ import com.databricks.sdk.core.ApiClient; import com.databricks.sdk.support.Generated; +import java.util.HashMap; +import java.util.Map; /** Package-local implementation of Shares */ @Generated @@ -16,43 +18,61 @@ public SharesImpl(ApiClient apiClient) { @Override public ShareInfo create(CreateShare request) { String path = "/api/2.1/unity-catalog/shares"; - return apiClient.POST(path, request, ShareInfo.class); + Map headers = new HashMap<>(); + headers.put("Accept", "application/json"); + headers.put("Content-Type", "application/json"); + return apiClient.POST(path, request, ShareInfo.class, headers); } @Override public void delete(DeleteShareRequest request) { String path = String.format("/api/2.1/unity-catalog/shares/%s", request.getName()); - apiClient.DELETE(path, request, Void.class); + Map headers = new HashMap<>(); + headers.put("Accept", "application/json"); + apiClient.DELETE(path, request, Void.class, headers); } @Override public ShareInfo get(GetShareRequest request) { String path = String.format("/api/2.1/unity-catalog/shares/%s", request.getName()); - return apiClient.GET(path, request, ShareInfo.class); + Map headers = new HashMap<>(); + headers.put("Accept", "application/json"); + return apiClient.GET(path, request, ShareInfo.class, headers); } @Override public ListSharesResponse list() { String path = "/api/2.1/unity-catalog/shares"; - return apiClient.GET(path, ListSharesResponse.class); + Map headers = new HashMap<>(); + headers.put("Accept", "application/json"); + return apiClient.GET(path, ListSharesResponse.class, headers); } @Override public com.databricks.sdk.service.catalog.PermissionsList sharePermissions( SharePermissionsRequest request) { String path = String.format("/api/2.1/unity-catalog/shares/%s/permissions", request.getName()); - return apiClient.GET(path, request, com.databricks.sdk.service.catalog.PermissionsList.class); + Map headers = new HashMap<>(); + headers.put("Accept", "application/json"); + return apiClient.GET( + path, request, com.databricks.sdk.service.catalog.PermissionsList.class, headers); } @Override public ShareInfo update(UpdateShare request) { String path = String.format("/api/2.1/unity-catalog/shares/%s", request.getName()); - return apiClient.PATCH(path, request, ShareInfo.class); + Map headers = new HashMap<>(); + headers.put("Accept", "application/json"); + headers.put("Content-Type", "application/json"); + return apiClient.PATCH(path, request, ShareInfo.class, headers); } @Override public void updatePermissions(UpdateSharePermissions request) { String path = String.format("/api/2.1/unity-catalog/shares/%s/permissions", request.getName()); - apiClient.PATCH(path, request, Void.class); + Map headers = new HashMap<>(); + headers.put("Accept", "application/json"); + headers.put("Content-Type", "application/json"); + apiClient.PATCH(path, request, Void.class, headers); } } diff --git a/databricks-sdk-java/src/main/java/com/databricks/sdk/service/sql/AlertsImpl.java b/databricks-sdk-java/src/main/java/com/databricks/sdk/service/sql/AlertsImpl.java index 0452f20ef..365333011 100755 --- a/databricks-sdk-java/src/main/java/com/databricks/sdk/service/sql/AlertsImpl.java +++ b/databricks-sdk-java/src/main/java/com/databricks/sdk/service/sql/AlertsImpl.java @@ -4,6 +4,8 @@ import com.databricks.sdk.core.ApiClient; import com.databricks.sdk.support.Generated; import java.util.Collection; +import java.util.HashMap; +import java.util.Map; /** Package-local implementation of Alerts */ @Generated @@ -17,30 +19,42 @@ public AlertsImpl(ApiClient apiClient) { @Override public Alert create(CreateAlert request) { String path = "/api/2.0/preview/sql/alerts"; - return apiClient.POST(path, request, Alert.class); + Map headers = new HashMap<>(); + headers.put("Accept", "application/json"); + headers.put("Content-Type", "application/json"); + return apiClient.POST(path, request, Alert.class, headers); } @Override public void delete(DeleteAlertRequest request) { String path = String.format("/api/2.0/preview/sql/alerts/%s", request.getAlertId()); - apiClient.DELETE(path, request, Void.class); + Map headers = new HashMap<>(); + headers.put("Accept", "application/json"); + apiClient.DELETE(path, request, Void.class, headers); } @Override public Alert get(GetAlertRequest request) { String path = String.format("/api/2.0/preview/sql/alerts/%s", request.getAlertId()); - return apiClient.GET(path, request, Alert.class); + Map headers = new HashMap<>(); + headers.put("Accept", "application/json"); + return apiClient.GET(path, request, Alert.class, headers); } @Override public Collection list() { String path = "/api/2.0/preview/sql/alerts"; - return apiClient.getCollection(path, null, Alert.class); + Map headers = new HashMap<>(); + headers.put("Accept", "application/json"); + return apiClient.getCollection(path, null, Alert.class, headers); } @Override public void update(EditAlert request) { String path = String.format("/api/2.0/preview/sql/alerts/%s", request.getAlertId()); - apiClient.PUT(path, request, Void.class); + Map headers = new HashMap<>(); + headers.put("Accept", "application/json"); + headers.put("Content-Type", "application/json"); + apiClient.PUT(path, request, Void.class, headers); } } diff --git a/databricks-sdk-java/src/main/java/com/databricks/sdk/service/sql/ChannelName.java b/databricks-sdk-java/src/main/java/com/databricks/sdk/service/sql/ChannelName.java index 3b9d4cbb0..8e0f647a1 100755 --- a/databricks-sdk-java/src/main/java/com/databricks/sdk/service/sql/ChannelName.java +++ b/databricks-sdk-java/src/main/java/com/databricks/sdk/service/sql/ChannelName.java @@ -4,6 +4,7 @@ import com.databricks.sdk.support.Generated; +/** Name of the channel */ @Generated public enum ChannelName { CHANNEL_NAME_CURRENT, diff --git a/databricks-sdk-java/src/main/java/com/databricks/sdk/service/sql/DashboardsImpl.java b/databricks-sdk-java/src/main/java/com/databricks/sdk/service/sql/DashboardsImpl.java index 04a291165..be0cd0c80 100755 --- a/databricks-sdk-java/src/main/java/com/databricks/sdk/service/sql/DashboardsImpl.java +++ b/databricks-sdk-java/src/main/java/com/databricks/sdk/service/sql/DashboardsImpl.java @@ -3,6 +3,8 @@ import com.databricks.sdk.core.ApiClient; import com.databricks.sdk.support.Generated; +import java.util.HashMap; +import java.util.Map; /** Package-local implementation of Dashboards */ @Generated @@ -16,31 +18,42 @@ public DashboardsImpl(ApiClient apiClient) { @Override public Dashboard create(CreateDashboardRequest request) { String path = "/api/2.0/preview/sql/dashboards"; - return apiClient.POST(path, request, Dashboard.class); + Map headers = new HashMap<>(); + headers.put("Accept", "application/json"); + headers.put("Content-Type", "application/json"); + return apiClient.POST(path, request, Dashboard.class, headers); } @Override public void delete(DeleteDashboardRequest request) { String path = String.format("/api/2.0/preview/sql/dashboards/%s", request.getDashboardId()); - apiClient.DELETE(path, request, Void.class); + Map headers = new HashMap<>(); + headers.put("Accept", "application/json"); + apiClient.DELETE(path, request, Void.class, headers); } @Override public Dashboard get(GetDashboardRequest request) { String path = String.format("/api/2.0/preview/sql/dashboards/%s", request.getDashboardId()); - return apiClient.GET(path, request, Dashboard.class); + Map headers = new HashMap<>(); + headers.put("Accept", "application/json"); + return apiClient.GET(path, request, Dashboard.class, headers); } @Override public ListResponse list(ListDashboardsRequest request) { String path = "/api/2.0/preview/sql/dashboards"; - return apiClient.GET(path, request, ListResponse.class); + Map headers = new HashMap<>(); + headers.put("Accept", "application/json"); + return apiClient.GET(path, request, ListResponse.class, headers); } @Override public void restore(RestoreDashboardRequest request) { String path = String.format("/api/2.0/preview/sql/dashboards/trash/%s", request.getDashboardId()); - apiClient.POST(path, request, Void.class); + Map headers = new HashMap<>(); + headers.put("Accept", "application/json"); + apiClient.POST(path, request, Void.class, headers); } } diff --git a/databricks-sdk-java/src/main/java/com/databricks/sdk/service/sql/DataSourcesImpl.java b/databricks-sdk-java/src/main/java/com/databricks/sdk/service/sql/DataSourcesImpl.java index 7a373f3ce..13a373ff9 100755 --- a/databricks-sdk-java/src/main/java/com/databricks/sdk/service/sql/DataSourcesImpl.java +++ b/databricks-sdk-java/src/main/java/com/databricks/sdk/service/sql/DataSourcesImpl.java @@ -4,6 +4,8 @@ import com.databricks.sdk.core.ApiClient; import com.databricks.sdk.support.Generated; import java.util.Collection; +import java.util.HashMap; +import java.util.Map; /** Package-local implementation of DataSources */ @Generated @@ -17,6 +19,8 @@ public DataSourcesImpl(ApiClient apiClient) { @Override public Collection list() { String path = "/api/2.0/preview/sql/data_sources"; - return apiClient.getCollection(path, null, DataSource.class); + Map headers = new HashMap<>(); + headers.put("Accept", "application/json"); + return apiClient.getCollection(path, null, DataSource.class, headers); } } diff --git a/databricks-sdk-java/src/main/java/com/databricks/sdk/service/sql/DbsqlPermissionsImpl.java b/databricks-sdk-java/src/main/java/com/databricks/sdk/service/sql/DbsqlPermissionsImpl.java index b4659a5df..a49470dbc 100755 --- a/databricks-sdk-java/src/main/java/com/databricks/sdk/service/sql/DbsqlPermissionsImpl.java +++ b/databricks-sdk-java/src/main/java/com/databricks/sdk/service/sql/DbsqlPermissionsImpl.java @@ -3,6 +3,8 @@ import com.databricks.sdk.core.ApiClient; import com.databricks.sdk.support.Generated; +import java.util.HashMap; +import java.util.Map; /** Package-local implementation of DbsqlPermissions */ @Generated @@ -19,7 +21,9 @@ public GetResponse get(GetDbsqlPermissionRequest request) { String.format( "/api/2.0/preview/sql/permissions/%s/%s", request.getObjectType(), request.getObjectId()); - return apiClient.GET(path, request, GetResponse.class); + Map headers = new HashMap<>(); + headers.put("Accept", "application/json"); + return apiClient.GET(path, request, GetResponse.class, headers); } @Override @@ -28,7 +32,10 @@ public SetResponse set(SetRequest request) { String.format( "/api/2.0/preview/sql/permissions/%s/%s", request.getObjectType(), request.getObjectId()); - return apiClient.POST(path, request, SetResponse.class); + Map headers = new HashMap<>(); + headers.put("Accept", "application/json"); + headers.put("Content-Type", "application/json"); + return apiClient.POST(path, request, SetResponse.class, headers); } @Override @@ -37,6 +44,9 @@ public Success transferOwnership(TransferOwnershipRequest request) { String.format( "/api/2.0/preview/sql/permissions/%s/%s/transfer", request.getObjectType(), request.getObjectId()); - return apiClient.POST(path, request, Success.class); + Map headers = new HashMap<>(); + headers.put("Accept", "application/json"); + headers.put("Content-Type", "application/json"); + return apiClient.POST(path, request, Success.class, headers); } } diff --git a/databricks-sdk-java/src/main/java/com/databricks/sdk/service/sql/QueriesImpl.java b/databricks-sdk-java/src/main/java/com/databricks/sdk/service/sql/QueriesImpl.java index 2ae271c9b..91951e493 100755 --- a/databricks-sdk-java/src/main/java/com/databricks/sdk/service/sql/QueriesImpl.java +++ b/databricks-sdk-java/src/main/java/com/databricks/sdk/service/sql/QueriesImpl.java @@ -3,6 +3,8 @@ import com.databricks.sdk.core.ApiClient; import com.databricks.sdk.support.Generated; +import java.util.HashMap; +import java.util.Map; /** Package-local implementation of Queries */ @Generated @@ -16,36 +18,50 @@ public QueriesImpl(ApiClient apiClient) { @Override public Query create(QueryPostContent request) { String path = "/api/2.0/preview/sql/queries"; - return apiClient.POST(path, request, Query.class); + Map headers = new HashMap<>(); + headers.put("Accept", "application/json"); + headers.put("Content-Type", "application/json"); + return apiClient.POST(path, request, Query.class, headers); } @Override public void delete(DeleteQueryRequest request) { String path = String.format("/api/2.0/preview/sql/queries/%s", request.getQueryId()); - apiClient.DELETE(path, request, Void.class); + Map headers = new HashMap<>(); + headers.put("Accept", "application/json"); + apiClient.DELETE(path, request, Void.class, headers); } @Override public Query get(GetQueryRequest request) { String path = String.format("/api/2.0/preview/sql/queries/%s", request.getQueryId()); - return apiClient.GET(path, request, Query.class); + Map headers = new HashMap<>(); + headers.put("Accept", "application/json"); + return apiClient.GET(path, request, Query.class, headers); } @Override public QueryList list(ListQueriesRequest request) { String path = "/api/2.0/preview/sql/queries"; - return apiClient.GET(path, request, QueryList.class); + Map headers = new HashMap<>(); + headers.put("Accept", "application/json"); + return apiClient.GET(path, request, QueryList.class, headers); } @Override public void restore(RestoreQueryRequest request) { String path = String.format("/api/2.0/preview/sql/queries/trash/%s", request.getQueryId()); - apiClient.POST(path, request, Void.class); + Map headers = new HashMap<>(); + headers.put("Accept", "application/json"); + apiClient.POST(path, request, Void.class, headers); } @Override public Query update(QueryEditContent request) { String path = String.format("/api/2.0/preview/sql/queries/%s", request.getQueryId()); - return apiClient.POST(path, request, Query.class); + Map headers = new HashMap<>(); + headers.put("Accept", "application/json"); + headers.put("Content-Type", "application/json"); + return apiClient.POST(path, request, Query.class, headers); } } diff --git a/databricks-sdk-java/src/main/java/com/databricks/sdk/service/sql/QueryHistoryImpl.java b/databricks-sdk-java/src/main/java/com/databricks/sdk/service/sql/QueryHistoryImpl.java index 73d2436c2..1a07ce143 100755 --- a/databricks-sdk-java/src/main/java/com/databricks/sdk/service/sql/QueryHistoryImpl.java +++ b/databricks-sdk-java/src/main/java/com/databricks/sdk/service/sql/QueryHistoryImpl.java @@ -3,6 +3,8 @@ import com.databricks.sdk.core.ApiClient; import com.databricks.sdk.support.Generated; +import java.util.HashMap; +import java.util.Map; /** Package-local implementation of QueryHistory */ @Generated @@ -16,6 +18,8 @@ public QueryHistoryImpl(ApiClient apiClient) { @Override public ListQueriesResponse list(ListQueryHistoryRequest request) { String path = "/api/2.0/sql/history/queries"; - return apiClient.GET(path, request, ListQueriesResponse.class); + Map headers = new HashMap<>(); + headers.put("Accept", "application/json"); + return apiClient.GET(path, request, ListQueriesResponse.class, headers); } } diff --git a/databricks-sdk-java/src/main/java/com/databricks/sdk/service/sql/StatementExecutionImpl.java b/databricks-sdk-java/src/main/java/com/databricks/sdk/service/sql/StatementExecutionImpl.java index 195d57113..d0955f3eb 100755 --- a/databricks-sdk-java/src/main/java/com/databricks/sdk/service/sql/StatementExecutionImpl.java +++ b/databricks-sdk-java/src/main/java/com/databricks/sdk/service/sql/StatementExecutionImpl.java @@ -3,6 +3,8 @@ import com.databricks.sdk.core.ApiClient; import com.databricks.sdk.support.Generated; +import java.util.HashMap; +import java.util.Map; /** Package-local implementation of StatementExecution */ @Generated @@ -16,19 +18,25 @@ public StatementExecutionImpl(ApiClient apiClient) { @Override public void cancelExecution(CancelExecutionRequest request) { String path = String.format("/api/2.0/sql/statements/%s/cancel", request.getStatementId()); - apiClient.POST(path, request, Void.class); + Map headers = new HashMap<>(); + apiClient.POST(path, request, Void.class, headers); } @Override public ExecuteStatementResponse executeStatement(ExecuteStatementRequest request) { String path = "/api/2.0/sql/statements/"; - return apiClient.POST(path, request, ExecuteStatementResponse.class); + Map headers = new HashMap<>(); + headers.put("Accept", "application/json"); + headers.put("Content-Type", "application/json"); + return apiClient.POST(path, request, ExecuteStatementResponse.class, headers); } @Override public GetStatementResponse getStatement(GetStatementRequest request) { String path = String.format("/api/2.0/sql/statements/%s", request.getStatementId()); - return apiClient.GET(path, request, GetStatementResponse.class); + Map headers = new HashMap<>(); + headers.put("Accept", "application/json"); + return apiClient.GET(path, request, GetStatementResponse.class, headers); } @Override @@ -37,6 +45,8 @@ public ResultData getStatementResultChunkN(GetStatementResultChunkNRequest reque String.format( "/api/2.0/sql/statements/%s/result/chunks/%s", request.getStatementId(), request.getChunkIndex()); - return apiClient.GET(path, request, ResultData.class); + Map headers = new HashMap<>(); + headers.put("Accept", "application/json"); + return apiClient.GET(path, request, ResultData.class, headers); } } diff --git a/databricks-sdk-java/src/main/java/com/databricks/sdk/service/sql/WarehousesImpl.java b/databricks-sdk-java/src/main/java/com/databricks/sdk/service/sql/WarehousesImpl.java index f28abaa7c..77de24693 100755 --- a/databricks-sdk-java/src/main/java/com/databricks/sdk/service/sql/WarehousesImpl.java +++ b/databricks-sdk-java/src/main/java/com/databricks/sdk/service/sql/WarehousesImpl.java @@ -3,6 +3,8 @@ import com.databricks.sdk.core.ApiClient; import com.databricks.sdk.support.Generated; +import java.util.HashMap; +import java.util.Map; /** Package-local implementation of Warehouses */ @Generated @@ -16,25 +18,35 @@ public WarehousesImpl(ApiClient apiClient) { @Override public CreateWarehouseResponse create(CreateWarehouseRequest request) { String path = "/api/2.0/sql/warehouses"; - return apiClient.POST(path, request, CreateWarehouseResponse.class); + Map headers = new HashMap<>(); + headers.put("Accept", "application/json"); + headers.put("Content-Type", "application/json"); + return apiClient.POST(path, request, CreateWarehouseResponse.class, headers); } @Override public void delete(DeleteWarehouseRequest request) { String path = String.format("/api/2.0/sql/warehouses/%s", request.getId()); - apiClient.DELETE(path, request, Void.class); + Map headers = new HashMap<>(); + headers.put("Accept", "application/json"); + apiClient.DELETE(path, request, Void.class, headers); } @Override public void edit(EditWarehouseRequest request) { String path = String.format("/api/2.0/sql/warehouses/%s/edit", request.getId()); - apiClient.POST(path, request, Void.class); + Map headers = new HashMap<>(); + headers.put("Accept", "application/json"); + headers.put("Content-Type", "application/json"); + apiClient.POST(path, request, Void.class, headers); } @Override public GetWarehouseResponse get(GetWarehouseRequest request) { String path = String.format("/api/2.0/sql/warehouses/%s", request.getId()); - return apiClient.GET(path, request, GetWarehouseResponse.class); + Map headers = new HashMap<>(); + headers.put("Accept", "application/json"); + return apiClient.GET(path, request, GetWarehouseResponse.class, headers); } @Override @@ -43,54 +55,75 @@ public GetWarehousePermissionLevelsResponse getWarehousePermissionLevels( String path = String.format( "/api/2.0/permissions/warehouses/%s/permissionLevels", request.getWarehouseId()); - return apiClient.GET(path, request, GetWarehousePermissionLevelsResponse.class); + Map headers = new HashMap<>(); + headers.put("Accept", "application/json"); + return apiClient.GET(path, request, GetWarehousePermissionLevelsResponse.class, headers); } @Override public WarehousePermissions getWarehousePermissions(GetWarehousePermissionsRequest request) { String path = String.format("/api/2.0/permissions/warehouses/%s", request.getWarehouseId()); - return apiClient.GET(path, request, WarehousePermissions.class); + Map headers = new HashMap<>(); + headers.put("Accept", "application/json"); + return apiClient.GET(path, request, WarehousePermissions.class, headers); } @Override public GetWorkspaceWarehouseConfigResponse getWorkspaceWarehouseConfig() { String path = "/api/2.0/sql/config/warehouses"; - return apiClient.GET(path, GetWorkspaceWarehouseConfigResponse.class); + Map headers = new HashMap<>(); + headers.put("Accept", "application/json"); + return apiClient.GET(path, GetWorkspaceWarehouseConfigResponse.class, headers); } @Override public ListWarehousesResponse list(ListWarehousesRequest request) { String path = "/api/2.0/sql/warehouses"; - return apiClient.GET(path, request, ListWarehousesResponse.class); + Map headers = new HashMap<>(); + headers.put("Accept", "application/json"); + return apiClient.GET(path, request, ListWarehousesResponse.class, headers); } @Override public WarehousePermissions setWarehousePermissions(WarehousePermissionsRequest request) { String path = String.format("/api/2.0/permissions/warehouses/%s", request.getWarehouseId()); - return apiClient.PUT(path, request, WarehousePermissions.class); + Map headers = new HashMap<>(); + headers.put("Accept", "application/json"); + headers.put("Content-Type", "application/json"); + return apiClient.PUT(path, request, WarehousePermissions.class, headers); } @Override public void setWorkspaceWarehouseConfig(SetWorkspaceWarehouseConfigRequest request) { String path = "/api/2.0/sql/config/warehouses"; - apiClient.PUT(path, request, Void.class); + Map headers = new HashMap<>(); + headers.put("Accept", "application/json"); + headers.put("Content-Type", "application/json"); + apiClient.PUT(path, request, Void.class, headers); } @Override public void start(StartRequest request) { String path = String.format("/api/2.0/sql/warehouses/%s/start", request.getId()); - apiClient.POST(path, request, Void.class); + Map headers = new HashMap<>(); + headers.put("Accept", "application/json"); + apiClient.POST(path, request, Void.class, headers); } @Override public void stop(StopRequest request) { String path = String.format("/api/2.0/sql/warehouses/%s/stop", request.getId()); - apiClient.POST(path, request, Void.class); + Map headers = new HashMap<>(); + headers.put("Accept", "application/json"); + apiClient.POST(path, request, Void.class, headers); } @Override public WarehousePermissions updateWarehousePermissions(WarehousePermissionsRequest request) { String path = String.format("/api/2.0/permissions/warehouses/%s", request.getWarehouseId()); - return apiClient.PATCH(path, request, WarehousePermissions.class); + Map headers = new HashMap<>(); + headers.put("Accept", "application/json"); + headers.put("Content-Type", "application/json"); + return apiClient.PATCH(path, request, WarehousePermissions.class, headers); } } diff --git a/databricks-sdk-java/src/main/java/com/databricks/sdk/service/workspace/GitCredentialsImpl.java b/databricks-sdk-java/src/main/java/com/databricks/sdk/service/workspace/GitCredentialsImpl.java index 85838459e..556130d6f 100755 --- a/databricks-sdk-java/src/main/java/com/databricks/sdk/service/workspace/GitCredentialsImpl.java +++ b/databricks-sdk-java/src/main/java/com/databricks/sdk/service/workspace/GitCredentialsImpl.java @@ -3,6 +3,8 @@ import com.databricks.sdk.core.ApiClient; import com.databricks.sdk.support.Generated; +import java.util.HashMap; +import java.util.Map; /** Package-local implementation of GitCredentials */ @Generated @@ -16,30 +18,41 @@ public GitCredentialsImpl(ApiClient apiClient) { @Override public CreateCredentialsResponse create(CreateCredentials request) { String path = "/api/2.0/git-credentials"; - return apiClient.POST(path, request, CreateCredentialsResponse.class); + Map headers = new HashMap<>(); + headers.put("Accept", "application/json"); + headers.put("Content-Type", "application/json"); + return apiClient.POST(path, request, CreateCredentialsResponse.class, headers); } @Override public void delete(DeleteGitCredentialRequest request) { String path = String.format("/api/2.0/git-credentials/%s", request.getCredentialId()); - apiClient.DELETE(path, request, Void.class); + Map headers = new HashMap<>(); + apiClient.DELETE(path, request, Void.class, headers); } @Override public CredentialInfo get(GetGitCredentialRequest request) { String path = String.format("/api/2.0/git-credentials/%s", request.getCredentialId()); - return apiClient.GET(path, request, CredentialInfo.class); + Map headers = new HashMap<>(); + headers.put("Accept", "application/json"); + return apiClient.GET(path, request, CredentialInfo.class, headers); } @Override public GetCredentialsResponse list() { String path = "/api/2.0/git-credentials"; - return apiClient.GET(path, GetCredentialsResponse.class); + Map headers = new HashMap<>(); + headers.put("Accept", "application/json"); + return apiClient.GET(path, GetCredentialsResponse.class, headers); } @Override public void update(UpdateCredentials request) { String path = String.format("/api/2.0/git-credentials/%s", request.getCredentialId()); - apiClient.PATCH(path, request, Void.class); + Map headers = new HashMap<>(); + headers.put("Accept", "application/json"); + headers.put("Content-Type", "application/json"); + apiClient.PATCH(path, request, Void.class, headers); } } diff --git a/databricks-sdk-java/src/main/java/com/databricks/sdk/service/workspace/ReposImpl.java b/databricks-sdk-java/src/main/java/com/databricks/sdk/service/workspace/ReposImpl.java index cc11e840d..3bd7948a9 100755 --- a/databricks-sdk-java/src/main/java/com/databricks/sdk/service/workspace/ReposImpl.java +++ b/databricks-sdk-java/src/main/java/com/databricks/sdk/service/workspace/ReposImpl.java @@ -3,6 +3,8 @@ import com.databricks.sdk.core.ApiClient; import com.databricks.sdk.support.Generated; +import java.util.HashMap; +import java.util.Map; /** Package-local implementation of Repos */ @Generated @@ -16,19 +18,25 @@ public ReposImpl(ApiClient apiClient) { @Override public RepoInfo create(CreateRepo request) { String path = "/api/2.0/repos"; - return apiClient.POST(path, request, RepoInfo.class); + Map headers = new HashMap<>(); + headers.put("Accept", "application/json"); + headers.put("Content-Type", "application/json"); + return apiClient.POST(path, request, RepoInfo.class, headers); } @Override public void delete(DeleteRepoRequest request) { String path = String.format("/api/2.0/repos/%s", request.getRepoId()); - apiClient.DELETE(path, request, Void.class); + Map headers = new HashMap<>(); + apiClient.DELETE(path, request, Void.class, headers); } @Override public RepoInfo get(GetRepoRequest request) { String path = String.format("/api/2.0/repos/%s", request.getRepoId()); - return apiClient.GET(path, request, RepoInfo.class); + Map headers = new HashMap<>(); + headers.put("Accept", "application/json"); + return apiClient.GET(path, request, RepoInfo.class, headers); } @Override @@ -36,36 +44,51 @@ public GetRepoPermissionLevelsResponse getRepoPermissionLevels( GetRepoPermissionLevelsRequest request) { String path = String.format("/api/2.0/permissions/repos/%s/permissionLevels", request.getRepoId()); - return apiClient.GET(path, request, GetRepoPermissionLevelsResponse.class); + Map headers = new HashMap<>(); + headers.put("Accept", "application/json"); + return apiClient.GET(path, request, GetRepoPermissionLevelsResponse.class, headers); } @Override public RepoPermissions getRepoPermissions(GetRepoPermissionsRequest request) { String path = String.format("/api/2.0/permissions/repos/%s", request.getRepoId()); - return apiClient.GET(path, request, RepoPermissions.class); + Map headers = new HashMap<>(); + headers.put("Accept", "application/json"); + return apiClient.GET(path, request, RepoPermissions.class, headers); } @Override public ListReposResponse list(ListReposRequest request) { String path = "/api/2.0/repos"; - return apiClient.GET(path, request, ListReposResponse.class); + Map headers = new HashMap<>(); + headers.put("Accept", "application/json"); + return apiClient.GET(path, request, ListReposResponse.class, headers); } @Override public RepoPermissions setRepoPermissions(RepoPermissionsRequest request) { String path = String.format("/api/2.0/permissions/repos/%s", request.getRepoId()); - return apiClient.PUT(path, request, RepoPermissions.class); + Map headers = new HashMap<>(); + headers.put("Accept", "application/json"); + headers.put("Content-Type", "application/json"); + return apiClient.PUT(path, request, RepoPermissions.class, headers); } @Override public void update(UpdateRepo request) { String path = String.format("/api/2.0/repos/%s", request.getRepoId()); - apiClient.PATCH(path, request, Void.class); + Map headers = new HashMap<>(); + headers.put("Accept", "application/json"); + headers.put("Content-Type", "application/json"); + apiClient.PATCH(path, request, Void.class, headers); } @Override public RepoPermissions updateRepoPermissions(RepoPermissionsRequest request) { String path = String.format("/api/2.0/permissions/repos/%s", request.getRepoId()); - return apiClient.PATCH(path, request, RepoPermissions.class); + Map headers = new HashMap<>(); + headers.put("Accept", "application/json"); + headers.put("Content-Type", "application/json"); + return apiClient.PATCH(path, request, RepoPermissions.class, headers); } } diff --git a/databricks-sdk-java/src/main/java/com/databricks/sdk/service/workspace/SecretsImpl.java b/databricks-sdk-java/src/main/java/com/databricks/sdk/service/workspace/SecretsImpl.java index b4ea8b377..ab31bf60e 100755 --- a/databricks-sdk-java/src/main/java/com/databricks/sdk/service/workspace/SecretsImpl.java +++ b/databricks-sdk-java/src/main/java/com/databricks/sdk/service/workspace/SecretsImpl.java @@ -3,6 +3,8 @@ import com.databricks.sdk.core.ApiClient; import com.databricks.sdk.support.Generated; +import java.util.HashMap; +import java.util.Map; /** Package-local implementation of Secrets */ @Generated @@ -16,66 +18,94 @@ public SecretsImpl(ApiClient apiClient) { @Override public void createScope(CreateScope request) { String path = "/api/2.0/secrets/scopes/create"; - apiClient.POST(path, request, Void.class); + Map headers = new HashMap<>(); + headers.put("Accept", "application/json"); + headers.put("Content-Type", "application/json"); + apiClient.POST(path, request, Void.class, headers); } @Override public void deleteAcl(DeleteAcl request) { String path = "/api/2.0/secrets/acls/delete"; - apiClient.POST(path, request, Void.class); + Map headers = new HashMap<>(); + headers.put("Accept", "application/json"); + headers.put("Content-Type", "application/json"); + apiClient.POST(path, request, Void.class, headers); } @Override public void deleteScope(DeleteScope request) { String path = "/api/2.0/secrets/scopes/delete"; - apiClient.POST(path, request, Void.class); + Map headers = new HashMap<>(); + headers.put("Accept", "application/json"); + headers.put("Content-Type", "application/json"); + apiClient.POST(path, request, Void.class, headers); } @Override public void deleteSecret(DeleteSecret request) { String path = "/api/2.0/secrets/delete"; - apiClient.POST(path, request, Void.class); + Map headers = new HashMap<>(); + headers.put("Accept", "application/json"); + headers.put("Content-Type", "application/json"); + apiClient.POST(path, request, Void.class, headers); } @Override public AclItem getAcl(GetAclRequest request) { String path = "/api/2.0/secrets/acls/get"; - return apiClient.GET(path, request, AclItem.class); + Map headers = new HashMap<>(); + headers.put("Accept", "application/json"); + return apiClient.GET(path, request, AclItem.class, headers); } @Override public GetSecretResponse getSecret(GetSecretRequest request) { String path = "/api/2.0/secrets/get"; - return apiClient.GET(path, request, GetSecretResponse.class); + Map headers = new HashMap<>(); + headers.put("Accept", "application/json"); + return apiClient.GET(path, request, GetSecretResponse.class, headers); } @Override public ListAclsResponse listAcls(ListAclsRequest request) { String path = "/api/2.0/secrets/acls/list"; - return apiClient.GET(path, request, ListAclsResponse.class); + Map headers = new HashMap<>(); + headers.put("Accept", "application/json"); + return apiClient.GET(path, request, ListAclsResponse.class, headers); } @Override public ListScopesResponse listScopes() { String path = "/api/2.0/secrets/scopes/list"; - return apiClient.GET(path, ListScopesResponse.class); + Map headers = new HashMap<>(); + headers.put("Accept", "application/json"); + return apiClient.GET(path, ListScopesResponse.class, headers); } @Override public ListSecretsResponse listSecrets(ListSecretsRequest request) { String path = "/api/2.0/secrets/list"; - return apiClient.GET(path, request, ListSecretsResponse.class); + Map headers = new HashMap<>(); + headers.put("Accept", "application/json"); + return apiClient.GET(path, request, ListSecretsResponse.class, headers); } @Override public void putAcl(PutAcl request) { String path = "/api/2.0/secrets/acls/put"; - apiClient.POST(path, request, Void.class); + Map headers = new HashMap<>(); + headers.put("Accept", "application/json"); + headers.put("Content-Type", "application/json"); + apiClient.POST(path, request, Void.class, headers); } @Override public void putSecret(PutSecret request) { String path = "/api/2.0/secrets/put"; - apiClient.POST(path, request, Void.class); + Map headers = new HashMap<>(); + headers.put("Accept", "application/json"); + headers.put("Content-Type", "application/json"); + apiClient.POST(path, request, Void.class, headers); } } diff --git a/databricks-sdk-java/src/main/java/com/databricks/sdk/service/workspace/WorkspaceImpl.java b/databricks-sdk-java/src/main/java/com/databricks/sdk/service/workspace/WorkspaceImpl.java index df66c495d..b6310eb57 100755 --- a/databricks-sdk-java/src/main/java/com/databricks/sdk/service/workspace/WorkspaceImpl.java +++ b/databricks-sdk-java/src/main/java/com/databricks/sdk/service/workspace/WorkspaceImpl.java @@ -3,6 +3,8 @@ import com.databricks.sdk.core.ApiClient; import com.databricks.sdk.support.Generated; +import java.util.HashMap; +import java.util.Map; /** Package-local implementation of Workspace */ @Generated @@ -16,19 +18,26 @@ public WorkspaceImpl(ApiClient apiClient) { @Override public void delete(Delete request) { String path = "/api/2.0/workspace/delete"; - apiClient.POST(path, request, Void.class); + Map headers = new HashMap<>(); + headers.put("Accept", "application/json"); + headers.put("Content-Type", "application/json"); + apiClient.POST(path, request, Void.class, headers); } @Override public ExportResponse export(ExportRequest request) { String path = "/api/2.0/workspace/export"; - return apiClient.GET(path, request, ExportResponse.class); + Map headers = new HashMap<>(); + headers.put("Accept", "application/json"); + return apiClient.GET(path, request, ExportResponse.class, headers); } @Override public ObjectInfo getStatus(GetStatusRequest request) { String path = "/api/2.0/workspace/get-status"; - return apiClient.GET(path, request, ObjectInfo.class); + Map headers = new HashMap<>(); + headers.put("Accept", "application/json"); + return apiClient.GET(path, request, ObjectInfo.class, headers); } @Override @@ -38,7 +47,9 @@ public GetWorkspaceObjectPermissionLevelsResponse getWorkspaceObjectPermissionLe String.format( "/api/2.0/permissions/%s/%s/permissionLevels", request.getWorkspaceObjectType(), request.getWorkspaceObjectId()); - return apiClient.GET(path, request, GetWorkspaceObjectPermissionLevelsResponse.class); + Map headers = new HashMap<>(); + headers.put("Accept", "application/json"); + return apiClient.GET(path, request, GetWorkspaceObjectPermissionLevelsResponse.class, headers); } @Override @@ -48,25 +59,35 @@ public WorkspaceObjectPermissions getWorkspaceObjectPermissions( String.format( "/api/2.0/permissions/%s/%s", request.getWorkspaceObjectType(), request.getWorkspaceObjectId()); - return apiClient.GET(path, request, WorkspaceObjectPermissions.class); + Map headers = new HashMap<>(); + headers.put("Accept", "application/json"); + return apiClient.GET(path, request, WorkspaceObjectPermissions.class, headers); } @Override public void importContent(Import request) { String path = "/api/2.0/workspace/import"; - apiClient.POST(path, request, Void.class); + Map headers = new HashMap<>(); + headers.put("Accept", "application/json"); + headers.put("Content-Type", "application/json"); + apiClient.POST(path, request, Void.class, headers); } @Override public ListResponse list(ListWorkspaceRequest request) { String path = "/api/2.0/workspace/list"; - return apiClient.GET(path, request, ListResponse.class); + Map headers = new HashMap<>(); + headers.put("Accept", "application/json"); + return apiClient.GET(path, request, ListResponse.class, headers); } @Override public void mkdirs(Mkdirs request) { String path = "/api/2.0/workspace/mkdirs"; - apiClient.POST(path, request, Void.class); + Map headers = new HashMap<>(); + headers.put("Accept", "application/json"); + headers.put("Content-Type", "application/json"); + apiClient.POST(path, request, Void.class, headers); } @Override @@ -76,7 +97,10 @@ public WorkspaceObjectPermissions setWorkspaceObjectPermissions( String.format( "/api/2.0/permissions/%s/%s", request.getWorkspaceObjectType(), request.getWorkspaceObjectId()); - return apiClient.PUT(path, request, WorkspaceObjectPermissions.class); + Map headers = new HashMap<>(); + headers.put("Accept", "application/json"); + headers.put("Content-Type", "application/json"); + return apiClient.PUT(path, request, WorkspaceObjectPermissions.class, headers); } @Override @@ -86,6 +110,9 @@ public WorkspaceObjectPermissions updateWorkspaceObjectPermissions( String.format( "/api/2.0/permissions/%s/%s", request.getWorkspaceObjectType(), request.getWorkspaceObjectId()); - return apiClient.PATCH(path, request, WorkspaceObjectPermissions.class); + Map headers = new HashMap<>(); + headers.put("Accept", "application/json"); + headers.put("Content-Type", "application/json"); + return apiClient.PATCH(path, request, WorkspaceObjectPermissions.class, headers); } } diff --git a/databricks-sdk-java/src/test/java/com/databricks/sdk/core/ApiClientTest.java b/databricks-sdk-java/src/test/java/com/databricks/sdk/core/ApiClientTest.java index 51987fc0f..52a9c6d5e 100644 --- a/databricks-sdk-java/src/test/java/com/databricks/sdk/core/ApiClientTest.java +++ b/databricks-sdk-java/src/test/java/com/databricks/sdk/core/ApiClientTest.java @@ -61,7 +61,7 @@ private void runApiClientTest( Class clazz, T expectedResponse) { ApiClient client = getApiClient(request, responses); - T response = client.GET(request.getUri().getPath(), clazz); + T response = client.GET(request.getUri().getPath(), clazz, Collections.emptyMap()); assertEquals(response, expectedResponse); } @@ -70,7 +70,8 @@ private void runFailingApiClientTest( ApiClient client = getApiClient(request, responses); DatabricksException exception = assertThrows( - DatabricksException.class, () -> client.GET(request.getUri().getPath(), clazz)); + DatabricksException.class, + () -> client.GET(request.getUri().getPath(), clazz, Collections.emptyMap())); assertEquals(exception.getMessage(), expectedMessage); }