59 lines
1.6 KiB
Bash
59 lines
1.6 KiB
Bash
#!/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}"
|