初始化

app/core/database.py

@@ -0,0 +1,58 @@
+from sqlalchemy import create_engine, text, MetaData
+from sqlalchemy.ext.declarative import declarative_base
+from sqlalchemy.orm import sessionmaker
+from sqlalchemy.pool import QueuePool
+from .config import settings
+
+# Railway数据库引擎（账号表）
+railway_engine = create_engine(
+    settings.RAILWAY_DATABASE_URL,
+    poolclass=QueuePool,
+    pool_pre_ping=True,
+    echo=False,
+    pool_size=20,
+    max_overflow=30,
+    pool_timeout=60,
+    pool_recycle=3600,
+    pool_reset_on_return='commit'
+)
+
+# Tunnel数据库引擎（业务数据表）
+tunnel_engine = create_engine(
+    settings.TUNNEL_DATABASE_URL,
+    poolclass=QueuePool,
+    pool_pre_ping=True,
+    echo=False,
+    pool_size=100,
+    max_overflow=200,
+    pool_timeout=60,
+    pool_recycle=3600,
+    pool_reset_on_return='commit'
+)
+
+RailwaySessionLocal = sessionmaker(autocommit=False, autoflush=False, bind=railway_engine)
+TunnelSessionLocal = sessionmaker(autocommit=False, autoflush=False, bind=tunnel_engine)
+
+RailwayBase = declarative_base()
+TunnelBase = declarative_base()
+
+def get_railway_db():
+    """Railway数据库依赖注入"""
+    db = RailwaySessionLocal()
+    try:
+        yield db
+    finally:
+        db.close()
+
+def get_tunnel_db():
+    """Tunnel数据库依赖注入"""
+    db = TunnelSessionLocal()
+    try:
+        yield db
+    finally:
+        db.close()
+
+def init_db():
+    """初始化数据库"""
+    RailwayBase.metadata.create_all(bind=railway_engine)
+    TunnelBase.metadata.create_all(bind=tunnel_engine)