src/judas_client/connector.py

@@ -4,6 +4,7 @@ from __future__ import annotations
 import logging as lg
 import socket
 import time
+from typing import Callable
 
 from judas_protocol import Message
 
@@ -17,6 +18,7 @@ class Connector:
         *,
         connect_timeout: float = 5.0,
         ack_timeout: float | None = None,
+        on_message: Callable[[Message], None],
     ) -> None:
         self.logger: lg.Logger = lg.getLogger(
             f"{__name__}.{self.__class__.__name__}"
@@ -35,6 +37,8 @@ class Connector:
 
         self.mac_address: str = mac_address
 
+        self.on_message: Callable[[Message], None] = on_message
+
     def _send_ack(self) -> None:
         self.logger.debug("[>] Sending ACK...")
         try: