mirror of
https://github.com/OMGeeky/flucto-heisskleber.git
synced 2025-12-26 16:07:50 +01:00
8c985bdf3c
* Add start, stop and __repr__ to sink and source types. * Merge conflicts on mqtt async pub and resampler. * Add start() and stop() functions to udp and zmq. Change tests accordingly. * Rename broker, ip, interface to common config name "host". * Updated "host" entry in config files. * Add lazyload to mqtt-source.
71 lines
2.3 KiB
Python
71 lines
2.3 KiB
Python
from asyncio import Queue, Task, create_task, sleep
|
|
|
|
import aiomqtt
|
|
|
|
from heisskleber.core.packer import get_packer
|
|
from heisskleber.core.types import AsyncSink, Serializable
|
|
|
|
from .config import MqttConf
|
|
|
|
|
|
class AsyncMqttPublisher(AsyncSink):
|
|
"""
|
|
MQTT publisher class.
|
|
Can be used everywhere that a flucto style publishing connection is required.
|
|
|
|
Network message loop is handled in a separated thread.
|
|
"""
|
|
|
|
def __init__(self, config: MqttConf) -> None:
|
|
self.config = config
|
|
self.pack = get_packer(config.packstyle)
|
|
self._send_queue: Queue[tuple[dict[str, Serializable], str]] = Queue()
|
|
self._sender_task: Task[None] | None = None
|
|
|
|
async def send(self, data: dict[str, Serializable], topic: str) -> None:
|
|
"""
|
|
Takes python dictionary, serializes it according to the packstyle
|
|
and sends it to the broker.
|
|
|
|
Publishing is asynchronous
|
|
"""
|
|
if not self._sender_task:
|
|
self.start()
|
|
|
|
await self._send_queue.put((data, topic))
|
|
|
|
async def send_work(self) -> None:
|
|
"""
|
|
Takes python dictionary, serializes it according to the packstyle
|
|
and sends it to the broker.
|
|
|
|
Publishing is asynchronous
|
|
"""
|
|
while True:
|
|
try:
|
|
async with aiomqtt.Client(
|
|
hostname=self.config.host,
|
|
port=self.config.port,
|
|
username=self.config.user,
|
|
password=self.config.password,
|
|
timeout=float(self.config.timeout_s),
|
|
) as client:
|
|
while True:
|
|
data, topic = await self._send_queue.get()
|
|
payload = self.pack(data)
|
|
await client.publish(topic, payload)
|
|
except aiomqtt.MqttError:
|
|
print("Connection to MQTT broker failed. Retrying in 5 seconds")
|
|
await sleep(5)
|
|
|
|
def __repr__(self) -> str:
|
|
return f"{self.__class__.__name__}(broker={self.config.host}, port={self.config.port})"
|
|
|
|
def start(self) -> None:
|
|
self._sender_task = create_task(self.send_work())
|
|
|
|
def stop(self) -> None:
|
|
if self._sender_task:
|
|
self._sender_task.cancel()
|
|
self._sender_task = None
|