Examples with HotRodServerRouteDestination org.infinispan.server.router.routes.hotrod.HotRodServerRouteDestination used on opensource projects

Example 1 with HotRodServerRouteDestination

use of org.infinispan.server.router.routes.hotrod.HotRodServerRouteDestination in project infinispan by infinispan.

the class Server method run.

public synchronized CompletableFuture<ExitStatus> run() {
    CompletableFuture<ExitStatus> r = exitHandler.getExitFuture();
    if (status == ComponentStatus.RUNNING) {
        return r;
    }
    protocolServers = new ConcurrentHashMap<>(4);
    try {
        // Load any server extensions
        extensions = new Extensions();
        extensions.load(classLoader);
        // Create the cache manager
        cacheManager = new DefaultCacheManager(configurationBuilderHolder, false);
        // Retrieve the server configuration
        serverConfiguration = SecurityActions.getCacheManagerConfiguration(cacheManager).module(ServerConfiguration.class);
        serverConfiguration.setServer(this);
        // Initialize the data sources
        dataSources = new HashMap<>();
        InitialContext initialContext = new InitialContext();
        for (DataSourceConfiguration dataSourceConfiguration : serverConfiguration.dataSources().values()) {
            DataSource dataSource = DataSourceFactory.create(dataSourceConfiguration);
            dataSources.put(dataSourceConfiguration.name(), dataSource);
            initialContext.bind(dataSourceConfiguration.jndiName(), dataSource);
        }
        // Start the cache manager
        SecurityActions.startCacheManager(cacheManager);
        BasicComponentRegistry bcr = SecurityActions.getGlobalComponentRegistry(cacheManager).getComponent(BasicComponentRegistry.class.getName());
        blockingManager = bcr.getComponent(BlockingManager.class).running();
        serverStateManager = new ServerStateManagerImpl(this, cacheManager, bcr.getComponent(GlobalConfigurationManager.class).running());
        bcr.registerComponent(ServerStateManager.class, serverStateManager, false);
        ScheduledExecutorService timeoutExecutor = bcr.getComponent(KnownComponentNames.TIMEOUT_SCHEDULE_EXECUTOR, ScheduledExecutorService.class).running();
        // BlockingManager of single container used for writing the global manifest, but this will need to change
        // when multiple containers are supported by the server. Similarly, the default cache manager is used to create
        // the clustered locks.
        Path dataRoot = serverRoot.toPath().resolve(properties.getProperty(INFINISPAN_SERVER_DATA_PATH));
        backupManager = new BackupManagerImpl(blockingManager, cacheManager, dataRoot);
        backupManager.init();
        // Register the task manager
        taskManager = bcr.getComponent(TaskManager.class).running();
        taskManager.registerTaskEngine(extensions.getServerTaskEngine(cacheManager));
        // Initialize the OpenTracing integration
        RequestTracer.start();
        for (EndpointConfiguration endpoint : serverConfiguration.endpoints().endpoints()) {
            // Start the protocol servers
            SinglePortRouteSource routeSource = new SinglePortRouteSource();
            Set<Route<? extends RouteSource, ? extends RouteDestination>> routes = ConcurrentHashMap.newKeySet();
            endpoint.connectors().parallelStream().forEach(configuration -> {
                try {
                    Class<? extends ProtocolServer> protocolServerClass = configuration.getClass().getAnnotation(ConfigurationFor.class).value().asSubclass(ProtocolServer.class);
                    ProtocolServer protocolServer = Util.getInstance(protocolServerClass);
                    protocolServer.setServerManagement(this, endpoint.admin());
                    if (configuration instanceof HotRodServerConfiguration) {
                        ElytronSASLAuthenticationProvider.init((HotRodServerConfiguration) configuration, serverConfiguration, timeoutExecutor);
                    } else if (configuration instanceof RestServerConfiguration) {
                        ElytronHTTPAuthenticator.init((RestServerConfiguration) configuration, serverConfiguration);
                    } else if (configuration instanceof RespServerConfiguration) {
                        ElytronRESPAuthenticator.init((RespServerConfiguration) configuration, serverConfiguration, blockingManager);
                    }
                    protocolServers.put(protocolServer.getName() + "-" + configuration.name(), protocolServer);
                    SecurityActions.startProtocolServer(protocolServer, configuration, cacheManager);
                    ProtocolServerConfiguration protocolConfig = protocolServer.getConfiguration();
                    if (protocolConfig.startTransport()) {
                        log.protocolStarted(protocolServer.getName(), configuration.socketBinding(), protocolConfig.host(), protocolConfig.port());
                    } else {
                        if (protocolServer instanceof HotRodServer) {
                            routes.add(new Route<>(routeSource, new HotRodServerRouteDestination(protocolServer.getName(), (HotRodServer) protocolServer)));
                            extensions.apply((HotRodServer) protocolServer);
                        } else if (protocolServer instanceof RestServer) {
                            routes.add(new Route<>(routeSource, new RestServerRouteDestination(protocolServer.getName(), (RestServer) protocolServer)));
                        } else if (protocolServer instanceof RespServer) {
                            routes.add(new Route<>(routeSource, new RespServerRouteDestination(protocolServer.getName(), (RespServer) protocolServer)));
                        }
                        log.protocolStarted(protocolServer.getName());
                    }
                } catch (Throwable t) {
                    throw t instanceof RuntimeException ? (RuntimeException) t : new RuntimeException(t);
                }
            });
            // Next we start the single-port endpoints
            SinglePortRouterConfiguration singlePortRouter = endpoint.singlePortRouter();
            SinglePortEndpointRouter endpointServer = new SinglePortEndpointRouter(singlePortRouter);
            endpointServer.start(new RoutingTable(routes), cacheManager);
            protocolServers.put("endpoint-" + endpoint.socketBinding(), endpointServer);
            log.protocolStarted(endpointServer.getName(), singlePortRouter.socketBinding(), singlePortRouter.host(), singlePortRouter.port());
            log.endpointUrl(Util.requireNonNullElse(cacheManager.getAddress(), "local"), singlePortRouter.ssl().enabled() ? "https" : "http", singlePortRouter.host(), singlePortRouter.port());
        }
        serverStateManager.start();
        // Change status
        this.status = ComponentStatus.RUNNING;
        log.serverStarted(Version.getBrandName(), Version.getBrandVersion(), timeService.timeDuration(startTime, TimeUnit.MILLISECONDS));
    } catch (Exception e) {
        r.completeExceptionally(e);
    }
    r = r.handle((status, t) -> {
        if (t != null) {
            Server.log.serverFailedToStart(Version.getBrandName(), t);
        }
        localShutdown(status);
        return null;
    });
    return r;
}

