use of org.elasticsearch.client.Response in project storm by apache.
the class EsLookupBolt method lookupValuesInEs.
private Collection<Values> lookupValuesInEs(Tuple tuple) throws IOException {
String index = tupleMapper.getIndex(tuple);
String type = tupleMapper.getType(tuple);
String id = tupleMapper.getId(tuple);
Map<String, String> params = tupleMapper.getParams(tuple, new HashMap<>());
Response response = client.performRequest("get", getEndpoint(index, type, id), params);
return output.toValues(response);
}
use of org.elasticsearch.client.Response in project storm by apache.
the class EsTestUtil method generateMockResponseException.
/**
* Generate a mock {@link ResponseException}.
* @return a mock {@link ResponseException}.
* @throws IOException
*/
public static ResponseException generateMockResponseException() throws IOException {
Response response = mock(Response.class);
RequestLine requestLine = mock(RequestLine.class);
StatusLine statusLine = mock(StatusLine.class);
when(response.getRequestLine()).thenReturn(requestLine);
when(response.getStatusLine()).thenReturn(statusLine);
return new ResponseException(response);
}
use of org.elasticsearch.client.Response in project graylog2-server by Graylog2.
the class FieldMappingApi method fieldTypes.
public Map<String, FieldMapping> fieldTypes(String index) {
final JsonNode result = client.execute((c, requestOptions) -> {
final Response response = c.getLowLevelClient().performRequest(request(index));
return objectMapper.readTree(response.getEntity().getContent());
}, "Unable to retrieve field types of index " + index);
final JsonNode fields = result.path(index).path("mappings").path("properties");
// noinspection UnstableApiUsage
return Streams.stream(fields.fields()).collect(Collectors.toMap(Map.Entry::getKey, entry -> FieldMapping.create(entry.getValue().path("type").asText(), entry.getValue().path("fielddata").asBoolean())));
}
use of org.elasticsearch.client.Response in project graylog2-server by Graylog2.
the class StatsApi method stats.
private JsonNode stats(Collection<String> indices, Collection<String> metrics, Consumer<Request> prepareRequest) {
final StringBuilder endpoint = new StringBuilder();
if (!indices.isEmpty()) {
final String joinedIndices = String.join(",", indices);
endpoint.append("/");
endpoint.append(joinedIndices);
}
endpoint.append("/_stats");
if (!metrics.isEmpty()) {
final String joinedMetrics = String.join(",", metrics);
endpoint.append("/");
endpoint.append(joinedMetrics);
}
final Request request = new Request("GET", endpoint.toString());
prepareRequest.accept(request);
return client.execute((c, requestOptions) -> {
request.setOptions(requestOptions);
final Response response = c.getLowLevelClient().performRequest(request);
return objectMapper.readTree(response.getEntity().getContent());
}, "Unable to retrieve index stats for " + String.join(",", indices));
}
use of org.elasticsearch.client.Response in project graylog2-server by Graylog2.
the class CatApi method perform.
private <R> R perform(Request request, TypeReference<R> responseClass, String errorMessage) {
return client.execute((c, requestOptions) -> {
request.setOptions(requestOptions);
final Response response = c.getLowLevelClient().performRequest(request);
return returnType(response, responseClass);
}, errorMessage);
}
Aggregations