mirror of
https://github.com/xfgryujk/blivechat.git
synced 2024-12-25 20:30:28 +08:00
33 lines
864 B
Python
33 lines
864 B
Python
# -*- coding: utf-8 -*-
|
||
from typing import *
|
||
|
||
import sqlalchemy.orm
|
||
|
||
import config
|
||
|
||
_engine: Optional[sqlalchemy.Engine] = None
|
||
|
||
|
||
class OrmBase(sqlalchemy.orm.DeclarativeBase):
|
||
pass
|
||
|
||
|
||
def init():
|
||
cfg = config.get_config()
|
||
global _engine
|
||
_engine = sqlalchemy.create_engine(
|
||
cfg.database_url,
|
||
pool_size=5, # 保持的连接数
|
||
max_overflow=5, # 临时的额外连接数
|
||
pool_timeout=3, # 连接数达到最大时获取新连接的超时时间
|
||
# pool_pre_ping=True, # 获取连接时先检测是否可用
|
||
pool_recycle=60 * 60, # 回收超过1小时的连接，防止数据库服务器主动断开不活跃的连接
|
||
# echo=cfg.debug, # 输出SQL语句
|
||
)
|
||
|
||
OrmBase.metadata.create_all(_engine)
|
||
|
||
|
||
def get_session() -> sqlalchemy.orm.Session:
|
||
return sqlalchemy.orm.Session(_engine)
|