anna/IM-server-new
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

Added to the server's MQTTHandler the handleing of DISCONNECT message

Browse Source
This commit is contained in:
selva.andre
2012-08-10 19:31:14 +02:00
parent 5be34ec932
commit 8e3f83403f
1 changed files with 16 additions and 0 deletions
Download Patch File Download Diff File Expand all files Collapse all files
parser/src/main/java/org/dna/mqtt/moquette/server/MQTTHandler.java
+16
View File
@@ -12,6 +12,7 @@ import static org.dna.mqtt.moquette.proto.messages.AbstractMessage.*;
import org.dna.mqtt.moquette.proto.messages.AbstractMessage.QOSType;
import org.dna.mqtt.moquette.proto.messages.ConnAckMessage;
import org.dna.mqtt.moquette.proto.messages.ConnectMessage;
import org.dna.mqtt.moquette.proto.messages.DisconnectMessage;
import org.dna.mqtt.moquette.proto.messages.PublishMessage;
import org.dna.mqtt.moquette.proto.messages.SubAckMessage;
import org.dna.mqtt.moquette.proto.messages.SubscribeMessage;
@@ -47,6 +48,9 @@ public class MQTTHandler extends IoHandlerAdapter implements INotifier {
case PUBLISH:
handlePublish(session, (PublishMessage) msg);
break;
case DISCONNECT:
handleDisconnect(session, (DisconnectMessage) msg);
}
} catch (Exception ex) {
LOG.error("Bad error in processing the message", ex);
@@ -72,6 +76,7 @@ public class MQTTHandler extends IoHandlerAdapter implements INotifier {
//if an old client with the same ID already exists close its session.
if (m_clientIDs.containsKey(msg.getClientID())) {
//TODO also clean the subscriptions if it was in cleanSession = true
m_clientIDs.get(msg.getClientID()).getSession().close(false);
}
@@ -138,6 +143,17 @@ public class MQTTHandler extends IoHandlerAdapter implements INotifier {
message.getQos(), message.isRetainFlag());
}
protected void handleDisconnect(IoSession session, DisconnectMessage disconnectMessage) {
String clientID = (String) session.getAttribute(ATTR_CLIENTID);
//remove from clientIDs
m_clientIDs.remove(clientID);
boolean cleanSession = (Boolean) session.getAttribute("cleanSession");
if (cleanSession) {
//cleanup topic subscriptions
m_messaging.removeSubscriptions(clientID);
}
}
@Override
public void sessionIdle(IoSession session, IdleStatus status) {
session.close(false);