update

db/docker-entrypoint.sh

@@ -0,0 +1,58 @@
+#!/bin/bash
+set -e
+
+cd "$(dirname "$0")"
+
+DATA_DIR="/app/data"
+DB_PATH="${DATABASE_URL#sqlite:///}"
+if [ "$DB_PATH" = "$DATABASE_URL" ]; then
+    # URL 不是 sqlite:/// 格式，使用默认路径
+    DB_PATH="$DATA_DIR/database.db"
+fi
+
+echo "数据目录: $DATA_DIR"
+echo "数据库路径: $DB_PATH"
+
+# 确保数据目录存在并可写
+mkdir -p "$(dirname "$DB_PATH")"
+
+# 初始化数据库表（如果不存在）
+if [[ "$DATABASE_URL" == sqlite:* ]]; then
+    echo "初始化数据库 schema..."
+    python -c "
+import os
+from sqlalchemy import create_engine, text
+
+db_url = os.environ.get('DATABASE_URL', 'sqlite:////app/data/database.db')
+engine = create_engine(
+    db_url,
+    connect_args={'check_same_thread': False} if db_url.startswith('sqlite') else {},
+)
+# 应用 SQLite PRAGMA 优化
+from sqlalchemy import event
+@event.listens_for(engine, 'connect')
+def _configure_sqlite(dbapi_conn, _):
+    cur = dbapi_conn.cursor()
+    cur.execute('PRAGMA journal_mode=WAL')
+    cur.execute('PRAGMA synchronous=NORMAL')
+    cur.execute('PRAGMA foreign_keys=ON')
+    cur.execute('PRAGMA busy_timeout=5000')
+    cur.close()
+
+with open('create_sqlite3.sql', 'r') as f:
+    schema_sql = f.read()
+with engine.connect() as conn:
+    for stmt in schema_sql.split(';'):
+        stmt = stmt.strip()
+        if stmt:
+            try:
+                conn.execute(text(stmt))
+            except Exception as e:
+                pass
+    conn.commit()
+print('Schema 初始化完成.')
+"
+fi
+
+echo "启动 dbv2 服务 ${LISTEN_ADDR:-0.0.0.0}:${LISTEN_PORT:-8005}..."
+exec python dbv2.py --addr "${LISTEN_ADDR:-0.0.0.0}" --port "${LISTEN_PORT:-8005}" --workers "${WORKERS:-4}"