Fixed drop messages in clean session phase, introduced ref counter

2016-09-23 09:45:47 +02:00
parent 5efad350d8
commit 4d637a349b
6 changed files with 99 additions and 6 deletions
@@ -100,4 +100,18 @@ public class MemoryMessagesStore implements IMessagesStore {
    public void cleanRetained(String topic) {
        m_retainedStore.remove(topic);
    }
+
+    @Override
+    public void incUsageCounter(String guid) {
+        IMessagesStore.StoredMessage storedMessage = m_persistentMessageStore.get(guid);
+        storedMessage.incReferenceCounter();
+        m_persistentMessageStore.replace(guid, storedMessage);
+    }
+
+    @Override
+    public void decUsageCounter(String guid) {
+        IMessagesStore.StoredMessage storedMessage = m_persistentMessageStore.get(guid);
+        storedMessage.decReferenceCounter();
+        m_persistentMessageStore.replace(guid, storedMessage);
+    }
 }