From 4a4fbdc949e02ed5f3a39fc19bd62e1a7e661725 Mon Sep 17 00:00:00 2001
From: =?UTF-8?q?=E7=94=B2=E8=BE=B0=E7=94=9F=E4=BA=A7?= <jiachenlong@local>
Date: Wed, 8 Apr 2026 09:14:44 +0800
Subject: [PATCH] =?UTF-8?q?v1.2.58=20-=20=E6=81=A2=E5=A4=8Dcoolbot=5Fdb?=
 =?UTF-8?q?=E6=A8=A1=E5=9D=97?=
MIME-Version: 1.0
Content-Type: text/plain; charset=UTF-8
Content-Transfer-Encoding: 8bit

---
 backend/app/core/coolbot_db.py | 37 ++++++++++++++++++++++++++++++++++
 1 file changed, 37 insertions(+)
 create mode 100644 backend/app/core/coolbot_db.py

diff --git a/backend/app/core/coolbot_db.py b/backend/app/core/coolbot_db.py
new file mode 100644
index 0000000..a0b9014
--- /dev/null
+++ b/backend/app/core/coolbot_db.py
@@ -0,0 +1,37 @@
+import os
+from sqlalchemy import create_engine
+from sqlalchemy.orm import sessionmaker
+from sqlalchemy.pool import QueuePool
+
+COOLBOT_DB_URL = os.getenv(
+    "COOLBOT_DB_URL",
+    "postgresql://coolbot:Coolbot123@pgm-bp1t1008h019ez6cno.pg.rds.aliyuncs.com:5432/coolbot_data"
+)
+
+coolbot_engine = create_engine(
+    COOLBOT_DB_URL,
+    poolclass=QueuePool,
+    pool_size=10,
+    max_overflow=20,
+    pool_timeout=30,
+    pool_recycle=1800,
+    pool_pre_ping=True,
+    echo=False,
+    connect_args={
+        "connect_timeout": 10,
+        "application_name": "zodiac-coolbot"
+    }
+)
+
+CoolbotSession = sessionmaker(autocommit=False, autoflush=False, bind=coolbot_engine)
+
+def get_coolbot_db():
+    """获取coolbot数据库会话"""
+    db = CoolbotSession()
+    try:
+        yield db
+    except Exception:
+        db.rollback()
+        raise
+    finally:
+        db.close()