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

Introduction

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

Prototype

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

Source Link

Usage

From source file:com.nexmo.client.auth.JWTAuthMethod.java

@Override
public RequestBuilder apply(RequestBuilder request) {
    String token = this.constructToken(System.currentTimeMillis() / 1000L, constructJTI());
    request.setHeader("Authorization", "Bearer " + token);
    return request;
}

---

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

/**
 * Add payload to request. By default, payload is sent as application/json
 *//*  w  ww.ja v a 2 s.  c om*/
protected int addPayload(RequestBuilder builder, String payload) {
    if (payload == null || payload.length() == 0) {
        return 0;
    }

    builder.setHeader(HttpHeaders.CONTENT_TYPE, contentType.getMimeType());
    builder.setEntity(new StringEntity(payload, contentType));
    return payload.length();
}

---

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

/**
 * Build a write request from a single record
 *///from w ww. jav a 2  s. c  o  m
private ApacheHttpRequest<GenericRecord> buildWriteRequest(BufferedRecord<GenericRecord> record) {
    if (record == null) {
        return null;
    }

    ApacheHttpRequest<GenericRecord> request = new ApacheHttpRequest<>();
    HttpOperation httpOperation = HttpUtils.toHttpOperation(record.getRecord());

    // Set uri
    URI uri = HttpUtils.buildURI(urlTemplate, httpOperation.getKeys(), httpOperation.getQueryParams());
    if (uri == null) {
        return null;
    }

    RequestBuilder builder = RequestBuilder.create(verb.toUpperCase());
    builder.setUri(uri);

    // Set headers
    Map<String, String> headers = httpOperation.getHeaders();
    if (headers != null && headers.size() != 0) {
        for (Map.Entry<String, String> header : headers.entrySet()) {
            builder.setHeader(header.getKey(), header.getValue());
        }
    }

    // Add payload
    int bytesWritten = addPayload(builder, httpOperation.getBody());
    if (bytesWritten == -1) {
        throw new RuntimeException("Fail to write payload into request");
    }

    request.setRawRequest(build(builder));
    request.markRecord(record, bytesWritten);
    return request;
}

---

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

private void setRequestEntity(RfRequestDTO requestDTO, RequestBuilder requestBuilder) {
    List<FormDataDTO> formParams = requestDTO.getFormParams();
    if (requestDTO.getApiBody() != null && !requestDTO.getApiBody().isEmpty()) {
        try {/*ww  w . j  a v a  2  s.com*/
            requestBuilder.setEntity(new StringEntity(requestDTO.getApiBody()));

        } catch (UnsupportedEncodingException e) {
            logger.error(e.getMessage(), e);
        }
    } else if (formParams != null && !formParams.isEmpty()) {
        // NOTE : http://stackoverflow.com/questions/12745710/apache-httpclient-4-2-1-post-request-to-fill-form-after-successful-login
        requestBuilder.setHeader(HttpHeaders.CONTENT_TYPE, MediaType.APPLICATION_FORM_URLENCODED_VALUE);
        List<NameValuePair> nvps2 = new ArrayList<NameValuePair>();
        for (FormDataDTO formDataDTO : formParams) {
            nvps2.add(new BasicNameValuePair(formDataDTO.getKey(), formDataDTO.getValue()));
        }
        try {
            requestBuilder.setEntity(new UrlEncodedFormEntity(nvps2));
        } catch (UnsupportedEncodingException e) {
        }
    }
}

---

From source file:org.eclipse.rdf4j.http.client.RDF4JProtocolSession.java

@Override
protected HttpUriRequest getQueryMethod(QueryLanguage ql, String query, String baseURI, Dataset dataset,
        boolean includeInferred, int maxQueryTime, Binding... bindings) {
    RequestBuilder builder = null;
    if (transactionURL != null) {
        builder = RequestBuilder.put(transactionURL);
        builder.setHeader("Content-Type", Protocol.SPARQL_QUERY_MIME_TYPE + "; charset=utf-8");
        builder.addParameter(Protocol.ACTION_PARAM_NAME, Action.QUERY.toString());
        for (NameValuePair nvp : getQueryMethodParameters(ql, null, baseURI, dataset, includeInferred,
                maxQueryTime, bindings)) {
            builder.addParameter(nvp);/*  ww  w .j a  va  2s . c  o m*/
        }
        // in a PUT request, we carry the actual query string as the entity
        // body rather than a parameter.
        builder.setEntity(new StringEntity(query, UTF8));
    } else {
        builder = RequestBuilder.post(getQueryURL());
        builder.setHeader("Content-Type", Protocol.FORM_MIME_TYPE + "; charset=utf-8");

        builder.setEntity(new UrlEncodedFormEntity(
                getQueryMethodParameters(ql, query, baseURI, dataset, includeInferred, maxQueryTime, bindings),
                UTF8));
    }
    // functionality to provide custom http headers as required by the
    // applications
    for (Map.Entry<String, String> additionalHeader : getAdditionalHttpHeaders().entrySet()) {
        builder.addHeader(additionalHeader.getKey(), additionalHeader.getValue());
    }
    return builder.build();
}

---

From source file:cn.wanghaomiao.seimi.http.hc.HcRequestGenerator.java