Example 2 with HotRodServerRouteDestination

use of org.infinispan.server.router.routes.hotrod.HotRodServerRouteDestination in project infinispan by infinispan.

the class SniRouteHandler method decode.

@Override
protected void decode(ChannelHandlerContext ctx, ByteBuf in, List<Object> out) throws Exception {
    super.decode(ctx, in, out);
    if (isHandShaked()) {
        // At this point Netty has replaced SNIHandler (formally this) with SSLHandler in the pipeline.
        // Now we need to add other handlers at the tail of the queue
        RouterLogger.SERVER.debugf("Handshaked with hostname %s", hostname());
        Optional<Route<SniRouteSource, HotRodServerRouteDestination>> route = routingTable.streamRoutes(SniRouteSource.class, HotRodServerRouteDestination.class).filter(r -> r.getRouteSource().getSniHostName().equals(this.hostname())).findAny();
        HotRodServerRouteDestination routeDestination = route.orElseThrow(() -> RouterLogger.SERVER.noRouteFound()).getRouteDestination();
        ChannelInitializer<Channel> channelInitializer = routeDestination.getProtocolServer().getInitializer();
        ctx.pipeline().addLast(channelInitializer);
        RouterLogger.SERVER.debug("Replaced with route destination's handlers");
    }
}

Example 3 with HotRodServerRouteDestination

use of org.infinispan.server.router.routes.hotrod.HotRodServerRouteDestination in project infinispan by infinispan.

the class ProtocolServerEndpointRouterTest method shouldRouteToProperHotRodServerBasedOnSniHostName.

/**
 * In this scenario we create 2 HotRod servers, each one with different credentials and SNI name. We also create a
 * new client for each server. The clients use proper TrustStores as well as SNI names.
 * <p>
 * The router should match properly SNI based routes and connect clients to proper server instances.
 */
