124 lines
3.1 KiB
Bash
Executable File
124 lines
3.1 KiB
Bash
Executable File
#!/usr/bin/env bash
|
||
|
||
set -euo pipefail
|
||
|
||
# 用法：
|
||
# sudo ./install-datamanagementd.sh --binary /path/to/datamanagementd
|
||
# 或：
|
||
# sudo ./install-datamanagementd.sh --source /path/to/sub2api/repo
|
||
|
||
BIN_PATH=""
|
||
SOURCE_PATH=""
|
||
INSTALL_DIR="/opt/sub2api"
|
||
DATA_DIR="/var/lib/sub2api/datamanagement"
|
||
SERVICE_FILE_NAME="sub2api-datamanagementd.service"
|
||
|
||
function print_help() {
|
||
cat <<'EOF'
|
||
用法:
|
||
install-datamanagementd.sh [--binary <datamanagementd二进制路径>] [--source <仓库路径>]
|
||
|
||
参数:
|
||
--binary 指定已构建的 datamanagementd 二进制路径
|
||
--source 指定 sub2api 仓库路径（脚本会执行 go build）
|
||
-h, --help 显示帮助
|
||
|
||
示例:
|
||
sudo ./install-datamanagementd.sh --binary ./datamanagement/datamanagementd
|
||
sudo ./install-datamanagementd.sh --source /opt/sub2api-src
|
||
EOF
|
||
}
|
||
|
||
while [[ $# -gt 0 ]]; do
|
||
case "$1" in
|
||
--binary)
|
||
BIN_PATH="${2:-}"
|
||
shift 2
|
||
;;
|
||
--source)
|
||
SOURCE_PATH="${2:-}"
|
||
shift 2
|
||
;;
|
||
-h|--help)
|
||
print_help
|
||
exit 0
|
||
;;
|
||
*)
|
||
echo "未知参数: $1"
|
||
print_help
|
||
exit 1
|
||
;;
|
||
esac
|
||
done
|
||
|
||
if [[ -n "$BIN_PATH" && -n "$SOURCE_PATH" ]]; then
|
||
echo "错误: --binary 与 --source 只能二选一"
|
||
exit 1
|
||
fi
|
||
|
||
if [[ -z "$BIN_PATH" && -z "$SOURCE_PATH" ]]; then
|
||
echo "错误: 必须提供 --binary 或 --source"
|
||
exit 1
|
||
fi
|
||
|
||
if [[ "$(id -u)" -ne 0 ]]; then
|
||
echo "错误: 请使用 root 权限执行（例如 sudo）"
|
||
exit 1
|
||
fi
|
||
|
||
if [[ -n "$SOURCE_PATH" ]]; then
|
||
if [[ ! -d "$SOURCE_PATH/datamanagement" ]]; then
|
||
echo "错误: 无效仓库路径，未找到 $SOURCE_PATH/datamanagement"
|
||
exit 1
|
||
fi
|
||
echo "[1/6] 从源码构建 datamanagementd..."
|
||
(cd "$SOURCE_PATH/datamanagement" && go build -o datamanagementd ./cmd/datamanagementd)
|
||
BIN_PATH="$SOURCE_PATH/datamanagement/datamanagementd"
|
||
fi
|
||
|
||
if [[ ! -f "$BIN_PATH" ]]; then
|
||
echo "错误: 二进制文件不存在: $BIN_PATH"
|
||
exit 1
|
||
fi
|
||
|
||
if ! id sub2api >/dev/null 2>&1; then
|
||
echo "[2/6] 创建系统用户 sub2api..."
|
||
useradd --system --no-create-home --shell /usr/sbin/nologin sub2api
|
||
else
|
||
echo "[2/6] 系统用户 sub2api 已存在，跳过创建"
|
||
fi
|
||
|
||
echo "[3/6] 安装 datamanagementd 二进制..."
|
||
mkdir -p "$INSTALL_DIR"
|
||
install -m 0755 "$BIN_PATH" "$INSTALL_DIR/datamanagementd"
|
||
|
||
echo "[4/6] 准备数据目录..."
|
||
mkdir -p "$DATA_DIR"
|
||
chown -R sub2api:sub2api /var/lib/sub2api
|
||
chmod 0750 "$DATA_DIR"
|
||
|
||
SCRIPT_DIR="$(cd "$(dirname "${BASH_SOURCE[0]}")" && pwd)"
|
||
SERVICE_TEMPLATE="$SCRIPT_DIR/$SERVICE_FILE_NAME"
|
||
if [[ ! -f "$SERVICE_TEMPLATE" ]]; then
|
||
echo "错误: 未找到服务模板 $SERVICE_TEMPLATE"
|
||
exit 1
|
||
fi
|
||
|
||
echo "[5/6] 安装 systemd 服务..."
|
||
cp "$SERVICE_TEMPLATE" "/etc/systemd/system/$SERVICE_FILE_NAME"
|
||
systemctl daemon-reload
|
||
systemctl enable --now sub2api-datamanagementd
|
||
|
||
echo "[6/6] 完成，当前状态："
|
||
systemctl --no-pager --full status sub2api-datamanagementd || true
|
||
|
||
cat <<'EOF'
|
||
|
||
下一步建议：
|
||
1. 查看日志：sudo journalctl -u sub2api-datamanagementd -f
|
||
2. 在 sub2api（容器部署时）挂载 socket:
|
||
/tmp/sub2api-datamanagement.sock:/tmp/sub2api-datamanagement.sock
|
||
3. 进入管理后台“数据管理”页面确认 agent=enabled
|
||
|
||
EOF
|