Example usage for org.apache.http.client.methods RequestBuilder addHeader

List of usage examples for org.apache.http.client.methods RequestBuilder addHeader

Introduction

In this page you can find the example usage for org.apache.http.client.methods RequestBuilder addHeader.

Prototype

public RequestBuilder addHeader(final String name, final String value) 

Source Link

Usage

From source file:org.dasein.cloud.azurepack.platform.AzurePackDatabaseRequests.java

private void addCommonHeaders(RequestBuilder requestBuilder) {
    requestBuilder.addHeader("x-ms-version", "2014-02-01");
    requestBuilder.addHeader("Accept", "application/json");
}

From source file:org.springframework.cloud.netflix.ribbon.apache.RibbonApacheHttpRequestTests.java

void testEntity(String entityValue, ByteArrayInputStream requestEntity, boolean addContentLengthHeader,
        String method) throws IOException {
    String lengthString = String.valueOf(entityValue.length());
    Long length = null;// ww w  . j ava 2 s. c o m
    URI uri = URI.create("http://example.com");
    LinkedMultiValueMap<String, String> headers = new LinkedMultiValueMap<>();
    if (addContentLengthHeader) {
        headers.add("Content-Length", lengthString);
        length = (long) entityValue.length();
    }

    RibbonRequestCustomizer requestCustomizer = new RibbonRequestCustomizer<RequestBuilder>() {
        @Override
        public boolean accepts(Class builderClass) {
            return builderClass == RequestBuilder.class;
        }

        @Override
        public void customize(RequestBuilder builder) {
            builder.addHeader("from-customizer", "foo");
        }
    };
    RibbonCommandContext context = new RibbonCommandContext("example", method, uri.toString(), false, headers,
            new LinkedMultiValueMap<String, String>(), requestEntity,
            Collections.singletonList(requestCustomizer));
    context.setContentLength(length);
    RibbonApacheHttpRequest httpRequest = new RibbonApacheHttpRequest(context);

    HttpUriRequest request = httpRequest.toRequest(RequestConfig.custom().build());

    assertThat("request is wrong type", request, is(instanceOf(HttpEntityEnclosingRequest.class)));
    assertThat("uri is wrong", request.getURI().toString(), startsWith(uri.toString()));
    if (addContentLengthHeader) {
        assertThat("Content-Length is missing", request.getFirstHeader("Content-Length"), is(notNullValue()));
        assertThat("Content-Length is wrong", request.getFirstHeader("Content-Length").getValue(),
                is(equalTo(lengthString)));
    }
    assertThat("from-customizer is missing", request.getFirstHeader("from-customizer"), is(notNullValue()));
    assertThat("from-customizer is wrong", request.getFirstHeader("from-customizer").getValue(),
            is(equalTo("foo")));

    HttpEntityEnclosingRequest entityRequest = (HttpEntityEnclosingRequest) request;
    assertThat("entity is missing", entityRequest.getEntity(), is(notNullValue()));
    HttpEntity entity = entityRequest.getEntity();
    assertThat("contentLength is wrong", entity.getContentLength(), is(equalTo((long) entityValue.length())));
    assertThat("content is missing", entity.getContent(), is(notNullValue()));
    String string = StreamUtils.copyToString(entity.getContent(), Charset.forName("UTF-8"));
    assertThat("content is wrong", string, is(equalTo(entityValue)));
}

From source file:org.dasein.cloud.azure.platform.AzureSQLDatabaseSupportRequests.java

private void addAzureCommonHeaders(RequestBuilder requestBuilder) {
    requestBuilder.addHeader("x-ms-version", "2012-03-01");
    requestBuilder.addHeader("Content-Type", "application/xml;charset=UTF-8");
}

From source file:org.apache.gobblin.http.ApacheHttpRequestBuilderTest.java

/**
 * Build a {@link HttpUriRequest} from a {@link GenericRecord}
 *//* w  w  w.ja  v  a 2  s.com*/
