Examples with Time org.apache.zookeeper.common.Time used on opensource projects

Example 1 with Time

use of org.apache.zookeeper.common.Time in project zookeeper by apache.

the class Learner method connectToLeader.

/**
 * Establish a connection with the LearnerMaster found by findLearnerMaster.
 * Followers only connect to Leaders, Observers can connect to any active LearnerMaster.
 * Retries until either initLimit time has elapsed or 5 tries have happened.
 * @param multiAddr - the address of the Peer to connect to.
 * @throws IOException - if the socket connection fails on the 5th attempt
 * if there is an authentication failure while connecting to leader
 */
protected void connectToLeader(MultipleAddresses multiAddr, String hostname) throws IOException {
    this.leaderAddr = multiAddr;
    Set<InetSocketAddress> addresses;
    if (self.isMultiAddressReachabilityCheckEnabled()) {
        // even if none of the addresses are reachable, we want to try to establish connection
        // see ZOOKEEPER-3758
        addresses = multiAddr.getAllReachableAddressesOrAll();
    } else {
        addresses = multiAddr.getAllAddresses();
    }
    ExecutorService executor = Executors.newFixedThreadPool(addresses.size());
    CountDownLatch latch = new CountDownLatch(addresses.size());
    AtomicReference<Socket> socket = new AtomicReference<>(null);
    addresses.stream().map(address -> new LeaderConnector(address, socket, latch)).forEach(executor::submit);
    try {
        latch.await();
    } catch (InterruptedException e) {
        LOG.warn("Interrupted while trying to connect to Leader", e);
    } finally {
        executor.shutdown();
        try {
            if (!executor.awaitTermination(1, TimeUnit.SECONDS)) {
                LOG.error("not all the LeaderConnector terminated properly");
            }
        } catch (InterruptedException ie) {
            LOG.error("Interrupted while terminating LeaderConnector executor.", ie);
        }
    }
    if (socket.get() == null) {
        throw new IOException("Failed connect to " + multiAddr);
    } else {
        sock = socket.get();
        sockBeingClosed.set(false);
    }
    self.authLearner.authenticate(sock, hostname);
    leaderIs = BinaryInputArchive.getArchive(new BufferedInputStream(sock.getInputStream()));
    bufferedOutput = new BufferedOutputStream(sock.getOutputStream());
    leaderOs = BinaryOutputArchive.getArchive(bufferedOutput);
    if (asyncSending) {
        startSendingThread();
    }
}