wehub-resource-sync
233 lines
8.3 KiB
Bash
233 lines
8.3 KiB
Bash
#!/usr/bin/env bash
|
||
# =============================================================
|
||
# BiSheng 运维管理脚本
|
||
# 用法: ./bisheng.sh <命令> [参数]
|
||
# =============================================================
|
||
|
||
set -euo pipefail
|
||
|
||
SCRIPT_DIR="$(cd "$(dirname "${BASH_SOURCE[0]}")" && pwd)"
|
||
COMPOSE_FILE="${SCRIPT_DIR}/docker-compose.yml"
|
||
COMPOSE_CMD="docker compose -f ${COMPOSE_FILE}"
|
||
|
||
# 容器名常量（与 docker-compose.yml 对应）
|
||
BACKEND_CONTAINER="bisheng-backend"
|
||
WORKER_CONTAINER="bisheng-backend-worker"
|
||
|
||
# 所有可管理的 compose service 名称
|
||
ALL_SERVICES=(backend backend_worker frontend mysql redis elasticsearch minio milvus etcd)
|
||
|
||
# ─── 颜色输出 ────────────────────────────────────────────────
|
||
RED='\033[0;31m'; GREEN='\033[0;32m'; YELLOW='\033[1;33m'
|
||
CYAN='\033[0;36m'; BOLD='\033[1m'; RESET='\033[0m'
|
||
|
||
info() { echo -e "${GREEN}[INFO]${RESET} $*"; }
|
||
warn() { echo -e "${YELLOW}[WARN]${RESET} $*"; }
|
||
error() { echo -e "${RED}[ERROR]${RESET} $*" >&2; }
|
||
header() { echo -e "${CYAN}${BOLD}$*${RESET}"; }
|
||
|
||
# ─── 帮助 ────────────────────────────────────────────────────
|
||
usage() {
|
||
header "═══════════════════════════════════════════════"
|
||
header " BiSheng 运维管理脚本"
|
||
header "═══════════════════════════════════════════════"
|
||
echo ""
|
||
echo -e "${BOLD}查看日志：${RESET}"
|
||
echo " $0 logs backend 实时跟踪 backend 日志（默认最近 200 行）"
|
||
echo " $0 logs worker 实时跟踪 backend_worker 日志（默认最近 200 行）"
|
||
echo " $0 logs backend -n 0 实时跟踪 backend 所有历史日志"
|
||
echo " $0 logs worker -n 500 查看 worker 最近 500 行日志"
|
||
echo ""
|
||
echo -e "${BOLD}镜像版本管理：${RESET}"
|
||
echo " $0 version 查看当前配置的镜像版本"
|
||
echo " $0 version v3.0.0 修改 backend、worker、frontend 的版本为 v3.0.0"
|
||
echo ""
|
||
echo -e "${BOLD}进入容器 Shell：${RESET}"
|
||
echo " $0 exec backend 进入 backend 容器"
|
||
echo " $0 exec worker 进入 backend_worker 容器"
|
||
echo ""
|
||
echo -e "${BOLD}更新镜像并重启：${RESET}"
|
||
echo " $0 update 拉取最新镜像并重启 backend + worker"
|
||
echo " $0 update backend 只更新并重启 backend"
|
||
echo " $0 update worker 只更新并重启 worker"
|
||
echo ""
|
||
echo -e "${BOLD}重启容器：${RESET}"
|
||
echo " $0 restart 重启 backend + worker"
|
||
echo " $0 restart backend 重启 backend"
|
||
echo " $0 restart worker 重启 worker"
|
||
echo " $0 restart frontend 重启 frontend"
|
||
echo " $0 restart <service...> 重启任意多个 service"
|
||
echo ""
|
||
echo -e "${BOLD}可用 service 名称：${RESET}"
|
||
echo " ${ALL_SERVICES[*]}"
|
||
echo ""
|
||
}
|
||
|
||
# ─── service 别名解析 ─────────────────────────────────────────
|
||
resolve_service() {
|
||
case "$1" in
|
||
backend) echo "backend" ;;
|
||
worker|backend_worker) echo "backend_worker" ;;
|
||
frontend) echo "frontend" ;;
|
||
mysql) echo "mysql" ;;
|
||
redis) echo "redis" ;;
|
||
es|elasticsearch) echo "elasticsearch" ;;
|
||
minio) echo "minio" ;;
|
||
milvus) echo "milvus" ;;
|
||
etcd) echo "etcd" ;;
|
||
*) echo "$1" ;; # 原样传入，让 docker compose 自行报错
|
||
esac
|
||
}
|
||
|
||
# ─── 查看日志 ─────────────────────────────────────────────────
|
||
cmd_logs() {
|
||
local target="${1:-}"
|
||
shift || true
|
||
|
||
local lines=200
|
||
|
||
# 解析可选 -n <行数>
|
||
while [[ $# -gt 0 ]]; do
|
||
case "$1" in
|
||
-n)
|
||
lines="${2:-200}"
|
||
shift 2
|
||
;;
|
||
*)
|
||
shift
|
||
;;
|
||
esac
|
||
done
|
||
|
||
local service
|
||
case "$target" in
|
||
backend) service="backend" ;;
|
||
worker|backend_worker) service="backend_worker" ;;
|
||
*)
|
||
error "未知目标 '${target}'，请使用 backend 或 worker"
|
||
exit 1
|
||
;;
|
||
esac
|
||
|
||
if [ "$lines" -eq 0 ]; then
|
||
info "实时跟踪 ${service} 所有历史日志（Ctrl+C 退出）..."
|
||
${COMPOSE_CMD} logs -f "${service}"
|
||
else
|
||
info "实时跟踪 ${service} 最近 ${lines} 行日志（Ctrl+C 退出）..."
|
||
${COMPOSE_CMD} logs -f --tail="${lines}" "${service}"
|
||
fi
|
||
}
|
||
|
||
# ─── 修改版本号 ───────────────────────────────────────────────
|
||
cmd_version() {
|
||
local new_version="${1:-}"
|
||
|
||
if [[ -z "$new_version" ]]; then
|
||
info "当前 docker-compose.yml 配置的版本："
|
||
grep -E "image:.*dataelement/bisheng-(backend|frontend):" "$COMPOSE_FILE" | awk '{$1=$1};1'
|
||
return 0
|
||
fi
|
||
|
||
info "正在将 backend, backend_worker, frontend 版本修改为: ${new_version}"
|
||
|
||
# 兼容 macOS 和 Linux 的 sed -i 用法
|
||
if sed --version 2>/dev/null | grep -q GNU; then
|
||
sed -i -E "s|(image: dataelement/bisheng-backend):.*|\1:${new_version}|g" "$COMPOSE_FILE"
|
||
sed -i -E "s|(image: dataelement/bisheng-frontend):.*|\1:${new_version}|g" "$COMPOSE_FILE"
|
||
else
|
||
# macOS/BSD sed
|
||
sed -i '' -E "s|(image: dataelement/bisheng-backend):.*|\1:${new_version}|g" "$COMPOSE_FILE"
|
||
sed -i '' -E "s|(image: dataelement/bisheng-frontend):.*|\1:${new_version}|g" "$COMPOSE_FILE"
|
||
fi
|
||
|
||
info "✅ 版本修改完成："
|
||
grep -E "image:.*dataelement/bisheng-(backend|frontend):" "$COMPOSE_FILE" | awk '{$1=$1};1'
|
||
warn "注意：只是修改了配置文件，若要生效请执行 '$0 update'"
|
||
}
|
||
|
||
# ─── 进入容器 ─────────────────────────────────────────────────
|
||
cmd_exec() {
|
||
local target="${1:-}"
|
||
local container
|
||
|
||
case "$target" in
|
||
backend) container="${BACKEND_CONTAINER}" ;;
|
||
worker|backend_worker) container="${WORKER_CONTAINER}" ;;
|
||
*)
|
||
error "未知目标 '${target}'，请使用 backend 或 worker"
|
||
exit 1
|
||
;;
|
||
esac
|
||
|
||
info "进入容器 ${container} ..."
|
||
docker exec -it "${container}" /bin/bash 2>/dev/null \
|
||
|| docker exec -it "${container}" /bin/sh
|
||
}
|
||
|
||
# ─── 更新镜像并重启 ───────────────────────────────────────────
|
||
cmd_update() {
|
||
local targets=()
|
||
|
||
if [[ $# -eq 0 ]]; then
|
||
targets=("backend" "backend_worker" "frontend")
|
||
else
|
||
for t in "$@"; do
|
||
targets+=("$(resolve_service "$t")")
|
||
done
|
||
fi
|
||
|
||
info "拉取最新镜像：${targets[*]}"
|
||
${COMPOSE_CMD} pull "${targets[@]}"
|
||
|
||
info "重启服务（不重建依赖）：${targets[*]}"
|
||
${COMPOSE_CMD} up -d --no-deps "${targets[@]}"
|
||
|
||
info "✅ 更新完成"
|
||
${COMPOSE_CMD} ps "${targets[@]}"
|
||
}
|
||
|
||
# ─── 重启容器 ─────────────────────────────────────────────────
|
||
cmd_restart() {
|
||
local targets=()
|
||
|
||
if [[ $# -eq 0 ]]; then
|
||
targets=("backend" "backend_worker" "frontend")
|
||
else
|
||
for t in "$@"; do
|
||
targets+=("$(resolve_service "$t")")
|
||
done
|
||
fi
|
||
|
||
info "重启服务：${targets[*]}"
|
||
${COMPOSE_CMD} restart "${targets[@]}"
|
||
|
||
info "✅ 重启完成"
|
||
${COMPOSE_CMD} ps "${targets[@]}"
|
||
}
|
||
|
||
# ─── 入口 ────────────────────────────────────────────────────
|
||
main() {
|
||
if [[ $# -eq 0 ]]; then
|
||
usage
|
||
exit 0
|
||
fi
|
||
|
||
local cmd="$1"; shift
|
||
|
||
case "$cmd" in
|
||
logs) cmd_logs "$@" ;;
|
||
version) cmd_version "$@" ;;
|
||
exec) cmd_exec "$@" ;;
|
||
update) cmd_update "$@" ;;
|
||
restart) cmd_restart "$@" ;;
|
||
help|-h|--help) usage ;;
|
||
*)
|
||
error "未知命令: ${cmd}"
|
||
usage
|
||
exit 1
|
||
;;
|
||
esac
|
||
}
|
||
|
||
main "$@"
|