add formating to the pipeline
parent
d8aff91f6f
commit
d9ef6690fc
@ -0,0 +1,31 @@
|
|||||||
|
"""
|
||||||
|
Format the alert message.
|
||||||
|
"""
|
||||||
|
|
||||||
|
import asyncio
|
||||||
|
from typing import (
|
||||||
|
Any,
|
||||||
|
NoReturn
|
||||||
|
)
|
||||||
|
|
||||||
|
def format_alert(
|
||||||
|
alert: dict[str, Any]
|
||||||
|
)->str:
|
||||||
|
"""
|
||||||
|
Format an alert in json format to a nice string.
|
||||||
|
"""
|
||||||
|
return str(alert)
|
||||||
|
|
||||||
|
async def format_alerts(
|
||||||
|
alert_queue: asyncio.Queue[dict[str,Any]],
|
||||||
|
message_queue: asyncio.Queue[str]
|
||||||
|
)->NoReturn:
|
||||||
|
"""
|
||||||
|
Read alerts from alert_queue, format them, and put them in message_queue.
|
||||||
|
"""
|
||||||
|
while True:
|
||||||
|
alert = await alert_queue.get()
|
||||||
|
message = format_alert(alert)
|
||||||
|
await message_queue.put(message)
|
||||||
|
alert_queue.task_done()
|
||||||
|
|
Loading…
Reference in New Issue