Diff: MQTTSocket.h

Revision:

45:b7cebc104c6c

Parent:

43:21da1f744243

--- a/MQTTSocket.h	Mon Oct 06 11:41:05 2014 +0000
+++ b/MQTTSocket.h	Sun Jun 28 22:44:19 2015 +0000
@@ -6,33 +6,41 @@
 
 class MQTTSocket
 {
-public:    
+public: 
+    ~MQTTSocket()
+    {   
+        if(mysock)
+            delete mysock;
+    }  
+      
     int connect(char* hostname, int port, int timeout=1000)
     {
-        mysock.set_blocking(false, timeout);    // 1 second Timeout 
-        return mysock.connect(hostname, port);
+        mysock->set_blocking(false, timeout);    // 1 second Timeout 
+        return mysock->connect(hostname, port);
     }
 
     int read(unsigned char* buffer, int len, int timeout)
     {
-        mysock.set_blocking(false, timeout);  
-        return mysock.receive((char*)buffer, len);
+        mysock->set_blocking(false, timeout);  
+        return mysock->receive((char*)buffer, len);
     }
     
     int write(unsigned char* buffer, int len, int timeout)
     {
-        mysock.set_blocking(false, timeout);  
-        return mysock.send((char*)buffer, len);
+        mysock->set_blocking(false, timeout);  
+        return mysock->send((char*)buffer, len);
     }
     
     int disconnect()
     {
-        return mysock.close();
+        return mysock->close();
     }
-    
+
+protected:
+    void createSocket() { mysock = new TCPSocketConnection(); }
 private:
 
-    TCPSocketConnection mysock; 
+    TCPSocketConnection *mysock; 
     
 };