Search in sources :

Example 1 with RestRouteSource

use of org.infinispan.server.router.routes.rest.RestRouteSource in project infinispan by infinispan.

the class RestEndpointRouterTest method shouldRouteToProperRestServerBasedOnPath.

/**
 * In this scenario we create 2 REST servers, each one with different REST Path: <ul> <li>REST1 -
 * http://127.0.0.1:8080/rest/rest1</li> <li>REST2 - http://127.0.0.1:8080/rest/rest2</li> </ul>
 * <p>
 * The router should match requests based on path and redirect them to proper server.
 */
@Test
public void shouldRouteToProperRestServerBasedOnPath() {
    // given
    restServer1 = RestTestingUtil.createDefaultRestServer("rest1", "default");
    restServer2 = RestTestingUtil.createDefaultRestServer("rest2", "default");
    RestServerRouteDestination rest1Destination = new RestServerRouteDestination("rest1", restServer1);
    RestRouteSource rest1Source = new RestRouteSource("rest1");
    Route<RestRouteSource, RestServerRouteDestination> routeToRest1 = new Route<>(rest1Source, rest1Destination);
    RestServerRouteDestination rest2Destination = new RestServerRouteDestination("rest2", restServer2);
    RestRouteSource rest2Source = new RestRouteSource("rest2");
    Route<RestRouteSource, RestServerRouteDestination> routeToRest2 = new Route<>(rest2Source, rest2Destination);
    RouterConfigurationBuilder routerConfigurationBuilder = new RouterConfigurationBuilder();
    routerConfigurationBuilder.rest().port(8080).ip(InetAddress.getLoopbackAddress()).routing().add(routeToRest1).add(routeToRest2);
    router = new Router(routerConfigurationBuilder.build());
    router.start();
    int port = router.getRouter(EndpointRouter.Protocol.REST).get().getPort();
    // when
    ServerConfigurationBuilder builder = new RestClientConfigurationBuilder().addServer().host("127.0.0.1").port(port);
    restClient = RestClient.forConfiguration(builder.build());
    RestRawClient rawClient = restClient.raw();
    String path1 = "/rest/rest1/v2/caches/default/test";
    String path2 = "/rest/rest2/v2/caches/default/test";
    join(rawClient.putValue(path1, emptyMap(), "rest1", TEXT_PLAIN_TYPE));
    join(rawClient.putValue(path2, emptyMap(), "rest2", TEXT_PLAIN_TYPE));
    String valueReturnedFromRest1 = join(rawClient.get(path1)).getBody();
    String valueReturnedFromRest2 = join(rawClient.get(path2)).getBody();
    // then
    assertThat(valueReturnedFromRest1).isEqualTo("rest1");
    assertThat(valueReturnedFromRest2).isEqualTo("rest2");
}
Also used : RestClientConfigurationBuilder(org.infinispan.client.rest.configuration.RestClientConfigurationBuilder) RestRawClient(org.infinispan.client.rest.RestRawClient) RestRouteSource(org.infinispan.server.router.routes.rest.RestRouteSource) Router(org.infinispan.server.router.Router) EndpointRouter(org.infinispan.server.router.router.EndpointRouter) RestServerRouteDestination(org.infinispan.server.router.routes.rest.RestServerRouteDestination) RouterConfigurationBuilder(org.infinispan.server.router.configuration.builder.RouterConfigurationBuilder) Route(org.infinispan.server.router.routes.Route) ServerConfigurationBuilder(org.infinispan.client.rest.configuration.ServerConfigurationBuilder) Test(org.junit.Test)

Aggregations

RestRawClient (org.infinispan.client.rest.RestRawClient)1 RestClientConfigurationBuilder (org.infinispan.client.rest.configuration.RestClientConfigurationBuilder)1 ServerConfigurationBuilder (org.infinispan.client.rest.configuration.ServerConfigurationBuilder)1 Router (org.infinispan.server.router.Router)1 RouterConfigurationBuilder (org.infinispan.server.router.configuration.builder.RouterConfigurationBuilder)1 EndpointRouter (org.infinispan.server.router.router.EndpointRouter)1 Route (org.infinispan.server.router.routes.Route)1 RestRouteSource (org.infinispan.server.router.routes.rest.RestRouteSource)1 RestServerRouteDestination (org.infinispan.server.router.routes.rest.RestServerRouteDestination)1 Test (org.junit.Test)1