Commit 2c2455ad authored by Hines, Jesse's avatar Hines, Jesse
Browse files

Fixes to topic creation

parent 03c0ee7e

simulation_server/server/main.py

+6 −5
Original line number Diff line number Diff line
@@ -50,9 +50,8 @@ async def lifespan(api: FastAPI): 


    if settings.env == 'dev':

        kafka_admin = get_kafka_admin()

        existing_topics = {t.topic for t in kafka_admin.list_topics().topics}

        topic_config = {"compression.type": "snappy"}

        new_topics = [

        topics = [

            NewTopic("svc-event-exadigit-sim", 1, 1, config = topic_config),

            NewTopic("svc-ts-exadigit-schedulersimsystem", 4, 1, config = topic_config),

            NewTopic("svc-event-exadigit-schedulersimjob", 2, 1, config = topic_config),
@@ -60,7 +59,9 @@ async def lifespan(api: FastAPI): 
            NewTopic("svc-ts-exadigit-coolingsimcep", 2, 1, config = topic_config),

            NewTopic("svc-ts-exadigit-jobpowerhistory", 4, 1, config = topic_config),

        ]

        new_topics = [t for t in new_topics if t.topic not in existing_topics]

        existing_topics = set(kafka_admin.list_topics().topics.keys())

        new_topics = [t for t in topics if t.topic not in existing_topics]

        if new_topics:

            logger.info(f"Creating kafka topics {', '.join(t.topic for t in new_topics)}")

            kafka_admin.create_topics(new_topics)