李顺东
53288327a1
- Add logs-kafka command to view Kafka container logs separately - Implement reset command to clean data volumes and reinitialize services - Add confirmation prompt for destructive reset operation - Update help text to clarify logs command shows application logs - Improve command case statement alignment for better readability - Add documentation for new reset command with data volume cleanup - Separate clean command documentation to focus on image pruning only
154 lines
3.6 KiB
Bash
154 lines
3.6 KiB
Bash
#!/bin/bash
|
||
# 部署脚本
|
||
|
||
set -e
|
||
|
||
# 配置
|
||
IMAGE_NAME="job-crawler"
|
||
IMAGE_TAG="latest"
|
||
CONTAINER_NAME="job-crawler"
|
||
REGISTRY="" # 私有仓库地址，如: registry.example.com
|
||
|
||
# 颜色输出
|
||
RED='\033[0;31m'
|
||
GREEN='\033[0;32m'
|
||
YELLOW='\033[1;33m'
|
||
NC='\033[0m'
|
||
|
||
log_info() { echo -e "${GREEN}[INFO]${NC} $1"; }
|
||
log_warn() { echo -e "${YELLOW}[WARN]${NC} $1"; }
|
||
log_error() { echo -e "${RED}[ERROR]${NC} $1"; }
|
||
|
||
# 显示帮助
|
||
show_help() {
|
||
echo "用法: ./deploy.sh [命令]"
|
||
echo ""
|
||
echo "命令:"
|
||
echo " build 构建镜像"
|
||
echo " push 推送镜像到仓库"
|
||
echo " up 启动所有服务 (docker-compose)"
|
||
echo " down 停止所有服务"
|
||
echo " restart 重启应用服务"
|
||
echo " logs 查看应用日志"
|
||
echo " logs-kafka 查看Kafka日志"
|
||
echo " status 查看服务状态"
|
||
echo " clean 清理无用镜像"
|
||
echo " reset 清理数据卷并重新启动（慎用）"
|
||
echo " help 显示帮助"
|
||
}
|
||
|
||
# 构建镜像
|
||
build() {
|
||
log_info "构建镜像: ${IMAGE_NAME}:${IMAGE_TAG}"
|
||
docker build -t ${IMAGE_NAME}:${IMAGE_TAG} .
|
||
log_info "构建完成"
|
||
}
|
||
|
||
# 推送镜像
|
||
push() {
|
||
if [ -z "$REGISTRY" ]; then
|
||
log_error "请先配置 REGISTRY 变量"
|
||
exit 1
|
||
fi
|
||
FULL_IMAGE="${REGISTRY}/${IMAGE_NAME}:${IMAGE_TAG}"
|
||
log_info "推送镜像: ${FULL_IMAGE}"
|
||
docker tag ${IMAGE_NAME}:${IMAGE_TAG} ${FULL_IMAGE}
|
||
docker push ${FULL_IMAGE}
|
||
log_info "推送完成"
|
||
}
|
||
|
||
# 启动服务
|
||
up() {
|
||
log_info "启动服务..."
|
||
|
||
# 检查配置文件
|
||
if [ ! -f "config/config.yml" ]; then
|
||
log_warn "配置文件不存在，从模板复制"
|
||
cp config/config.yml.docker config/config.yml
|
||
log_warn "请编辑 config/config.yml 填入账号密码后重新运行"
|
||
exit 1
|
||
fi
|
||
|
||
docker-compose up -d
|
||
log_info "服务已启动"
|
||
log_info "API地址: http://localhost:8000"
|
||
log_info "API文档: http://localhost:8000/docs"
|
||
}
|
||
|
||
# 停止服务
|
||
down() {
|
||
log_info "停止服务..."
|
||
docker-compose down
|
||
log_info "服务已停止"
|
||
}
|
||
|
||
# 重启应用
|
||
restart() {
|
||
log_info "重启应用服务..."
|
||
docker-compose restart app
|
||
log_info "重启完成"
|
||
}
|
||
|
||
# 查看应用日志
|
||
logs() {
|
||
docker-compose logs -f --tail=100 app
|
||
}
|
||
|
||
# 查看Kafka日志
|
||
logs_kafka() {
|
||
docker-compose logs -f --tail=100 kafka
|
||
}
|
||
|
||
# 查看状态
|
||
status() {
|
||
echo "=== 容器状态 ==="
|
||
docker-compose ps
|
||
echo ""
|
||
echo "=== 采集状态 ==="
|
||
curl -s http://localhost:8000/status 2>/dev/null | python -m json.tool || echo "服务未启动"
|
||
}
|
||
|
||
# 清理镜像
|
||
clean() {
|
||
log_info "清理无用镜像..."
|
||
docker image prune -f
|
||
log_info "清理完成"
|
||
}
|
||
|
||
# 重置：清理数据卷并重新启动
|
||
reset() {
|
||
log_warn "即将清理所有数据卷（Kafka数据、采集进度等），是否继续？[y/N]"
|
||
read -r confirm
|
||
if [ "$confirm" != "y" ] && [ "$confirm" != "Y" ]; then
|
||
log_info "已取消"
|
||
exit 0
|
||
fi
|
||
|
||
log_info "停止服务..."
|
||
docker-compose down -v
|
||
|
||
log_info "清理数据卷..."
|
||
docker volume rm job_crawler_kafka_data job_crawler_zookeeper_data job_crawler_app_data 2>/dev/null || true
|
||
|
||
log_info "重新启动..."
|
||
docker-compose up -d
|
||
|
||
log_info "重置完成"
|
||
}
|
||
|
||
# 主逻辑
|
||
case "${1:-help}" in
|
||
build) build ;;
|
||
push) push ;;
|
||
up) up ;;
|
||
down) down ;;
|
||
restart) restart ;;
|
||
logs) logs ;;
|
||
logs-kafka) logs_kafka ;;
|
||
status) status ;;
|
||
clean) clean ;;
|
||
reset) reset ;;
|
||
help) show_help ;;
|
||
*) show_help ;;
|
||
esac
|