84 lines
3.6 KiB
Java
84 lines
3.6 KiB
Java
package io.moquette.spi.impl;
|
|
|
|
|
|
import io.moquette.parser.proto.messages.AbstractMessage;
|
|
import io.moquette.parser.proto.messages.PublishMessage;
|
|
import io.moquette.spi.ClientSession;
|
|
import io.moquette.spi.IMessagesStore;
|
|
import org.slf4j.Logger;
|
|
import org.slf4j.LoggerFactory;
|
|
|
|
import java.nio.ByteBuffer;
|
|
import java.util.ArrayList;
|
|
import java.util.Collection;
|
|
import java.util.List;
|
|
import java.util.concurrent.BlockingQueue;
|
|
|
|
class InternalRepublisher {
|
|
|
|
private static final Logger LOG = LoggerFactory.getLogger(InternalRepublisher.class);
|
|
|
|
private final PersistentQueueMessageSender messageSender;
|
|
|
|
InternalRepublisher(PersistentQueueMessageSender messageSender) {
|
|
this.messageSender = messageSender;
|
|
}
|
|
|
|
void publishRetained(ClientSession targetSession, Collection<IMessagesStore.StoredMessage> messages) {
|
|
for (IMessagesStore.StoredMessage storedMsg : messages) {
|
|
//fire as retained the message
|
|
Integer packetID = storedMsg.getQos() == AbstractMessage.QOSType.MOST_ONE ? null : targetSession.nextPacketId();
|
|
if (packetID != null) {
|
|
if (LOG.isDebugEnabled()) {
|
|
LOG.debug("Adding message to inflight zone. MqttClientId = {}, packetId = {}, messageId = {}, topic = {}.",
|
|
targetSession.clientID, packetID, storedMsg.getMessageID(), storedMsg.getTopic());
|
|
}
|
|
targetSession.inFlightAckWaiting(storedMsg.getGuid(), packetID);
|
|
}
|
|
PublishMessage publishMsg = retainedPublish(storedMsg);
|
|
//set the PacketIdentifier only for QoS > 0
|
|
if (publishMsg.getQos() != AbstractMessage.QOSType.MOST_ONE) {
|
|
publishMsg.setMessageID(packetID);
|
|
}
|
|
this.messageSender.sendPublish(targetSession, publishMsg);
|
|
}
|
|
}
|
|
|
|
void publishStored(ClientSession clientSession, BlockingQueue<IMessagesStore.StoredMessage> publishedEvents) {
|
|
List<IMessagesStore.StoredMessage> storedPublishes = new ArrayList<>();
|
|
publishedEvents.drainTo(storedPublishes);
|
|
|
|
for (IMessagesStore.StoredMessage pubEvt : storedPublishes) {
|
|
//put in flight zone
|
|
if (LOG.isDebugEnabled()) {
|
|
LOG.debug("Adding message ot inflight zone. MqttClientId = {}, guid = {}, messageId = {}, topic = {}.",
|
|
clientSession.clientID, pubEvt.getGuid(), pubEvt.getMessageID(), pubEvt.getTopic());
|
|
}
|
|
clientSession.inFlightAckWaiting(pubEvt.getGuid(), pubEvt.getMessageID());
|
|
PublishMessage publishMsg = notRetainedPublish(pubEvt);
|
|
//set the PacketIdentifier only for QoS > 0
|
|
if (publishMsg.getQos() != AbstractMessage.QOSType.MOST_ONE) {
|
|
publishMsg.setMessageID(pubEvt.getMessageID());
|
|
}
|
|
this.messageSender.sendPublish(clientSession, publishMsg);
|
|
}
|
|
}
|
|
|
|
private PublishMessage notRetainedPublish(IMessagesStore.StoredMessage storedMessage) {
|
|
return createPublishForQos(storedMessage.getTopic(), storedMessage.getQos(), storedMessage.getMessage(), false);
|
|
}
|
|
|
|
private PublishMessage retainedPublish(IMessagesStore.StoredMessage storedMessage) {
|
|
return createPublishForQos(storedMessage.getTopic(), storedMessage.getQos(), storedMessage.getMessage(), true);
|
|
}
|
|
|
|
public static PublishMessage createPublishForQos(String topic, AbstractMessage.QOSType qos, ByteBuffer message, boolean retained) {
|
|
PublishMessage pubMessage = new PublishMessage();
|
|
pubMessage.setRetainFlag(retained);
|
|
pubMessage.setTopicName(topic);
|
|
pubMessage.setQos(qos);
|
|
pubMessage.setPayload(message);
|
|
return pubMessage;
|
|
}
|
|
}
|