diff --git a/broker/src/main/java/org/dna/mqtt/moquette/messaging/spi/impl/ProtocolProcessor.java b/broker/src/main/java/org/dna/mqtt/moquette/messaging/spi/impl/ProtocolProcessor.java index 56a65079..08a65c6a 100644 --- a/broker/src/main/java/org/dna/mqtt/moquette/messaging/spi/impl/ProtocolProcessor.java +++ b/broker/src/main/java/org/dna/mqtt/moquette/messaging/spi/impl/ProtocolProcessor.java @@ -303,5 +303,11 @@ class ProtocolProcessor { m_clientIDs.get(clientID).getSession().write(pubRelMessage); } + + void processPubComp(String clientID, int messageID) { + //once received the PUBCOMP then remove the message from the temp memory + String publishKey = String.format("%s%d", clientID, messageID); + m_storageService.cleanInFlight(publishKey); + } } diff --git a/broker/src/main/java/org/dna/mqtt/moquette/messaging/spi/impl/SimpleMessaging.java b/broker/src/main/java/org/dna/mqtt/moquette/messaging/spi/impl/SimpleMessaging.java index a5c11aa9..d9262f4a 100644 --- a/broker/src/main/java/org/dna/mqtt/moquette/messaging/spi/impl/SimpleMessaging.java +++ b/broker/src/main/java/org/dna/mqtt/moquette/messaging/spi/impl/SimpleMessaging.java @@ -81,11 +81,6 @@ public class SimpleMessaging implements IMessaging, EventHandler { } -/* public void setNotifier(INotifier notifier) { - m_notifier= notifier; - } */ - - private void disruptorPublish(MessagingEvent msgEvent) { long sequence = m_ringBuffer.next(); ValueEvent event = m_ringBuffer.get(sequence); @@ -189,7 +184,7 @@ public class SimpleMessaging implements IMessaging, EventHandler { } else if (message instanceof PubCompMessage) { String clientID = (String) session.getAttribute(Constants.ATTR_CLIENTID); int messageID = ((PubCompMessage) message).getMessageID(); - processPubComp(clientID, messageID); + m_processor.processPubComp(clientID, messageID); } else { throw new RuntimeException("Illegal message received " + message); } @@ -212,12 +207,6 @@ public class SimpleMessaging implements IMessaging, EventHandler { m_processor.processPublish(evt); } - private void processPubComp(String clientID, int messageID) { - //once received the PUBCOMP then remove the message from the temp memory - String publishKey = String.format("%s%d", clientID, messageID); - m_storageService.cleanInFlight(publishKey); - } - private void subscribeSingleTopic(Subscription newSubscription, final String topic) { subscriptions.add(newSubscription);