MOved the method nextClientID from IMessagesStore to the ISessionsStore because it's session related

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

@@ -83,18 +83,18 @@ public class MapDBPersistentStoreTest {
 
     @Test
     public void testNextPacketID_notExistingClientSession() {
-        int packetId = m_messagesStore.nextPacketID("NOT_EXISTING_CLI");
+        int packetId = m_sessionsStore.nextPacketID("NOT_EXISTING_CLI");
         assertEquals(1, packetId);
     }
 
     @Test
     public void testNextPacketID_existingClientSession() {
         //Force creation of inflight map for the CLIENT session
-        int packetId = m_messagesStore.nextPacketID("CLIENT");
+        int packetId = m_sessionsStore.nextPacketID("CLIENT");
         assertEquals(1, packetId);
 
         //request a second packetID
-        packetId = m_messagesStore.nextPacketID("CLIENT");
+        packetId = m_sessionsStore.nextPacketID("CLIENT");
         assertEquals(2, packetId);
     }
 
@@ -102,7 +102,7 @@ public class MapDBPersistentStoreTest {
     public void testNextPacketID() {
         //request a first ID
 
-        int packetId = m_messagesStore.nextPacketID("CLIENT");
+        int packetId = m_sessionsStore.nextPacketID("CLIENT");
         m_sessionsStore.inFlight("CLIENT", packetId, "ABCDE"); //simulate an inflight
         assertEquals(1, packetId);
 
@@ -110,7 +110,7 @@ public class MapDBPersistentStoreTest {
         m_sessionsStore.inFlightAck("CLIENT", packetId);
 
         //request a second packetID, counter restarts from 0
-        packetId = m_messagesStore.nextPacketID("CLIENT");
+        packetId = m_sessionsStore.nextPacketID("CLIENT");
         assertEquals(1, packetId);
     }
 

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

@@ -39,6 +39,7 @@ public class MemorySessionStore implements ISessionsStore {
 
     //maps clientID->[MessageId -> guid]
     private Map<String, Map<Integer, String>> m_inflightStore = new HashMap<>();
+//    private Map<String, Set<Integer>> m_inflightIDs = new HashMap<>();
     //maps clientID->[guid*]
     private Map<String, Set<String>> m_enqueuedStore = new HashMap<>();
     //maps clientID->[messageID*]
@@ -169,6 +170,24 @@ public class MemorySessionStore implements ISessionsStore {
         this.m_inflightStore.put(clientID, m);
     }
 
+    /**
+     * Return the next valid packetIdentifier for the given client session.
+     * */
+    @Override
+    public int nextPacketID(String clientID) {
+        Map<Integer, String> m = this.m_inflightStore.get(clientID);
+        if (m == null) {
+            m = new HashMap<>();
+            int nextPacketId = 1;
+            m.put(nextPacketId, null);
+            return nextPacketId;
+        }
+        int maxId = Collections.max(m.keySet());
+        int nextPacketId = (maxId + 1) % 0xFFFF;
+        m.put(nextPacketId, null);
+        return nextPacketId;
+    }
+
     @Override
     public void bindToDeliver(String guid, String clientID) {
         Set<String> guids = Utils.defaultGet(m_enqueuedStore, clientID, new HashSet<String>());