Examples with Server org.apache.druid.client.selector.Server used on opensource projects

Example 16 with Server

use of org.apache.druid.client.selector.Server in project druid by druid-io.

the class QueryHostFinder method pickDefaultServer.

public Server pickDefaultServer() {
    Server server = findDefaultServer();
    assertServerFound(server, "No default server found!");
    return server;
}

Example 17 with Server

use of org.apache.druid.client.selector.Server in project druid by druid-io.

the class QueryHostFinder method findServerAvatica.

public Server findServerAvatica(String connectionId) {
    Server chosenServer = avaticaConnectionBalancer.pickServer(getAllServers(), connectionId);
    assertServerFound(chosenServer, "No server found for Avatica request with connectionId[%s]", connectionId);
    log.debug("Balancer class [%s] sending request with connectionId[%s] to server: %s", avaticaConnectionBalancer.getClass(), connectionId, chosenServer.getHost());
    return chosenServer;
}

Example 18 with Server

use of org.apache.druid.client.selector.Server in project druid by druid-io.

the class QueryHostFinder method findServerInner.

private Server findServerInner(final Pair<String, Server> selected) {
    if (selected == null) {
        log.error("Unable to find any brokers!");
    }
    final String serviceName = selected == null ? hostSelector.getDefaultServiceName() : selected.lhs;
    Server server = selected == null ? null : selected.rhs;
    if (server == null) {
        log.error("No server found for serviceName[%s]. Using backup", serviceName);
        server = serverBackup.get(serviceName);
        if (server == null) {
            log.error("No backup found for serviceName[%s]. Using default[%s]", serviceName, hostSelector.getDefaultServiceName());
            server = serverBackup.get(hostSelector.getDefaultServiceName());
        }
    }
    if (server != null) {
        serverBackup.put(serviceName, server);
    }
    return server;
}

Example 19 with Server

use of org.apache.druid.client.selector.Server in project druid by druid-io.

the class QueryHostFinder method pickServer.

public <T> Server pickServer(Query<T> query) {
    Server server = findServer(query);
    assertServerFound(server, "No server found for query[%s]", query);
    return server;
}

Example 20 with Server

use of org.apache.druid.client.selector.Server in project druid by druid-io.

the class QueryHostFinderTest method setUp.

@Before
public void setUp() {
    brokerSelector = EasyMock.createMock(TieredBrokerHostSelector.class);
    server = new Server() {

        @Override
        public String getScheme() {
            return null;
        }

        @Override
        public String getHost() {
            return "foo";
        }

        @Override
        public String getAddress() {
            return null;
        }

        @Override
        public int getPort() {
            return 0;
        }
    };
    EasyMock.expect(brokerSelector.select(EasyMock.anyObject(Query.class))).andReturn(Pair.of("service", server));
    EasyMock.replay(brokerSelector);
}