Examples with ResponseEntity org.springframework.http.ResponseEntity used on opensource projects

Example 11 with ResponseEntity

use of org.springframework.http.ResponseEntity in project spring-framework by spring-projects.

the class ResponseBodyEmitterReturnValueHandler method handleReturnValue.

@Override
public void handleReturnValue(Object returnValue, MethodParameter returnType, ModelAndViewContainer mavContainer, NativeWebRequest webRequest) throws Exception {
    if (returnValue == null) {
        mavContainer.setRequestHandled(true);
        return;
    }
    HttpServletResponse response = webRequest.getNativeResponse(HttpServletResponse.class);
    ServerHttpResponse outputMessage = new ServletServerHttpResponse(response);
    if (returnValue instanceof ResponseEntity) {
        ResponseEntity<?> responseEntity = (ResponseEntity<?>) returnValue;
        response.setStatus(responseEntity.getStatusCodeValue());
        outputMessage.getHeaders().putAll(responseEntity.getHeaders());
        returnValue = responseEntity.getBody();
        if (returnValue == null) {
            mavContainer.setRequestHandled(true);
            outputMessage.flush();
            return;
        }
    }
    ServletRequest request = webRequest.getNativeRequest(ServletRequest.class);
    ShallowEtagHeaderFilter.disableContentCaching(request);
    ResponseBodyEmitterAdapter adapter = getAdapterFor(returnValue.getClass());
    if (adapter == null) {
        throw new IllegalStateException("Could not find ResponseBodyEmitterAdapter for return value type: " + returnValue.getClass());
    }
    ResponseBodyEmitter emitter = adapter.adaptToEmitter(returnValue, outputMessage);
    emitter.extendResponse(outputMessage);
    // Commit the response and wrap to ignore further header changes
    outputMessage.getBody();
    outputMessage.flush();
    outputMessage = new StreamingServletServerHttpResponse(outputMessage);
    DeferredResult<?> deferredResult = new DeferredResult<>(emitter.getTimeout());
    WebAsyncUtils.getAsyncManager(webRequest).startDeferredResultProcessing(deferredResult, mavContainer);
    HttpMessageConvertingHandler handler = new HttpMessageConvertingHandler(outputMessage, deferredResult);
    emitter.initialize(handler);
}

Example 12 with ResponseEntity

use of org.springframework.http.ResponseEntity in project spring-framework by spring-projects.

the class StreamingResponseBodyReturnValueHandler method handleReturnValue.

@Override
public void handleReturnValue(Object returnValue, MethodParameter returnType, ModelAndViewContainer mavContainer, NativeWebRequest webRequest) throws Exception {
    if (returnValue == null) {
        mavContainer.setRequestHandled(true);
        return;
    }
    HttpServletResponse response = webRequest.getNativeResponse(HttpServletResponse.class);
    ServerHttpResponse outputMessage = new ServletServerHttpResponse(response);
    if (returnValue instanceof ResponseEntity) {
        ResponseEntity<?> responseEntity = (ResponseEntity<?>) returnValue;
        response.setStatus(responseEntity.getStatusCodeValue());
        outputMessage.getHeaders().putAll(responseEntity.getHeaders());
        returnValue = responseEntity.getBody();
        if (returnValue == null) {
            mavContainer.setRequestHandled(true);
            outputMessage.flush();
            return;
        }
    }
    ServletRequest request = webRequest.getNativeRequest(ServletRequest.class);
    ShallowEtagHeaderFilter.disableContentCaching(request);
    Assert.isInstanceOf(StreamingResponseBody.class, returnValue, "StreamingResponseBody expected");
    StreamingResponseBody streamingBody = (StreamingResponseBody) returnValue;
    Callable<Void> callable = new StreamingResponseBodyTask(outputMessage.getBody(), streamingBody);
    WebAsyncUtils.getAsyncManager(webRequest).startCallableProcessing(callable, mavContainer);
}

Example 13 with ResponseEntity

use of org.springframework.http.ResponseEntity in project spring-framework by spring-projects.

the class HttpEntityMethodProcessorMockTests method shouldFailHandlingWhenConverterCannotWrite.

@Test
public void shouldFailHandlingWhenConverterCannotWrite() throws Exception {
    String body = "Foo";
    ResponseEntity<String> returnValue = new ResponseEntity<>(body, HttpStatus.OK);
    MediaType accepted = MediaType.TEXT_PLAIN;
    servletRequest.addHeader("Accept", accepted.toString());
    given(stringHttpMessageConverter.canWrite(String.class, null)).willReturn(true);
    given(stringHttpMessageConverter.getSupportedMediaTypes()).willReturn(Collections.singletonList(MediaType.TEXT_PLAIN));
    given(stringHttpMessageConverter.canWrite(String.class, accepted)).willReturn(false);
    this.thrown.expect(HttpMediaTypeNotAcceptableException.class);
    processor.handleReturnValue(returnValue, returnTypeResponseEntityProduces, mavContainer, webRequest);
}

Example 14 with ResponseEntity

use of org.springframework.http.ResponseEntity in project spring-framework by spring-projects.

the class HttpEntityMethodProcessorMockTests method shouldFailHandlingWhenContentTypeNotSupported.

@Test
public void shouldFailHandlingWhenContentTypeNotSupported() throws Exception {
    String body = "Foo";
    ResponseEntity<String> returnValue = new ResponseEntity<>(body, HttpStatus.OK);
    MediaType accepted = MediaType.APPLICATION_ATOM_XML;
    servletRequest.addHeader("Accept", accepted.toString());
    given(stringHttpMessageConverter.canWrite(String.class, null)).willReturn(true);
    given(stringHttpMessageConverter.getSupportedMediaTypes()).willReturn(Collections.singletonList(MediaType.TEXT_PLAIN));
    this.thrown.expect(HttpMediaTypeNotAcceptableException.class);
    processor.handleReturnValue(returnValue, returnTypeResponseEntity, mavContainer, webRequest);
}

Example 15 with ResponseEntity

use of org.springframework.http.ResponseEntity in project spring-framework by spring-projects.

the class HttpEntityMethodProcessorMockTests method shouldHandleReturnValue.

@Test
public void shouldHandleReturnValue() throws Exception {
    String body = "Foo";
    ResponseEntity<String> returnValue = new ResponseEntity<>(body, HttpStatus.OK);
    MediaType accepted = MediaType.TEXT_PLAIN;
    servletRequest.addHeader("Accept", accepted.toString());
    initStringMessageConversion(accepted);
    processor.handleReturnValue(returnValue, returnTypeResponseEntity, mavContainer, webRequest);
    assertTrue(mavContainer.isRequestHandled());
    verify(stringHttpMessageConverter).write(eq(body), eq(accepted), isA(HttpOutputMessage.class));
}