Examples with PeerStatus org.apache.nifi.remote.PeerStatus used on opensource projects

Example 16 with PeerStatus

use of org.apache.nifi.remote.PeerStatus in project nifi by apache.

the class PeerSelector method recoverPersistedPeerStatuses.

private static Set<PeerStatus> recoverPersistedPeerStatuses(final File file) throws IOException {
    if (!file.exists()) {
        return null;
    }
    final Set<PeerStatus> statuses = new HashSet<>();
    try (final InputStream fis = new FileInputStream(file);
        final BufferedReader reader = new BufferedReader(new InputStreamReader(fis))) {
        String line;
        while ((line = reader.readLine()) != null) {
            final String[] splits = line.split(Pattern.quote(":"));
            if (splits.length != 3 && splits.length != 4) {
                continue;
            }
            final String hostname = splits[0];
            final int port = Integer.parseInt(splits[1]);
            final boolean secure = Boolean.parseBoolean(splits[2]);
            final boolean supportQueryForPeer = splits.length == 4 && Boolean.parseBoolean(splits[3]);
            statuses.add(new PeerStatus(new PeerDescription(hostname, port, secure), 1, supportQueryForPeer));
        }
    }
    return statuses;
}

Example 17 with PeerStatus

use of org.apache.nifi.remote.PeerStatus in project nifi by apache.

the class PeerSelector method getPeerStatuses.

private Set<PeerStatus> getPeerStatuses() {
    final PeerStatusCache cache = this.peerStatusCache;
    if (cache == null || cache.getStatuses() == null || cache.getStatuses().isEmpty()) {
        return null;
    }
    if (cache.getTimestamp() + PEER_CACHE_MILLIS < systemTime.currentTimeMillis()) {
        final Set<PeerStatus> equalizedSet = new HashSet<>(cache.getStatuses().size());
        for (final PeerStatus status : cache.getStatuses()) {
            final PeerStatus equalizedStatus = new PeerStatus(status.getPeerDescription(), 1, status.isQueryForPeers());
            equalizedSet.add(equalizedStatus);
        }
        return equalizedSet;
    }
    return cache.getStatuses();
}

Example 18 with PeerStatus

use of org.apache.nifi.remote.PeerStatus in project nifi by apache.

the class PeerSelector method persistPeerStatuses.

private void persistPeerStatuses(final Set<PeerStatus> statuses) {
    if (persistenceFile == null) {
        return;
    }
    try (final OutputStream fos = new FileOutputStream(persistenceFile);
        final OutputStream out = new BufferedOutputStream(fos)) {
        for (final PeerStatus status : statuses) {
            final PeerDescription description = status.getPeerDescription();
            final String line = description.getHostname() + ":" + description.getPort() + ":" + description.isSecure() + ":" + status.isQueryForPeers() + "\n";
            out.write(line.getBytes(StandardCharsets.UTF_8));
        }
    } catch (final IOException e) {
        error(logger, eventReporter, "Failed to persist list of Peers due to {}; if restarted and peer's NCM is down," + " may be unable to transfer data until communications with NCM are restored", e.toString());
        logger.error("", e);
    }
}