Example usage for org.springframework.http HttpEntity getHeaders

List of usage examples for org.springframework.http HttpEntity getHeaders

Introduction

In this page you can find the example usage for org.springframework.http HttpEntity getHeaders.

Prototype

public HttpHeaders getHeaders() 

Source Link

Document

Returns the headers of this entity.

Usage

From source file:org.springframework.http.converter.FormHttpMessageConverter.java

@SuppressWarnings("unchecked")
private void writePart(String name, HttpEntity<?> partEntity, OutputStream os) throws IOException {
    Object partBody = partEntity.getBody();
    if (partBody == null) {
        throw new IllegalStateException("Empty body for part '" + name + "': " + partEntity);
    }/*w w  w  .  j a  v a 2s.c  om*/
    Class<?> partType = partBody.getClass();
    HttpHeaders partHeaders = partEntity.getHeaders();
    MediaType partContentType = partHeaders.getContentType();
    for (HttpMessageConverter<?> messageConverter : this.partConverters) {
        if (messageConverter.canWrite(partType, partContentType)) {
            Charset charset = isFilenameCharsetSet() ? StandardCharsets.US_ASCII : this.charset;
            HttpOutputMessage multipartMessage = new MultipartHttpOutputMessage(os, charset);
            multipartMessage.getHeaders().setContentDispositionFormData(name, getFilename(partBody));
            if (!partHeaders.isEmpty()) {
                multipartMessage.getHeaders().putAll(partHeaders);
            }
            ((HttpMessageConverter<Object>) messageConverter).write(partBody, partContentType,
                    multipartMessage);
            return;
        }
    }
    throw new HttpMessageNotWritableException("Could not write request: no suitable HttpMessageConverter "
            + "found for request type [" + partType.getName() + "]");
}

From source file:org.springframework.security.oauth2.provider.error.DefaultOAuth2ExceptionRenderer.java

public void handleHttpEntityResponse(HttpEntity<?> responseEntity, ServletWebRequest webRequest)
        throws Exception {
    if (responseEntity == null) {
        return;//w  w w  .ja  va 2 s  .c o m
    }
    HttpInputMessage inputMessage = createHttpInputMessage(webRequest);
    HttpOutputMessage outputMessage = createHttpOutputMessage(webRequest);
    if (responseEntity instanceof ResponseEntity && outputMessage instanceof ServerHttpResponse) {
        ((ServerHttpResponse) outputMessage)
                .setStatusCode(((ResponseEntity<?>) responseEntity).getStatusCode());
    }
    HttpHeaders entityHeaders = responseEntity.getHeaders();
    if (!entityHeaders.isEmpty()) {
        outputMessage.getHeaders().putAll(entityHeaders);
    }
    Object body = responseEntity.getBody();
    if (body != null) {
        writeWithMessageConverters(body, inputMessage, outputMessage);
    } else {
        // flush headers
        outputMessage.getBody();
    }
}

From source file:org.springframework.vault.authentication.AuthenticationStepsExecutor.java

private static HttpEntity<?> getEntity(HttpEntity<?> entity, @Nullable Object state) {

    if (entity == null) {
        return state == null ? HttpEntity.EMPTY : new HttpEntity<>(state);
    }//w ww  . j  ava 2 s  . c  o  m

    if (entity.getBody() == null && state != null) {
        return new HttpEntity<>(state, entity.getHeaders());
    }

    return entity;
}

From source file:org.springframework.vault.authentication.AuthenticationStepsOperator.java

private Mono<Object> doHttpRequest(HttpRequestNode<Object> step, Object state) {

    HttpRequest<Object> definition = step.getDefinition();
    HttpEntity<?> entity = getEntity(definition.getEntity(), state);

    RequestBodySpec spec;//from   w w w .j av  a2 s.c o m
    if (definition.getUri() == null) {

        spec = webClient.method(definition.getMethod()).uri(definition.getUriTemplate(),
                (Object[]) definition.getUrlVariables());
    } else {
        spec = webClient.method(definition.getMethod()).uri(definition.getUri());
    }

    for (Entry<String, List<String>> header : entity.getHeaders().entrySet()) {
        spec = spec.header(header.getKey(), header.getValue().get(0));
    }

    if (entity.getBody() != null && !entity.getBody().equals(Undefinded.INSTANCE)) {
        return spec.syncBody(entity.getBody()).retrieve().bodyToMono(definition.getResponseType());
    }

    return spec.retrieve().bodyToMono(definition.getResponseType());
}

