Added isStarted() to XMPPServer, which cleans PubSubEngine.start() up some.

git-svn-id: http://svn.igniterealtime.org/svn/repos/openfire/branches/pep@8773 b35dd754-fafc-0310-a699-88a17e54d16e

src/java/org/jivesoftware/openfire/XMPPServer.java

@@ -101,6 +101,7 @@ public class XMPPServer {
     private Version version;
     private Date startDate;
     private boolean initialized = false;
+    private boolean started = false;
     private NodeID nodeID;
     private static final NodeID DEFAULT_NODE_ID = new NodeID(new byte[0]);
 
@@ -451,6 +452,8 @@ public class XMPPServer {
             Log.info(startupBanner);
             System.out.println(startupBanner);
 
+            started = true;
+            
             // Notify server listeners that the server has been started
             for (XMPPServerListener listener : listeners) {
                 listener.serverStarted();
@@ -1383,4 +1386,13 @@ public class XMPPServer {
     public void setRemoteSessionLocator(RemoteSessionLocator remoteSessionLocator) {
         this.remoteSessionLocator = remoteSessionLocator;
     }
+
+    /**
+     * Returns whether or not the server has been started.
+     * 
+     * @return whether or not the server has been started.
+     */
+    public boolean isStarted() {
+        return started;
+    }
 }

src/java/org/jivesoftware/openfire/disco/IQDiscoInfoHandler.java

@@ -293,7 +293,8 @@ public class IQDiscoInfoHandler extends IQHandler implements ClusterEventListene
                 serverIdentities.add(it.next());
             }
         }
-        // 
+        
+        // FIXME: Create a UserIdentityProvider interface
         if (server.getIQPEPHandler() != null) {
             Element userIdentity = DocumentHelper.createElement("identity");
             userIdentity.addAttribute("category", "pubsub");

src/java/org/jivesoftware/openfire/pep/IQPEPHandler.java

@@ -92,6 +92,7 @@ public class IQPEPHandler extends IQHandler implements ServerIdentitiesProvider,
         // TODO: Much to be done here...
 
         if (packet.getTo() == null) {
+            // TODO: Do not allow anonymous users to create a service
             String jidFrom = packet.getFrom().toBareJID();
             
             PEPService pepService = pepServices.get(jidFrom);
@@ -100,9 +101,13 @@ public class IQPEPHandler extends IQHandler implements ServerIdentitiesProvider,
             if (pepService == null) {
                 pepService = new PEPService(XMPPServer.getInstance(), jidFrom);
                 pepServices.put(jidFrom, pepService);
-                pubSubEngine.start(pepService);     // Keep DB synced
+                
+                // Probe presences
+                pubSubEngine.start(pepService);
+                if (Log.isDebugEnabled()) {
                     Log.debug("PEP: " + jidFrom + " had a PEPService created");
                 }
+            }
 
             // If publishing a node, and the node doesn't exist, create it.
             if (packet.getType() == IQ.Type.set) {
@@ -117,18 +122,24 @@ public class IQPEPHandler extends IQHandler implements ServerIdentitiesProvider,
                         LeafNode newNode = new LeafNode(pepService, null, nodeID, creator);
                         newNode.addOwner(creator);
                         newNode.saveToDB();
+                        if (Log.isDebugEnabled()) {
                             Log.debug("PEP: Created node ('" + nodeID + "') for " + jidFrom);
                         }
                     }
                 }                        
+            }
             
             // Process with PubSub as usual.
             if (pubSubEngine.process(pepService, packet)) {
+                if (Log.isDebugEnabled()) {
                     Log.debug("PEP: The pubSubEngine processed a packet for " + jidFrom + "'s pepService.");
                 }
+            }
             else {
+                if (Log.isDebugEnabled()) {
                     Log.debug("PEP: The pubSubEngine did not process a packet for " + jidFrom + "'s pepService.");
                 }
+            }
 
         }
         else {

src/java/org/jivesoftware/openfire/pubsub/PubSubEngine.java

@@ -1673,28 +1673,37 @@ public class PubSubEngine {
         return completedForm;
     }
 
-    public void start(PubSubService service) {
+    public void start(final PubSubService service) {
         // Probe presences of users that this service has subscribed to (once the server
         // has started)
-        final PubSubService tempService = service;    // TODO: Needs to be tested for correctness
+        
+        if (XMPPServer.getInstance().isStarted()) {
+            probePresences(service);
+        }
+        else {
             XMPPServer.getInstance().addServerListener(new XMPPServerListener() {
                 public void serverStarted() {
+                    probePresences(service);
+                }
+
+                public void serverStopping() {
+                }
+            });
+        }        
+    }
+
+    private void probePresences(final PubSubService service) {
         Set<JID> affiliates = new HashSet<JID>();
-                for (Node node : tempService.getNodes()) {
+        for (Node node : service.getNodes()) {
             affiliates.addAll(node.getPresenceBasedSubscribers());
         }
         for (JID jid : affiliates) {
             // Send probe presence
             Presence subscription = new Presence(Presence.Type.probe);
             subscription.setTo(jid);
-                    subscription.setFrom(tempService.getAddress());
-                    tempService.send(subscription);
-                }
-            }
-
-            public void serverStopping() {
+            subscription.setFrom(service.getAddress());
+            service.send(subscription);
         }
-        });
     }
 
     public void shutdown(PubSubService service) {