unknown changes
This commit is contained in:
1 parent
bec3fcaa17
commit
538417922c
13 files changed
+428
-20
No files matched your search
@@ -0,0 +1,32 @@
|
||||
import asyncio
|
||||
import os
|
||||
from enum import Enum
|
||||
|
||||
from typing import Any
|
||||
|
||||
from pydantic import BaseModel, Field
|
||||
from faststream import FastStream
|
||||
from faststream.nats import NatsBroker
|
||||
|
||||
from models import ProbeRegister, ProbeUnregister, ProbeHeartBeat
|
||||
|
||||
NATS_SERVERS = os.getenv('NATS_URL', 'nats://localhost:4222').split(',')
|
||||
NATS_TOKEN = os.getenv('NATS_TOKEN', '')
|
||||
|
||||
broker = NatsBroker(servers=NATS_SERVERS, token=NATS_TOKEN)
|
||||
app = FastStream(broker)
|
||||
|
||||
|
||||
|
||||
@broker.subscriber(f"discovery.probe")
|
||||
async def probe_discovery(data: ProbeRegister | ProbeHeartBeat | ProbeUnregister):
|
||||
print(20*'-')
|
||||
print(type(data), "->", data)
|
||||
return True
|
||||
|
||||
async def main():
|
||||
print("START")
|
||||
await app.run()
|
||||
|
||||
if __name__ == "__main__":
|
||||
asyncio.run(main())
|
||||
Reference in new issue
Block a user