Added remove of subscribed topic during processing of UNSUBSCRIBE message (fix issue #50)

broker/src/main/java/org/eclipse/moquette/spi/ISessionsStore.java

@@ -32,6 +32,11 @@ public interface ISessionsStore {
      */
     void addNewSubscription(Subscription newSubscription);
 
+    /**
+     * Removed a specific subscription
+     * */
+    void removeSubscription(String topic, String clientID);
+
     /**
      * Remove all the subscriptions of the session
      */

broker/src/main/java/org/eclipse/moquette/spi/impl/MemoryStorageService.java

@@ -158,6 +158,27 @@ public class MemoryStorageService implements IMessagesStore, ISessionsStore {
         return nextPacketId;
     }
 
+    @Override
+    public void removeSubscription(String topic, String clientID) {
+        LOG.debug("removeSubscription topic filter: {} for clientID: {}", topic, clientID);
+        if (!m_persistentSubscriptions.containsKey(clientID)) {
+            return;
+        }
+        Set<Subscription> clientSubscriptions = m_persistentSubscriptions.get(clientID);
+        //search for the subscription to remove
+        Subscription toBeRemoved = null;
+        for (Subscription sub : clientSubscriptions) {
+            if (sub.getTopicFilter().equals(topic)) {
+                toBeRemoved = sub;
+                break;
+            }
+        }
+
+        if (toBeRemoved != null) {
+            clientSubscriptions.remove(toBeRemoved);
+        }
+    }
+
     @Override
     public void addNewSubscription(Subscription newSubscription) {
         final String clientID = newSubscription.getClientId();

broker/src/main/java/org/eclipse/moquette/spi/impl/ProtocolProcessor.java

@@ -587,7 +587,9 @@ class ProtocolProcessor implements EventHandler<ValueEvent> {
 
         for (String topic : topics) {
             subscriptions.removeSubscription(topic, clientID);
+            m_sessionsStore.removeSubscription(topic, clientID);
         }
+
         //ack the client
         UnsubAckMessage ackMessage = new UnsubAckMessage();
         ackMessage.setMessageID(messageID);

broker/src/main/java/org/eclipse/moquette/spi/persistence/MapDBPersistentStore.java

@@ -221,6 +221,29 @@ public class MapDBPersistentStore implements IMessagesStore, ISessionsStore {
         return nextPacketId;
     }
 
+    @Override
+    public void removeSubscription(String topic, String clientID) {
+        LOG.debug("removeSubscription topic filter: {} for clientID: {}", topic, clientID);
+        if (!m_persistentSubscriptions.containsKey(clientID)) {
+            return;
+        }
+        Set<Subscription> clientSubscriptions = m_persistentSubscriptions.get(clientID);
+        //search for the subscription to remove
+        Subscription toBeRemoved = null;
+        for (Subscription sub : clientSubscriptions) {
+            if (sub.getTopicFilter().equals(topic)) {
+                toBeRemoved = sub;
+                break;
+            }
+        }
+
+        if (toBeRemoved != null) {
+            clientSubscriptions.remove(toBeRemoved);
+        }
+        m_persistentSubscriptions.put(clientID, clientSubscriptions);
+        m_db.commit();
+    }
+
     public void addNewSubscription(Subscription newSubscription) {
         LOG.debug("addNewSubscription invoked with subscription {}", newSubscription);
         final String clientID = newSubscription.getClientId();