Further improve EventHandler <-> EventHandlerBase port situation

This commit is contained in:
Kristóf Tóth 2019-07-03 16:48:17 +02:00
parent 49844c7b58
commit ca697b41b3
4 changed files with 32 additions and 22 deletions

View File

@ -1,10 +1,9 @@
from tfw.event_handlers import EventHandlerBase
from tfw.networking import Scope
from .tfw_server_connector import TFWServerConnector
class EventHandler(EventHandlerBase):
# pylint: disable=abstract-method
def __init__(self, key, scope=Scope.ZMQ):
super().__init__(key, TFWServerConnector(), scope=scope)
def _build_server_connector(self):
return TFWServerConnector()

View File

@ -1,12 +1,10 @@
from abc import ABC
from tfw.components import FSMAware
from tfw.networking import Scope
from .event_handler import EventHandler
class FSMAwareEventHandler(EventHandler, FSMAware, ABC):
class FSMAwareEventHandler(EventHandler, FSMAware):
# pylint: disable=abstract-method
"""
Abstract base class for EventHandlers which automatically

View File

@ -1,19 +1,25 @@
from functools import partial
from tfw.networking import ServerUplinkConnector, ServerConnector
from tfw.config import TFWENV
UPLINK_CONN_ADDR = f'tcp://localhost:{TFWENV.PULL_PORT}'
DOWNLINK_CONN_ADDR = f'tcp://localhost:{TFWENV.PUB_PORT}'
class ConnAddrMixin:
@property
def uplink_conn_addr(self):
return f'tcp://localhost:{TFWENV.PULL_PORT}'
@property
def downlink_conn_addr(self):
return f'tcp://localhost:{TFWENV.PUB_PORT}'
TFWServerUplinkConnector = partial(
ServerUplinkConnector,
connect_addr=UPLINK_CONN_ADDR
)
TFWServerConnector = partial(
ServerConnector,
downlink_connect_addr=DOWNLINK_CONN_ADDR,
uplink_connect_addr=UPLINK_CONN_ADDR
)
class TFWServerUplinkConnector(ServerUplinkConnector, ConnAddrMixin):
def __init__(self):
super().__init__(self.uplink_conn_addr)
class TFWServerConnector(ServerConnector, ConnAddrMixin):
def __init__(self):
super().__init__(
self.downlink_conn_addr,
self.uplink_conn_addr
)

View File

@ -1,10 +1,13 @@
import logging
from abc import ABC, abstractmethod
from typing import Iterable
from tfw.networking import Scope
LOG = logging.getLogger(__name__)
class EventHandlerBase:
class EventHandlerBase(ABC):
"""
Abstract base class for all Python based EventHandlers. Useful implementation template
for other languages.
@ -13,9 +16,9 @@ class EventHandlerBase:
"""
_instances = set()
def __init__(self, key, server_connector, scope):
def __init__(self, key, scope=Scope.ZMQ):
type(self)._instances.add(self)
self.server_connector = server_connector
self.server_connector = self._build_server_connector()
self.scope = scope
self.keys = []
if isinstance(key, str):
@ -26,6 +29,10 @@ class EventHandlerBase:
self.subscribe(*self.keys)
self.server_connector.register_callback(self.event_handler_callback)
@abstractmethod
def _build_server_connector(self):
raise NotImplementedError()
def subscribe(self, *keys):
"""
Subscribe this EventHandler to receive events for given keys.