Examples with RestStatusException org.springframework.data.elasticsearch.RestStatusException used on opensource projects

Example 1 with RestStatusException

use of org.springframework.data.elasticsearch.RestStatusException in project spring-data-elasticsearch by spring-projects.

the class DefaultReactiveElasticsearchClient method contentOrError.

/**
 * checks if the given content body contains an {@link ElasticsearchException}, if yes it is returned in a Mono.error.
 * Otherwise the content is returned in the Mono
 *
 * @param content the content to analyze
 * @param mediaType the returned media type
 * @param status the response status
 * @return a Mono with the content or an Mono.error
 */
private static Mono<String> contentOrError(String content, String mediaType, RestStatus status) {
    ElasticsearchException exception = getElasticsearchException(content, mediaType, status);
    if (exception != null) {
        StringBuilder sb = new StringBuilder();
        buildExceptionMessages(sb, exception);
        return Mono.error(new RestStatusException(status.getStatus(), sb.toString(), exception));
    }
    return Mono.just(content);
}

Example 2 with RestStatusException

use of org.springframework.data.elasticsearch.RestStatusException in project spring-data-elasticsearch by spring-projects.

the class DefaultReactiveElasticsearchClientTest method shouldThrowRestStatusExceptionOnServer5xxWithEmptyBody.

// #1712
@Test
@DisplayName("should throw RestStatusException on server 5xx with empty body")
void shouldThrowRestStatusExceptionOnServer5xxWithEmptyBody() {
    when(hostProvider.getActive(any())).thenReturn(Mono.just(webClient));
    WebClient.RequestBodyUriSpec requestBodyUriSpec = mock(WebClient.RequestBodyUriSpec.class);
    when(requestBodyUriSpec.uri((Function<UriBuilder, URI>) any())).thenReturn(requestBodyUriSpec);
    when(requestBodyUriSpec.attribute(any(), any())).thenReturn(requestBodyUriSpec);
    when(requestBodyUriSpec.headers(any())).thenReturn(requestBodyUriSpec);
    when(webClient.method(any())).thenReturn(requestBodyUriSpec);
    when(requestBodyUriSpec.exchangeToMono(any())).thenAnswer(invocationOnMock -> {
        Function<ClientResponse, ? extends Mono<?>> responseHandler = invocationOnMock.getArgument(0);
        ClientResponse clientResponse = mock(ClientResponse.class);
        when(clientResponse.statusCode()).thenReturn(HttpStatus.SERVICE_UNAVAILABLE);
        ClientResponse.Headers headers = mock(ClientResponse.Headers.class);
        when(headers.contentType()).thenReturn(Optional.empty());
        when(clientResponse.headers()).thenReturn(headers);
        when(clientResponse.body(any())).thenReturn(Mono.empty());
        return responseHandler.apply(clientResponse);
    });
    ReactiveElasticsearchClient client = new DefaultReactiveElasticsearchClient(hostProvider, requestCreator);
    // 
    client.get(new GetRequest("42")).as(// 
    StepVerifier::create).expectError(// 
    RestStatusException.class).verify();
}

Example 3 with RestStatusException

use of org.springframework.data.elasticsearch.RestStatusException in project spring-data-elasticsearch by spring-projects.

the class ElasticsearchExceptionTranslator method isSeqNoConflict.

private boolean isSeqNoConflict(Exception exception) {
    Integer status = null;
    String message = null;
    if (exception instanceof ElasticsearchStatusException) {
        ElasticsearchStatusException statusException = (ElasticsearchStatusException) exception;
        status = statusException.status().getStatus();
        message = statusException.getMessage();
    }
    if (exception instanceof RestStatusException) {
        RestStatusException statusException = (RestStatusException) exception;
        status = statusException.getStatus();
        message = statusException.getMessage();
    }
    if (status != null && message != null) {
        return status == 409 && message.contains("type=version_conflict_engine_exception") && message.contains("version conflict, required seqNo");
    }
    if (exception instanceof VersionConflictEngineException) {
        VersionConflictEngineException versionConflictEngineException = (VersionConflictEngineException) exception;
        return versionConflictEngineException.getMessage() != null && versionConflictEngineException.getMessage().contains("version conflict, required seqNo");
    }
    return false;
}

Example 4 with RestStatusException

use of org.springframework.data.elasticsearch.RestStatusException in project spring-data-elasticsearch by spring-projects.

the class ElasticsearchExceptionTranslator method translateExceptionIfPossible.

@Override
public DataAccessException translateExceptionIfPossible(RuntimeException ex) {
    if (isSeqNoConflict(ex)) {
        return new OptimisticLockingFailureException("Cannot index a document due to seq_no+primary_term conflict", ex);
    }
    if (ex instanceof ElasticsearchException) {
        ElasticsearchException elasticsearchException = (ElasticsearchException) ex;
        if (!indexAvailable(elasticsearchException)) {
            return new NoSuchIndexException(ObjectUtils.nullSafeToString(elasticsearchException.getMetadata("es.index")), ex);
        }
        if (elasticsearchException instanceof ElasticsearchStatusException) {
            ElasticsearchStatusException elasticsearchStatusException = (ElasticsearchStatusException) elasticsearchException;
            return new RestStatusException(elasticsearchStatusException.status().getStatus(), elasticsearchStatusException.getMessage(), elasticsearchStatusException);
        }
        return new UncategorizedElasticsearchException(ex.getMessage(), ex);
    }
    if (ex instanceof RestStatusException) {
        RestStatusException restStatusException = (RestStatusException) ex;
        Throwable cause = restStatusException.getCause();
        if (cause instanceof ElasticsearchException) {
            ElasticsearchException elasticsearchException = (ElasticsearchException) cause;
            if (!indexAvailable(elasticsearchException)) {
                return new NoSuchIndexException(ObjectUtils.nullSafeToString(elasticsearchException.getMetadata("es.index")), ex);
            }
        }
    }
    if (ex instanceof ValidationException) {
        return new DataIntegrityViolationException(ex.getMessage(), ex);
    }
    Throwable cause = ex.getCause();
    if (cause instanceof IOException) {
        return new DataAccessResourceFailureException(ex.getMessage(), ex);
    }
    return null;
}

Example 5 with RestStatusException

use of org.springframework.data.elasticsearch.RestStatusException in project spring-data-elasticsearch by spring-projects.

the class DefaultReactiveElasticsearchClient method handleServerError.

// endregion
// region error and exception handling
private <T> Publisher<? extends T> handleServerError(Request request, ClientResponse response) {
    int statusCode = response.statusCode().value();
    RestStatus status = RestStatus.fromCode(statusCode);
    String mediaType = response.headers().contentType().map(MediaType::toString).orElse(XContentType.JSON.mediaType());
    return // 
    response.body(BodyExtractors.toMono(byte[].class)).switchIfEmpty(Mono.error(new RestStatusException(status.getStatus(), String.format("%s request to %s returned error code %s and no body.", request.getMethod(), request.getEndpoint(), statusCode)))).map(// 
    bytes -> new String(bytes, StandardCharsets.UTF_8)).flatMap(content -> contentOrError(content, mediaType, status)).flatMap(unused -> Mono.error(new RestStatusException(status.getStatus(), String.format("%s request to %s returned error code %s.", request.getMethod(), request.getEndpoint(), statusCode))));
}