Minor change, moved method argument from exploded arguments to carrier object

This commit is contained in:
andrea
2015-06-20 15:52:00 +02:00
parent 92a76c9de1
commit c4ebd75008
3 changed files with 44 additions and 40 deletions
+1
View File
@@ -15,6 +15,7 @@ Version 0.7:
Added ACL file for topic read|write operations. Added ACL file for topic read|write operations.
Removed cause of leakage of Netty Sessions (fix issue #39) Removed cause of leakage of Netty Sessions (fix issue #39)
Cleaned Netty's stuff reference from Events used in Disruptor (fix issue #40) Cleaned Netty's stuff reference from Events used in Disruptor (fix issue #40)
Bad assignment of PacketID during publish to subscribed QoS0 topics (fix issue #38)
Version 0.6: Version 0.6:
Fix issue #42, bad password file path composition Fix issue #42, bad password file path composition
@@ -291,32 +291,33 @@ class ProtocolProcessor implements EventHandler<ValueEvent> {
@MQTTMessage(message = PublishMessage.class) @MQTTMessage(message = PublishMessage.class)
void processPublish(ServerChannel session, PublishMessage msg) { void processPublish(ServerChannel session, PublishMessage msg) {
LOG.trace("PUB --PUBLISH--> SRV processPublish invoked with {}", msg); LOG.trace("PUB --PUBLISH--> SRV executePublish invoked with {}", msg);
String clientID = (String) session.getAttribute(NettyChannel.ATTR_KEY_CLIENTID); String clientID = (String) session.getAttribute(NettyChannel.ATTR_KEY_CLIENTID);
final String topic = msg.getTopicName(); final String topic = msg.getTopicName();
final AbstractMessage.QOSType qos = msg.getQos();
final ByteBuffer message = msg.getPayload();
boolean retain = msg.isRetainFlag();
//check if the topic can be wrote //check if the topic can be wrote
String user = (String) session.getAttribute(NettyChannel.ATTR_KEY_USERNAME); String user = (String) session.getAttribute(NettyChannel.ATTR_KEY_USERNAME);
if (m_authorizator.canWrite(topic, user, clientID)) { if (m_authorizator.canWrite(topic, user, clientID)) {
processPublish(clientID, topic, qos, message, retain, msg.getMessageID()); executePublish(clientID, msg);
} else { } else {
LOG.debug("topic {} doesn't have write credentials", topic); LOG.debug("topic {} doesn't have write credentials", topic);
} }
} }
private void processPublish(String clientID, String topic, QOSType qos, ByteBuffer message, boolean retain, Integer messageID) { private void executePublish(String clientID, PublishMessage msg) {
final String topic = msg.getTopicName();
final AbstractMessage.QOSType qos = msg.getQos();
final ByteBuffer message = msg.getPayload();
boolean retain = msg.isRetainFlag();
final Integer messageID = msg.getMessageID();
LOG.info("PUBLISH from clientID <{}> on topic <{}> with QoS {}", clientID, topic, qos); LOG.info("PUBLISH from clientID <{}> on topic <{}> with QoS {}", clientID, topic, qos);
PublishEvent publishEvt = new PublishEvent(clientID, msg);
if (qos == AbstractMessage.QOSType.MOST_ONE) { //QoS0 if (qos == AbstractMessage.QOSType.MOST_ONE) { //QoS0
forward2Subscribers(topic, qos, message, retain, messageID); forward2Subscribers(publishEvt);
} else if (qos == AbstractMessage.QOSType.LEAST_ONE) { //QoS1 } else if (qos == AbstractMessage.QOSType.LEAST_ONE) { //QoS1
PublishEvent inFlightEvt = new PublishEvent(topic, qos, message, retain,
clientID, messageID);
//TODO use a message store for TO PUBLISH MESSAGES it has nothing to do with inFlight!! //TODO use a message store for TO PUBLISH MESSAGES it has nothing to do with inFlight!!
m_messagesStore.addInFlight(inFlightEvt, clientID, messageID); m_messagesStore.addInFlight(publishEvt, clientID, messageID);
forward2Subscribers(topic, qos, message, retain, messageID); forward2Subscribers(publishEvt);
m_messagesStore.cleanInFlight(clientID, messageID); m_messagesStore.cleanInFlight(clientID, messageID);
//NB the PUB_ACK could be sent also after the addInFlight //NB the PUB_ACK could be sent also after the addInFlight
sendPubAck(new PubAckEvent(messageID, clientID)); sendPubAck(new PubAckEvent(messageID, clientID));
@@ -324,9 +325,7 @@ class ProtocolProcessor implements EventHandler<ValueEvent> {
} else if (qos == AbstractMessage.QOSType.EXACTLY_ONCE) { //QoS2 } else if (qos == AbstractMessage.QOSType.EXACTLY_ONCE) { //QoS2
String publishKey = String.format("%s%d", clientID, messageID); String publishKey = String.format("%s%d", clientID, messageID);
//store the message in temp store //store the message in temp store
PublishEvent qos2Persistent = new PublishEvent(topic, qos, message, retain, m_messagesStore.persistQoS2Message(publishKey, publishEvt);
clientID, messageID);
m_messagesStore.persistQoS2Message(publishKey, qos2Persistent);
sendPubRec(clientID, messageID); sendPubRec(clientID, messageID);
//Next the client will send us a pub rel //Next the client will send us a pub rel
//NB publish to subscribers for QoS 2 happen upon PUBREL from publisher //NB publish to subscribers for QoS 2 happen upon PUBREL from publisher
@@ -347,25 +346,27 @@ class ProtocolProcessor implements EventHandler<ValueEvent> {
*/ */
private void forwardPublishWill(WillMessage will, String clientID) { private void forwardPublishWill(WillMessage will, String clientID) {
//it has just to publish the message downstream to the subscribers //it has just to publish the message downstream to the subscribers
final String topic = will.getTopic();
final AbstractMessage.QOSType qos = will.getQos();
final ByteBuffer message = will.getPayload();
boolean retain = will.isRetained();
//NB it's a will publish, it needs a PacketIdentifier for this conn, default to 1 //NB it's a will publish, it needs a PacketIdentifier for this conn, default to 1
if (qos == AbstractMessage.QOSType.MOST_ONE) { Integer messageId = null;
forward2Subscribers(topic, qos, message, retain, null); if (will.getQos() != AbstractMessage.QOSType.MOST_ONE) {
} else { messageId = m_messagesStore.nextPacketID(clientID);
int messageId = m_messagesStore.nextPacketID(clientID);
forward2Subscribers(topic, qos, message, retain, messageId);
} }
PublishEvent pub = new PublishEvent(will.getTopic(), will.getQos(), will.getPayload(), will.isRetained(),
clientID, messageId);
forward2Subscribers(pub);
} }
/** /**
* Flood the subscribers with the message to notify. MessageID is optional and should only used for QoS 1 and 2 * Flood the subscribers with the message to notify. MessageID is optional and should only used for QoS 1 and 2
* */ * */
private void forward2Subscribers(String topic, AbstractMessage.QOSType qos, ByteBuffer origMessage, private void forward2Subscribers(PublishEvent pubEvt) {
boolean retain, Integer messageID) { final String topic = pubEvt.getTopic();
AbstractMessage.QOSType qos = pubEvt.getQos();
final ByteBuffer origMessage = pubEvt.getMessage();
boolean retain = pubEvt.isRetain();
final Integer messageID = pubEvt.getMessageID();
LOG.debug("forward2Subscribers republishing to existing subscribers that matches the topic {}", topic); LOG.debug("forward2Subscribers republishing to existing subscribers that matches the topic {}", topic);
if (LOG.isDebugEnabled()) { if (LOG.isDebugEnabled()) {
LOG.debug("content <{}>", DebugUtils.payload2Str(origMessage)); LOG.debug("content <{}>", DebugUtils.payload2Str(origMessage));
@@ -491,15 +492,12 @@ class ProtocolProcessor implements EventHandler<ValueEvent> {
LOG.debug("PUB --PUBREL--> SRV processPubRel invoked for clientID {} ad messageID {}", clientID, messageID); LOG.debug("PUB --PUBREL--> SRV processPubRel invoked for clientID {} ad messageID {}", clientID, messageID);
String publishKey = String.format("%s%d", clientID, messageID); String publishKey = String.format("%s%d", clientID, messageID);
PublishEvent evt = m_messagesStore.retrieveQoS2Message(publishKey); PublishEvent evt = m_messagesStore.retrieveQoS2Message(publishKey);
forward2Subscribers(evt);
final String topic = evt.getTopic();
final AbstractMessage.QOSType qos = evt.getQos();
forward2Subscribers(topic, qos, evt.getMessage(), evt.isRetain(), evt.getMessageID());
m_messagesStore.removeQoS2Message(publishKey); m_messagesStore.removeQoS2Message(publishKey);
if (evt.isRetain()) { if (evt.isRetain()) {
final String topic = evt.getTopic();
final AbstractMessage.QOSType qos = evt.getQos();
m_messagesStore.storeRetained(topic, evt.getMessage(), qos); m_messagesStore.storeRetained(topic, evt.getMessage(), qos);
} }
@@ -561,13 +559,6 @@ class ProtocolProcessor implements EventHandler<ValueEvent> {
void processConnectionLost(LostConnectionEvent evt) { void processConnectionLost(LostConnectionEvent evt) {
String clientID = evt.clientID; String clientID = evt.clientID;
// if (m_clientIDs.containsKey(clientID)) {
// if (!m_clientIDs.get(clientID).getSession().equals(evt.session)) {
// LOG.info("Received a lost connection with client <{}> for a not matching session", clientID);
// return;
// }
// }
//If already removed a disconnect message was already processed for this clientID //If already removed a disconnect message was already processed for this clientID
if (m_clientIDs.remove(clientID) != null) { if (m_clientIDs.remove(clientID) != null) {
@@ -16,6 +16,7 @@
package org.eclipse.moquette.spi.impl.events; package org.eclipse.moquette.spi.impl.events;
import org.eclipse.moquette.proto.messages.AbstractMessage.QOSType; import org.eclipse.moquette.proto.messages.AbstractMessage.QOSType;
import org.eclipse.moquette.proto.messages.PublishMessage;
import java.nio.ByteBuffer; import java.nio.ByteBuffer;
@@ -44,7 +45,18 @@ public class PublishEvent extends MessagingEvent {
m_msgID = msgID; m_msgID = msgID;
} }
} }
public PublishEvent(String clientID, PublishMessage msg) {
m_clientID = clientID;
m_topic = msg.getTopicName();
m_qos = msg.getQos();
m_message = msg.getPayload();
m_retain = msg.isRetainFlag();
if (msg.getQos() != QOSType.MOST_ONE) {
m_msgID = msg.getMessageID();
}
}
public String getTopic() { public String getTopic() {
return m_topic; return m_topic;
} }