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

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

  private void expectRoutingException(String uri,
                                      ProtocolVersion version,
                                      String httpMethod,
                                      String restliMethod) throws URISyntaxException
  {
    RestRequestBuilder builder = createRequestBuilder(uri, httpMethod, version);
    if (restliMethod != null)
    {
      builder.setHeader("X-RestLi-Method", restliMethod);
    }
    RestRequest request = builder.build();
    try
    {
      RoutingResult r = _router.process(request, new RequestContext());
      fail("Expected RoutingException, got: " + r.toString());
    }

View Full Code Here

  }


  private RestRequestBuilder createRequestBuilder(String uri, String method, ProtocolVersion version)
    throws URISyntaxException
  {
    return new RestRequestBuilder(new URI(uri)).setMethod(method)
                                               .setHeader(RestConstants.HEADER_RESTLI_PROTOCOL_VERSION, version.toString());
  }

View Full Code Here

    );
  }


  private void fireRestRequest(FilterChain fc)
  {
    fc.onRestRequest(new RestRequestBuilder(URI.create("test")).build(),
                     createRequestContext(), createWireAttributes()
    );
  }

View Full Code Here

  @Test
  public void testRestToHttpErrorMessage() throws TimeoutException, InterruptedException
  {
    URI uri = URI.create("http://some.host/thisShouldAppearInTheErrorMessage");


    RestRequest r = new RestRequestBuilder(uri).build();


    FutureCallback<RestResponse> futureCallback = new FutureCallback<RestResponse>();
    TransportCallback<RestResponse> callback = new TransportCallbackAdapter<RestResponse>(futureCallback);
    TransportCallback<RestResponse> bridgeCallback = HttpBridge.restToHttpCallback(callback, r);

View Full Code Here

    for (Map.Entry<String, Map<String,Long>> d2Service : trafficProportion.entrySet())
    {
      for (Map.Entry<String, Long> partition : d2Service.getValue().entrySet())
      {
        final URI uri = new URI("d2://" + d2Service.getKey() + "?partitionId=" + partition.getKey());
        RestRequestBuilder requestBuilder = new RestRequestBuilder(uri).setMethod("get");
        if (delay != null)
        {
          requestBuilder.setHeader("delay", delay.toString());
        }
        RestRequest request = requestBuilder.build();
        Long queryPerSecond = partition.getValue();
        for (int i = 0; i < queryPerSecond; i++)
        {
          //we don't care about the result from the server after all,
          //you can see the traffic hits the echo server from stdout

View Full Code Here

      clients.add(new TransportClientAdapter(factory.getClient(Collections.<String, String>emptyMap())));
    }


    for (Client c : clients)
    {
      RestRequest r = new RestRequestBuilder(_testServer.getRequestURI()).build();
      c.restRequest(r).get(30, TimeUnit.SECONDS);
    }


    for (Client c : clients)
    {

View Full Code Here

      clients.add(new TransportClientAdapter(factory.getClient(Collections.<String, String>emptyMap())));
    }


    for (Client c : clients)
    {
      RestRequest r = new RestRequestBuilder(_testServer.getRequestURI()).build();
      c.restRequest(r).get(30, TimeUnit.SECONDS);
    }


    FutureCallback<None> factoryShutdown = new FutureCallback<None>();
    factory.shutdown(factoryShutdown);

View Full Code Here

      clients.add(new TransportClientAdapter(factory.getClient(Collections.<String, String>emptyMap())));
    }


    for (Client c : clients)
    {
      RestRequest r = new RestRequestBuilder(_testServer.getRequestURI()).build();
      c.restRequest(r).get(30, TimeUnit.SECONDS);
    }


    FutureCallback<None> factoryShutdown = new FutureCallback<None>();
    factory.shutdown(factoryShutdown, 1, TimeUnit.SECONDS);

View Full Code Here


  @Test(dataProvider = "headerConstant")
  public void testExtractProtocolVersion(String headerConstant) throws URISyntaxException
  {
    ProtocolVersion p1 = new ProtocolVersion("1.2.3");
    RestRequestBuilder requestBuilder = new RestRequestBuilder(new URI("/test/1")).setHeader(headerConstant, p1.toString());
    ProtocolVersion p2 = ProtocolVersionUtil.extractProtocolVersion(requestBuilder.getHeaders());
    Assert.assertEquals(p2, p1);
  }

View Full Code Here

      clients.add(new TransportClientAdapter(factory.getClient(Collections.<String, String>emptyMap())));
    }


    for (Client c : clients)
    {
      RestRequest r = new RestRequestBuilder(_testServer.getRequestURI()).build();
      c.restRequest(r).get(30, TimeUnit.SECONDS);
    }


    FutureCallback<None> factoryShutdown = new FutureCallback<None>();
    factory.shutdown(factoryShutdown);

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.