Add simple connection management capabilities to the broker. An external codebase will now be able to monitor and close the opened connections.

broker/src/test/java/io/moquette/spi/impl/MemoryMessagesStore.java

@@ -98,4 +98,13 @@ public class MemoryMessagesStore implements IMessagesStore {
     public void cleanRetained(String topic) {
         m_retainedStore.remove(topic);
     }
+
+    @Override
+    public int getPendingPublishMessages(String clientID) {
+        Map<Integer, MessageGUID> messageToGuids = m_messageToGuids.get(clientID);
+        if (messageToGuids == null)
+            return 0;
+        else
+            return messageToGuids.size();
+    }
 }

broker/src/test/java/io/moquette/spi/persistence/MemorySessionStore.java

@@ -23,6 +23,7 @@ import io.moquette.spi.ISessionsStore;
 import io.moquette.spi.MessageGUID;
 import io.moquette.spi.impl.Utils;
 import io.moquette.spi.impl.subscriptions.Subscription;
+import io.moquette.spi.persistence.MapDBPersistentStore.PersistentSession;
 
 import org.slf4j.Logger;
 import org.slf4j.LoggerFactory;
@@ -129,6 +130,15 @@ public class MemorySessionStore implements ISessionsStore {
         MapDBPersistentStore.PersistentSession storedSession = m_persistentSessions.get(clientID);
         return new ClientSession(clientID, m_messagesStore, this, storedSession.cleanSession);
     }
+    
+    @Override
+    public Collection<ClientSession> getAllSessions() {
+        Collection<ClientSession> result = new ArrayList<ClientSession>();
+        for (Map.Entry<String, PersistentSession> entry : m_persistentSessions.entrySet()) {
+            result.add(new ClientSession(entry.getKey(), m_messagesStore, this, entry.getValue().cleanSession));
+        }
+        return result;
+    }
 
     @Override
     public void updateCleanStatus(String clientID, boolean cleanSession) {
@@ -252,4 +262,28 @@ public class MemorySessionStore implements ISessionsStore {
 	public StoredMessage getInflightMessage( String clientID, int messageID ) {
 		return null;
 	}
+
+    @Override
+    public int getInflightMessagesNo(String clientID) {
+        Map<Integer, MessageGUID> inflightMessages = m_inflightStore.get(clientID);
+        if (inflightMessages == null)
+            return 0;
+        else
+            return inflightMessages.size();
+    }
+
+    @Override
+    public int getPendingPublishMessagesNo(String clientID) {
+        return m_messagesStore.getPendingPublishMessages(clientID);
+    }
+
+    @Override
+    public int getSecondPhaseAckPendingMessages(String clientID) {
+        Map<Integer, MessageGUID> pendingAcks = m_secondPhaseStore.get(clientID);
+        if (pendingAcks == null)
+            return 0;
+        else
+            return pendingAcks.size();
+    }
+
 }