All Downloads are FREE. Search and download functionalities are using the official Maven repository.
Many resources are needed to download a project. Please understand that we have to compensate our server costs. Thank you in advance.
Project price only 1 $
You can buy this project and download/modify it how often you want.
io.camunda.operate.opensearch.ExtendedOpenSearchClient Maven / Gradle / Ivy
/*
* Copyright Camunda Services GmbH and/or licensed to Camunda Services GmbH under
* one or more contributor license agreements. See the NOTICE file distributed
* with this work for additional information regarding copyright ownership.
* Licensed under the Camunda License 1.0. You may not use this file
* except in compliance with the Camunda License 1.0.
*/
package io.camunda.operate.opensearch;
import static java.lang.String.format;
import static java.lang.String.join;
import com.fasterxml.jackson.core.JsonProcessingException;
import com.fasterxml.jackson.core.type.TypeReference;
import com.fasterxml.jackson.databind.ObjectMapper;
import jakarta.json.stream.JsonGenerator;
import java.io.ByteArrayOutputStream;
import java.io.IOException;
import java.util.HashMap;
import java.util.List;
import java.util.Map;
import java.util.regex.Matcher;
import java.util.regex.Pattern;
import org.opensearch.client.json.JsonpDeserializer;
import org.opensearch.client.json.jackson.JacksonJsonpMapper;
import org.opensearch.client.opensearch.OpenSearchClient;
import org.opensearch.client.opensearch._types.ErrorResponse;
import org.opensearch.client.opensearch._types.OpenSearchException;
import org.opensearch.client.opensearch.core.SearchRequest;
import org.opensearch.client.opensearch.core.SearchResponse;
import org.opensearch.client.transport.JsonEndpoint;
import org.opensearch.client.transport.OpenSearchTransport;
import org.opensearch.client.transport.endpoints.EndpointWithResponseMapperAttr;
import org.opensearch.client.transport.endpoints.SimpleEndpoint;
public class ExtendedOpenSearchClient extends OpenSearchClient {
private static final Pattern SEARCH_AFTER_PATTERN =
Pattern.compile("(\"search_after\":\\[[^\\]]*\\])");
private static final String DOCUMENT_ATTR =
"org.opensearch.client:Deserializer:_global.search.TDocument";
public ExtendedOpenSearchClient(final OpenSearchTransport transport) {
super(transport);
}
private static SimpleEndpoint, R> arbitraryEndpoint(
final String method, final String path, final JsonpDeserializer responseParser) {
final boolean hasBody = !List.of("GET", "DELETE").contains(method.toUpperCase());
return new SimpleEndpoint<>(
request -> method, // Request method
request -> path, // Request path
request -> Map.of(), // Request parameters
request -> Map.of(), // Headers
hasBody, // Has body
responseParser);
}
private ObjectMapper objectMapper() {
return ((JacksonJsonpMapper) transport.jsonpMapper()).objectMapper();
}
private Map jsonToMap(final String json) throws JsonProcessingException {
return objectMapper().readValue(json, new TypeReference<>() {});
}
private String json(final SearchRequest request) {
final ByteArrayOutputStream baos = new ByteArrayOutputStream();
final JsonGenerator generator = transport.jsonpMapper().jsonProvider().createGenerator(baos);
request.serialize(generator, transport.jsonpMapper());
generator.close();
return baos.toString();
}
/**
* Fixes searchAfter block in SearchRequest request by replacing minimum long values represented
* as String with Long representations.
*
* @param json SearchRequest json
* @return fixed json
*/
/*
*/
private String fixSearchAfter(final String json) {
final Matcher m = SEARCH_AFTER_PATTERN.matcher(json);
if (m.find()) {
final var searchAfter = m.group(1); // Find "searchAfter" block in search request
// Replace all occurrences of minimum long value string representations with long
// representations inside searchAfter
final var fixedSearchAfter =
searchAfter.replaceAll(format("\"%s\"", Long.MIN_VALUE), String.valueOf(Long.MIN_VALUE));
return json.replace(searchAfter, fixedSearchAfter);
} else {
return json;
}
}
public SearchResponse fixedSearch(
final SearchRequest request, final Class tDocumentClass)
throws IOException, OpenSearchException {
final var path = format("/%s/_search", join(",", request.index()));
JsonEndpoint, SearchResponse, ErrorResponse> endpoint =
arbitraryEndpoint("POST", path, SearchResponse._DESERIALIZER);
endpoint =
new EndpointWithResponseMapperAttr<>(
endpoint, DOCUMENT_ATTR, getDeserializer(tDocumentClass));
String requestJson = json(request);
requestJson = fixSearchAfter(requestJson);
return (SearchResponse) arbitraryRequest(requestJson, endpoint);
}
public Map searchAsMap(final SearchRequest request)
throws IOException, OpenSearchException {
final JsonEndpoint endpoint =
SearchRequest._ENDPOINT.withResponseDeserializer(getDeserializer(HashMap.class));
return transport.performRequest(request, endpoint, transport.options());
}
public Map arbitraryRequest(
final String method, final String path, final String json)
throws IOException, OpenSearchException {
final JsonEndpoint, HashMap, ErrorResponse> endpoint =
arbitraryEndpoint(method, path, getDeserializer(HashMap.class));
return arbitraryRequest(json, endpoint);
}
private R arbitraryRequest(
final String json, final JsonEndpoint, R, ErrorResponse> endpoint)
throws IOException, OpenSearchException {
return transport.performRequest(jsonToMap(json), endpoint, transport.options());
}
}
