From 346b337f59ba9b9d40f9d76f7276dbebaefd0c1d Mon Sep 17 00:00:00 2001
From: Sem van der Hoeven <semmer99@gmail.com>
Date: Thu, 25 May 2023 17:04:43 +0200
Subject: [PATCH] handle disconnection

---
 src/api_communication/api_communication/api_listener.py | 5 +++++
 1 file changed, 5 insertions(+)

diff --git a/src/api_communication/api_communication/api_listener.py b/src/api_communication/api_communication/api_listener.py
index bed7da63..7270a057 100644
--- a/src/api_communication/api_communication/api_listener.py
+++ b/src/api_communication/api_communication/api_listener.py
@@ -168,6 +168,11 @@ class ApiListener(Node):
 
         except websockets.exceptions.ConnectionClosed:
             self.get_logger().info('Connection closed')
+            self.websocket = None
+        except Exception as e:
+            self.get_logger().error('Something went wrong!')
+            self.get_logger().error(str(e))
+            self.websocket = None
 
 
 def main(args=None):