Examples of com.linkedin.r2.message.rest.RestRequestBuilder

com.linkedin.r2.message.rest.RestRequestBuilder
@author Chris Pettitt @version $Revision$

  }


  @Test
  public void testChainBuildRestRequestFromRestRequestBuilder()
  {
    final RestRequest req = new RestRequestBuilder(URI.create("test"))
            .setEntity(new byte[] {1,2,3,4})
            .setHeader("k1", "v1")
            .setMethod(RestMethod.PUT)
            .build()
            .builder()

View Full Code Here

  }


  @Test
  public void testChainBuildRestRequestFromRequestBuilder()
  {
    final Request req = new RestRequestBuilder(URI.create("test"))
            .setEntity(new byte[] {1,2,3,4})
            .setHeader("k1", "v1")
            .setMethod(RestMethod.PUT)
            .build()
            .requestBuilder()

View Full Code Here

  }


  @Test
  public void testChainBuildRestRequestFromRestBuilder()
  {
    final RestMessage req = new RestRequestBuilder(URI.create("test"))
            .setEntity(new byte[] {1,2,3,4})
            .setHeader("k1", "v1")
            .setMethod(RestMethod.PUT)
            .build()
            .restBuilder()

View Full Code Here

  }


  @Test
  public void testChainBuildRestRequestFromMessageBuilder()
  {
    final MessageBuilder<?> builder = new RestRequestBuilder(URI.create("test"))
            .setEntity(new byte[] {1,2,3,4})
            .setHeader("k1", "v1")
            .setMethod(RestMethod.PUT)
            .build()
            .builder();

View Full Code Here

      String serviceName = proportion.getKey();
      Random random = new Random();
      for (int i = 0; i < queryPerSecond; i++)
      {
        final URI uri = new URI("d2://" + serviceName + "?user=" + random.nextInt());
        RestRequestBuilder requestBuilder = new RestRequestBuilder(uri).setMethod("get");
        RestRequest request = requestBuilder.build();
        //we don't care about the result from the server after all,
        //you can see the traffic hits the echo server from stdout
        d2Client.restRequest(request, new Callback<RestResponse>()
        {
          @Override

View Full Code Here

      throws URISyntaxException
  {
    for (Map.Entry<String, Long> entry : trafficProportion.entrySet())
    {
      URI uri = new URI("d2://" + entry.getKey());
      RestRequest request = new RestRequestBuilder(uri).setMethod("get").build();
      for (long i = 0; i < entry.getValue(); i++)
      {
        //we don't care about the result from the server after all,
        //you can see the traffic hits the echo server from stdout
        d2Client.restRequest(request);

View Full Code Here

            .build();
  }


  public static RestRequest simpleRestRequest()
  {
    return new RestRequestBuilder(URI.create("simple_uri"))
            .build();
  }

View Full Code Here

        // Create a new request at this point from the debug request by removing the path suffix
        // starting with "__debug".
        String fullPath = request.getURI().getPath();
        int debugSegmentIndex = fullPath.indexOf(DEBUG_PATH_SEGMENT);


        RestRequestBuilder requestBuilder = new RestRequestBuilder(request);


        UriBuilder uriBuilder = UriBuilder.fromUri(request.getURI());
        uriBuilder.replacePath(request.getURI().getPath().substring(0, debugSegmentIndex - 1));
        requestBuilder.setURI(uriBuilder.build());


        handleResourceRequest(requestBuilder.build(), requestContext, callback, true);
      }
    }, callback);
  }

View Full Code Here

      default:
        throw new IllegalArgumentException("Unsupported enum value: " + _requestOptions.getProtocolVersionOption());
    }


    URI uri = RestliUriBuilderUtil.createUriBuilder(request, "", protocolVersion).build();
    RestRequestBuilder requestBuilder = new RestRequestBuilder(uri);
    requestBuilder.setMethod(request.getMethod().getHttpMethod().name());


    // unfortunately some headers get set in RestClient, and since we're not using rest client, we
    // replicate that behavior here
    requestBuilder.setHeader(RestConstants.HEADER_ACCEPT, RestConstants.HEADER_VALUE_APPLICATION_JSON);


    requestBuilder.setHeader(RestConstants.HEADER_RESTLI_PROTOCOL_VERSION, protocolVersion.toString());


    if (request.getMethod().getHttpMethod() == HttpMethod.POST)
    {
      requestBuilder.setHeader(RestConstants.HEADER_RESTLI_REQUEST_METHOD, request.getMethod().toString());
    }


    if (request.getInputRecord() != null)
    {
      requestBuilder.setHeader(RestConstants.HEADER_CONTENT_TYPE, RestConstants.HEADER_VALUE_APPLICATION_JSON);
      writeEntity(request, protocolVersion, requestBuilder);
    }


    return requestBuilder.build();
  }

View Full Code Here

  @Test
  public void testParseqTraceDebugGetRequestHandlerTracevis()
      throws URISyntaxException, ExecutionException, InterruptedException
  {
    RestRequest request =
        new RestRequestBuilder(
            new URI(URI_PREFIX + "greetingsPromise/1/__debug/parseqtrace/tracevis"))
              .setMethod("GET")
              .build();


    sendRequestAndVerifyParseqTracevisResponse(request);

View Full Code Here

0 1 2 3 4 5 6 7 8 9

TOP

Related Classes of com.linkedin.r2.message.rest.RestRequestBuilder

com.example.d2.client.ProfileClientExample

com.example.fortune.inject.FortunesGuiceServletConfig

com.example.fortune.RestLiFortunesClient

com.linkedin.d2.balancer.clients.DynamicClientTest

com.linkedin.d2.balancer.clients.RewriteClientTest

com.linkedin.d2.balancer.clients.TrackerClientTest

com.linkedin.d2.balancer.simple.SimpleLoadBalancerStrawMan

com.linkedin.d2.balancer.simple.SimpleLoadBalancerTest

com.linkedin.d2.balancer.strategies.degrader.DegraderLoadBalancerTest

com.linkedin.d2.balancer.util.LoadBalancerClientCli

All source code are property of their respective owners. Java is a trademark of Sun Microsystems, Inc and owned by ORACLE Inc. Contact coftware#gmail.com.