Modified #getSessionCount() to use AtomicInteger. JM-720

git-svn-id: http://svn.igniterealtime.org/svn/repos/wildfire/trunk@4021 b35dd754-fafc-0310-a699-88a17e54d16e

src/java/org/jivesoftware/wildfire/SessionManager.java

@@ -38,6 +38,7 @@ import org.xmpp.packet.Presence;
 import java.util.*;
 import java.util.concurrent.ConcurrentHashMap;
 import java.util.concurrent.CopyOnWriteArrayList;
+import java.util.concurrent.atomic.AtomicInteger;
 
 /**
  * Manages the sessions associated with an account. The information
@@ -57,6 +58,12 @@ public class SessionManager extends BasicModule {
     private UserManager userManager;
     private int conflictLimit;
 
+    /**
+     * Counter of user sessions. A session is counted just after it was created and not
+     * after the user came available.
+     */
+    private final AtomicInteger usersSessionsCounter = new AtomicInteger(0);
+
     private ClientSessionListener clientSessionListener = new ClientSessionListener();
     private ComponentSessionListener componentSessionListener = new ComponentSessionListener();
     private IncomingServerSessionListener incomingServerListener = new IncomingServerSessionListener();
@@ -511,6 +518,8 @@ public class SessionManager extends BasicModule {
 
         // Add to pre-authenticated sessions.
         preAuthenticatedSessions.put(session.getAddress().getResource(), session);
+        // Increment the counter of user sessions
+        usersSessionsCounter.incrementAndGet();
         return session;
     }
 
@@ -1199,12 +1208,7 @@ public class SessionManager extends BasicModule {
      * @return number of client sessions that are connected to the server.
      */
     public int getSessionCount() {
-        int sessionCount = 0;
-        for (String username : getSessionUsers()) {
-            sessionCount += getSessionCount(username);
-        }
-        sessionCount += anonymousSessions.size();
-        return sessionCount;
+        return usersSessionsCounter.get();
     }
 
     /**
@@ -1420,6 +1424,8 @@ public class SessionManager extends BasicModule {
                 finally {
                     // Remove the session
                     removeSession(session);
+                    // Decrement the counter of user sessions
+                    usersSessionsCounter.decrementAndGet();
                 }
             }
             catch (Exception e) {