46 lines
1.2 KiB
Python
46 lines
1.2 KiB
Python
import uvicorn
|
|
from aioredis import create_redis_pool
|
|
from fastapi import FastAPI
|
|
from fastapi.middleware.cors import CORSMiddleware
|
|
|
|
from routers import point, user, event
|
|
from settings import settings
|
|
|
|
from utils.ta_sdk import TGAnalytics, ToKafka
|
|
|
|
app = FastAPI()
|
|
|
|
app.add_middleware(
|
|
CORSMiddleware,
|
|
allow_origins=['*'],
|
|
allow_credentials=True,
|
|
allow_methods=["*"],
|
|
allow_headers=["*"],
|
|
)
|
|
|
|
def register_redis(app: FastAPI) -> None:
|
|
@app.on_event('startup')
|
|
async def startup_event():
|
|
app.state.redis = await create_redis_pool(**settings.REDIS_CONF)
|
|
|
|
@app.on_event('shutdown')
|
|
async def shutdown_event():
|
|
app.state.redis.close()
|
|
await app.state.redis.wait_closed()
|
|
|
|
|
|
def register_ta(app: FastAPI) -> None:
|
|
@app.on_event('startup')
|
|
def startup_event():
|
|
app.state.ta = TGAnalytics(ToKafka(settings.KAFKA_CONF))
|
|
|
|
|
|
app.include_router(point.router, prefix='/v1')
|
|
app.include_router(user.router, prefix='/v1')
|
|
app.include_router(event.router, prefix='/v1')
|
|
register_redis(app)
|
|
register_ta(app)
|
|
|
|
if __name__ == '__main__':
|
|
uvicorn.run(app='main:app', host="0.0.0.0", port=6666, reload=True, debug=True)
|