Manage currently connected nodes.
[jSite2.git] / src / net / pterodactylus / jsite / core / NodeManager.java
index 40e4190..957e1bc 100644 (file)
@@ -31,10 +31,12 @@ import java.net.UnknownHostException;
 import java.util.ArrayList;
 import java.util.Collections;
 import java.util.HashMap;
+import java.util.HashSet;
 import java.util.Iterator;
 import java.util.List;
 import java.util.Map;
 import java.util.Properties;
+import java.util.Set;
 import java.util.logging.Level;
 import java.util.logging.Logger;
 
@@ -66,20 +68,23 @@ public class NodeManager implements Iterable<Node>, PropertyChangeListener, High
        /** Object used for synchronization. */
        private final Object syncObject = new Object();
 
-       /** Node listeners. */
-       private List<NodeListener> nodeListeners = Collections.synchronizedList(new ArrayList<NodeListener>());
+       /** Node listener support. */
+       private final NodeListenerSupport nodeListenerSupport = new NodeListenerSupport();
 
        /** All nodes. */
-       private List<Node> nodes = Collections.synchronizedList(new ArrayList<Node>());
+       private final List<Node> nodes = Collections.synchronizedList(new ArrayList<Node>());
 
        /** Map from node ID to node. */
-       private Map<String, Node> idNodes = Collections.synchronizedMap(new HashMap<String, Node>());
+       private final Map<String, Node> idNodes = Collections.synchronizedMap(new HashMap<String, Node>());
 
        /** All FCP connections. */
-       private Map<Node, HighLevelClient> nodeClients = Collections.synchronizedMap(new HashMap<Node, HighLevelClient>());
+       private final Map<Node, HighLevelClient> nodeClients = Collections.synchronizedMap(new HashMap<Node, HighLevelClient>());
 
        /** Maps nodes to high-level clients. */
-       private Map<HighLevelClient, Node> clientNodes = Collections.synchronizedMap(new HashMap<HighLevelClient, Node>());
+       private final Map<HighLevelClient, Node> clientNodes = Collections.synchronizedMap(new HashMap<HighLevelClient, Node>());
+
+       /** Collection of currently connected nodes. */
+       private final Set<Node> connectedNodes = Collections.synchronizedSet(new HashSet<Node>());
 
        /**
         * Creates a new FCP collector.
@@ -105,7 +110,7 @@ public class NodeManager implements Iterable<Node>, PropertyChangeListener, High
         *            The listener to add
         */
        public void addNodeListener(NodeListener nodeListener) {
-               nodeListeners.add(nodeListener);
+               nodeListenerSupport.addListener(nodeListener);
        }
 
        /**
@@ -115,77 +120,7 @@ public class NodeManager implements Iterable<Node>, PropertyChangeListener, High
         *            The listener to remove
         */
        public void removeNodeListener(NodeListener nodeListener) {
-               nodeListeners.remove(nodeListener);
-       }
-
-       /**
-        * Notifies all listeners that a node was added.
-        *
-        * @param node
-        *            The node that was added.
-        */
-       private void fireNodeAdded(Node node) {
-               logger.log(Level.FINEST, "firing nodeAdded event with [node=" + node + "]");
-               for (NodeListener nodeListener : nodeListeners) {
-                       nodeListener.nodeAdded(node);
-               }
-       }
-
-       /**
-        * Notifies all listeners that a node was removed.
-        *
-        * @param node
-        *            The node that was removed
-        */
-       private void fireNodeRemoved(Node node) {
-               logger.log(Level.FINEST, "firing nodeRemoved event with [node=" + node + "]");
-               for (NodeListener nodeListener : nodeListeners) {
-                       nodeListener.nodeRemoved(node);
-               }
-       }
-
-       /**
-        * Notifies all listeners that the given node was connected.
-        *
-        * @param node
-        *            The node that is now connected
-        */
-       private void fireNodeConnected(Node node) {
-               logger.log(Level.FINEST, "firing nodeConnected event with [node=" + node + "]");
-               for (NodeListener nodeListener : nodeListeners) {
-                       nodeListener.nodeConnected(node);
-               }
-       }
-
-       /**
-        * Notifies all listeners that a connection to a node has failed.
-        *
-        * @param node
-        *            The node that could not be connected
-        * @param cause
-        *            The cause of the failure
-        */
-       private void fireNodeConnectionFailed(Node node, Throwable cause) {
-               logger.log(Level.FINEST, "firing nodeConnectionFailed event with [node=" + node + ",cause=" + cause + "]");
-               for (NodeListener nodeListener : nodeListeners) {
-                       nodeListener.nodeConnectionFailed(node, cause);
-               }
-       }
-
-       /**
-        * Notifies all listeners that the given node was disconnected.
-        *
-        * @param node
-        *            The node that is now disconnected
-        * @param throwable
-        *            The exception that caused the disconnect, or <code>null</code>
-        *            if there was no exception
-        */
-       private void fireNodeDisconnected(Node node, Throwable throwable) {
-               logger.log(Level.FINEST, "firing nodeDisconnected event with [node=" + node + ",throwable=" + throwable + "]");
-               for (NodeListener nodeListener : nodeListeners) {
-                       nodeListener.nodeDisconnected(node, throwable);
-               }
+               nodeListenerSupport.removeListener(nodeListener);
        }
 
        //
