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

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

Introduction

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

Prototype

public RequestBuilder setUri(final String uri) 

Source Link

Usage

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

/**
 * Build a {@link HttpUriRequest} from a {@link GenericRecord}
 *///from   www . j av a2  s . c om
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: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 w w w .ja v a 2 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:org.dasein.cloud.azurepack.platform.AzurePackDatabaseRequests.java

private RequestBuilder getListRequestBuilder(String uri) {
    RequestBuilder requestBuilder = RequestBuilder.get();
    addCommonHeaders(requestBuilder);//  w  w  w.  j  a v  a 2s . c  o  m
    requestBuilder.setUri(String.format(uri, this.provider.getContext().getCloud().getEndpoint(),
            this.provider.getContext().getAccountNumber()));
    return requestBuilder;
}

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

private RequestBuilder getCreateDBRequestBuilder(String uri, WAPDatabaseModel databaseModel) {
    RequestBuilder requestBuilder = RequestBuilder.post();
    addCommonHeaders(requestBuilder);/*  w w  w  .java 2s . co m*/
    requestBuilder.setUri(String.format(uri, this.provider.getContext().getCloud().getEndpoint(),
            this.provider.getContext().getAccountNumber()));
    requestBuilder.setEntity(new DaseinObjectToJsonEntity<WAPDatabaseModel>(databaseModel));
    return requestBuilder;
}

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

private RequestBuilder getDeleteDBRequestBuilder(String uri, String databaseName) {
    RequestBuilder requestBuilder = RequestBuilder.delete();
    addCommonHeaders(requestBuilder);//from   w  w w  .  j  av  a2 s.  com
    requestBuilder.setUri(String.format(uri, this.provider.getContext().getCloud().getEndpoint(),
            this.provider.getContext().getAccountNumber(), databaseName));
    return requestBuilder;
}

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

/**
 * Build a write request from a single record
 *//*from w  w w  . j  av  a2s.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:org.dasein.cloud.azure.platform.AzureSQLDatabaseSupportRequests.java

public RequestBuilder listServers() {
    RequestBuilder requestBuilder = RequestBuilder.get();
    addAzureCommonHeaders(requestBuilder);
    requestBuilder.setUri(String.format(RESOURCE_SERVERS, this.provider.getContext().getAccountNumber()));
    return requestBuilder;
}

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

public RequestBuilder listServersNonGen() {
    RequestBuilder requestBuilder = RequestBuilder.get();
    addAzureCommonHeaders(requestBuilder);
    requestBuilder
            .setUri(String.format(RESOURCE_SERVERS_NONGEN, this.provider.getContext().getAccountNumber()));
    return requestBuilder;
}

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

public RequestBuilder listDatabases(String serverName) {
    RequestBuilder requestBuilder = RequestBuilder.get();
    addAzureCommonHeaders(requestBuilder);
    requestBuilder.setUri(
            String.format(RESOURCE_LIST_DATABASES, this.provider.getContext().getAccountNumber(), serverName));
    return requestBuilder;
}

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

public RequestBuilder subscriptionMetaRequest() {
    RequestBuilder requestBuilder = RequestBuilder.get();
    addAzureCommonHeaders(requestBuilder);
    requestBuilder
            .setUri(String.format(RESOURCE_SUBSCRIPTION_META, this.provider.getContext().getAccountNumber()));
    return requestBuilder;
}