public void testBuildWriteRequest() throws IOException {
    String urlTemplate = "http://www.test.com/a/part1:${part1}/a/part2:${part2}";
    String verb = "post";
    ApacheHttpRequestBuilder builder = spy(new ApacheHttpRequestBuilder(urlTemplate, verb, "application/json"));
    ArgumentCaptor<RequestBuilder> requestBuilderArgument = ArgumentCaptor.forClass(RequestBuilder.class);

    Queue<BufferedRecord<GenericRecord>> queue = HttpTestUtils.createQueue(1, false);
    AsyncRequest<GenericRecord, HttpUriRequest> request = builder.buildRequest(queue);
    verify(builder).build(requestBuilderArgument.capture());

    RequestBuilder expected = RequestBuilder.post();
    expected.setUri("http://www.test.com/a/part1:01/a/part2:02?param1=01");
    String payloadStr = "{\"id\":\"id0\"}";
    expected.addHeader(HttpHeaders.CONTENT_TYPE, ContentType.APPLICATION_JSON.getMimeType())
            .setEntity(new StringEntity(payloadStr, ContentType.APPLICATION_JSON));

    // Compare HttpUriRequest
    HttpTestUtils.assertEqual(requestBuilderArgument.getValue(), expected);
    Assert.assertEquals(request.getRecordCount(), 1);
    Assert.assertEquals(queue.size(), 0);
}

From source file:net.duckling.ddl.util.RESTClient.java

public JsonObject httpGet(String url, List<NameValuePair> params) throws IOException {
    CloseableHttpClient httpclient = HttpClients.createDefault();
    try {//from   ww  w .jav a2s. co  m
        RequestBuilder rb = RequestBuilder.get().setUri(new URI(url));
        for (NameValuePair hp : params) {
            rb.addParameter(hp.getName(), hp.getValue());
        }
        rb.addHeader("accept", "application/json");
        HttpUriRequest uriRequest = rb.build();
        HttpResponse response = httpclient.execute(uriRequest);
        if (response.getStatusLine().getStatusCode() != 200) {
            throw new RuntimeException(
                    "Failed : HTTP error code : " + response.getStatusLine().getStatusCode());
        }
        BufferedReader br = new BufferedReader(new InputStreamReader((response.getEntity().getContent())));
        JsonParser jp = new JsonParser();
        JsonElement je = jp.parse(br);
        return je.getAsJsonObject();
    } catch (ClientProtocolException e) {
        e.printStackTrace();
    } catch (IOException e) {
        e.printStackTrace();
    } catch (URISyntaxException e) {
        e.printStackTrace();
    } finally {
        httpclient.close();
    }
    return null;
}

From source file:com.salesforce.dva.argus.service.callback.DefaultCallbackService.java

private HttpUriRequest buildRequest(DefaultAlertService.NotificationContext context,
        CallbackService.Request request) {
    RequestBuilder builder = RequestBuilder.create(request.method().name()).setUri(request.uri())
            .setEntity(body(context, request));
    request.header().forEach((k, v) -> builder.addHeader(k, v));
    return builder.build();
}

From source file:org.springframework.cloud.netflix.ribbon.apache.RibbonApacheHttpRequest.java

public HttpUriRequest toRequest(final RequestConfig requestConfig) {
    final RequestBuilder builder = RequestBuilder.create(this.context.getMethod());
    builder.setUri(this.uri);
    for (final String name : this.context.getHeaders().keySet()) {
        final List<String> values = this.context.getHeaders().get(name);
        for (final String value : values) {
            builder.addHeader(name, value);
        }//from  ww w .  j a  v  a2  s . c  om
    }

    for (final String name : this.context.getParams().keySet()) {
        final List<String> values = this.context.getParams().get(name);
        for (final String value : values) {
            builder.addParameter(name, value);
        }
    }

    if (this.context.getRequestEntity() != null) {
        final BasicHttpEntity entity;
        entity = new BasicHttpEntity();
        entity.setContent(this.context.getRequestEntity());
        // if the entity contentLength isn't set, transfer-encoding will be set
        // to chunked in org.apache.http.protocol.RequestContent. See gh-1042
        if (this.context.getContentLength() != null) {
            entity.setContentLength(this.context.getContentLength());
        } else if ("GET".equals(this.context.getMethod())) {
            entity.setContentLength(0);
        }
        builder.setEntity(entity);
    }

    customize(this.context.getRequestCustomizers(), builder);

    builder.setConfig(requestConfig);
    return builder.build();
}