public static RequestBuilder getHttpRequestBuilder(Request request, CrawlerModel crawlerModel) {
    RequestBuilder requestBuilder;
    BaseSeimiCrawler crawler = crawlerModel.getInstance();
    if (request.isUseSeimiAgent()) {
        if (StringUtils.isBlank(crawler.seimiAgentHost())) {
            throw new SeimiProcessExcepiton("SeimiAgentHost is blank.");
        }/*from   w  ww  .  jav  a 2s .co  m*/
        String seimiAgentUrl = "http://" + crawler.seimiAgentHost()
                + (crawler.seimiAgentPort() != 80 ? (":" + crawler.seimiAgentPort()) : "") + "/doload";
        requestBuilder = RequestBuilder.post().setUri(seimiAgentUrl);
        requestBuilder.addParameter("url", request.getUrl());
        if (StringUtils.isNotBlank(crawler.proxy())) {
            requestBuilder.addParameter("proxy", crawler.proxy());
        }
        if (request.getSeimiAgentRenderTime() > 0) {
            requestBuilder.addParameter("renderTime", String.valueOf(request.getSeimiAgentRenderTime()));
        }
        if (StringUtils.isNotBlank(request.getSeimiAgentScript())) {
            requestBuilder.addParameter("script", request.getSeimiAgentScript());
        }
        //SeimiAgent?cookie
        if ((request.isSeimiAgentUseCookie() == null && crawlerModel.isUseCookie())
                || (request.isSeimiAgentUseCookie() != null && request.isSeimiAgentUseCookie())) {
            requestBuilder.addParameter("useCookie", "1");
        }
        if (request.getParams() != null && request.getParams().size() > 0) {
            requestBuilder.addParameter("postParam", JSON.toJSONString(request.getParams()));
        }
        if (request.getSeimiAgentContentType().val() > SeimiAgentContentType.HTML.val()) {
            requestBuilder.addParameter("contentType", request.getSeimiAgentContentType().typeVal());
        }
    } else {
        if (HttpMethod.POST.equals(request.getHttpMethod())) {
            requestBuilder = RequestBuilder.post().setUri(request.getUrl());
        } else {
            requestBuilder = RequestBuilder.get().setUri(request.getUrl());
        }
        RequestConfig config = RequestConfig.custom().setProxy(crawlerModel.getProxy())
                .setCircularRedirectsAllowed(true).build();

        if (request.getParams() != null) {
            for (Map.Entry<String, String> entry : request.getParams().entrySet()) {
                requestBuilder.addParameter(entry.getKey(), entry.getValue());
            }
        }
        requestBuilder.setConfig(config).setHeader("User-Agent",
                crawlerModel.isUseCookie() ? crawlerModel.getCurrentUA() : crawler.getUserAgent());
        requestBuilder.setHeader("Accept",
                "text/html,application/xhtml+xml,application/xml;q=0.9,image/webp,*/*;q=0.8");
        requestBuilder.setHeader("Accept-Language", "zh-CN,zh;q=0.8,en;q=0.6");
    }
    if (!CollectionUtils.isEmpty(request.getHeader())) {
        for (Map.Entry<String, String> entry : request.getHeader().entrySet()) {
            requestBuilder.setHeader(entry.getKey(), entry.getValue());
        }
    }
    return requestBuilder;
}

---

From source file:com.hp.octane.integrations.services.rest.OctaneRestClientImpl.java

/**
 * This method should be the ONLY mean that creates Http Request objects
 *
 * @param octaneRequest Request data as it is maintained in Octane related flavor
 * @return pre-configured HttpUriRequest
 *///w  w w.j a  va  2 s. com
private HttpUriRequest createHttpRequest(OctaneRequest octaneRequest) {
    HttpUriRequest request;
    RequestBuilder requestBuilder;

    //  create base request by METHOD
    if (octaneRequest.getMethod().equals(HttpMethod.GET)) {
        requestBuilder = RequestBuilder.get(octaneRequest.getUrl());
    } else if (octaneRequest.getMethod().equals(HttpMethod.DELETE)) {
        requestBuilder = RequestBuilder.delete(octaneRequest.getUrl());
    } else if (octaneRequest.getMethod().equals(HttpMethod.POST)) {
        requestBuilder = RequestBuilder.post(octaneRequest.getUrl());
        requestBuilder
                .addHeader(new BasicHeader(RestService.CONTENT_ENCODING_HEADER, RestService.GZIP_ENCODING));
        requestBuilder.setEntity(new GzipCompressingEntity(
                new InputStreamEntity(octaneRequest.getBody(), ContentType.APPLICATION_JSON)));
    } else if (octaneRequest.getMethod().equals(HttpMethod.PUT)) {
        requestBuilder = RequestBuilder.put(octaneRequest.getUrl());
        requestBuilder
                .addHeader(new BasicHeader(RestService.CONTENT_ENCODING_HEADER, RestService.GZIP_ENCODING));
        requestBuilder.setEntity(new GzipCompressingEntity(
                new InputStreamEntity(octaneRequest.getBody(), ContentType.APPLICATION_JSON)));
    } else {
        throw new RuntimeException("HTTP method " + octaneRequest.getMethod() + " not supported");
    }

    //  set custom headers
    if (octaneRequest.getHeaders() != null) {
        for (Map.Entry<String, String> e : octaneRequest.getHeaders().entrySet()) {
            requestBuilder.setHeader(e.getKey(), e.getValue());
        }
    }

    //  set system headers
    requestBuilder.setHeader(CLIENT_TYPE_HEADER, CLIENT_TYPE_VALUE);

    request = requestBuilder.build();
    return request;
}

---