diff --git a/broker/src/main/java/io/moquette/spi/persistence/MapDBMessagesStore.java b/broker/src/main/java/io/moquette/spi/persistence/MapDBMessagesStore.java index 54668069..4eb497e0 100644 --- a/broker/src/main/java/io/moquette/spi/persistence/MapDBMessagesStore.java +++ b/broker/src/main/java/io/moquette/spi/persistence/MapDBMessagesStore.java @@ -130,13 +130,13 @@ class MapDBMessagesStore implements IMessagesStore { public void incUsageCounter(MessageGUID guid) { IMessagesStore.StoredMessage storedMessage = m_persistentMessageStore.get(guid); storedMessage.incReferenceCounter(); - m_persistentMessageStore.replace(guid, storedMessage); + m_persistentMessageStore.put(guid, storedMessage); } @Override public void decUsageCounter(MessageGUID guid) { IMessagesStore.StoredMessage storedMessage = m_persistentMessageStore.get(guid); storedMessage.decReferenceCounter(); - m_persistentMessageStore.replace(guid, storedMessage); + m_persistentMessageStore.put(guid, storedMessage); } } diff --git a/broker/src/test/java/io/moquette/spi/impl/MemoryMessagesStore.java b/broker/src/test/java/io/moquette/spi/impl/MemoryMessagesStore.java index 75af2e00..3cf8d5cb 100644 --- a/broker/src/test/java/io/moquette/spi/impl/MemoryMessagesStore.java +++ b/broker/src/test/java/io/moquette/spi/impl/MemoryMessagesStore.java @@ -112,13 +112,13 @@ public class MemoryMessagesStore implements IMessagesStore { public void incUsageCounter(MessageGUID guid) { IMessagesStore.StoredMessage storedMessage = m_persistentMessageStore.get(guid); storedMessage.incReferenceCounter(); - m_persistentMessageStore.replace(guid, storedMessage); + m_persistentMessageStore.put(guid, storedMessage); } @Override public void decUsageCounter(MessageGUID guid) { IMessagesStore.StoredMessage storedMessage = m_persistentMessageStore.get(guid); storedMessage.decReferenceCounter(); - m_persistentMessageStore.replace(guid, storedMessage); + m_persistentMessageStore.put(guid, storedMessage); } }