[PECO-1542] Add support for proxy

.codegen.json

@@ -11,9 +11,14 @@
     ".codegen/api.java.tmpl": "databricks-sdk-java/src/main/java/com/databricks/sdk/service/{{.Package.Name}}/{{.PascalName}}API.java",
     ".codegen/impl.java.tmpl": "databricks-sdk-java/src/main/java/com/databricks/sdk/service/{{.Package.Name}}/{{.PascalName}}Impl.java"
   },
+  "exception_types": {
+    ".codegen/exception.java.tmpl": "databricks-sdk-java/src/main/java/com/databricks/sdk/core/error/platform/{{.PascalName}}.java"
+  },
   "batch": {
     ".codegen/workspace.java.tmpl": "databricks-sdk-java/src/main/java/com/databricks/sdk/WorkspaceClient.java",
-    ".codegen/account.java.tmpl": "databricks-sdk-java/src/main/java/com/databricks/sdk/AccountClient.java"
+    ".codegen/account.java.tmpl": "databricks-sdk-java/src/main/java/com/databricks/sdk/AccountClient.java",
+    ".codegen/error-mapper.java.tmpl": "databricks-sdk-java/src/main/java/com/databricks/sdk/core/error/ErrorMapper.java",
+    ".codegen/error-overrides.java.tmpl": "databricks-sdk-java/src/main/java/com/databricks/sdk/core/error/ErrorOverrides.java"
   },
   "version": {
     "pom.xml": "<artifactId>databricks-sdk-parent</artifactId>\n  <version>$VERSION</version>",

.codegen/_openapi_sha

@@ -1 +1 @@
-23fe1139476cb1a8c7b8ff2ff1fbc0132380f7a5
+06d330f43d92c1be864d4638c672cd0723e20a51

.codegen/api.java.tmpl

@@ -146,47 +146,55 @@ public class {{.PascalName}}API {
 {{- end}}
 
 {{define "method-call-paginated" -}}
-  {{- if .Pagination.MultiRequest -}}
-    {{- if and .Pagination.Offset (not (eq .Path "/api/2.0/clusters/events")) -}}
-    request.set{{.Pagination.Offset.PascalName}}(
-    {{- if eq .Pagination.Increment 1 -}}
-      1
-    {{- else if contains .Path "/scim/v2/" -}}
-      1
-    {{- else -}}
-      0
-    {{- end}}L);{{end -}}
-    {{if and .Pagination.Limit (contains .Path "/scim/v2/")}}
-    if (request.get{{.Pagination.Limit.PascalName}}() == null) {
-       request.set{{.Pagination.Limit.PascalName}}(100L);
-    }{{end -}}
-    return new Paginator<>(request, impl::{{template "java-name" .}}, {{template "type" .Response}}::get{{.Pagination.Results.PascalName}}, response -> {
-      {{if eq .Path "/api/2.0/clusters/events" -}}
-      return response.getNextPage();
+  {{- if and .Pagination.Offset (not (eq .Path "/api/2.0/clusters/events")) -}}
+  request.set{{.Pagination.Offset.PascalName}}(
+  {{- if eq .Pagination.Increment 1 -}}
+    1
+  {{- else if contains .Path "/scim/v2/" -}}
+    1
+  {{- else -}}
+    0
+  {{- end}}L);{{end -}}
+  {{if and .Pagination.Limit (contains .Path "/scim/v2/")}}
+  if (request.get{{.Pagination.Limit.PascalName}}() == null) {
+     request.set{{.Pagination.Limit.PascalName}}(100L);
+  }{{end -}}
+  return new Paginator<>(
+    {{ if .Request }}request{{ else }}null{{ end }},
+    {{ if .Request }}impl::{{template "java-name" .}}{{ else }}(Void v) -> impl.{{template "java-name" .}}(){{ end }},
+    {{template "type" .Response}}::get{{.Pagination.Results.PascalName}},
+    response ->
+      {{ if not .Pagination.MultiRequest }}
+      null
+      {{- else if eq .Path "/api/2.0/clusters/events" -}}
+      response.getNextPage()
       {{- else if .Pagination.Token -}}
-      String token = response.get{{.Pagination.Token.Bind.PascalName}}();
-      if (token == null) {
-        return null;
+      {
+        String token = response.get{{.Pagination.Token.Bind.PascalName}}();
+        if (token == null) {
+          return null;
+        }
+        return request.set{{.Pagination.Token.PollField.PascalName}}(token);
       }
-      return request.set{{.Pagination.Token.PollField.PascalName}}(token);
       {{- else if eq .Pagination.Increment 1 -}}
-      Long page = request.get{{.Pagination.Offset.PascalName}}();
-      if (page == null) {
-        page = 1L; // redash uses 1-based pagination
+      {
+        Long page = request.get{{.Pagination.Offset.PascalName}}();
+        if (page == null) {
+          page = 1L; // redash uses 1-based pagination
+        }
+        return request.set{{.Pagination.Offset.PascalName}}(page+1L);
       }
-      return request.set{{.Pagination.Offset.PascalName}}(page+1L);
       {{- else -}}
-      Long offset = request.get{{.Pagination.Offset.PascalName}}();
-      if (offset == null) {
-        offset = 0L;
+      {
+        Long offset = request.get{{.Pagination.Offset.PascalName}}();
+        if (offset == null) {
+          offset = 0L;
+        }
+        offset += response.get{{.Pagination.Results.PascalName}}().size();
+        return request.set{{.Pagination.Offset.PascalName}}(offset);
       }
-      offset += response.get{{.Pagination.Results.PascalName}}().size();
-      return request.set{{.Pagination.Offset.PascalName}}(offset);
       {{- end}}
-    }){{if .NeedsOffsetDedupe -}}.withDedupe({{.Pagination.Entity.PascalName}}::get{{.IdentifierField.PascalName}}){{end}};
-  {{- else -}}
-    return impl.{{template "java-name" .}}({{if .Request}}request{{end}}){{with .Pagination.Results}}.get{{.PascalName}}(){{end}};
-  {{- end -}}
+  ){{if .NeedsOffsetDedupe -}}.withDedupe({{.Pagination.Entity.PascalName}}::get{{.IdentifierField.PascalName}}){{end}};
 {{- end}}
 
 {{define "method-call-retried" -}}

.codegen/error-mapper.java.tmpl

@@ -0,0 +1,15 @@
+// Code generated from OpenAPI specs by Databricks SDK Generator. DO NOT EDIT.
+
+package com.databricks.sdk.core.error;
+
+import com.databricks.sdk.support.Generated;
+
+@Generated
+class ErrorMapper extends AbstractErrorMapper {
+  public ErrorMapper() {
+    {{range .ErrorStatusCodeMapping}}statusCode({{.StatusCode}}, com.databricks.sdk.core.error.platform.{{.PascalName}}::new);
+    {{end}}
+    {{range .ErrorCodeMapping}}errorCode("{{.ErrorCode}}", com.databricks.sdk.core.error.platform.{{.PascalName}}::new);
+    {{end}}
+  }
+}

.codegen/error-overrides.java.tmpl

@@ -0,0 +1,24 @@
+// Code generated from OpenAPI specs by Databricks SDK Generator. DO NOT EDIT.
+
+package com.databricks.sdk.core.error;
+
+import java.util.Arrays;
+import java.util.List;
+
+import com.databricks.sdk.support.Generated;
+
+@Generated
+class ErrorOverrides {
+  static final List<ErrorOverride<?>> ALL_OVERRIDES = Arrays.asList(
+{{- range $i, $x := .ErrorOverrides }}
+    {{if not (eq $i 0)}}, {{end}}new ErrorOverride<>(
+        "{{$x.Name}}",
+        "{{ replaceAll "\\" "\\\\" $x.PathRegex}}",
+        "{{$x.Verb}}",
+        "{{ replaceAll "\\" "\\\\" $x.StatusCodeMatcher}}",
+        "{{ replaceAll "\\" "\\\\" $x.ErrorCodeMatcher}}",
+        "{{ replaceAll "\\" "\\\\" $x.MessageMatcher}}",
+        com.databricks.sdk.core.error.platform.{{$x.OverrideErrorCode.PascalName}}.class)
+{{- end}}
+  );
+}

.codegen/exception.java.tmpl

@@ -0,0 +1,25 @@
+// Code generated from OpenAPI specs by Databricks SDK Generator. DO NOT EDIT.
+
+package com.databricks.sdk.core.error.platform;
+
+import com.databricks.sdk.core.DatabricksError;
+import com.databricks.sdk.core.error.ErrorDetail;
+import com.databricks.sdk.support.Generated;
+
+import java.util.List;
+
+/**
+ {{.Comment " * " 80}}
+ */
+@Generated
+public class {{.PascalName}} extends {{if .Inherit -}}{{.Inherit.PascalName}}{{else}}DatabricksError{{end}} {
+  public {{.PascalName}}(String message, List<ErrorDetail> details) {
+    super("{{.Name}}", message, {{if not .Inherit}}{{.StatusCode}}, {{end}}details);
+  }
+
+  {{if not .Inherit}}
+  public {{.PascalName}}(String errorCode, String message, List<ErrorDetail> details) {
+      super(errorCode, message, {{.StatusCode}}, details);
+  }
+  {{end}}
+}

.codegen/impl.java.tmpl

@@ -9,6 +9,7 @@ import java.util.HashMap;
 
 import com.databricks.sdk.core.ApiClient;
 import com.databricks.sdk.core.DatabricksException;
+import com.databricks.sdk.core.http.Encoding;
 import com.databricks.sdk.support.Generated;
 
 {{range .Package.ImportedEntities}}
@@ -25,9 +26,7 @@ class {{.PascalName}}Impl implements {{.PascalName}}Service {
 	{{range .Methods}}
   @Override
   public {{if not .Response.IsEmpty -}}{{template "type" .Response}}{{else}}void{{end}} {{.CamelName}}{{if .IsNameReserved}}Content{{end}}({{if .Request}}{{template "type" .Request}} request{{end}}) {
-    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}};
+    String path = {{ template "path" . }};
     {{ template "headers" . -}}
     {{ if .Response.IsEmpty -}}
     {{ template "api-call" . }}
@@ -39,6 +38,23 @@ class {{.PascalName}}Impl implements {{.PascalName}}Service {
   {{end}}
 }
 
+{{- define "path" -}}
+{{- if .PathParts -}}
+  String.format("{{range  .PathParts -}}
+    {{- .Prefix -}}
+    {{- if or .Field .IsAccountId -}}%s{{- end -}}
+  {{- end -}}"
+  {{- range .PathParts -}}
+    {{- if and .Field .Field.IsPathMultiSegment -}}, Encoding.encodeMultiSegmentPathParameter(request.get{{.Field.PascalName}}())
+    {{- else if .Field -}}, request.get{{.Field.PascalName}}()
+    {{- else if .IsAccountId -}}, apiClient.configuredAccountID()
+    {{- end -}}
+  {{- end -}})
+{{- else -}}
+  "{{.Path}}"
+{{- end -}}
+{{- end -}}
+
 {{ define "api-call" }}
 apiClient.{{.Verb}}(path
   {{- if .Request}}, {{ template "request-param" .}}{{end}}