@@ -214,6 +149,18 @@ public class NodeManager implements Iterable<Node>, PropertyChangeListener, High
        }
 
        /**
+        * Returns whether the given node is currently connected.
+        *
+        * @param node
+        *            The node to check
+        * @return <code>true</code> if the node is currently connected,
+        *         <code>false</code> otherwise
+        */
+       public boolean isNodeConnected(Node node) {
+               return connectedNodes.contains(node);
+       }
+
+       /**
         * {@inheritDoc}
         */
        public Iterator<Node> iterator() {
@@ -347,7 +294,7 @@ public class NodeManager implements Iterable<Node>, PropertyChangeListener, High
                clientNodes.put(highLevelClient, node);
                nodeClients.put(node, highLevelClient);
                highLevelClient.addHighLevelClientListener(this);
-               fireNodeAdded(node);
+               nodeListenerSupport.fireNodeAdded(node);
                return true;
        }
 
@@ -370,7 +317,7 @@ public class NodeManager implements Iterable<Node>, PropertyChangeListener, High
                        nodes.remove(node);
                        idNodes.remove(node.getId());
                        node.removePropertyChangeListener(this);
-                       fireNodeRemoved(node);
+                       nodeListenerSupport.fireNodeRemoved(node);
                }
        }
 
@@ -391,9 +338,9 @@ public class NodeManager implements Iterable<Node>, PropertyChangeListener, High
                try {
                        highLevelClient.connect(node.getHostname(), node.getPort());
                } catch (UnknownHostException uhe1) {
-                       fireNodeConnectionFailed(node, uhe1);
+                       nodeListenerSupport.fireNodeConnectionFailed(node, uhe1);
                } catch (IOException ioe1) {
-                       fireNodeConnectionFailed(node, ioe1);
+                       nodeListenerSupport.fireNodeConnectionFailed(node, ioe1);
                }
        }
 
@@ -505,7 +452,8 @@ public class NodeManager implements Iterable<Node>, PropertyChangeListener, High
                        logger.log(Level.WARNING, "got event for unknown client");
                        return;
                }
-               fireNodeConnected(node);
+               connectedNodes.add(node);
+               nodeListenerSupport.fireNodeConnected(node);
        }
 
        /**
@@ -519,7 +467,8 @@ public class NodeManager implements Iterable<Node>, PropertyChangeListener, High
                                logger.log(Level.WARNING, "got event for unknown client");
                                return;
                        }
-                       fireNodeDisconnected(node, throwable);
+                       connectedNodes.remove(node);
+                       nodeListenerSupport.fireNodeDisconnected(node, throwable);
                }
        }
 
@@ -546,9 +495,9 @@ public class NodeManager implements Iterable<Node>, PropertyChangeListener, High
                                        try {
                                                highLevelClient.connect(node.getHostname(), node.getPort());
                                        } catch (UnknownHostException uhe1) {
-                                               fireNodeConnectionFailed(node, uhe1);
+                                               nodeListenerSupport.fireNodeConnectionFailed(node, uhe1);
                                        } catch (IOException ioe1) {
-                                               fireNodeConnectionFailed(node, ioe1);
+                                               nodeListenerSupport.fireNodeConnectionFailed(node, ioe1);
                                        }
                                }
                        }