Merge pull request #33 from bcmi-labs/umqtt_improvements

umqtt: Make umqtt socket non-blocking by default.

Author: iabdalkader

arduino_iot_cloud/umqtt.py

@@ -27,10 +27,12 @@
     import usocket as socket
     import ustruct as struct
     import ulogging as logging
+    import uselect as select
 except ImportError:
     import socket
     import struct
     import logging
+    import select
     from arduino_iot_cloud.ussl import wrap_socket
 
 
@@ -90,7 +92,7 @@ def set_last_will(self, topic, msg, retain=False, qos=0):
         self.lw_qos = qos
         self.lw_retain = retain
 
-    def connect(self, clean_session=True):
+    def connect(self, clean_session=True, timeout=5.0):
         addr = socket.getaddrinfo(self.server, self.port)[0][-1]
 
         if self.sock is not None:
@@ -99,11 +101,13 @@ def connect(self, clean_session=True):
 
         try:
             self.sock = socket.socket()
+            self.sock.settimeout(timeout)
             self.sock = wrap_socket(self.sock, **self.ssl_params)
             self.sock.connect(addr)
         except Exception:
             self.sock.close()
             self.sock = socket.socket()
+            self.sock.settimeout(timeout)
             self.sock.connect(addr)
             self.sock = wrap_socket(self.sock, **self.ssl_params)
 
@@ -214,9 +218,8 @@ def subscribe(self, topic, qos=0):
     # messages processed internally.
     def wait_msg(self):
         res = self.sock.read(1)
-        if res == b"" or res is None:
+        if res is None or res == b"":
             return None
-        self.sock.setblocking(True)
         if res == b"\xd0":  # PINGRESP
             sz = self.sock.read(1)[0]
             assert sz == 0
@@ -246,5 +249,6 @@ def wait_msg(self):
     # If not, returns immediately with None. Otherwise, does
     # the same processing as wait_msg.
     def check_msg(self):
-        self.sock.setblocking(False)
-        return self.wait_msg()
+        r, w, e = select.select([self.sock], [], [], 0.05)
+        if (len(r)):
+            return self.wait_msg()