Skip to content

Commit 505fabb

Browse files
committed
Cleanup collections and stringmap methods
1 parent f9ed3e1 commit 505fabb

30 files changed

+63
-163
lines changed

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

Lines changed: 5 additions & 57 deletions
Original file line numberDiff line numberDiff line change
@@ -161,77 +161,25 @@ public static <I> void setQuery(Request in, I entity) {
161161
}
162162
}
163163

164-
private static <I> void setHeaders(Request in, Map<String, String> headers) {
165-
if (headers == null) {
166-
return;
167-
}
168-
for (Map.Entry<String, String> e : headers.entrySet()) {
169-
in.withHeader(e.getKey(), e.getValue());
170-
}
171-
}
172-
173-
public <I, O> Collection<O> getCollection(
174-
String path, I in, Class<O> element, Map<String, String> headers) {
164+
public <O> Collection<O> getCollection(Request req, Class<O> element) {
175165
return withJavaType(
176-
path,
177-
in,
178-
mapper.getTypeFactory().constructCollectionType(Collection.class, element),
179-
headers);
166+
req, mapper.getTypeFactory().constructCollectionType(Collection.class, element));
180167
}
181168

182-
public <I> Map<String, String> getStringMap(String path, I in, Map<String, String> headers) {
169+
public Map<String, String> getStringMap(Request req) {
183170
return withJavaType(
184-
path,
185-
in,
186-
mapper.getTypeFactory().constructMapType(Map.class, String.class, String.class),
187-
headers);
171+
req, mapper.getTypeFactory().constructMapType(Map.class, String.class, String.class));
188172
}
189173

190-
protected <I, O> O withJavaType(
191-
String path, I in, JavaType javaType, Map<String, String> headers) {
174+
protected <I, O> O withJavaType(Request request, JavaType javaType) {
192175
try {
193-
Request request = prepareRequest("GET", path, in, headers);
194176
Response response = getResponse(request);
195177
return deserialize(response.getBody(), javaType);
196178
} catch (IOException e) {
197179
throw new DatabricksException("IO error: " + e.getMessage(), e);
198180
}
199181
}
200182

201-
public <I, O> O execute(
202-
String method, String path, I in, Class<O> target, Map<String, String> headers) {
203-
try {
204-
return execute(prepareRequest(method, path, in, headers), target);
205-
} catch (IOException e) {
206-
throw new DatabricksException("IO error: " + e.getMessage(), e);
207-
}
208-
}
209-
210-
private boolean hasBody(String method) {
211-
return !method.equals("GET") && !method.equals("DELETE") && !method.equals("HEAD");
212-
}
213-
214-
private <I> Request prepareBaseRequest(String method, String path, I in)
215-
throws JsonProcessingException {
216-
if (in == null || !hasBody(method)) {
217-
return new Request(method, path);
218-
} else if (InputStream.class.isAssignableFrom(in.getClass())) {
219-
InputStream body = (InputStream) in;
220-
return new Request(method, path, body);
221-
} else {
222-
String body = (in instanceof String) ? (String) in : serialize(in);
223-
return new Request(method, path, body);
224-
}
225-
}
226-
227-
private <I> Request prepareRequest(String method, String path, I in, Map<String, String> headers)
228-
throws JsonProcessingException {
229-
Request req = prepareBaseRequest(method, path, in);
230-
setQuery(req, in);
231-
setHeaders(req, headers);
232-
return req;
233-
}
234-
235183
/**
236184
* Executes HTTP request with retries and converts it to proper POJO
237185
*

databricks-sdk-java/src/main/java/com/databricks/sdk/service/catalog/AccountMetastoresImpl.java

Lines changed: 0 additions & 1 deletion
Some generated files are not rendered by default. Learn more about customizing how changed files appear on GitHub.

databricks-sdk-java/src/main/java/com/databricks/sdk/service/catalog/MetastoresImpl.java

Lines changed: 0 additions & 3 deletions
Some generated files are not rendered by default. Learn more about customizing how changed files appear on GitHub.

databricks-sdk-java/src/main/java/com/databricks/sdk/service/compute/ClustersImpl.java

Lines changed: 0 additions & 3 deletions
Some generated files are not rendered by default. Learn more about customizing how changed files appear on GitHub.

databricks-sdk-java/src/main/java/com/databricks/sdk/service/compute/GlobalInitScriptsImpl.java

Lines changed: 0 additions & 1 deletion
Some generated files are not rendered by default. Learn more about customizing how changed files appear on GitHub.

databricks-sdk-java/src/main/java/com/databricks/sdk/service/compute/InstancePoolsImpl.java

Lines changed: 0 additions & 1 deletion
Some generated files are not rendered by default. Learn more about customizing how changed files appear on GitHub.

databricks-sdk-java/src/main/java/com/databricks/sdk/service/compute/InstanceProfilesImpl.java

Lines changed: 0 additions & 1 deletion
Some generated files are not rendered by default. Learn more about customizing how changed files appear on GitHub.

databricks-sdk-java/src/main/java/com/databricks/sdk/service/compute/LibrariesImpl.java

Lines changed: 0 additions & 1 deletion
Some generated files are not rendered by default. Learn more about customizing how changed files appear on GitHub.

databricks-sdk-java/src/main/java/com/databricks/sdk/service/iam/CurrentUserImpl.java

Lines changed: 0 additions & 1 deletion
Some generated files are not rendered by default. Learn more about customizing how changed files appear on GitHub.

databricks-sdk-java/src/main/java/com/databricks/sdk/service/iam/UsersImpl.java

Lines changed: 0 additions & 2 deletions
Some generated files are not rendered by default. Learn more about customizing how changed files appear on GitHub.

0 commit comments

Comments
 (0)