@Test
public void shouldRouteToProperHotRodServerBasedOnSniHostName() {
    // given
    hotrodServer1 = HotRodTestingUtil.startHotRodServerWithoutTransport("default");
    hotrodServer2 = HotRodTestingUtil.startHotRodServerWithoutTransport("default");
    HotRodServerRouteDestination hotrod1Destination = new HotRodServerRouteDestination("HotRod1", hotrodServer1);
    SniNettyRouteSource hotrod1Source = new SniNettyRouteSource("hotrod1", KEYSTORE_LOCATION_FOR_HOTROD_1, "secret".toCharArray());
    Route<SniNettyRouteSource, HotRodServerRouteDestination> routeToHotrod1 = new Route<>(hotrod1Source, hotrod1Destination);
    HotRodServerRouteDestination hotrod2Destination = new HotRodServerRouteDestination("HotRod2", hotrodServer2);
    SniNettyRouteSource hotrod2Source = new SniNettyRouteSource("hotrod2", KEYSTORE_LOCATION_FOR_HOTROD_2, "secret".toCharArray());
    Route<SniNettyRouteSource, HotRodServerRouteDestination> routeToHotrod2 = new Route<>(hotrod2Source, hotrod2Destination);
    RouterConfigurationBuilder routerConfigurationBuilder = new RouterConfigurationBuilder();
    routerConfigurationBuilder.hotrod().port(0).ip(InetAddress.getLoopbackAddress()).routing().add(routeToHotrod1).add(routeToHotrod2);
    router = new Router(routerConfigurationBuilder.build());
    router.start();
    InetAddress routerIp = router.getRouter(EndpointRouter.Protocol.HOT_ROD).get().getIp();
    int routerPort = router.getRouter(EndpointRouter.Protocol.HOT_ROD).get().getPort();
    // when
    hotrod1Client = HotRodClientTestingUtil.createWithSni(routerIp, routerPort, "hotrod1", TRUSTSTORE_LOCATION_FOR_HOTROD_1, "secret".toCharArray());
    hotrod2Client = HotRodClientTestingUtil.createWithSni(routerIp, routerPort, "hotrod2", TRUSTSTORE_LOCATION_FOR_HOTROD_2, "secret".toCharArray());
    hotrod1Client.getCache("default").put("test", "hotrod1");
    hotrod2Client.getCache("default").put("test", "hotrod2");
    // then
    Cache<String, String> hotrod1Cache = hotrodServer1.getCacheManager().getCache("default");
    Cache<String, String> hotrod2Cache = hotrodServer2.getCacheManager().getCache("default");
    assertThat(hotrod1Cache.size()).isEqualTo(1);
    assertThat(hotrod2Cache.size()).isEqualTo(1);
    assertThat(hotrod1Cache.get("test")).isEqualTo("hotrod1");
    assertThat(hotrod2Cache.get("test")).isEqualTo("hotrod2");
}

Example 4 with HotRodServerRouteDestination

use of org.infinispan.server.router.routes.hotrod.HotRodServerRouteDestination in project infinispan by infinispan.

the class TwoServersWithSslSni method initRoutes.

@Override
public Optional<Set<Route<? extends RouteSource, ? extends RouteDestination>>> initRoutes(List<HotRodServer> servers) {
    Set<Route<? extends RouteSource, ? extends RouteDestination>> routes = new HashSet<>();
    HotRodServerRouteDestination hotrod1Destination = new HotRodServerRouteDestination("hotrod1", servers.get(0));
    SniNettyRouteSource hotrod1Source = new SniNettyRouteSource("hotrod1", KEYSTORE_LOCATION_FOR_HOTROD_1, KEYSTORE_PASSWORD);
    routes.add(new Route<>(hotrod1Source, hotrod1Destination));
    HotRodServerRouteDestination hotrod2Destination = new HotRodServerRouteDestination("hotrod2", servers.get(1));
    SniNettyRouteSource hotrod2Source = new SniNettyRouteSource("hotrod2", KEYSTORE_LOCATION_FOR_HOTROD_2, KEYSTORE_PASSWORD);
    routes.add(new Route<>(hotrod2Source, hotrod2Destination));
    return Optional.of(routes);
}

Example 5 with HotRodServerRouteDestination

use of org.infinispan.server.router.routes.hotrod.HotRodServerRouteDestination in project keycloak by keycloak.

the class HotRodUtils method createHotRodMapStoreServer.

/**
 * Not suitable for a production usage. Only for development and test purposes.
 * Also do not use in clustered environment.
 * @param hotRodServer HotRodServer
 * @param hotRodCacheManager DefaultCacheManager
 * @param embeddedPort int
 */
public static void createHotRodMapStoreServer(HotRodServer hotRodServer, DefaultCacheManager hotRodCacheManager, int embeddedPort) {
    HotRodServerConfigurationBuilder hotRodServerConfigurationBuilder = new HotRodServerConfigurationBuilder();
    hotRodServerConfigurationBuilder.startTransport(false);
    hotRodServerConfigurationBuilder.port(embeddedPort);
    hotRodServer.start(hotRodServerConfigurationBuilder.build(), hotRodCacheManager);
    RestServerConfigurationBuilder restServerConfigurationBuilder = new RestServerConfigurationBuilder();
    restServerConfigurationBuilder.startTransport(false);
    restServerConfigurationBuilder.port(embeddedPort);
    RestServer restServer = new RestServer();
    restServer.start(restServerConfigurationBuilder.build(), hotRodCacheManager);
    SinglePortRouteSource routeSource = new SinglePortRouteSource();
    Set<Route<? extends RouteSource, ? extends RouteDestination>> routes = new HashSet<>();
    routes.add(new Route<>(routeSource, new HotRodServerRouteDestination("hotrod", hotRodServer)));
    routes.add(new Route<>(routeSource, new RestServerRouteDestination("rest", restServer)));
    SinglePortRouterConfiguration singlePortRouter = new SinglePortServerConfigurationBuilder().port(embeddedPort).build();
    SinglePortEndpointRouter endpointServer = new SinglePortEndpointRouter(singlePortRouter);
    endpointServer.start(new RoutingTable(routes));
}