feat(connector.py): add on_message handler function argument to init
This commit is contained in:
@@ -4,6 +4,7 @@ from __future__ import annotations
|
|||||||
import logging as lg
|
import logging as lg
|
||||||
import socket
|
import socket
|
||||||
import time
|
import time
|
||||||
|
from typing import Callable
|
||||||
|
|
||||||
from judas_protocol import Message
|
from judas_protocol import Message
|
||||||
|
|
||||||
@@ -17,6 +18,7 @@ class Connector:
|
|||||||
*,
|
*,
|
||||||
connect_timeout: float = 5.0,
|
connect_timeout: float = 5.0,
|
||||||
ack_timeout: float | None = None,
|
ack_timeout: float | None = None,
|
||||||
|
on_message: Callable[[Message], None],
|
||||||
) -> None:
|
) -> None:
|
||||||
self.logger: lg.Logger = lg.getLogger(
|
self.logger: lg.Logger = lg.getLogger(
|
||||||
f"{__name__}.{self.__class__.__name__}"
|
f"{__name__}.{self.__class__.__name__}"
|
||||||
@@ -35,6 +37,8 @@ class Connector:
|
|||||||
|
|
||||||
self.mac_address: str = mac_address
|
self.mac_address: str = mac_address
|
||||||
|
|
||||||
|
self.on_message: Callable[[Message], None] = on_message
|
||||||
|
|
||||||
def _send_ack(self) -> None:
|
def _send_ack(self) -> None:
|
||||||
self.logger.debug("[>] Sending ACK...")
|
self.logger.debug("[>] Sending ACK...")
|
||||||
try:
|
try:
|
||||||
|
|||||||
Reference in New Issue
Block a user