feat: fix
This commit is contained in:
6
main.py
6
main.py
@@ -9,8 +9,8 @@ from starlette.staticfiles import StaticFiles
|
||||
|
||||
import routers
|
||||
from constants import API_ROOT
|
||||
from external.kafka import producer, consume_messages
|
||||
from external.kafka.producer import init_producer
|
||||
from external.kafka import consume_messages
|
||||
from external.kafka.producer import init_producer, get_producer
|
||||
|
||||
origins = [
|
||||
'http://localhost:5173'
|
||||
@@ -21,6 +21,7 @@ origins = [
|
||||
async def lifespan(app: FastAPI):
|
||||
try:
|
||||
await init_producer()
|
||||
producer = await get_producer()
|
||||
if producer:
|
||||
await producer.start()
|
||||
except KafkaConnectionError as e:
|
||||
@@ -29,6 +30,7 @@ async def lifespan(app: FastAPI):
|
||||
consumer_task = asyncio.create_task(consume_messages())
|
||||
|
||||
yield
|
||||
producer = await get_producer()
|
||||
if producer:
|
||||
await producer.stop()
|
||||
consumer_task.cancel()
|
||||
|
||||
Reference in New Issue
Block a user