382 lines
9.7 KiB
Bash
Executable File
382 lines
9.7 KiB
Bash
Executable File
#!/usr/bin/env bash
|
||
|
||
set -Eeuo pipefail
|
||
|
||
ROOT_DIR="$(cd "$(dirname "${BASH_SOURCE[0]}")" && pwd)"
|
||
BACKEND_DIR="${ROOT_DIR}/backend"
|
||
FRONTEND_DIR="${ROOT_DIR}/frontend"
|
||
BACKEND_PORT=17861
|
||
FRONTEND_PORT=16801
|
||
BACKEND_PYTHON=""
|
||
BACKEND_PID=""
|
||
FRONTEND_PID=""
|
||
SHUTTING_DOWN=0
|
||
CHECK_ONLY=0
|
||
DATABASE_URL_SOURCE=""
|
||
|
||
info() {
|
||
printf '[启动] %s\n' "$*"
|
||
}
|
||
|
||
warn() {
|
||
printf '[警告] %s\n' "$*" >&2
|
||
}
|
||
|
||
fail() {
|
||
printf '[错误] %s\n' "$*" >&2
|
||
exit 1
|
||
}
|
||
|
||
usage() {
|
||
cat <<'EOF'
|
||
用法：bash ./start.sh [--check] [--help]
|
||
|
||
--check 仅检查运行环境和端口，不启动服务
|
||
--help 显示帮助
|
||
|
||
脚本会继承当前终端的环境变量，例如：
|
||
DATABASE_HOST='db.example.com' bash ./start.sh
|
||
EOF
|
||
}
|
||
|
||
parse_args() {
|
||
if [ "$#" -gt 1 ]; then
|
||
usage >&2
|
||
exit 2
|
||
fi
|
||
|
||
case "${1:-}" in
|
||
"") ;;
|
||
--check) CHECK_ONLY=1 ;;
|
||
--help|-h)
|
||
usage
|
||
exit 0
|
||
;;
|
||
*)
|
||
usage >&2
|
||
exit 2
|
||
;;
|
||
esac
|
||
}
|
||
|
||
select_backend_python() {
|
||
if [ -x "${BACKEND_DIR}/.venv/bin/python" ]; then
|
||
BACKEND_PYTHON="${BACKEND_DIR}/.venv/bin/python"
|
||
elif command -v python3 >/dev/null 2>&1; then
|
||
BACKEND_PYTHON="$(command -v python3)"
|
||
warn "未找到 backend/.venv，将使用 ${BACKEND_PYTHON}。"
|
||
else
|
||
fail "未找到 Python 3。请先创建 backend/.venv 并安装后端依赖。"
|
||
fi
|
||
}
|
||
|
||
load_database_url() {
|
||
local env_file="${ROOT_DIR}/docker/app/.env"
|
||
local value=""
|
||
|
||
if ! value="$(DATABASE_CONFIG_FILE="${env_file}" "${BACKEND_PYTHON}" -c '
|
||
import ipaddress
|
||
import os
|
||
from pathlib import Path
|
||
from urllib.parse import quote
|
||
|
||
|
||
def read_env(path: str) -> dict[str, str]:
|
||
values: dict[str, str] = {}
|
||
env_path = Path(path)
|
||
if not env_path.is_file():
|
||
return values
|
||
for raw_line in env_path.read_text(encoding="utf-8").splitlines():
|
||
line = raw_line.strip()
|
||
if not line or line.startswith("#") or "=" not in line:
|
||
continue
|
||
key, value = line.split("=", 1)
|
||
value = value.strip()
|
||
if len(value) >= 2 and value[0] == value[-1] and value[0] in {"\x27", "\""}:
|
||
value = value[1:-1]
|
||
values[key.strip()] = value
|
||
return values
|
||
|
||
|
||
file_values = read_env(os.environ["DATABASE_CONFIG_FILE"])
|
||
|
||
|
||
def setting(*names: str, default: str = "") -> str:
|
||
for name in names:
|
||
env_value = os.environ.get(name, "").strip()
|
||
if env_value:
|
||
return env_value
|
||
for name in names:
|
||
file_value = file_values.get(name, "").strip()
|
||
if file_value:
|
||
return file_value
|
||
return default
|
||
|
||
|
||
shell_url = os.environ.get("DATABASE_URL", "").strip()
|
||
shell_host = os.environ.get("DATABASE_HOST", "").strip()
|
||
file_host = file_values.get("DATABASE_HOST", "").strip()
|
||
file_url = file_values.get("DATABASE_URL", "").strip()
|
||
|
||
candidate = shell_url or shell_host or file_host or file_url
|
||
if not candidate:
|
||
raise SystemExit(0)
|
||
|
||
if "://" in candidate:
|
||
print(candidate)
|
||
raise SystemExit(0)
|
||
|
||
host = candidate.strip().rstrip("/")
|
||
if not host or any(character in host for character in "/@"):
|
||
raise SystemExit(1)
|
||
|
||
port_text = setting("DATABASE_PORT", default="5432")
|
||
try:
|
||
port = int(port_text)
|
||
except ValueError:
|
||
raise SystemExit(1)
|
||
if not 1 <= port <= 65535:
|
||
raise SystemExit(1)
|
||
|
||
username = setting("DATABASE_USER", "POSTGRES_USER", default="yg_ft")
|
||
password = setting("DATABASE_PASSWORD", "POSTGRES_PASSWORD", default="change_me")
|
||
database = setting("DATABASE_NAME", "POSTGRES_DB", default="yg_ft")
|
||
if not username or not database:
|
||
raise SystemExit(1)
|
||
|
||
try:
|
||
parsed_host = ipaddress.ip_address(host)
|
||
url_host = f"[{host}]" if parsed_host.version == 6 else host
|
||
except ValueError:
|
||
url_host = host
|
||
|
||
credentials = quote(username, safe="")
|
||
if password:
|
||
credentials += ":" + quote(password, safe="")
|
||
encoded_database = quote(database, safe="")
|
||
print(f"postgresql+psycopg://{credentials}@{url_host}:{port}/{encoded_database}")
|
||
' 2>/dev/null)"; then
|
||
fail "数据库分项配置无效，请检查 DATABASE_HOST 和 DATABASE_PORT。"
|
||
fi
|
||
|
||
if [ -z "${value}" ]; then
|
||
return
|
||
fi
|
||
|
||
if [ -n "${DATABASE_URL:-}" ] || [ -n "${DATABASE_HOST:-}" ]; then
|
||
DATABASE_URL_SOURCE="当前终端环境变量"
|
||
elif [ -f "${env_file}" ] && grep -Eq '^[[:space:]]*DATABASE_HOST=' "${env_file}"; then
|
||
DATABASE_URL_SOURCE="docker/app/.env 的 DATABASE_HOST"
|
||
else
|
||
DATABASE_URL_SOURCE="docker/app/.env 的 DATABASE_URL"
|
||
fi
|
||
export DATABASE_URL="${value}"
|
||
}
|
||
|
||
validate_database_url() {
|
||
if [ -z "${DATABASE_URL:-}" ]; then
|
||
return
|
||
fi
|
||
|
||
case "${DATABASE_URL}" in
|
||
*"@postgres:"*|*"//postgres:"*)
|
||
fail "DATABASE_URL 使用了 Docker 服务名 postgres；本地启动请填写真实数据库主机名或 IP。"
|
||
;;
|
||
esac
|
||
|
||
if ! "${BACKEND_PYTHON}" -c '
|
||
import os
|
||
from urllib.parse import urlsplit
|
||
|
||
try:
|
||
parsed = urlsplit(os.environ["DATABASE_URL"])
|
||
port = parsed.port or 5432
|
||
except (KeyError, ValueError):
|
||
raise SystemExit(1)
|
||
|
||
valid = (
|
||
parsed.scheme in {"postgresql", "postgresql+psycopg"}
|
||
and bool(parsed.hostname)
|
||
and bool(parsed.path.strip("/"))
|
||
and 1 <= port <= 65535
|
||
)
|
||
raise SystemExit(0 if valid else 1)
|
||
'; then
|
||
fail "${DATABASE_URL_SOURCE:-DATABASE_URL} 中的数据库连接串格式无效。格式应为：postgresql+psycopg://用户名:密码@主机:端口/数据库名"
|
||
fi
|
||
|
||
info "已使用 ${DATABASE_URL_SOURCE:-DATABASE_URL} 中的数据库连接配置（敏感信息已隐藏）。"
|
||
}
|
||
|
||
check_backend_dependencies() {
|
||
if ! "${BACKEND_PYTHON}" -c \
|
||
'import sys; raise SystemExit(0 if sys.version_info >= (3, 12) else 1)'; then
|
||
fail "后端要求 Python >= 3.12，当前解释器为 $(${BACKEND_PYTHON} --version 2>&1)。"
|
||
fi
|
||
|
||
if ! "${BACKEND_PYTHON}" -c \
|
||
'import alembic, dotenv, fastapi, httpx, jwt, multipart, passlib, psycopg, pydantic, redis, sqlalchemy, uvicorn' \
|
||
>/dev/null 2>&1; then
|
||
fail "后端依赖不完整。请执行：${BACKEND_PYTHON} -m pip install -r backend/requirements.txt"
|
||
fi
|
||
}
|
||
|
||
check_frontend_dependencies() {
|
||
command -v node >/dev/null 2>&1 || fail "未找到 Node.js。"
|
||
command -v npm >/dev/null 2>&1 || fail "未找到 npm。"
|
||
|
||
if [ ! -x "${FRONTEND_DIR}/node_modules/.bin/vite" ]; then
|
||
fail "前端依赖未安装。请执行：cd frontend && npm ci"
|
||
fi
|
||
|
||
if ! npm --prefix "${FRONTEND_DIR}" ls --depth=0 >/dev/null 2>&1; then
|
||
fail "前端依赖状态异常。请执行：cd frontend && npm ci"
|
||
fi
|
||
}
|
||
|
||
check_port() {
|
||
local port="$1"
|
||
local service_name="$2"
|
||
|
||
if command -v lsof >/dev/null 2>&1 && \
|
||
lsof -nP -iTCP:"${port}" -sTCP:LISTEN >/dev/null 2>&1; then
|
||
fail "${service_name}端口 ${port} 已被占用，请先停止占用该端口的进程。"
|
||
fi
|
||
}
|
||
|
||
warn_if_default_database_is_unavailable() {
|
||
if [ -n "${DATABASE_URL:-}" ] || ! command -v lsof >/dev/null 2>&1; then
|
||
return
|
||
fi
|
||
|
||
if ! lsof -nP -iTCP:15432 -sTCP:LISTEN >/dev/null 2>&1; then
|
||
warn "未检测到默认 PostgreSQL 端口 15432；后端可启动，但健康检查和业务接口可能失败。"
|
||
fi
|
||
}
|
||
|
||
preflight() {
|
||
[ -d "${BACKEND_DIR}" ] || fail "缺少后端目录：${BACKEND_DIR}"
|
||
[ -d "${FRONTEND_DIR}" ] || fail "缺少前端目录：${FRONTEND_DIR}"
|
||
[ -f "${FRONTEND_DIR}/package-lock.json" ] || fail "缺少 frontend/package-lock.json"
|
||
|
||
select_backend_python
|
||
load_database_url
|
||
validate_database_url
|
||
check_backend_dependencies
|
||
check_frontend_dependencies
|
||
check_port "${BACKEND_PORT}" "后端"
|
||
check_port "${FRONTEND_PORT}" "前端"
|
||
warn_if_default_database_is_unavailable
|
||
|
||
info "环境检查通过：$(${BACKEND_PYTHON} --version 2>&1)，Node $(node --version)。"
|
||
}
|
||
|
||
terminate_process_tree() {
|
||
local pid="$1"
|
||
local child=""
|
||
|
||
if [ -z "${pid}" ] || ! kill -0 "${pid}" 2>/dev/null; then
|
||
return
|
||
fi
|
||
|
||
if command -v pgrep >/dev/null 2>&1; then
|
||
for child in $(pgrep -P "${pid}" 2>/dev/null || true); do
|
||
terminate_process_tree "${child}"
|
||
done
|
||
fi
|
||
|
||
kill -TERM "${pid}" 2>/dev/null || true
|
||
}
|
||
|
||
cleanup() {
|
||
local exit_code="${1:-0}"
|
||
|
||
if [ "${SHUTTING_DOWN}" -eq 1 ]; then
|
||
return
|
||
fi
|
||
SHUTTING_DOWN=1
|
||
trap - EXIT INT TERM
|
||
|
||
if [ -n "${BACKEND_PID}" ] || [ -n "${FRONTEND_PID}" ]; then
|
||
info "正在停止前端和后端服务……"
|
||
fi
|
||
terminate_process_tree "${FRONTEND_PID}"
|
||
terminate_process_tree "${BACKEND_PID}"
|
||
[ -z "${FRONTEND_PID}" ] || wait "${FRONTEND_PID}" 2>/dev/null || true
|
||
[ -z "${BACKEND_PID}" ] || wait "${BACKEND_PID}" 2>/dev/null || true
|
||
|
||
exit "${exit_code}"
|
||
}
|
||
|
||
start_backend() {
|
||
info "启动后端：http://127.0.0.1:${BACKEND_PORT}/modelTF/health"
|
||
(
|
||
cd "${BACKEND_DIR}"
|
||
exec "${BACKEND_PYTHON}" -m uvicorn app.main:app \
|
||
--reload \
|
||
--host 127.0.0.1 \
|
||
--port "${BACKEND_PORT}"
|
||
) &
|
||
BACKEND_PID=$!
|
||
}
|
||
|
||
start_frontend() {
|
||
info "启动前端：http://localhost:${FRONTEND_PORT}"
|
||
(
|
||
cd "${FRONTEND_DIR}"
|
||
exec npm run dev -- --strictPort
|
||
) &
|
||
FRONTEND_PID=$!
|
||
}
|
||
|
||
wait_for_services() {
|
||
local exit_code=0
|
||
|
||
while :; do
|
||
if ! kill -0 "${BACKEND_PID}" 2>/dev/null; then
|
||
if wait "${BACKEND_PID}"; then
|
||
exit_code=0
|
||
else
|
||
exit_code=$?
|
||
fi
|
||
warn "后端服务已退出（状态码 ${exit_code}）。"
|
||
[ "${exit_code}" -ne 0 ] || exit_code=1
|
||
return "${exit_code}"
|
||
fi
|
||
|
||
if ! kill -0 "${FRONTEND_PID}" 2>/dev/null; then
|
||
if wait "${FRONTEND_PID}"; then
|
||
exit_code=0
|
||
else
|
||
exit_code=$?
|
||
fi
|
||
warn "前端服务已退出（状态码 ${exit_code}）。"
|
||
[ "${exit_code}" -ne 0 ] || exit_code=1
|
||
return "${exit_code}"
|
||
fi
|
||
|
||
sleep 1
|
||
done
|
||
}
|
||
|
||
main() {
|
||
parse_args "$@"
|
||
preflight
|
||
|
||
if [ "${CHECK_ONLY}" -eq 1 ]; then
|
||
info "前后端均可启动。"
|
||
return
|
||
fi
|
||
|
||
trap 'cleanup $?' EXIT
|
||
trap 'cleanup 130' INT
|
||
trap 'cleanup 143' TERM
|
||
|
||
start_backend
|
||
start_frontend
|
||
info "前后端已启动，按 Ctrl+C 同时停止。"
|
||
wait_for_services
|
||
}
|
||
|
||
main "$@"
|