74 lines
1.9 KiB
Python
74 lines
1.9 KiB
Python
#!/usr/bin/python
|
||
# -*- coding: utf-8 -*-
|
||
# @version : 1.0
|
||
# @Creaet Time : 2022/3/21 11:03
|
||
# @File : event.py
|
||
# @IDE : PyCharm
|
||
# @desc : 全局事件
|
||
|
||
|
||
from fastapi import FastAPI
|
||
from aioredis import from_url
|
||
from application.settings import REDIS_DB_URL, MONGO_DB_URL, MONGO_DB_NAME
|
||
from core.mongo import db
|
||
from utils.cache import cache_aliyun_settings
|
||
|
||
|
||
def register_redis(app: FastAPI) -> None:
|
||
"""
|
||
把 redis 挂载到 app 对象上面
|
||
|
||
博客：https://blog.csdn.net/wgPython/article/details/107668521
|
||
博客：https://www.cnblogs.com/emunshe/p/15761597.html
|
||
官网：https://aioredis.readthedocs.io/en/latest/getting-started/
|
||
@param app:
|
||
@return:
|
||
"""
|
||
|
||
@app.on_event('startup')
|
||
async def startup_event():
|
||
"""
|
||
获取链接
|
||
@return:
|
||
"""
|
||
print("Connecting to Redis")
|
||
app.state.redis = from_url(REDIS_DB_URL, decode_responses=True, health_check_interval=1)
|
||
await cache_aliyun_settings(app.state.redis)
|
||
|
||
@app.on_event('shutdown')
|
||
async def shutdown_event():
|
||
"""
|
||
关闭
|
||
@return:
|
||
"""
|
||
print("Redis connection closed")
|
||
await app.state.redis.close()
|
||
|
||
|
||
def register_mongo(app: FastAPI) -> None:
|
||
"""
|
||
把 mongo 挂载到 app 对象上面
|
||
|
||
博客：https://www.cnblogs.com/aduner/p/13532504.html
|
||
mongodb 官网：https://www.mongodb.com/docs/drivers/motor/
|
||
motor 文档：https://motor.readthedocs.io/en/stable/
|
||
@param app:
|
||
@return:
|
||
"""
|
||
|
||
@app.on_event('startup')
|
||
async def startup_event():
|
||
"""
|
||
获取 mongodb 连接
|
||
@return:
|
||
"""
|
||
await db.connect_to_database(path=MONGO_DB_URL, db_name=MONGO_DB_NAME)
|
||
|
||
@app.on_event('shutdown')
|
||
async def shutdown_event():
|
||
"""
|
||
关闭
|
||
@return:
|
||
"""
|
||
await db.close_database_connection()
|