项目打包
This commit is contained in:
237
docker-entrypoint.sh
Executable file
237
docker-entrypoint.sh
Executable file
@@ -0,0 +1,237 @@
|
||||
#!/bin/bash
|
||||
|
||||
# BigDataTool Docker 启动脚本
|
||||
# 用于容器化部署的入口脚本
|
||||
|
||||
set -e
|
||||
|
||||
# 颜色定义
|
||||
RED='\033[0;31m'
|
||||
GREEN='\033[0;32m'
|
||||
YELLOW='\033[1;33m'
|
||||
BLUE='\033[0;34m'
|
||||
NC='\033[0m' # No Color
|
||||
|
||||
# 日志函数
|
||||
log_info() {
|
||||
echo -e "${BLUE}[INFO]${NC} $1"
|
||||
}
|
||||
|
||||
log_warn() {
|
||||
echo -e "${YELLOW}[WARN]${NC} $1"
|
||||
}
|
||||
|
||||
log_error() {
|
||||
echo -e "${RED}[ERROR]${NC} $1"
|
||||
}
|
||||
|
||||
log_success() {
|
||||
echo -e "${GREEN}[SUCCESS]${NC} $1"
|
||||
}
|
||||
|
||||
# 显示启动信息
|
||||
show_banner() {
|
||||
echo "======================================"
|
||||
echo " BigDataTool Container Startup"
|
||||
echo "======================================"
|
||||
echo "Version: 2.0"
|
||||
echo "Python: $(python --version)"
|
||||
echo "Working Directory: $(pwd)"
|
||||
echo "User: $(whoami)"
|
||||
echo "======================================"
|
||||
}
|
||||
|
||||
# 检查环境变量
|
||||
check_environment() {
|
||||
log_info "检查环境变量..."
|
||||
|
||||
# 设置默认值
|
||||
export FLASK_ENV=${FLASK_ENV:-production}
|
||||
export FLASK_DEBUG=${FLASK_DEBUG:-False}
|
||||
export SECRET_KEY=${SECRET_KEY:-$(python -c "import secrets; print(secrets.token_hex(32))")}
|
||||
export DATABASE_URL=${DATABASE_URL:-sqlite:///config_groups.db}
|
||||
|
||||
log_info "FLASK_ENV: $FLASK_ENV"
|
||||
log_info "FLASK_DEBUG: $FLASK_DEBUG"
|
||||
log_info "数据库URL: $DATABASE_URL"
|
||||
|
||||
# 检查必要的环境变量
|
||||
if [ -z "$SECRET_KEY" ]; then
|
||||
log_warn "SECRET_KEY 未设置，使用随机生成的密钥"
|
||||
fi
|
||||
}
|
||||
|
||||
# 初始化数据库
|
||||
initialize_database() {
|
||||
log_info "初始化数据库..."
|
||||
|
||||
# 检查数据库文件是否存在
|
||||
if [ ! -f "config_groups.db" ]; then
|
||||
log_info "数据库文件不存在，将自动创建"
|
||||
python -c "
|
||||
from modules.database import ensure_database
|
||||
if ensure_database():
|
||||
print('数据库初始化成功')
|
||||
else:
|
||||
print('数据库初始化失败')
|
||||
exit(1)
|
||||
"
|
||||
if [ $? -eq 0 ]; then
|
||||
log_success "数据库初始化完成"
|
||||
else
|
||||
log_error "数据库初始化失败"
|
||||
exit 1
|
||||
fi
|
||||
else
|
||||
log_info "数据库文件已存在"
|
||||
fi
|
||||
}
|
||||
|
||||
# 创建必要目录
|
||||
create_directories() {
|
||||
log_info "创建必要目录..."
|
||||
|
||||
# 创建日志目录
|
||||
if [ ! -d "logs" ]; then
|
||||
mkdir -p logs
|
||||
log_info "创建日志目录: logs"
|
||||
fi
|
||||
|
||||
# 创建配置目录
|
||||
if [ ! -d "config" ]; then
|
||||
mkdir -p config
|
||||
log_info "创建配置目录: config"
|
||||
fi
|
||||
|
||||
# 设置权限
|
||||
chmod -R 755 logs config || true
|
||||
}
|
||||
|
||||
# 健康检查函数
|
||||
health_check() {
|
||||
log_info "执行健康检查..."
|
||||
|
||||
local max_attempts=30
|
||||
local attempt=1
|
||||
|
||||
while [ $attempt -le $max_attempts ]; do
|
||||
if wget --no-verbose --tries=1 --spider http://localhost:5000/api/health >/dev/null 2>&1; then
|
||||
log_success "应用健康检查通过"
|
||||
return 0
|
||||
fi
|
||||
|
||||
log_info "等待应用启动... ($attempt/$max_attempts)"
|
||||
sleep 2
|
||||
((attempt++))
|
||||
done
|
||||
|
||||
log_error "健康检查失败，应用可能未正常启动"
|
||||
return 1
|
||||
}
|
||||
|
||||
# 信号处理函数
|
||||
cleanup() {
|
||||
log_warn "收到退出信号，正在清理..."
|
||||
|
||||
# 这里可以添加清理逻辑
|
||||
# 例如：保存缓存、关闭数据库连接等
|
||||
|
||||
log_info "清理完成，退出应用"
|
||||
exit 0
|
||||
}
|
||||
|
||||
# 设置信号处理
|
||||
trap cleanup SIGTERM SIGINT
|
||||
|
||||
# 主启动函数
|
||||
start_application() {
|
||||
log_info "启动BigDataTool应用..."
|
||||
|
||||
# 根据环境变量选择启动方式
|
||||
if [ "$FLASK_ENV" = "development" ]; then
|
||||
log_info "以开发模式启动"
|
||||
python app.py
|
||||
else
|
||||
log_info "以生产模式启动"
|
||||
|
||||
# 检查是否安装了gunicorn
|
||||
if command -v gunicorn >/dev/null 2>&1; then
|
||||
log_info "使用Gunicorn启动应用"
|
||||
exec gunicorn \
|
||||
--bind 0.0.0.0:5000 \
|
||||
--workers 4 \
|
||||
--worker-class sync \
|
||||
--worker-connections 1000 \
|
||||
--max-requests 1000 \
|
||||
--max-requests-jitter 50 \
|
||||
--timeout 120 \
|
||||
--keep-alive 5 \
|
||||
--log-level info \
|
||||
--access-logfile - \
|
||||
--error-logfile - \
|
||||
app:app
|
||||
else
|
||||
log_warn "Gunicorn未安装，使用Flask开发服务器"
|
||||
python app.py
|
||||
fi
|
||||
fi
|
||||
}
|
||||
|
||||
# 显示帮助信息
|
||||
show_help() {
|
||||
echo "BigDataTool Docker 启动脚本"
|
||||
echo ""
|
||||
echo "用法: $0 [选项]"
|
||||
echo ""
|
||||
echo "选项:"
|
||||
echo " start 启动应用（默认）"
|
||||
echo " health-check 执行健康检查"
|
||||
echo " init-db 仅初始化数据库"
|
||||
echo " shell 进入交互式shell"
|
||||
echo " help 显示此帮助信息"
|
||||
echo ""
|
||||
echo "环境变量:"
|
||||
echo " FLASK_ENV Flask运行环境 (development/production)"
|
||||
echo " FLASK_DEBUG 是否启用调试模式 (True/False)"
|
||||
echo " SECRET_KEY 应用密钥"
|
||||
echo " DATABASE_URL 数据库连接URL"
|
||||
echo ""
|
||||
}
|
||||
|
||||
# 主逻辑
|
||||
main() {
|
||||
case "${1:-start}" in
|
||||
start)
|
||||
show_banner
|
||||
check_environment
|
||||
create_directories
|
||||
initialize_database
|
||||
start_application
|
||||
;;
|
||||
health-check)
|
||||
health_check
|
||||
;;
|
||||
init-db)
|
||||
log_info "仅初始化数据库模式"
|
||||
check_environment
|
||||
create_directories
|
||||
initialize_database
|
||||
log_success "数据库初始化完成"
|
||||
;;
|
||||
shell)
|
||||
log_info "进入交互式shell"
|
||||
exec /bin/bash
|
||||
;;
|
||||
help|--help|-h)
|
||||
show_help
|
||||
;;
|
||||
*)
|
||||
log_error "未知选项: $1"
|
||||
show_help
|
||||
exit 1
|
||||
;;
|
||||
esac
|
||||
}
|
||||
|
||||
# 执行主函数
|
||||
main "$@"
|
||||
Reference in New Issue
Block a user