Examples with SinglePortRouteSource org.infinispan.server.router.routes.singleport.SinglePortRouteSource used on opensource projects

Example 6 with SinglePortRouteSource

use of org.infinispan.server.router.routes.singleport.SinglePortRouteSource in project infinispan by infinispan.

the class SinglePortTest method shouldUpgradeToHotRodThroughHTTP11UpgradeHeaders.

@Test
public void shouldUpgradeToHotRodThroughHTTP11UpgradeHeaders() {
    // given
    EmbeddedCacheManager cacheManager = TestCacheManagerFactory.createCacheManager(hotRodCacheConfiguration());
    // Initialize a transport-less Hot Rod server
    HotRodServerConfigurationBuilder hotRodServerBuilder = new HotRodServerConfigurationBuilder();
    hotRodServerBuilder.startTransport(false);
    hotRodServerBuilder.name(TestResourceTracker.getCurrentTestName());
    hotrodServer = HotRodClientTestingUtil.startHotRodServer(cacheManager, hotRodServerBuilder);
    // Initialize a transport-less REST server
    restServer = new RestServer();
    RestServerConfigurationBuilder restServerConfigurationBuilder = new RestServerConfigurationBuilder();
    restServerConfigurationBuilder.startTransport(false);
    restServerConfigurationBuilder.name(TestResourceTracker.getCurrentTestName());
    restServer.setServerManagement(new DummyServerManagement(), true);
    restServer.start(restServerConfigurationBuilder.build(), cacheManager);
    // Initialize a Single Port server with routes to the Hot Rod and REST servers
    HotRodServerRouteDestination hotrodDestination = new HotRodServerRouteDestination("hotrod", hotrodServer);
    RestServerRouteDestination restDestination = new RestServerRouteDestination("rest", restServer);
    SinglePortRouteSource singlePortSource = new SinglePortRouteSource();
    Route<SinglePortRouteSource, HotRodServerRouteDestination> routeToHotRod = new Route<>(singlePortSource, hotrodDestination);
    Route<SinglePortRouteSource, RestServerRouteDestination> routeToRest = new Route<>(singlePortSource, restDestination);
    RouterConfigurationBuilder routerConfigurationBuilder = new RouterConfigurationBuilder();
    routerConfigurationBuilder.singlePort().port(0).ip(InetAddress.getLoopbackAddress()).routing().add(routeToRest).add(routeToHotRod);
    router = new Router(routerConfigurationBuilder.build());
    router.start();
    EndpointRouter endpointRouter = router.getRouter(EndpointRouter.Protocol.SINGLE_PORT).get();
    String host = endpointRouter.getHost();
    int port = endpointRouter.getPort();
    // First off we verify that the HTTP side of things works
    RestClientConfigurationBuilder builder = new RestClientConfigurationBuilder();
    builder.addServer().host(host).port(port).protocol(Protocol.HTTP_11);
    httpClient = RestClient.forConfiguration(builder.build());
    CompletionStage<RestResponse> response = httpClient.cache(cacheManager.getCacheManagerConfiguration().defaultCacheName().get()).post("key", VALUE);
    ResponseAssertion.assertThat(response).hasNoContent();
    Assertions.assertThat(restServer.getCacheManager().getCache().size()).isEqualTo(1);
    // Next up, the RemoteCacheManager
    ConfigurationBuilder configurationBuilder = new ConfigurationBuilder();
    configurationBuilder.marshaller(new UTF8StringMarshaller());
    configurationBuilder.addServer().host(host).port(port);
    hotRodClient = new RemoteCacheManager(configurationBuilder.build());
    Object value = hotRodClient.getCache().withDataFormat(DataFormat.builder().keyType(TEXT_PLAIN).valueType(TEXT_PLAIN).build()).get("key");
    Assertions.assertThat(value).isEqualTo("test");
}