From source file:gobblin.writer.http.SalesforceRestWriter.java

private HttpUriRequest newRequest(RequestBuilder builder, JsonElement payload) {
    try {// ww w  . j a  va  2s .  c om
        builder.addHeader(HttpHeaders.CONTENT_TYPE, ContentType.APPLICATION_JSON.getMimeType())
                .addHeader(HttpHeaders.AUTHORIZATION, "OAuth " + accessToken)
                .setEntity(new StringEntity(payload.toString(), ContentType.APPLICATION_JSON));
    } catch (Exception e) {
        throw new RuntimeException(e);
    }
    if (getLog().isDebugEnabled()) {
        getLog().debug("Request builder: "
                + ToStringBuilder.reflectionToString(builder, ToStringStyle.SHORT_PREFIX_STYLE));
    }
    return builder.build();
}

From source file:com.searchbox.collection.oppfin.IdealISTCollection.java

private Document httpGet(RequestBuilder builder) {
    HttpClient client = HttpClientBuilder.create().build();

    HttpUriRequest request = builder
            .addHeader("User-Agent", env.getProperty(CRAWLER_USER_AGENT, CRAWLER_USER_AGENT_DEFAULT)).build();

    try {/*ww  w. j a  va2s.co m*/
        HttpResponse httpResponse = client.execute(request);
        InputStream ips = httpResponse.getEntity().getContent();
        BufferedReader buf = new BufferedReader(new InputStreamReader(ips, "UTF-8"));
        if (httpResponse.getStatusLine().getStatusCode() != HttpStatus.SC_OK) {
            LOGGER.error("could read response ({}) for url: {}", httpResponse.getStatusLine().getReasonPhrase(),
                    builder.getUri());
        }
        StringBuilder sb = new StringBuilder();
        String s;
        while (true) {
            s = buf.readLine();
            if (s == null || s.length() == 0)
                break;
            sb.append(s);

        }
        buf.close();
        ips.close();

        return buildXMLDocument(sb.toString());

    } catch (Exception e) {
        LOGGER.error("Could not get XML from {}", builder.getUri(), e);
    }
    return null;
}

From source file:com.restfiddle.handler.http.builder.RfRequestBuilder.java

private void setHeaders(RfRequestDTO requestDTO, RequestBuilder requestBuilder) {
    List<RfHeaderDTO> headers = requestDTO.getHeaders();
    if (headers != null && !headers.isEmpty()) {
        boolean contentTypeFound = false;
        for (RfHeaderDTO rfHeaderDTO : headers) {
            if (rfHeaderDTO.getHeaderName() != null && rfHeaderDTO.getHeaderValue() != null) {
                requestBuilder.addHeader(rfHeaderDTO.getHeaderName(), rfHeaderDTO.getHeaderValue());
            }/*from  ww  w . j a  v  a  2s.co  m*/
            if (HttpHeaders.CONTENT_TYPE.equalsIgnoreCase(rfHeaderDTO.getHeaderName())) {
                contentTypeFound = true;
            }
        }
        if (!contentTypeFound) {
            requestBuilder.addHeader(HttpHeaders.CONTENT_TYPE, MediaType.APPLICATION_JSON_VALUE);
        }
    } else {
        requestBuilder.addHeader(HttpHeaders.CONTENT_TYPE, MediaType.APPLICATION_JSON_VALUE);
    }
}