From source file:org.springframework.vault.authentication.AuthenticationStepsOperator.java

private static HttpEntity<?> getEntity(HttpEntity<?> entity, Object state) {

    if (entity == null) {
        return state == null ? HttpEntity.EMPTY : new HttpEntity<>(state);
    }/*w w w. j  a  v a2 s.c om*/

    if (entity.getBody() == null && state != null) {
        return new HttpEntity<>(state, entity.getHeaders());
    }

    return entity;
}

From source file:org.springframework.web.client.RestTemplateIntegrationTests.java

@Test
public void exchangePost() throws Exception {
    HttpHeaders requestHeaders = new HttpHeaders();
    requestHeaders.set("MyHeader", "MyValue");
    requestHeaders.setContentType(MediaType.TEXT_PLAIN);
    HttpEntity<String> requestEntity = new HttpEntity<String>(helloWorld, requestHeaders);
    HttpEntity<Void> result = template.exchange(baseUrl + "/{method}", HttpMethod.POST, requestEntity,
            Void.class, "post");
    assertEquals("Invalid location", new URI(baseUrl + "/post/1"), result.getHeaders().getLocation());
    assertFalse(result.hasBody());//from w  w w  . j  a va2 s.co  m
}

From source file:org.venice.beachfront.bfapi.services.OAuthServiceTests.java

@Test
public void testRequestOAuthProfileSuccess() throws UserException, IOException {
    String mockAccessToken = "mock-access-token-321";
    String mockResponseBody = IOUtils.toString(this.getClass().getClassLoader().getResourceAsStream(
            String.format("%s%s%s", "model", File.separator, "geoaxis-single-string.json")), "UTF-8");

    Mockito.when(this.restTemplate.exchange(Mockito.eq(this.oauthProfileUrl), Mockito.eq(HttpMethod.GET),
            Mockito.any(), Mockito.eq(String.class))).then(new Answer<ResponseEntity<String>>() {
                @Override//from w w w .  j  a va  2  s . co  m
                public ResponseEntity<String> answer(InvocationOnMock invocation)
                        throws JsonProcessingException {
                    HttpEntity<?> entity = invocation.getArgumentAt(2, HttpEntity.class);
                    List<String> headerValues = entity.getHeaders().get("Authorization");
                    assertEquals(1, headerValues.size());
                    assertEquals("Bearer " + mockAccessToken, headerValues.get(0));
                    return new ResponseEntity<>(mockResponseBody, HttpStatus.OK);
                }
            });

    ProfileResponseBody receivedResponseBody = this.oauthService.requestOAuthProfile(mockAccessToken);

    assertEquals("testuser.localdomain", receivedResponseBody.getCommonName().toString());
    assertEquals("distinguished-name.test.localdomain", receivedResponseBody.getDn());
    assertEquals("testuser.localdomain", receivedResponseBody.getComputedUserName());
    assertEquals("distinguished-name.test.localdomain", receivedResponseBody.getComputedUserId());
}

From source file:software.coolstuff.springframework.owncloud.service.impl.rest.AbstractOwncloudRestServiceImpl.java

protected <T extends Ocs, E> T exchange(String url, HttpMethod method, HttpEntity<E> httpEntity, Class<T> clazz,
        OwncloudResponseStatusChecker statusChecker, Object... urlVariables) {
    log.trace("Exchange Data by a {} Request with URL {}. Requested Class of returned Data is {}", method, url,
            clazz);//from  w  w w.j  a v a  2  s  . com
    ResponseEntity<T> response = restTemplate.exchange(url, method, httpEntity, clazz, urlVariables);
    T result = response.getBody();
    log.trace("Returned Meta-Data: {}", result.getMeta());
    log.debug("Check the returned Meta-Data for Errors");
    String authorizationUser = getAuthorizationUserFromHeaders(httpEntity.getHeaders());
    statusChecker.checkForFailure(authorizationUser, url, result.getMeta());
    return result;
}