increase read try counter

Florian-Barthel · Aug 13, 2024 · ab2658e · ab2658e
1 parent 38ed217
commit ab2658e
Show file tree

Hide file tree

Showing 2 changed files with 14 additions and 3 deletions.
diff --git a/splatviz_network/splatviz_network.py b/splatviz_network/splatviz_network.py
@@ -39,8 +39,19 @@ def try_connect(self):
 
     def read(self):
         messageLength = self.conn.recv(4)
-        messageLength = int.from_bytes(messageLength, 'little')
-        message = self.conn.recv(messageLength)
+        expected_bytes = int.from_bytes(messageLength, 'little')
+
+        current_bytes = 0
+        try_counter = 10
+        counter = 0
+        message = bytes()
+        while current_bytes < expected_bytes:
+            message += self.conn.recv(expected_bytes - current_bytes)
+            current_bytes = len(message)
+            counter += 1
+            if counter > try_counter:
+                print("Package loss")
+                break
         return json.loads(message.decode("utf-8"))
 
     def send(self, message_bytes, training_stats):

diff --git a/viz_renderer/attach_renderer.py b/viz_renderer/attach_renderer.py
@@ -58,7 +58,7 @@ def read(self, resolution):
         try:
             current_bytes = 0
             expected_bytes = resolution * resolution * 3
-            try_counter = 3
+            try_counter = 10
             counter = 0
             message = bytes()
             while current_bytes < expected_bytes: