feat: fixed socket typing for communication agent and ping router- automatically try to reconnect with robot.

ref: N25B-151

src/control_backend/agents/ri_communication_agent.py

@@ -4,6 +4,7 @@ import logging
 from spade.agent import Agent
 from spade.behaviour import CyclicBehaviour
 import zmq
+import zmq.asyncio
 
 
 from control_backend.core.config import settings
@@ -16,7 +17,8 @@ logger = logging.getLogger(__name__)
 
 
 class RICommunicationAgent(Agent):
-    req_socket: zmq.Socket
+    _pub_socket: zmq.asyncio.Socket 
+    req_socket: zmq.asyncio.Socket | None
     _address = ""
     _bind = True
     connected = False
@@ -25,14 +27,18 @@ class RICommunicationAgent(Agent):
         self,
         jid: str,
         password: str,
+        pub_socket: zmq.asyncio.Socket,
         port: int = 5222,
         verify_security: bool = False,
         address="tcp://localhost:0000",
         bind=False,
+        
     ):
         super().__init__(jid, password, port, verify_security)
         self._address = address
         self._bind = bind
+        self.req_socket = None
+        self._pub_socket = pub_socket
 
     class ListenBehaviour(CyclicBehaviour):
         async def run(self):
@@ -43,7 +49,7 @@ class RICommunicationAgent(Agent):
 
             # We need to listen and sent pings.
             message = {"endpoint": "ping", "data": {"id": "e.g. some reference id"}}
-            seconds_to_wait_total = 4.0
+            seconds_to_wait_total = 1.0
             try:
                 await asyncio.wait_for(
                     self.agent.req_socket.send_json(message), timeout=seconds_to_wait_total / 2
@@ -62,16 +68,12 @@ class RICommunicationAgent(Agent):
 
             # We didnt get a reply :(
             except TimeoutError as e:
-                logger.info(f"No ping back retrieved in {seconds_to_wait_total/2} seconds totalling {seconds_to_wait_total} of time, killing myself.")
-                self.agent.connected = False
+                logger.info(f"No ping back retrieved in {seconds_to_wait_total/2} seconds totalling {seconds_to_wait_total} of time, killing myself (or maybe just laying low).")
                 # TODO: Send event to UI letting know that we've lost connection
                 topic = b"ping"
                 data = json.dumps(False).encode()
-                pub_socket = context.socket(zmq.PUB)
-                pub_socket.connect(settings.zmq_settings.internal_comm_address)
-                pub_socket.send_multipart([topic, data])
-
-                self.kill()
+                self.agent._pub_socket.send_multipart([topic, data])
+                await self.agent.setup()
 
             except Exception as e:
                 logger.debug(f"Differennt exception: {e}")
@@ -90,25 +92,38 @@ class RICommunicationAgent(Agent):
                         "Received message with topic different than ping, while ping expected."
                     )
 
-    async def setup(self, max_retries: int = 5):
-        """
-        Try to setup the communication agent, we have 5 retries in case we dont have a response yet.
-        """
-        logger.info("Setting up %s", self.jid)
-        retries = 0
 
-
-        # Let's try a certain amount of times before failing connection
-        while retries < max_retries:
-            # Bind request socket
+    async def setup_req_socket(self, force = False):
+        """
+        Sets up request socket for communication agent.
+        """
+        if self.req_socket is None or force:
             self.req_socket = context.socket(zmq.REQ)
             if self._bind:
                 self.req_socket.bind(self._address)
             else:
                 self.req_socket.connect(self._address)
 
+        
+    async def setup(self, max_retries: int = 5):
+        """
+        Try to setup the communication agent, we have 5 retries in case we dont have a response yet.
+        """
+        logger.info("Setting up %s", self.jid)
+
+        # Bind request socket
+        await self.setup_req_socket()
+        
+        retries = 0
+        # Let's try a certain amount of times before failing connection
+        while retries < max_retries:
+
+            # Make sure the socket is properly setup.
+            if self.req_socket is None:
+                continue
+            
             # Send our message and receive one back:)
-            message = {"endpoint": "negotiate/ports", "data": None}
+            message = {"endpoint": "negotiate/ports", "data": {}}
             await self.req_socket.send_json(message)
 
             try:
@@ -190,5 +205,8 @@ class RICommunicationAgent(Agent):
         self.add_behaviour(listen_behaviour)
 
         # TODO: Let UI know that we're connected >:)
+        topic = b"ping"
+        data = json.dumps(True).encode()
+        await self._pub_socket.send_multipart([topic, data])
         self.connected = True
         logger.info("Finished setting up %s", self.jid)