From 11ff2fa0a1371e366341e911bc269b8ec3fe395a Mon Sep 17 00:00:00 2001
From: "selva.andre" <andrea@sempai>
Date: Mon, 23 Jul 2012 21:04:20 +0200
Subject: [PATCH] Minor code layout change

---
 .../java/org/dna/mqtt/moquette/client/Client.java     | 11 +++++++++--
 1 file changed, 9 insertions(+), 2 deletions(-)

diff --git a/parser/src/main/java/org/dna/mqtt/moquette/client/Client.java b/parser/src/main/java/org/dna/mqtt/moquette/client/Client.java
index 336c9bf5..34a63e6f 100644
--- a/parser/src/main/java/org/dna/mqtt/moquette/client/Client.java
+++ b/parser/src/main/java/org/dna/mqtt/moquette/client/Client.java
@@ -19,8 +19,10 @@ import org.dna.mqtt.moquette.ConnectionException;
 import org.dna.mqtt.moquette.MQTTException;
 import org.dna.mqtt.moquette.proto.ConnAckDecoder;
 import org.dna.mqtt.moquette.proto.ConnectEncoder;
+import org.dna.mqtt.moquette.proto.DisconnectEncoder;
 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.server.Server;
 
 /**
@@ -62,6 +64,7 @@ public final class Client {
 
         DemuxingProtocolEncoder encoder = new DemuxingProtocolEncoder();
         encoder.addMessageEncoder(ConnectMessage.class, new ConnectEncoder());
+        encoder.addMessageEncoder(DisconnectMessage.class, new DisconnectEncoder());
 
         m_connector = new NioSocketConnector();
 
@@ -145,14 +148,18 @@ public final class Client {
         }
     }
     
+    
+    /**
+     *  TODO extract this SPI method in a SPI
+     */
     protected void connectionAckCallback(byte returnCode) {
         m_returnCode = returnCode;
         m_connectBarrier.countDown();
     }
     
     public void close() {
-        //TODO send the CLOSE message
-        
+        //send the CLOSE message
+        m_session.write(new DisconnectMessage());
         
         // wait until the summation is done
         m_session.getCloseFuture().awaitUninterruptibly();