websocket: raise alarm on connectivity loss

daemon.py

@@ -44,7 +44,7 @@ def run():
     else:
         ser_handler = SerialHandler(db_queue, request_queue, serial_output_queue, alarm_input_queue)
     db_handler = DbClient(db_queue)
-    websocket_handler = WebsocketHandler(serial_output_queue)
+    websocket_handler = WebsocketHandler(serial_output_queue, alarm_input_queue)
     alarm_handler = AlarmHandler(alarm_input_queue,serial_output_queue, request_queue)
     request_handler = RequestHandler(api_request, request_queue)
 

ventilator_alarm.py

@@ -42,7 +42,7 @@ def run(self, name):
                 msg = None
 
             if msg != None:
-                if msg['type'] == "ALARM":
+                if msg['type'] == proto.alarm:
                     self.time_last_kick_received == cur_time
                     if msg['val'] != 0:
                         self.request_queue.put({'type': 'error', 'value': msg['val']})

ventilator_websocket.py

@@ -23,6 +23,9 @@ def handle_settings(self, settings):
                 print("send setting {}".format(key))
                 self.serial_queue.put({'type': key, 'val': settings[key]})
 
+    def raise_alarm(self):
+        self.alarm_queue.put({'type': proto.alarm, 'val': 1})
+
     def subscribe(self, path):
         """
         Subscribe to updates
@@ -39,6 +42,12 @@ def do_handshake(self):
         self.send_msg(hello_msg)
         reply = self.ws.recv()
         print(reply)
+        try:
+            msg = json.loads(reply)
+            self.interval = msg['heartbeat']['interval']
+            self.ws.settimeout(self.interval/1000)
+        except:
+            print("Could not parse handshake response")
 
     def run(self, name):
         print("Starting {}".format(name))
@@ -50,7 +59,17 @@ def run(self, name):
         self.subscribe('settings')
 
         while True:
-            json_msg = self.ws.recv()
+            try:
+                json_msg = self.ws.recv()
+            except:
+                print("Timeout or socket closed: reconnecting")
+                self.raise_alarm()
+                self.ws.close()
+                self.attempt_reconnect()
+                self.do_handshake()
+                self.subscribe('settings')
+                continue
+
             try:
                 msg = json.loads(json_msg)
                 if msg['type'] == "ping":
@@ -73,10 +92,11 @@ def attempt_reconnect(self):
             except:
                 continue
 
-    def __init__(self, serial_queue, addr='localhost', port=3001):
+    def __init__(self, serial_queue, alarm_queue, addr='localhost', port=3001):
         self.url = "ws://" + addr + ":" + str(port) + "/"
         self.id = 1
         self.serial_queue = serial_queue
+        self.alarm_queue = alarm_queue
 
 
 if __name__ == "__main__":