Files
cds-ai/atvm/watcher-service/start-atvm-runner.sh

64 lines
1.6 KiB
Bash

#!/usr/bin/env bash
set -euo pipefail
usage() {
cat <<'EOF'
Usage:
start-atvm-runner.sh --build-name <name> --runner-command <text> [options]
Options:
--build-name <name>
--runner-command <text>
--workdir <path> Default: /root/cdc-e2e-cyp-12.17.4
--log-path <path> Default: /tmp/<build-name>.log
--state-root <path> Default: /var/lib/atvm-run-watcher
EOF
}
BUILD_NAME=""
RUNNER_COMMAND=""
RUNNER_WORKDIR="/root/cdc-e2e-cyp-12.17.4"
RUNNER_LOG=""
STATE_ROOT="/var/lib/atvm-run-watcher"
while [[ $# -gt 0 ]]; do
case "$1" in
--build-name) BUILD_NAME="${2:-}"; shift 2 ;;
--runner-command) RUNNER_COMMAND="${2:-}"; shift 2 ;;
--workdir) RUNNER_WORKDIR="${2:-}"; shift 2 ;;
--log-path) RUNNER_LOG="${2:-}"; shift 2 ;;
--state-root) STATE_ROOT="${2:-}"; shift 2 ;;
-h|--help) usage; exit 0 ;;
*) echo "Unknown argument: $1" >&2; usage >&2; exit 1 ;;
esac
done
if [[ -z "$BUILD_NAME" ]]; then
echo "--build-name is required" >&2
usage >&2
exit 1
fi
if [[ -z "$RUNNER_COMMAND" ]]; then
echo "--runner-command is required" >&2
usage >&2
exit 1
fi
if [[ -z "$RUNNER_LOG" ]]; then
RUNNER_LOG="/tmp/${BUILD_NAME}.log"
fi
RUN_DIR="${STATE_ROOT}/${BUILD_NAME}"
mkdir -p "$RUN_DIR"
cat >"${RUN_DIR}/run.env" <<EOF
ATVM_RUNNER_COMMAND=${RUNNER_COMMAND@Q}
ATVM_RUNNER_WORKDIR=${RUNNER_WORKDIR@Q}
ATVM_RUNNER_LOG=${RUNNER_LOG@Q}
EOF
systemctl stop "atvm-runner@${BUILD_NAME}.service" >/dev/null 2>&1 || true
systemctl start "atvm-runner@${BUILD_NAME}.service"
systemctl status --no-pager "atvm-runner@${BUILD_NAME}.service" || true