little update
This commit is contained in:
44
src/scripts/fire_game_cleanup.py
Normal file
44
src/scripts/fire_game_cleanup.py
Normal file
@@ -0,0 +1,44 @@
|
||||
"""Publish a game cleanup request to NATS JetStream."""
|
||||
|
||||
import argparse
|
||||
import asyncio
|
||||
import json
|
||||
import os
|
||||
import sys
|
||||
from pathlib import Path
|
||||
from uuid import UUID
|
||||
|
||||
import nats
|
||||
|
||||
if __package__ is None or __package__ == "":
|
||||
sys.path.insert(0, str(Path(__file__).resolve().parents[1]))
|
||||
|
||||
from scripts.init_nats import ensure_stream
|
||||
|
||||
|
||||
async def publish(url: str, stream: str, subject: str, start_subject: str, game_id: UUID) -> None:
|
||||
await ensure_stream(url, stream, [start_subject, subject])
|
||||
nc = await nats.connect(url)
|
||||
try:
|
||||
await nc.jetstream().publish(
|
||||
subject,
|
||||
json.dumps({"game_id": str(game_id)}).encode(),
|
||||
)
|
||||
print(f"published cleanup for game {game_id} to {subject} ({stream})")
|
||||
finally:
|
||||
await nc.drain()
|
||||
|
||||
|
||||
def main() -> None:
|
||||
parser = argparse.ArgumentParser(description="Queue cleanup of one game")
|
||||
parser.add_argument("game_id", type=UUID)
|
||||
parser.add_argument("--url", default=os.getenv("NATS_URL", "nats://127.0.0.1:4222"))
|
||||
parser.add_argument("--stream", default=os.getenv("NATS_STREAM", "GAME"))
|
||||
parser.add_argument("--subject", default=os.getenv("NATS_CLEANUP_SUBJECT", "game.cleanup"))
|
||||
parser.add_argument("--start-subject", default=os.getenv("NATS_SUBJECT", "game.start"))
|
||||
args = parser.parse_args()
|
||||
asyncio.run(publish(args.url, args.stream, args.subject, args.start_subject, args.game_id))
|
||||
|
||||
|
||||
if __name__ == "__main__":
|
||||
main()
|
||||
Reference in New Issue
Block a user