main.py

from fastapi import FastAPI
import uvicorn
import asyncio
import uuid
import random
import json
import utils
from sqlalchemy.exc import OperationalError
from infrastructure.consumers import subscribe_to_topic
from modules.orders.application.events.events import EventOrderCreated, OrderCreatedPayload
from modules.orders.application.commands.commands import CommandCheckInventoryOrder
from config.db import Base, engine, initialize_base
from modules.orders.application.logic import create_order
from infrastructure.dispatchers import Dispatcher

app = FastAPI()

tasks = list()
initialize_base()
try:
    Base.metadata.create_all(bind=engine)
except OperationalError:
    Base.metadata.create_all(bind=engine)

@app.on_event("startup")
async def app_startup():
    global tasks
    task1 = asyncio.ensure_future(subscribe_to_topic(
        "order-events", "sub-inbound", EventOrderCreated))
    task2 = asyncio.ensure_future(subscribe_to_topic(
        "order-commands", "sub-com-inbound", CommandCheckInventoryOrder))
    tasks.append(task1)
    tasks.append(task2)


@app.on_event("shutdown")
def shutdown_event():
    global tasks
    for task in tasks:
        task.cancel()

@app.get("/orders")
def create_order_endpoint():
    event_payload = OrderCreatedPayload(
        order_id = str(uuid.uuid4()),
        customer_id = str(uuid.uuid4()),
        order_date = str("2023-02-27T08:05:08.464634"),
        order_status = "Created",
        order_items = json.dumps([{ "product_id": "9cad4dc7-50c0-44d7-9ed9-3f887a9d565b", "supplier_id": "987eba3c-ae2b-4382-86f9-7ea238733e05", "name": "product1", "description": "Test Desc", "price": 33000.0, "quantity": 5 } ]),
        order_total = float(random.randint(2, 15000)),
        order_version = int(random.randint(1,10))
    )

    event = EventOrderCreated(
        time = utils.time_millis(),
        ingestion = utils.time_millis(),
        datacontenttype = OrderCreatedPayload.__name__,
        data_payload = event_payload,
        type = "CommandCreateOrder"
    )

    event2 = EventOrderCreated(
        time = utils.time_millis(),
        ingestion = utils.time_millis(),
        datacontenttype = OrderCreatedPayload.__name__,
        data_payload = event_payload,
        type = "CancelOrder"
    )

    dispatcher = Dispatcher()
    dispatcher.publish_message(event, "order-events")
    import time
    time.sleep(1)
    dispatcher.publish_message(event2, "order-events")
    return {"message": "Order created successfully"}


if __name__ == "__main__":
    uvicorn.run(app, host="localhost", port=6969)