Examples with ServerLocation org.apache.geode.distributed.internal.ServerLocation used on opensource projects

Example 36 with ServerLocation

use of org.apache.geode.distributed.internal.ServerLocation in project geode by apache.

the class ConnectionFactoryImpl method createClientToServerConnection.

public Connection createClientToServerConnection(Set excludedServers) throws GemFireSecurityException {
    final Set origExcludedServers = excludedServers;
    excludedServers = new HashSet(excludedServers);
    Set blackListedServers = blackList.getBadServers();
    excludedServers.addAll(blackListedServers);
    Connection conn = null;
    // long startTime = System.currentTimeMillis();
    RuntimeException fatalException = null;
    boolean tryBlackList = true;
    do {
        ServerLocation server = source.findServer(excludedServers);
        if (server == null) {
            if (tryBlackList) {
                // Nothing worked! Let's try without the blacklist.
                tryBlackList = false;
                int size = excludedServers.size();
                excludedServers.removeAll(blackListedServers);
                // make sure we didn't remove any of the ones that the caller set not to use
                excludedServers.addAll(origExcludedServers);
                if (excludedServers.size() < size) {
                    // We are able to remove some exclusions, so lets give this another whirl.
                    continue;
                }
            }
            if (logger.isDebugEnabled()) {
                logger.debug("Source was unable to locate any servers");
            }
            if (fatalException != null) {
                throw fatalException;
            }
            return null;
        }
        try {
            conn = createClientToServerConnection(server, false);
        } catch (CancelException e) {
            // propagate this up immediately
            throw e;
        } catch (GemFireSecurityException e) {
            // propagate this up immediately
            throw e;
        } catch (GatewayConfigurationException e) {
            // propagate this up immediately
            throw e;
        } catch (ServerRefusedConnectionException srce) {
            fatalException = srce;
            if (logger.isDebugEnabled()) {
                logger.debug("ServerRefusedConnectionException attempting to connect to {}", server, srce);
            }
        } catch (Exception e) {
            logger.warn(LocalizedMessage.create(LocalizedStrings.ConnectException_COULD_NOT_CONNECT_TO_0, server), e);
        }
        excludedServers.add(server);
    } while (conn == null);
    // }
    return conn;
}

Example 37 with ServerLocation

use of org.apache.geode.distributed.internal.ServerLocation in project geode by apache.

the class ConnectionFactoryImpl method findBestServer.

public ServerLocation findBestServer(ServerLocation currentServer, Set excludedServers) {
    if (currentServer != null && source.isBalanced()) {
        return currentServer;
    }
    final Set origExcludedServers = excludedServers;
    excludedServers = new HashSet(excludedServers);
    Set blackListedServers = blackList.getBadServers();
    excludedServers.addAll(blackListedServers);
    ServerLocation server = source.findReplacementServer(currentServer, excludedServers);
    if (server == null) {
        // Nothing worked! Let's try without the blacklist.
        if (excludedServers.size() > origExcludedServers.size()) {
            // We had some guys black listed so lets give this another whirl.
            server = source.findReplacementServer(currentServer, origExcludedServers);
        }
    }
    if (server == null && logger.isDebugEnabled()) {
        logger.debug("Source was unable to findForReplacement any servers");
    }
    return server;
}

Example 38 with ServerLocation

use of org.apache.geode.distributed.internal.ServerLocation in project geode by apache.

the class OpExecutorImpl method getNextOpServerLocation.

public ServerLocation getNextOpServerLocation() {
    ServerLocation retVal = null;
    Connection conn = (Connection) (threadLocalConnections ? localConnection.get() : null);
    if (conn == null || conn.isDestroyed()) {
        conn = connectionManager.borrowConnection(serverTimeout);
        retVal = conn.getServer();
        this.connectionManager.returnConnection(conn);
    } else {
        retVal = conn.getServer();
    }
    return retVal;
}

Example 39 with ServerLocation

use of org.apache.geode.distributed.internal.ServerLocation in project geode by apache.

the class ConnectionPoolImplJUnitTest method testCreatePool.

@Test
public void testCreatePool() throws Exception {
    CacheServer server1 = cache.addCacheServer();
    int port1 = port;
    server1.setPort(port1);
    server1.start();
    PoolFactory cpf = PoolManager.createFactory();
    cpf.addServer("localhost", port1);
    cpf.setSubscriptionEnabled(true);
    cpf.setSubscriptionRedundancy(0);
    PoolImpl pool = (PoolImpl) cpf.create("pool1");
    ServerLocation location1 = new ServerLocation("localhost", port1);
    Op testOp = new Op() {

        public Object attempt(Connection cnx) throws Exception {
            return cnx.getServer();
        }

        @Override
        public boolean useThreadLocalConnection() {
            return true;
        }
    };
    assertEquals(location1, pool.executeOnPrimary(testOp));
    assertEquals(location1, pool.executeOnQueuesAndReturnPrimaryResult(testOp));
}

Example 40 with ServerLocation

use of org.apache.geode.distributed.internal.ServerLocation in project geode by apache.

the class ConnectionPoolImplJUnitTest method testExecuteOp.

@Test
public void testExecuteOp() throws Exception {
    CacheServer server1 = cache.addCacheServer();
    CacheServer server2 = cache.addCacheServer();
    int[] ports = AvailablePortHelper.getRandomAvailableTCPPorts(2);
    int port1 = ports[0];
    int port2 = ports[1];
    server1.setPort(port1);
    server2.setPort(port2);
    server1.start();
    server2.start();
    PoolFactory cpf = PoolManager.createFactory();
    cpf.addServer("localhost", port2);
    cpf.addServer("localhost", port1);
    PoolImpl pool = (PoolImpl) cpf.create("pool1");
    ServerLocation location1 = new ServerLocation("localhost", port1);
    ServerLocation location2 = new ServerLocation("localhost", port2);
    Op testOp = new Op() {

        int attempts = 0;

        public Object attempt(Connection cnx) throws Exception {
            if (attempts == 0) {
                attempts++;
                throw new SocketTimeoutException();
            } else {
                return cnx.getServer();
            }
        }

        @Override
        public boolean useThreadLocalConnection() {
            return true;
        }
    };
    // TODO - set retry attempts, and throw in some assertions
    // about how many times we retry
    ServerLocation usedServer = (ServerLocation) pool.execute(testOp);
    assertTrue("expected " + location1 + " or " + location2 + ", got " + usedServer, location1.equals(usedServer) || location2.equals(usedServer));
    testOp = new Op() {

        public Object attempt(Connection cnx) throws Exception {
            throw new SocketTimeoutException();
        }

        @Override
        public boolean useThreadLocalConnection() {
            return true;
        }
    };
    try {
        usedServer = (ServerLocation) pool.execute(testOp);
        fail("Should have failed");
    } catch (ServerConnectivityException expected) {
    // do nothing
    }
}