Compare commits
14 Commits
dev_1.0.0_
...
2c799f2c1e
| Author | SHA1 | Date | |
|---|---|---|---|
| 2c799f2c1e | |||
| 1d4208ed3c | |||
| a1cdd05950 | |||
| 0b1ccbd87f | |||
| 1d9a8ec695 | |||
|
|
b0d451cbe7 | ||
|
|
835e81282f | ||
|
|
c4582c99bc | ||
|
|
299765ed40 | ||
|
|
a8bbf2d6e9 | ||
|
|
d1b89c0cf6 | ||
| 1a768bc837 | |||
| 31ccb0b1b8 | |||
|
|
8fbe107ac9 |
@@ -10,19 +10,28 @@ Usage: $0 [OPTIONS]
|
|||||||
Options:
|
Options:
|
||||||
--intranet Use intranet mirror for log/bind builds
|
--intranet Use intranet mirror for log/bind builds
|
||||||
--master-offline Build master offline image (requires src/master/offline_wheels.tar.gz)
|
--master-offline Build master offline image (requires src/master/offline_wheels.tar.gz)
|
||||||
|
--metric Build metric module images (ftp, prometheus, grafana, test nodes)
|
||||||
|
--no-cache Build all images without using Docker layer cache
|
||||||
|
--only LIST Comma-separated targets to build: core,master,metric,web,alert,all
|
||||||
-h, --help Show this help message
|
-h, --help Show this help message
|
||||||
|
|
||||||
Examples:
|
Examples:
|
||||||
$0 # Build with default sources
|
$0 # Build with default sources
|
||||||
$0 --intranet # Build with intranet mirror
|
$0 --intranet # Build with intranet mirror
|
||||||
$0 --master-offline # Additionally build argus-master:offline
|
$0 --master-offline # Additionally build argus-master:offline
|
||||||
$0 --intranet --master-offline
|
$0 --metric # Additionally build metric module images
|
||||||
|
$0 --intranet --master-offline --metric
|
||||||
EOF
|
EOF
|
||||||
}
|
}
|
||||||
|
|
||||||
use_intranet=false
|
use_intranet=false
|
||||||
|
build_core=true
|
||||||
build_master=true
|
build_master=true
|
||||||
build_master_offline=false
|
build_master_offline=false
|
||||||
|
build_metric=true
|
||||||
|
build_web=true
|
||||||
|
build_alert=true
|
||||||
|
no_cache=false
|
||||||
|
|
||||||
while [[ $# -gt 0 ]]; do
|
while [[ $# -gt 0 ]]; do
|
||||||
case $1 in
|
case $1 in
|
||||||
@@ -39,6 +48,34 @@ while [[ $# -gt 0 ]]; do
|
|||||||
build_master_offline=true
|
build_master_offline=true
|
||||||
shift
|
shift
|
||||||
;;
|
;;
|
||||||
|
--metric)
|
||||||
|
build_metric=true
|
||||||
|
shift
|
||||||
|
;;
|
||||||
|
--no-cache)
|
||||||
|
no_cache=true
|
||||||
|
shift
|
||||||
|
;;
|
||||||
|
--only)
|
||||||
|
if [[ -z ${2:-} ]]; then
|
||||||
|
echo "--only requires a target list" >&2; exit 1
|
||||||
|
fi
|
||||||
|
sel="$2"; shift 2
|
||||||
|
# reset all, then enable selected
|
||||||
|
build_core=false; build_master=false; build_metric=false; build_web=false; build_alert=false
|
||||||
|
IFS=',' read -ra parts <<< "$sel"
|
||||||
|
for p in "${parts[@]}"; do
|
||||||
|
case "$p" in
|
||||||
|
core) build_core=true ;;
|
||||||
|
master) build_master=true ;;
|
||||||
|
metric) build_metric=true ;;
|
||||||
|
web) build_web=true ;;
|
||||||
|
alert) build_alert=true ;;
|
||||||
|
all) build_core=true; build_master=true; build_metric=true; build_web=true; build_alert=true ;;
|
||||||
|
*) echo "Unknown --only target: $p" >&2; exit 1 ;;
|
||||||
|
esac
|
||||||
|
done
|
||||||
|
;;
|
||||||
-h|--help)
|
-h|--help)
|
||||||
show_help
|
show_help
|
||||||
exit 0
|
exit 0
|
||||||
@@ -65,6 +102,10 @@ cd "$root"
|
|||||||
load_build_user
|
load_build_user
|
||||||
build_args+=("--build-arg" "ARGUS_BUILD_UID=${ARGUS_BUILD_UID}" "--build-arg" "ARGUS_BUILD_GID=${ARGUS_BUILD_GID}")
|
build_args+=("--build-arg" "ARGUS_BUILD_UID=${ARGUS_BUILD_UID}" "--build-arg" "ARGUS_BUILD_GID=${ARGUS_BUILD_GID}")
|
||||||
|
|
||||||
|
if [[ "$no_cache" == true ]]; then
|
||||||
|
build_args+=("--no-cache")
|
||||||
|
fi
|
||||||
|
|
||||||
master_root="$root/src/master"
|
master_root="$root/src/master"
|
||||||
master_offline_tar="$master_root/offline_wheels.tar.gz"
|
master_offline_tar="$master_root/offline_wheels.tar.gz"
|
||||||
master_offline_dir="$master_root/offline_wheels"
|
master_offline_dir="$master_root/offline_wheels"
|
||||||
@@ -105,14 +146,22 @@ build_image() {
|
|||||||
local image_name=$1
|
local image_name=$1
|
||||||
local dockerfile_path=$2
|
local dockerfile_path=$2
|
||||||
local tag=$3
|
local tag=$3
|
||||||
|
local context="."
|
||||||
shift 3
|
shift 3
|
||||||
|
|
||||||
|
if [[ $# -gt 0 ]]; then
|
||||||
|
context=$1
|
||||||
|
shift
|
||||||
|
fi
|
||||||
|
|
||||||
local extra_args=("$@")
|
local extra_args=("$@")
|
||||||
|
|
||||||
echo "🔄 Building $image_name image..."
|
echo "🔄 Building $image_name image..."
|
||||||
echo " Dockerfile: $dockerfile_path"
|
echo " Dockerfile: $dockerfile_path"
|
||||||
echo " Tag: $tag"
|
echo " Tag: $tag"
|
||||||
|
echo " Context: $context"
|
||||||
|
|
||||||
if docker build "${build_args[@]}" "${extra_args[@]}" -f "$dockerfile_path" -t "$tag" .; then
|
if docker build "${build_args[@]}" "${extra_args[@]}" -f "$dockerfile_path" -t "$tag" "$context"; then
|
||||||
echo "✅ $image_name image built successfully"
|
echo "✅ $image_name image built successfully"
|
||||||
return 0
|
return 0
|
||||||
else
|
else
|
||||||
@@ -121,29 +170,59 @@ build_image() {
|
|||||||
fi
|
fi
|
||||||
}
|
}
|
||||||
|
|
||||||
|
pull_base_image() {
|
||||||
|
local image_ref=$1
|
||||||
|
local attempts=${2:-3}
|
||||||
|
local delay=${3:-5}
|
||||||
|
|
||||||
|
# If the image already exists locally, skip pulling.
|
||||||
|
if docker image inspect "$image_ref" >/dev/null 2>&1; then
|
||||||
|
echo " Local image present; skip pull: $image_ref"
|
||||||
|
return 0
|
||||||
|
fi
|
||||||
|
|
||||||
|
for ((i=1; i<=attempts; i++)); do
|
||||||
|
echo " Pulling base image ($i/$attempts): $image_ref"
|
||||||
|
if docker pull "$image_ref" >/dev/null; then
|
||||||
|
echo " Base image ready: $image_ref"
|
||||||
|
return 0
|
||||||
|
fi
|
||||||
|
echo " Pull failed: $image_ref"
|
||||||
|
if (( i < attempts )); then
|
||||||
|
echo " Retrying in ${delay}s..."
|
||||||
|
sleep "$delay"
|
||||||
|
fi
|
||||||
|
done
|
||||||
|
|
||||||
|
echo "❌ Unable to pull base image after ${attempts} attempts: $image_ref"
|
||||||
|
return 1
|
||||||
|
}
|
||||||
|
|
||||||
images_built=()
|
images_built=()
|
||||||
build_failed=false
|
build_failed=false
|
||||||
|
|
||||||
if build_image "Elasticsearch" "src/log/elasticsearch/build/Dockerfile" "argus-elasticsearch:latest"; then
|
if [[ "$build_core" == true ]]; then
|
||||||
|
if build_image "Elasticsearch" "src/log/elasticsearch/build/Dockerfile" "argus-elasticsearch:latest"; then
|
||||||
images_built+=("argus-elasticsearch:latest")
|
images_built+=("argus-elasticsearch:latest")
|
||||||
else
|
else
|
||||||
build_failed=true
|
build_failed=true
|
||||||
fi
|
fi
|
||||||
|
|
||||||
echo ""
|
echo ""
|
||||||
|
|
||||||
if build_image "Kibana" "src/log/kibana/build/Dockerfile" "argus-kibana:latest"; then
|
if build_image "Kibana" "src/log/kibana/build/Dockerfile" "argus-kibana:latest"; then
|
||||||
images_built+=("argus-kibana:latest")
|
images_built+=("argus-kibana:latest")
|
||||||
else
|
else
|
||||||
build_failed=true
|
build_failed=true
|
||||||
fi
|
fi
|
||||||
|
|
||||||
echo ""
|
echo ""
|
||||||
|
|
||||||
if build_image "BIND9" "src/bind/build/Dockerfile" "argus-bind9:latest"; then
|
if build_image "BIND9" "src/bind/build/Dockerfile" "argus-bind9:latest"; then
|
||||||
images_built+=("argus-bind9:latest")
|
images_built+=("argus-bind9:latest")
|
||||||
else
|
else
|
||||||
build_failed=true
|
build_failed=true
|
||||||
|
fi
|
||||||
fi
|
fi
|
||||||
|
|
||||||
echo ""
|
echo ""
|
||||||
@@ -159,6 +238,9 @@ if [[ "$build_master" == true ]]; then
|
|||||||
if [[ "$build_master_offline" == true ]]; then
|
if [[ "$build_master_offline" == true ]]; then
|
||||||
master_args+=("--offline")
|
master_args+=("--offline")
|
||||||
fi
|
fi
|
||||||
|
if [[ "$no_cache" == true ]]; then
|
||||||
|
master_args+=("--no-cache")
|
||||||
|
fi
|
||||||
if ./scripts/build_images.sh "${master_args[@]}"; then
|
if ./scripts/build_images.sh "${master_args[@]}"; then
|
||||||
if [[ "$build_master_offline" == true ]]; then
|
if [[ "$build_master_offline" == true ]]; then
|
||||||
images_built+=("argus-master:offline")
|
images_built+=("argus-master:offline")
|
||||||
@@ -171,6 +253,91 @@ if [[ "$build_master" == true ]]; then
|
|||||||
popd >/dev/null
|
popd >/dev/null
|
||||||
fi
|
fi
|
||||||
|
|
||||||
|
if [[ "$build_metric" == true ]]; then
|
||||||
|
echo ""
|
||||||
|
echo "Building Metric module images..."
|
||||||
|
|
||||||
|
metric_base_images=(
|
||||||
|
"ubuntu:22.04"
|
||||||
|
"ubuntu/prometheus:3-24.04_stable"
|
||||||
|
"grafana/grafana:11.1.0"
|
||||||
|
)
|
||||||
|
|
||||||
|
for base_image in "${metric_base_images[@]}"; do
|
||||||
|
if ! pull_base_image "$base_image"; then
|
||||||
|
build_failed=true
|
||||||
|
fi
|
||||||
|
done
|
||||||
|
|
||||||
|
metric_builds=(
|
||||||
|
"Metric FTP|src/metric/ftp/build/Dockerfile|argus-metric-ftp:latest|src/metric/ftp/build"
|
||||||
|
"Metric Prometheus|src/metric/prometheus/build/Dockerfile|argus-metric-prometheus:latest|src/metric/prometheus/build"
|
||||||
|
"Metric Grafana|src/metric/grafana/build/Dockerfile|argus-metric-grafana:latest|src/metric/grafana/build"
|
||||||
|
)
|
||||||
|
|
||||||
|
for build_spec in "${metric_builds[@]}"; do
|
||||||
|
IFS='|' read -r image_label dockerfile_path image_tag build_context <<< "$build_spec"
|
||||||
|
if build_image "$image_label" "$dockerfile_path" "$image_tag" "$build_context"; then
|
||||||
|
images_built+=("$image_tag")
|
||||||
|
else
|
||||||
|
build_failed=true
|
||||||
|
fi
|
||||||
|
echo ""
|
||||||
|
done
|
||||||
|
fi
|
||||||
|
|
||||||
|
# =======================================
|
||||||
|
# Web & Alert module images
|
||||||
|
# =======================================
|
||||||
|
|
||||||
|
if [[ "$build_web" == true || "$build_alert" == true ]]; then
|
||||||
|
echo ""
|
||||||
|
echo "Building Web and Alert module images..."
|
||||||
|
|
||||||
|
# Pre-pull commonly used base images for stability
|
||||||
|
web_alert_base_images=(
|
||||||
|
"node:20"
|
||||||
|
"ubuntu:24.04"
|
||||||
|
)
|
||||||
|
|
||||||
|
for base_image in "${web_alert_base_images[@]}"; do
|
||||||
|
if ! pull_base_image "$base_image"; then
|
||||||
|
build_failed=true
|
||||||
|
fi
|
||||||
|
done
|
||||||
|
|
||||||
|
if [[ "$build_web" == true ]]; then
|
||||||
|
web_builds=(
|
||||||
|
"Web Frontend|src/web/build_tools/frontend/Dockerfile|argus-web-frontend:latest|."
|
||||||
|
"Web Proxy|src/web/build_tools/proxy/Dockerfile|argus-web-proxy:latest|."
|
||||||
|
)
|
||||||
|
for build_spec in "${web_builds[@]}"; do
|
||||||
|
IFS='|' read -r image_label dockerfile_path image_tag build_context <<< "$build_spec"
|
||||||
|
if build_image "$image_label" "$dockerfile_path" "$image_tag" "$build_context"; then
|
||||||
|
images_built+=("$image_tag")
|
||||||
|
else
|
||||||
|
build_failed=true
|
||||||
|
fi
|
||||||
|
echo ""
|
||||||
|
done
|
||||||
|
fi
|
||||||
|
|
||||||
|
if [[ "$build_alert" == true ]]; then
|
||||||
|
alert_builds=(
|
||||||
|
"Alertmanager|src/alert/alertmanager/build/Dockerfile|argus-alertmanager:latest|."
|
||||||
|
)
|
||||||
|
for build_spec in "${alert_builds[@]}"; do
|
||||||
|
IFS='|' read -r image_label dockerfile_path image_tag build_context <<< "$build_spec"
|
||||||
|
if build_image "$image_label" "$dockerfile_path" "$image_tag" "$build_context"; then
|
||||||
|
images_built+=("$image_tag")
|
||||||
|
else
|
||||||
|
build_failed=true
|
||||||
|
fi
|
||||||
|
echo ""
|
||||||
|
done
|
||||||
|
fi
|
||||||
|
fi
|
||||||
|
|
||||||
echo "======================================="
|
echo "======================================="
|
||||||
echo "📦 Build Summary"
|
echo "📦 Build Summary"
|
||||||
echo "======================================="
|
echo "======================================="
|
||||||
@@ -197,7 +364,6 @@ if [[ "$build_master_offline" == true ]]; then
|
|||||||
echo ""
|
echo ""
|
||||||
echo "🧳 Master offline wheels 已解压到 $master_offline_dir"
|
echo "🧳 Master offline wheels 已解压到 $master_offline_dir"
|
||||||
fi
|
fi
|
||||||
|
|
||||||
echo ""
|
echo ""
|
||||||
echo "🚀 Next steps:"
|
echo "🚀 Next steps:"
|
||||||
echo " ./build/save_images.sh --compress # 导出镜像"
|
echo " ./build/save_images.sh --compress # 导出镜像"
|
||||||
|
|||||||
@@ -68,6 +68,12 @@ declare -A images=(
|
|||||||
["argus-kibana:latest"]="argus-kibana-latest.tar"
|
["argus-kibana:latest"]="argus-kibana-latest.tar"
|
||||||
["argus-bind9:latest"]="argus-bind9-latest.tar"
|
["argus-bind9:latest"]="argus-bind9-latest.tar"
|
||||||
["argus-master:offline"]="argus-master-offline.tar"
|
["argus-master:offline"]="argus-master-offline.tar"
|
||||||
|
["argus-metric-ftp:latest"]="argus-metric-ftp-latest.tar"
|
||||||
|
["argus-metric-prometheus:latest"]="argus-metric-prometheus-latest.tar"
|
||||||
|
["argus-metric-grafana:latest"]="argus-metric-grafana-latest.tar"
|
||||||
|
["argus-web-frontend:latest"]="argus-web-frontend-latest.tar"
|
||||||
|
["argus-web-proxy:latest"]="argus-web-proxy-latest.tar"
|
||||||
|
["argus-alertmanager:latest"]="argus-alertmanager-latest.tar"
|
||||||
)
|
)
|
||||||
|
|
||||||
# 函数:检查镜像是否存在
|
# 函数:检查镜像是否存在
|
||||||
|
|||||||
@@ -34,6 +34,18 @@ Agent 不再依赖配置文件;所有参数均由环境变量与主机名推
|
|||||||
| `MASTER_ENDPOINT` | 是 | N/A | Master 基础地址,可写 `http://host:3000` 或 `host:3000`(自动补全 `http://`)。 |
|
| `MASTER_ENDPOINT` | 是 | N/A | Master 基础地址,可写 `http://host:3000` 或 `host:3000`(自动补全 `http://`)。 |
|
||||||
| `REPORT_INTERVAL_SECONDS` | 否 | `60` | 状态上报间隔(秒)。必须为正整数。 |
|
| `REPORT_INTERVAL_SECONDS` | 否 | `60` | 状态上报间隔(秒)。必须为正整数。 |
|
||||||
| `AGENT_HOSTNAME` | 否 | `$(hostname)` | 覆盖容器内主机名,便于测试或特殊命名需求。 |
|
| `AGENT_HOSTNAME` | 否 | `$(hostname)` | 覆盖容器内主机名,便于测试或特殊命名需求。 |
|
||||||
|
| `AGENT_ENV` | 否 | 来源于主机名 | 运行环境标识(如 `dev`、`prod`)。与 `AGENT_USER`、`AGENT_INSTANCE` 必须同时设置。 |
|
||||||
|
| `AGENT_USER` | 否 | 来源于主机名 | 归属用户或团队标识。与 `AGENT_ENV`、`AGENT_INSTANCE` 必须同时设置。 |
|
||||||
|
| `AGENT_INSTANCE` | 否 | 来源于主机名 | 实例编号或别名。与 `AGENT_ENV`、`AGENT_USER` 必须同时设置。 |
|
||||||
|
|
||||||
|
主机名与元数据的解析优先级:
|
||||||
|
|
||||||
|
1. 若设置 `AGENT_ENV` / `AGENT_USER` / `AGENT_INSTANCE` 且全部存在,则直接使用这些值。
|
||||||
|
2. 否则检查历史 `node.json`(注册成功后由 Master 返回的信息),若包含 `env` / `user` / `instance` 则沿用。
|
||||||
|
3. 若以上均不可用,则按历史约定从主机名解析 `env-user-instance` 前缀。
|
||||||
|
4. 如果仍无法得到完整结果,Agent 启动会失败并提示需要提供上述环境变量。
|
||||||
|
|
||||||
|
> 提示:在首次部署时需确保环境变量或主机名能够提供完整信息。完成注册后,Agent 会把 Master 返回的元数据写入 `node.json`,后续重启无需再次提供环境变量就能保持一致性。
|
||||||
|
|
||||||
派生路径:
|
派生路径:
|
||||||
|
|
||||||
|
|||||||
@@ -18,13 +18,12 @@ _HOSTNAME_PATTERN = re.compile(r"^([^-]+)-([^-]+)-([^-]+)-.*$")
|
|||||||
def collect_metadata(config: AgentConfig) -> Dict[str, Any]:
|
def collect_metadata(config: AgentConfig) -> Dict[str, Any]:
|
||||||
"""汇总节点注册需要的静态信息。"""
|
"""汇总节点注册需要的静态信息。"""
|
||||||
hostname = config.hostname
|
hostname = config.hostname
|
||||||
env, user, instance = _parse_hostname(hostname)
|
|
||||||
meta = {
|
meta = {
|
||||||
"hostname": hostname,
|
"hostname": hostname,
|
||||||
"ip": _detect_ip_address(),
|
"ip": _detect_ip_address(),
|
||||||
"env": env,
|
"env": config.environment,
|
||||||
"user": user,
|
"user": config.user,
|
||||||
"instance": instance,
|
"instance": config.instance,
|
||||||
"cpu_number": _detect_cpu_count(),
|
"cpu_number": _detect_cpu_count(),
|
||||||
"memory_in_bytes": _detect_memory_bytes(),
|
"memory_in_bytes": _detect_memory_bytes(),
|
||||||
"gpu_number": _detect_gpu_count(),
|
"gpu_number": _detect_gpu_count(),
|
||||||
|
|||||||
@@ -6,14 +6,21 @@ from dataclasses import dataclass
|
|||||||
from pathlib import Path
|
from pathlib import Path
|
||||||
from typing import Final
|
from typing import Final
|
||||||
|
|
||||||
|
from .state import load_node_state
|
||||||
from .version import VERSION
|
from .version import VERSION
|
||||||
|
from .log import get_logger
|
||||||
|
|
||||||
DEFAULT_REPORT_INTERVAL_SECONDS: Final[int] = 60
|
DEFAULT_REPORT_INTERVAL_SECONDS: Final[int] = 60
|
||||||
|
|
||||||
|
LOGGER = get_logger("argus.agent.config")
|
||||||
|
|
||||||
|
|
||||||
@dataclass(frozen=True)
|
@dataclass(frozen=True)
|
||||||
class AgentConfig:
|
class AgentConfig:
|
||||||
hostname: str
|
hostname: str
|
||||||
|
environment: str
|
||||||
|
user: str
|
||||||
|
instance: str
|
||||||
node_file: str
|
node_file: str
|
||||||
version: str
|
version: str
|
||||||
master_endpoint: str
|
master_endpoint: str
|
||||||
@@ -47,11 +54,68 @@ def _resolve_hostname() -> str:
|
|||||||
return os.environ.get("AGENT_HOSTNAME") or socket.gethostname()
|
return os.environ.get("AGENT_HOSTNAME") or socket.gethostname()
|
||||||
|
|
||||||
|
|
||||||
|
def _load_metadata_from_state(node_file: str) -> tuple[str, str, str] | None:
|
||||||
|
state = load_node_state(node_file)
|
||||||
|
if not state:
|
||||||
|
return None
|
||||||
|
|
||||||
|
meta = state.get("meta_data") or {}
|
||||||
|
env = meta.get("env") or state.get("env")
|
||||||
|
user = meta.get("user") or state.get("user")
|
||||||
|
instance = meta.get("instance") or state.get("instance")
|
||||||
|
|
||||||
|
if env and user and instance:
|
||||||
|
LOGGER.debug("Metadata resolved from node state", extra={"node_file": node_file})
|
||||||
|
return env, user, instance
|
||||||
|
|
||||||
|
LOGGER.warning(
|
||||||
|
"node.json missing metadata fields; ignoring",
|
||||||
|
extra={"node_file": node_file, "meta_data": meta},
|
||||||
|
)
|
||||||
|
return None
|
||||||
|
|
||||||
|
|
||||||
|
def _resolve_metadata_fields(hostname: str, node_file: str) -> tuple[str, str, str]:
|
||||||
|
env = os.environ.get("AGENT_ENV")
|
||||||
|
user = os.environ.get("AGENT_USER")
|
||||||
|
instance = os.environ.get("AGENT_INSTANCE")
|
||||||
|
|
||||||
|
if env and user and instance:
|
||||||
|
return env, user, instance
|
||||||
|
|
||||||
|
if any([env, user, instance]):
|
||||||
|
LOGGER.warning(
|
||||||
|
"Incomplete metadata environment variables; falling back to persisted metadata",
|
||||||
|
extra={
|
||||||
|
"has_env": bool(env),
|
||||||
|
"has_user": bool(user),
|
||||||
|
"has_instance": bool(instance),
|
||||||
|
},
|
||||||
|
)
|
||||||
|
|
||||||
|
state_metadata = _load_metadata_from_state(node_file)
|
||||||
|
if state_metadata is not None:
|
||||||
|
return state_metadata
|
||||||
|
|
||||||
|
from .collector import _parse_hostname # Local import to avoid circular dependency
|
||||||
|
|
||||||
|
env, user, instance = _parse_hostname(hostname)
|
||||||
|
|
||||||
|
if not all([env, user, instance]):
|
||||||
|
raise ValueError(
|
||||||
|
"Failed to determine metadata fields; set AGENT_ENV/USER/INSTANCE or use supported hostname pattern"
|
||||||
|
)
|
||||||
|
|
||||||
|
return env, user, instance
|
||||||
|
|
||||||
|
|
||||||
def load_config() -> AgentConfig:
|
def load_config() -> AgentConfig:
|
||||||
"""从环境变量推导配置,移除了外部配置文件依赖。"""
|
"""从环境变量推导配置,移除了外部配置文件依赖。"""
|
||||||
|
|
||||||
hostname = _resolve_hostname()
|
hostname = _resolve_hostname()
|
||||||
node_file = f"/private/argus/agent/{hostname}/node.json"
|
node_file = f"/private/argus/agent/{hostname}/node.json"
|
||||||
|
environment, user, instance = _resolve_metadata_fields(hostname, node_file)
|
||||||
|
|
||||||
health_dir = f"/private/argus/agent/{hostname}/health/"
|
health_dir = f"/private/argus/agent/{hostname}/health/"
|
||||||
|
|
||||||
master_endpoint_env = os.environ.get("MASTER_ENDPOINT")
|
master_endpoint_env = os.environ.get("MASTER_ENDPOINT")
|
||||||
@@ -66,6 +130,9 @@ def load_config() -> AgentConfig:
|
|||||||
|
|
||||||
return AgentConfig(
|
return AgentConfig(
|
||||||
hostname=hostname,
|
hostname=hostname,
|
||||||
|
environment=environment,
|
||||||
|
user=user,
|
||||||
|
instance=instance,
|
||||||
node_file=node_file,
|
node_file=node_file,
|
||||||
version=VERSION,
|
version=VERSION,
|
||||||
master_endpoint=master_endpoint,
|
master_endpoint=master_endpoint,
|
||||||
|
|||||||
BIN
src/agent/dist/argus-agent
vendored
BIN
src/agent/dist/argus-agent
vendored
Binary file not shown.
@@ -12,6 +12,8 @@ VENV_DIR="$BUILD_ROOT/venv"
|
|||||||
|
|
||||||
AGENT_BUILD_IMAGE="${AGENT_BUILD_IMAGE:-python:3.11-slim-bullseye}"
|
AGENT_BUILD_IMAGE="${AGENT_BUILD_IMAGE:-python:3.11-slim-bullseye}"
|
||||||
AGENT_BUILD_USE_DOCKER="${AGENT_BUILD_USE_DOCKER:-1}"
|
AGENT_BUILD_USE_DOCKER="${AGENT_BUILD_USE_DOCKER:-1}"
|
||||||
|
# 默认在容器内忽略代理以避免公司内网代理在 Docker 网络不可达导致 pip 失败(可用 0 关闭)
|
||||||
|
AGENT_BUILD_IGNORE_PROXY="${AGENT_BUILD_IGNORE_PROXY:-1}"
|
||||||
USED_DOCKER=0
|
USED_DOCKER=0
|
||||||
|
|
||||||
run_host_build() {
|
run_host_build() {
|
||||||
@@ -71,6 +73,7 @@ run_docker_build() {
|
|||||||
pass_env_if_set http_proxy
|
pass_env_if_set http_proxy
|
||||||
pass_env_if_set https_proxy
|
pass_env_if_set https_proxy
|
||||||
pass_env_if_set no_proxy
|
pass_env_if_set no_proxy
|
||||||
|
pass_env_if_set AGENT_BUILD_IGNORE_PROXY
|
||||||
|
|
||||||
build_script=$(cat <<'INNER'
|
build_script=$(cat <<'INNER'
|
||||||
set -euo pipefail
|
set -euo pipefail
|
||||||
@@ -82,6 +85,10 @@ rm -rf build dist
|
|||||||
mkdir -p build/pyinstaller dist
|
mkdir -p build/pyinstaller dist
|
||||||
python3 -m venv --copies build/venv
|
python3 -m venv --copies build/venv
|
||||||
source build/venv/bin/activate
|
source build/venv/bin/activate
|
||||||
|
# 若指定忽略代理,则清空常见代理与 pip 镜像环境变量,避免容器内代理不可达
|
||||||
|
if [ "${AGENT_BUILD_IGNORE_PROXY:-1}" = "1" ]; then
|
||||||
|
unset http_proxy https_proxy HTTP_PROXY HTTPS_PROXY PIP_INDEX_URL PIP_EXTRA_INDEX_URL PIP_TRUSTED_HOST
|
||||||
|
fi
|
||||||
pip install --upgrade pip
|
pip install --upgrade pip
|
||||||
pip install .
|
pip install .
|
||||||
pip install pyinstaller==6.6.0
|
pip install pyinstaller==6.6.0
|
||||||
|
|||||||
0
src/agent/tests/__init__.py
Normal file
0
src/agent/tests/__init__.py
Normal file
@@ -60,6 +60,36 @@ services:
|
|||||||
ipv4_address: 172.28.0.20
|
ipv4_address: 172.28.0.20
|
||||||
restart: always
|
restart: always
|
||||||
|
|
||||||
|
agent_env:
|
||||||
|
image: ubuntu:22.04
|
||||||
|
container_name: argus-agent-env-e2e
|
||||||
|
hostname: host_abc
|
||||||
|
depends_on:
|
||||||
|
- master
|
||||||
|
- bind
|
||||||
|
environment:
|
||||||
|
- MASTER_ENDPOINT=http://master.argus.com:3000
|
||||||
|
- REPORT_INTERVAL_SECONDS=2
|
||||||
|
- AGENT_ENV=prod
|
||||||
|
- AGENT_USER=ml
|
||||||
|
- AGENT_INSTANCE=node-3
|
||||||
|
- AGENT_HOSTNAME=host_abc
|
||||||
|
- "ARGUS_BUILD_UID=${ARGUS_BUILD_UID:-2133}"
|
||||||
|
- "ARGUS_BUILD_GID=${ARGUS_BUILD_GID:-2015}"
|
||||||
|
volumes:
|
||||||
|
- ./private/argus/agent/host_abc:/private/argus/agent/host_abc
|
||||||
|
- ./private/argus/agent/host_abc/health:/private/argus/agent/host_abc/health
|
||||||
|
- ./private/argus/etc:/private/argus/etc
|
||||||
|
- ../dist/argus-agent:/usr/local/bin/argus-agent:ro
|
||||||
|
- ./scripts/agent_entrypoint.sh:/usr/local/bin/agent-entrypoint.sh:ro
|
||||||
|
- ../scripts/agent_deployment_verify.sh:/usr/local/bin/agent_deployment_verify.sh:ro
|
||||||
|
entrypoint:
|
||||||
|
- /usr/local/bin/agent-entrypoint.sh
|
||||||
|
networks:
|
||||||
|
default:
|
||||||
|
ipv4_address: 172.28.0.21
|
||||||
|
restart: always
|
||||||
|
|
||||||
networks:
|
networks:
|
||||||
default:
|
default:
|
||||||
driver: bridge
|
driver: bridge
|
||||||
|
|||||||
@@ -7,10 +7,10 @@ SCRIPTS=(
|
|||||||
"02_up.sh"
|
"02_up.sh"
|
||||||
"03_wait_and_assert_registration.sh"
|
"03_wait_and_assert_registration.sh"
|
||||||
"04_write_health_files.sh"
|
"04_write_health_files.sh"
|
||||||
"08_verify_agent.sh"
|
"05_verify_agent.sh"
|
||||||
"05_assert_status_on_master.sh"
|
"06_assert_status_on_master.sh"
|
||||||
"06_restart_agent_and_reregister.sh"
|
"07_restart_agent_and_reregister.sh"
|
||||||
"07_down.sh"
|
"08_down.sh"
|
||||||
)
|
)
|
||||||
|
|
||||||
for script in "${SCRIPTS[@]}"; do
|
for script in "${SCRIPTS[@]}"; do
|
||||||
|
|||||||
@@ -41,7 +41,7 @@ compose() {
|
|||||||
fi
|
fi
|
||||||
}
|
}
|
||||||
|
|
||||||
docker container rm -f argus-agent-e2e argus-master-agent-e2e argus-bind-agent-e2e >/dev/null 2>&1 || true
|
docker container rm -f argus-agent-e2e argus-agent-env-e2e argus-master-agent-e2e argus-bind-agent-e2e >/dev/null 2>&1 || true
|
||||||
|
|
||||||
docker network rm tests_default >/dev/null 2>&1 || true
|
docker network rm tests_default >/dev/null 2>&1 || true
|
||||||
|
|
||||||
|
|||||||
@@ -6,11 +6,14 @@ TEST_ROOT="$(cd "$SCRIPT_DIR/.." && pwd)"
|
|||||||
TMP_ROOT="$TEST_ROOT/tmp"
|
TMP_ROOT="$TEST_ROOT/tmp"
|
||||||
API_BASE="http://localhost:32300/api/v1/master"
|
API_BASE="http://localhost:32300/api/v1/master"
|
||||||
AGENT_HOSTNAME="dev-e2euser-e2einst-pod-0"
|
AGENT_HOSTNAME="dev-e2euser-e2einst-pod-0"
|
||||||
|
ENV_AGENT_HOSTNAME="host_abc"
|
||||||
NODE_FILE="$TEST_ROOT/private/argus/agent/$AGENT_HOSTNAME/node.json"
|
NODE_FILE="$TEST_ROOT/private/argus/agent/$AGENT_HOSTNAME/node.json"
|
||||||
|
ENV_NODE_FILE="$TEST_ROOT/private/argus/agent/$ENV_AGENT_HOSTNAME/node.json"
|
||||||
|
|
||||||
mkdir -p "$TMP_ROOT"
|
mkdir -p "$TMP_ROOT"
|
||||||
|
|
||||||
node_id=""
|
primary_node_id=""
|
||||||
|
env_node_id=""
|
||||||
for _ in {1..30}; do
|
for _ in {1..30}; do
|
||||||
sleep 2
|
sleep 2
|
||||||
response=$(curl -sS "$API_BASE/nodes" || true)
|
response=$(curl -sS "$API_BASE/nodes" || true)
|
||||||
@@ -19,24 +22,49 @@ for _ in {1..30}; do
|
|||||||
fi
|
fi
|
||||||
list_file="$TMP_ROOT/nodes_list.json"
|
list_file="$TMP_ROOT/nodes_list.json"
|
||||||
echo "$response" > "$list_file"
|
echo "$response" > "$list_file"
|
||||||
node_id=$(python3 - "$list_file" <<'PY'
|
readarray -t node_ids < <(python3 - "$list_file" "$AGENT_HOSTNAME" "$ENV_AGENT_HOSTNAME" <<'PY'
|
||||||
import json, sys
|
import json, sys
|
||||||
|
|
||||||
with open(sys.argv[1]) as handle:
|
with open(sys.argv[1]) as handle:
|
||||||
nodes = json.load(handle)
|
nodes = json.load(handle)
|
||||||
print(nodes[0]["id"] if nodes else "")
|
|
||||||
|
target_primary = sys.argv[2]
|
||||||
|
target_env = sys.argv[3]
|
||||||
|
|
||||||
|
primary_id = ""
|
||||||
|
env_id = ""
|
||||||
|
|
||||||
|
for node in nodes:
|
||||||
|
if node.get("name") == target_primary:
|
||||||
|
primary_id = node.get("id", "")
|
||||||
|
if node.get("name") == target_env:
|
||||||
|
env_id = node.get("id", "")
|
||||||
|
|
||||||
|
print(primary_id)
|
||||||
|
print(env_id)
|
||||||
PY
|
PY
|
||||||
)
|
)
|
||||||
if [[ -n "$node_id" ]]; then
|
|
||||||
|
primary_node_id="${node_ids[0]}"
|
||||||
|
env_node_id="${node_ids[1]}"
|
||||||
|
|
||||||
|
if [[ -n "$primary_node_id" && -n "$env_node_id" ]]; then
|
||||||
break
|
break
|
||||||
fi
|
fi
|
||||||
done
|
done
|
||||||
|
|
||||||
if [[ -z "$node_id" ]]; then
|
if [[ -z "$primary_node_id" ]]; then
|
||||||
echo "[ERROR] Agent did not register within timeout" >&2
|
echo "[ERROR] Primary agent did not register within timeout" >&2
|
||||||
exit 1
|
exit 1
|
||||||
fi
|
fi
|
||||||
|
|
||||||
echo "$node_id" > "$TMP_ROOT/node_id"
|
if [[ -z "$env_node_id" ]]; then
|
||||||
|
echo "[ERROR] Env-variable agent did not register within timeout" >&2
|
||||||
|
exit 1
|
||||||
|
fi
|
||||||
|
|
||||||
|
echo "$primary_node_id" > "$TMP_ROOT/node_id"
|
||||||
|
echo "$env_node_id" > "$TMP_ROOT/node_id_host_abc"
|
||||||
|
|
||||||
if [[ ! -f "$NODE_FILE" ]]; then
|
if [[ ! -f "$NODE_FILE" ]]; then
|
||||||
echo "[ERROR] node.json not created at $NODE_FILE" >&2
|
echo "[ERROR] node.json not created at $NODE_FILE" >&2
|
||||||
@@ -50,8 +78,20 @@ with open(sys.argv[1]) as handle:
|
|||||||
assert "id" in node and node["id"], "node.json missing id"
|
assert "id" in node and node["id"], "node.json missing id"
|
||||||
PY
|
PY
|
||||||
|
|
||||||
|
if [[ ! -f "$ENV_NODE_FILE" ]]; then
|
||||||
|
echo "[ERROR] node.json not created at $ENV_NODE_FILE" >&2
|
||||||
|
exit 1
|
||||||
|
fi
|
||||||
|
|
||||||
|
python3 - "$ENV_NODE_FILE" <<'PY'
|
||||||
|
import json, sys
|
||||||
|
with open(sys.argv[1]) as handle:
|
||||||
|
node = json.load(handle)
|
||||||
|
assert "id" in node and node["id"], "env agent node.json missing id"
|
||||||
|
PY
|
||||||
|
|
||||||
detail_file="$TMP_ROOT/initial_detail.json"
|
detail_file="$TMP_ROOT/initial_detail.json"
|
||||||
curl -sS "$API_BASE/nodes/$node_id" -o "$detail_file"
|
curl -sS "$API_BASE/nodes/$primary_node_id" -o "$detail_file"
|
||||||
python3 - "$detail_file" "$TMP_ROOT/initial_ip" <<'PY'
|
python3 - "$detail_file" "$TMP_ROOT/initial_ip" <<'PY'
|
||||||
import json, sys, pathlib
|
import json, sys, pathlib
|
||||||
with open(sys.argv[1]) as handle:
|
with open(sys.argv[1]) as handle:
|
||||||
@@ -62,4 +102,5 @@ if not ip:
|
|||||||
pathlib.Path(sys.argv[2]).write_text(ip)
|
pathlib.Path(sys.argv[2]).write_text(ip)
|
||||||
PY
|
PY
|
||||||
|
|
||||||
echo "[INFO] Agent registered with node id $node_id"
|
echo "[INFO] Agent registered with node id $primary_node_id"
|
||||||
|
echo "[INFO] Env-variable agent registered with node id $env_node_id"
|
||||||
|
|||||||
60
src/agent/tests/scripts/05_verify_agent.sh
Executable file
60
src/agent/tests/scripts/05_verify_agent.sh
Executable file
@@ -0,0 +1,60 @@
|
|||||||
|
#!/usr/bin/env bash
|
||||||
|
set -euo pipefail
|
||||||
|
|
||||||
|
SCRIPT_DIR="$(cd "$(dirname "${BASH_SOURCE[0]}")" && pwd)"
|
||||||
|
TEST_ROOT="$(cd "$SCRIPT_DIR/.." && pwd)"
|
||||||
|
REPO_ROOT="$(cd "$TEST_ROOT/.." && pwd)"
|
||||||
|
VERIFY_SCRIPT="$REPO_ROOT/scripts/agent_deployment_verify.sh"
|
||||||
|
ENV_NODE_ID_FILE="$TEST_ROOT/tmp/node_id_host_abc"
|
||||||
|
PRIMARY_CONTAINER="argus-agent-e2e"
|
||||||
|
ENV_CONTAINER="argus-agent-env-e2e"
|
||||||
|
PRIMARY_HOSTNAME="dev-e2euser-e2einst-pod-0"
|
||||||
|
ENV_HOSTNAME="host_abc"
|
||||||
|
|
||||||
|
if ! docker ps --format '{{.Names}}' | grep -q "^${PRIMARY_CONTAINER}$"; then
|
||||||
|
echo "[WARN] agent container not running; skip verification"
|
||||||
|
exit 0
|
||||||
|
fi
|
||||||
|
|
||||||
|
if docker exec -i "$PRIMARY_CONTAINER" bash -lc 'command -v curl >/dev/null && command -v jq >/dev/null'; then
|
||||||
|
echo "[INFO] curl/jq already installed in agent container"
|
||||||
|
else
|
||||||
|
echo "[INFO] Installing curl/jq in agent container"
|
||||||
|
docker exec -i "$PRIMARY_CONTAINER" bash -lc 'apt-get update >/dev/null 2>&1 && apt-get install -y curl jq >/dev/null 2>&1' || true
|
||||||
|
fi
|
||||||
|
|
||||||
|
if [[ ! -f "$VERIFY_SCRIPT" ]]; then
|
||||||
|
echo "[ERROR] Verification script missing at $VERIFY_SCRIPT" >&2
|
||||||
|
exit 1
|
||||||
|
fi
|
||||||
|
|
||||||
|
run_verifier() {
|
||||||
|
local container="$1" hostname="$2"
|
||||||
|
|
||||||
|
if ! docker ps --format '{{.Names}}' | grep -q "^${container}$"; then
|
||||||
|
echo "[WARN] container $container not running; skip"
|
||||||
|
return
|
||||||
|
fi
|
||||||
|
|
||||||
|
if ! docker exec -i "$container" bash -lc 'command -v /usr/local/bin/agent_deployment_verify.sh >/dev/null'; then
|
||||||
|
echo "[ERROR] /usr/local/bin/agent_deployment_verify.sh missing in $container" >&2
|
||||||
|
exit 1
|
||||||
|
fi
|
||||||
|
|
||||||
|
echo "[INFO] Running verification for $hostname in $container"
|
||||||
|
docker exec -i "$container" env VERIFY_HOSTNAME="$hostname" /usr/local/bin/agent_deployment_verify.sh
|
||||||
|
}
|
||||||
|
|
||||||
|
run_verifier "$PRIMARY_CONTAINER" "$PRIMARY_HOSTNAME"
|
||||||
|
|
||||||
|
if docker ps --format '{{.Names}}' | grep -q "^${ENV_CONTAINER}$"; then
|
||||||
|
if docker exec -i "$ENV_CONTAINER" bash -lc 'command -v curl >/dev/null && command -v jq >/dev/null'; then
|
||||||
|
echo "[INFO] curl/jq already installed in env agent container"
|
||||||
|
else
|
||||||
|
echo "[INFO] Installing curl/jq in env agent container"
|
||||||
|
docker exec -i "$ENV_CONTAINER" bash -lc 'apt-get update >/dev/null 2>&1 && apt-get install -y curl jq >/dev/null 2>&1' || true
|
||||||
|
fi
|
||||||
|
run_verifier "$ENV_CONTAINER" "$ENV_HOSTNAME"
|
||||||
|
else
|
||||||
|
echo "[WARN] env-driven agent container not running; skip secondary verification"
|
||||||
|
fi
|
||||||
@@ -6,6 +6,8 @@ TEST_ROOT="$(cd "$SCRIPT_DIR/.." && pwd)"
|
|||||||
TMP_ROOT="$TEST_ROOT/tmp"
|
TMP_ROOT="$TEST_ROOT/tmp"
|
||||||
API_BASE="http://localhost:32300/api/v1/master"
|
API_BASE="http://localhost:32300/api/v1/master"
|
||||||
NODE_ID="$(cat "$TMP_ROOT/node_id")"
|
NODE_ID="$(cat "$TMP_ROOT/node_id")"
|
||||||
|
ENV_NODE_ID="$(cat "$TMP_ROOT/node_id_host_abc")"
|
||||||
|
ENV_HOSTNAME="host_abc"
|
||||||
NODES_JSON="$TEST_ROOT/private/argus/metric/prometheus/nodes.json"
|
NODES_JSON="$TEST_ROOT/private/argus/metric/prometheus/nodes.json"
|
||||||
|
|
||||||
success=false
|
success=false
|
||||||
@@ -41,13 +43,36 @@ if [[ ! -f "$NODES_JSON" ]]; then
|
|||||||
exit 1
|
exit 1
|
||||||
fi
|
fi
|
||||||
|
|
||||||
python3 - "$NODES_JSON" <<'PY'
|
python3 - "$NODES_JSON" "$NODE_ID" "$ENV_NODE_ID" <<'PY'
|
||||||
import json, sys
|
import json, sys
|
||||||
with open(sys.argv[1]) as handle:
|
with open(sys.argv[1]) as handle:
|
||||||
nodes = json.load(handle)
|
nodes = json.load(handle)
|
||||||
assert len(nodes) == 1, nodes
|
|
||||||
entry = nodes[0]
|
expected_primary = sys.argv[2]
|
||||||
assert entry["node_id"], entry
|
expected_env = sys.argv[3]
|
||||||
|
|
||||||
|
ids = {entry.get("node_id") for entry in nodes}
|
||||||
|
assert expected_primary in ids, nodes
|
||||||
|
assert expected_env in ids, nodes
|
||||||
|
assert len(nodes) >= 2, nodes
|
||||||
PY
|
PY
|
||||||
|
|
||||||
echo "[INFO] Master reflects agent health and nodes.json entries"
|
echo "[INFO] Master reflects agent health and nodes.json entries"
|
||||||
|
|
||||||
|
env_detail_file="$TMP_ROOT/env_agent_detail.json"
|
||||||
|
curl -sS "$API_BASE/nodes/$ENV_NODE_ID" -o "$env_detail_file"
|
||||||
|
python3 - "$env_detail_file" "$ENV_HOSTNAME" <<'PY'
|
||||||
|
import json, sys
|
||||||
|
with open(sys.argv[1]) as handle:
|
||||||
|
node = json.load(handle)
|
||||||
|
|
||||||
|
expected_name = sys.argv[2]
|
||||||
|
|
||||||
|
assert node.get("name") == expected_name, node
|
||||||
|
meta = node.get("meta_data", {})
|
||||||
|
assert meta.get("env") == "prod", meta
|
||||||
|
assert meta.get("user") == "ml", meta
|
||||||
|
assert meta.get("instance") == "node-3", meta
|
||||||
|
PY
|
||||||
|
|
||||||
|
echo "[INFO] Env-variable agent reports expected metadata"
|
||||||
@@ -6,10 +6,20 @@ TEST_ROOT="$(cd "$SCRIPT_DIR/.." && pwd)"
|
|||||||
TMP_ROOT="$TEST_ROOT/tmp"
|
TMP_ROOT="$TEST_ROOT/tmp"
|
||||||
API_BASE="http://localhost:32300/api/v1/master"
|
API_BASE="http://localhost:32300/api/v1/master"
|
||||||
NODE_ID="$(cat "$TMP_ROOT/node_id")"
|
NODE_ID="$(cat "$TMP_ROOT/node_id")"
|
||||||
|
ENV_NODE_ID_FILE="$TMP_ROOT/node_id_host_abc"
|
||||||
|
if [[ ! -f "$ENV_NODE_ID_FILE" ]]; then
|
||||||
|
echo "[ERROR] Env agent node id file missing at $ENV_NODE_ID_FILE" >&2
|
||||||
|
exit 1
|
||||||
|
fi
|
||||||
|
|
||||||
|
ENV_NODE_ID="$(cat "$ENV_NODE_ID_FILE")"
|
||||||
AGENT_HOSTNAME="dev-e2euser-e2einst-pod-0"
|
AGENT_HOSTNAME="dev-e2euser-e2einst-pod-0"
|
||||||
|
ENV_AGENT_HOSTNAME="host_abc"
|
||||||
NETWORK_NAME="tests_default"
|
NETWORK_NAME="tests_default"
|
||||||
NEW_AGENT_IP="172.28.0.200"
|
NEW_AGENT_IP="172.28.0.200"
|
||||||
|
NEW_ENV_AGENT_IP="172.28.0.210"
|
||||||
ENTRYPOINT_SCRIPT="$SCRIPT_DIR/agent_entrypoint.sh"
|
ENTRYPOINT_SCRIPT="$SCRIPT_DIR/agent_entrypoint.sh"
|
||||||
|
VERIFY_SCRIPT="$TEST_ROOT/../scripts/agent_deployment_verify.sh"
|
||||||
ENV_FILE="$TEST_ROOT/.env"
|
ENV_FILE="$TEST_ROOT/.env"
|
||||||
|
|
||||||
# 中文提示:重启场景也需要同样的入口脚本,确保 DNS 注册逻辑一致
|
# 中文提示:重启场景也需要同样的入口脚本,确保 DNS 注册逻辑一致
|
||||||
@@ -18,6 +28,11 @@ if [[ ! -f "$ENTRYPOINT_SCRIPT" ]]; then
|
|||||||
exit 1
|
exit 1
|
||||||
fi
|
fi
|
||||||
|
|
||||||
|
if [[ ! -f "$VERIFY_SCRIPT" ]]; then
|
||||||
|
echo "[ERROR] agent verification script missing at $VERIFY_SCRIPT" >&2
|
||||||
|
exit 1
|
||||||
|
fi
|
||||||
|
|
||||||
if [[ ! -f "$TMP_ROOT/agent_binary_path" ]]; then
|
if [[ ! -f "$TMP_ROOT/agent_binary_path" ]]; then
|
||||||
echo "[ERROR] Agent binary path missing; rerun bootstrap" >&2
|
echo "[ERROR] Agent binary path missing; rerun bootstrap" >&2
|
||||||
exit 1
|
exit 1
|
||||||
@@ -74,15 +89,37 @@ if [[ "$prev_ip" != "$initial_ip" ]]; then
|
|||||||
exit 1
|
exit 1
|
||||||
fi
|
fi
|
||||||
|
|
||||||
|
env_before_file="$TMP_ROOT/env_before_restart.json"
|
||||||
|
curl -sS "$API_BASE/nodes/$ENV_NODE_ID" -o "$env_before_file"
|
||||||
|
env_prev_last_updated=$(python3 - "$env_before_file" <<'PY'
|
||||||
|
import json, sys
|
||||||
|
with open(sys.argv[1]) as handle:
|
||||||
|
node = json.load(handle)
|
||||||
|
print(node.get("last_updated", ""))
|
||||||
|
PY
|
||||||
|
)
|
||||||
|
env_prev_ip=$(python3 - "$env_before_file" <<'PY'
|
||||||
|
import json, sys
|
||||||
|
with open(sys.argv[1]) as handle:
|
||||||
|
node = json.load(handle)
|
||||||
|
print(node["meta_data"].get("ip", ""))
|
||||||
|
PY
|
||||||
|
)
|
||||||
|
|
||||||
pushd "$TEST_ROOT" >/dev/null
|
pushd "$TEST_ROOT" >/dev/null
|
||||||
compose rm -sf agent
|
compose rm -sf agent
|
||||||
|
compose rm -sf agent_env
|
||||||
popd >/dev/null
|
popd >/dev/null
|
||||||
|
|
||||||
docker container rm -f argus-agent-e2e >/dev/null 2>&1 || true
|
docker container rm -f argus-agent-e2e >/dev/null 2>&1 || true
|
||||||
|
docker container rm -f argus-agent-env-e2e >/dev/null 2>&1 || true
|
||||||
|
|
||||||
AGENT_DIR="$TEST_ROOT/private/argus/agent/$AGENT_HOSTNAME"
|
AGENT_DIR="$TEST_ROOT/private/argus/agent/$AGENT_HOSTNAME"
|
||||||
HEALTH_DIR="$TEST_ROOT/private/argus/agent/$AGENT_HOSTNAME/health"
|
HEALTH_DIR="$TEST_ROOT/private/argus/agent/$AGENT_HOSTNAME/health"
|
||||||
|
|
||||||
|
ENV_AGENT_DIR="$TEST_ROOT/private/argus/agent/$ENV_AGENT_HOSTNAME"
|
||||||
|
ENV_HEALTH_DIR="$TEST_ROOT/private/argus/agent/$ENV_AGENT_HOSTNAME/health"
|
||||||
|
|
||||||
# 先以 sleep 方式启动容器,确保我们掌握注册时的网络状态
|
# 先以 sleep 方式启动容器,确保我们掌握注册时的网络状态
|
||||||
if ! docker run -d \
|
if ! docker run -d \
|
||||||
--name argus-agent-e2e \
|
--name argus-agent-e2e \
|
||||||
@@ -94,6 +131,7 @@ if ! docker run -d \
|
|||||||
-v "$TEST_ROOT/private/argus/etc:/private/argus/etc" \
|
-v "$TEST_ROOT/private/argus/etc:/private/argus/etc" \
|
||||||
-v "$AGENT_BINARY:/usr/local/bin/argus-agent:ro" \
|
-v "$AGENT_BINARY:/usr/local/bin/argus-agent:ro" \
|
||||||
-v "$ENTRYPOINT_SCRIPT:/usr/local/bin/agent-entrypoint.sh:ro" \
|
-v "$ENTRYPOINT_SCRIPT:/usr/local/bin/agent-entrypoint.sh:ro" \
|
||||||
|
-v "$VERIFY_SCRIPT:/usr/local/bin/agent_deployment_verify.sh:ro" \
|
||||||
-e MASTER_ENDPOINT=http://master.argus.com:3000 \
|
-e MASTER_ENDPOINT=http://master.argus.com:3000 \
|
||||||
-e REPORT_INTERVAL_SECONDS=2 \
|
-e REPORT_INTERVAL_SECONDS=2 \
|
||||||
-e ARGUS_BUILD_UID="$AGENT_UID" \
|
-e ARGUS_BUILD_UID="$AGENT_UID" \
|
||||||
@@ -141,3 +179,76 @@ if [[ "$success" != true ]]; then
|
|||||||
fi
|
fi
|
||||||
|
|
||||||
echo "[INFO] Agent restart produced successful re-registration with IP change"
|
echo "[INFO] Agent restart produced successful re-registration with IP change"
|
||||||
|
|
||||||
|
# ---- Restart env-driven agent without metadata environment variables ----
|
||||||
|
|
||||||
|
if [[ ! -d "$ENV_AGENT_DIR" ]]; then
|
||||||
|
echo "[ERROR] Env agent data dir missing at $ENV_AGENT_DIR" >&2
|
||||||
|
exit 1
|
||||||
|
fi
|
||||||
|
|
||||||
|
if [[ ! -d "$ENV_HEALTH_DIR" ]]; then
|
||||||
|
mkdir -p "$ENV_HEALTH_DIR"
|
||||||
|
fi
|
||||||
|
|
||||||
|
if ! docker run -d \
|
||||||
|
--name argus-agent-env-e2e \
|
||||||
|
--hostname "$ENV_AGENT_HOSTNAME" \
|
||||||
|
--network "$NETWORK_NAME" \
|
||||||
|
--ip "$NEW_ENV_AGENT_IP" \
|
||||||
|
-v "$ENV_AGENT_DIR:/private/argus/agent/$ENV_AGENT_HOSTNAME" \
|
||||||
|
-v "$ENV_HEALTH_DIR:/private/argus/agent/$ENV_AGENT_HOSTNAME/health" \
|
||||||
|
-v "$TEST_ROOT/private/argus/etc:/private/argus/etc" \
|
||||||
|
-v "$AGENT_BINARY:/usr/local/bin/argus-agent:ro" \
|
||||||
|
-v "$ENTRYPOINT_SCRIPT:/usr/local/bin/agent-entrypoint.sh:ro" \
|
||||||
|
-v "$VERIFY_SCRIPT:/usr/local/bin/agent_deployment_verify.sh:ro" \
|
||||||
|
-e MASTER_ENDPOINT=http://master.argus.com:3000 \
|
||||||
|
-e REPORT_INTERVAL_SECONDS=2 \
|
||||||
|
-e ARGUS_BUILD_UID="$AGENT_UID" \
|
||||||
|
-e ARGUS_BUILD_GID="$AGENT_GID" \
|
||||||
|
--entrypoint /usr/local/bin/agent-entrypoint.sh \
|
||||||
|
ubuntu:22.04 >/dev/null; then
|
||||||
|
echo "[ERROR] Failed to start env-driven agent container without metadata env" >&2
|
||||||
|
exit 1
|
||||||
|
fi
|
||||||
|
|
||||||
|
env_success=false
|
||||||
|
env_detail_file="$TMP_ROOT/env_post_restart.json"
|
||||||
|
for _ in {1..20}; do
|
||||||
|
sleep 3
|
||||||
|
if ! curl -sS "$API_BASE/nodes/$ENV_NODE_ID" -o "$env_detail_file"; then
|
||||||
|
continue
|
||||||
|
fi
|
||||||
|
if python3 - "$env_detail_file" "$env_prev_last_updated" "$ENV_NODE_ID" "$env_prev_ip" "$NEW_ENV_AGENT_IP" <<'PY'
|
||||||
|
import json, sys
|
||||||
|
with open(sys.argv[1]) as handle:
|
||||||
|
node = json.load(handle)
|
||||||
|
prev_last_updated = sys.argv[2]
|
||||||
|
expected_id = sys.argv[3]
|
||||||
|
old_ip = sys.argv[4]
|
||||||
|
expected_ip = sys.argv[5]
|
||||||
|
last_updated = node.get("last_updated")
|
||||||
|
current_ip = node["meta_data"].get("ip")
|
||||||
|
meta = node.get("meta_data", {})
|
||||||
|
assert node["id"] == expected_id
|
||||||
|
if current_ip != expected_ip:
|
||||||
|
raise SystemExit(1)
|
||||||
|
if current_ip == old_ip:
|
||||||
|
raise SystemExit(1)
|
||||||
|
if not last_updated or last_updated == prev_last_updated:
|
||||||
|
raise SystemExit(1)
|
||||||
|
if meta.get("env") != "prod" or meta.get("user") != "ml" or meta.get("instance") != "node-3":
|
||||||
|
raise SystemExit(1)
|
||||||
|
PY
|
||||||
|
then
|
||||||
|
env_success=true
|
||||||
|
break
|
||||||
|
fi
|
||||||
|
done
|
||||||
|
|
||||||
|
if [[ "$env_success" != true ]]; then
|
||||||
|
echo "[ERROR] Env-driven agent did not reuse persisted metadata after restart" >&2
|
||||||
|
exit 1
|
||||||
|
fi
|
||||||
|
|
||||||
|
echo "[INFO] Env-driven agent restart succeeded with persisted metadata"
|
||||||
@@ -13,7 +13,7 @@ compose() {
|
|||||||
fi
|
fi
|
||||||
}
|
}
|
||||||
|
|
||||||
docker container rm -f argus-agent-e2e >/dev/null 2>&1 || true
|
docker container rm -f argus-agent-e2e argus-agent-env-e2e >/dev/null 2>&1 || true
|
||||||
|
|
||||||
pushd "$TEST_ROOT" >/dev/null
|
pushd "$TEST_ROOT" >/dev/null
|
||||||
compose down --remove-orphans
|
compose down --remove-orphans
|
||||||
@@ -1,26 +0,0 @@
|
|||||||
#!/usr/bin/env bash
|
|
||||||
set -euo pipefail
|
|
||||||
|
|
||||||
SCRIPT_DIR="$(cd "$(dirname "${BASH_SOURCE[0]}")" && pwd)"
|
|
||||||
TEST_ROOT="$(cd "$SCRIPT_DIR/.." && pwd)"
|
|
||||||
VERIFY_SCRIPT="$(cd "$TEST_ROOT/.." && pwd)/scripts/agent_deployment_verify.sh"
|
|
||||||
|
|
||||||
if ! docker ps --format '{{.Names}}' | grep -q '^argus-agent-e2e$'; then
|
|
||||||
echo "[WARN] agent container not running; skip verification"
|
|
||||||
exit 0
|
|
||||||
fi
|
|
||||||
|
|
||||||
if docker exec -i argus-agent-e2e bash -lc 'command -v curl >/dev/null && command -v jq >/dev/null'; then
|
|
||||||
echo "[INFO] curl/jq already installed in agent container"
|
|
||||||
else
|
|
||||||
echo "[INFO] Installing curl/jq in agent container"
|
|
||||||
docker exec -i argus-agent-e2e bash -lc 'apt-get update >/dev/null 2>&1 && apt-get install -y curl jq >/dev/null 2>&1' || true
|
|
||||||
fi
|
|
||||||
|
|
||||||
if docker exec -i argus-agent-e2e bash -lc 'command -v /usr/local/bin/agent_deployment_verify.sh >/dev/null'; then
|
|
||||||
docker exec -i argus-agent-e2e /usr/local/bin/agent_deployment_verify.sh
|
|
||||||
elif [[ -x "$VERIFY_SCRIPT" ]]; then
|
|
||||||
docker exec -i argus-agent-e2e "$VERIFY_SCRIPT"
|
|
||||||
else
|
|
||||||
echo "[WARN] agent_deployment_verify.sh not found"
|
|
||||||
fi
|
|
||||||
151
src/agent/tests/test_config_metadata.py
Normal file
151
src/agent/tests/test_config_metadata.py
Normal file
@@ -0,0 +1,151 @@
|
|||||||
|
from __future__ import annotations
|
||||||
|
|
||||||
|
import os
|
||||||
|
import unittest
|
||||||
|
from contextlib import contextmanager
|
||||||
|
from unittest.mock import patch
|
||||||
|
|
||||||
|
from app.config import AgentConfig, load_config
|
||||||
|
|
||||||
|
|
||||||
|
@contextmanager
|
||||||
|
def temp_env(**overrides: str | None):
|
||||||
|
originals: dict[str, str | None] = {}
|
||||||
|
try:
|
||||||
|
for key, value in overrides.items():
|
||||||
|
originals[key] = os.environ.get(key)
|
||||||
|
if value is None:
|
||||||
|
os.environ.pop(key, None)
|
||||||
|
else:
|
||||||
|
os.environ[key] = value
|
||||||
|
yield
|
||||||
|
finally:
|
||||||
|
for key, original in originals.items():
|
||||||
|
if original is None:
|
||||||
|
os.environ.pop(key, None)
|
||||||
|
else:
|
||||||
|
os.environ[key] = original
|
||||||
|
|
||||||
|
|
||||||
|
class LoadConfigMetadataTests(unittest.TestCase):
|
||||||
|
@patch("app.config.Path.mkdir")
|
||||||
|
def test_metadata_from_environment_variables(self, mock_mkdir):
|
||||||
|
with temp_env(
|
||||||
|
MASTER_ENDPOINT="http://master.local",
|
||||||
|
AGENT_HOSTNAME="dev-user-one-pod",
|
||||||
|
AGENT_ENV="prod",
|
||||||
|
AGENT_USER="ops",
|
||||||
|
AGENT_INSTANCE="node-1",
|
||||||
|
):
|
||||||
|
config = load_config()
|
||||||
|
|
||||||
|
self.assertEqual(config.environment, "prod")
|
||||||
|
self.assertEqual(config.user, "ops")
|
||||||
|
self.assertEqual(config.instance, "node-1")
|
||||||
|
mock_mkdir.assert_called()
|
||||||
|
|
||||||
|
@patch("app.config.Path.mkdir")
|
||||||
|
def test_metadata_falls_back_to_hostname(self, mock_mkdir):
|
||||||
|
with temp_env(
|
||||||
|
MASTER_ENDPOINT="http://master.local",
|
||||||
|
AGENT_HOSTNAME="qa-team-abc-pod-2",
|
||||||
|
AGENT_ENV=None,
|
||||||
|
AGENT_USER=None,
|
||||||
|
AGENT_INSTANCE=None,
|
||||||
|
):
|
||||||
|
config = load_config()
|
||||||
|
|
||||||
|
self.assertEqual(config.environment, "qa")
|
||||||
|
self.assertEqual(config.user, "team")
|
||||||
|
self.assertEqual(config.instance, "abc")
|
||||||
|
mock_mkdir.assert_called()
|
||||||
|
|
||||||
|
@patch("app.config._load_metadata_from_state", return_value=("prod", "ops", "node-1"))
|
||||||
|
@patch("app.config.Path.mkdir")
|
||||||
|
def test_metadata_from_node_state(self, mock_mkdir, mock_state):
|
||||||
|
with temp_env(
|
||||||
|
MASTER_ENDPOINT="http://master.local",
|
||||||
|
AGENT_HOSTNAME="host_abc",
|
||||||
|
AGENT_ENV=None,
|
||||||
|
AGENT_USER=None,
|
||||||
|
AGENT_INSTANCE=None,
|
||||||
|
):
|
||||||
|
config = load_config()
|
||||||
|
|
||||||
|
self.assertEqual(config.environment, "prod")
|
||||||
|
self.assertEqual(config.user, "ops")
|
||||||
|
self.assertEqual(config.instance, "node-1")
|
||||||
|
mock_state.assert_called_once()
|
||||||
|
mock_mkdir.assert_called()
|
||||||
|
|
||||||
|
@patch("app.config.Path.mkdir")
|
||||||
|
def test_partial_environment_variables_fallback(self, mock_mkdir):
|
||||||
|
with temp_env(
|
||||||
|
MASTER_ENDPOINT="http://master.local",
|
||||||
|
AGENT_HOSTNAME="stage-ml-001-node",
|
||||||
|
AGENT_ENV="prod",
|
||||||
|
AGENT_USER=None,
|
||||||
|
AGENT_INSTANCE=None,
|
||||||
|
):
|
||||||
|
config = load_config()
|
||||||
|
|
||||||
|
self.assertEqual(config.environment, "stage")
|
||||||
|
self.assertEqual(config.user, "ml")
|
||||||
|
self.assertEqual(config.instance, "001")
|
||||||
|
mock_mkdir.assert_called()
|
||||||
|
|
||||||
|
@patch("app.config.Path.mkdir")
|
||||||
|
def test_invalid_hostname_raises_error(self, mock_mkdir):
|
||||||
|
with temp_env(
|
||||||
|
MASTER_ENDPOINT="http://master.local",
|
||||||
|
AGENT_HOSTNAME="invalidhostname",
|
||||||
|
AGENT_ENV=None,
|
||||||
|
AGENT_USER=None,
|
||||||
|
AGENT_INSTANCE=None,
|
||||||
|
):
|
||||||
|
with self.assertRaises(ValueError):
|
||||||
|
load_config()
|
||||||
|
|
||||||
|
mock_mkdir.assert_not_called()
|
||||||
|
|
||||||
|
|
||||||
|
class CollectMetadataTests(unittest.TestCase):
|
||||||
|
@patch("app.collector._detect_ip_address", return_value="127.0.0.1")
|
||||||
|
@patch("app.collector._detect_gpu_count", return_value=0)
|
||||||
|
@patch("app.collector._detect_memory_bytes", return_value=1024)
|
||||||
|
@patch("app.collector._detect_cpu_count", return_value=8)
|
||||||
|
def test_collect_metadata_uses_config_fields(
|
||||||
|
self,
|
||||||
|
mock_cpu,
|
||||||
|
mock_memory,
|
||||||
|
mock_gpu,
|
||||||
|
mock_ip,
|
||||||
|
):
|
||||||
|
config = AgentConfig(
|
||||||
|
hostname="dev-user-001-pod",
|
||||||
|
environment="prod",
|
||||||
|
user="ops",
|
||||||
|
instance="node-1",
|
||||||
|
node_file="/tmp/node.json",
|
||||||
|
version="1.0.0",
|
||||||
|
master_endpoint="http://master.local",
|
||||||
|
report_interval_seconds=60,
|
||||||
|
health_dir="/tmp/health",
|
||||||
|
)
|
||||||
|
|
||||||
|
from app.collector import collect_metadata
|
||||||
|
|
||||||
|
metadata = collect_metadata(config)
|
||||||
|
|
||||||
|
self.assertEqual(metadata["env"], "prod")
|
||||||
|
self.assertEqual(metadata["user"], "ops")
|
||||||
|
self.assertEqual(metadata["instance"], "node-1")
|
||||||
|
self.assertEqual(metadata["hostname"], "dev-user-001-pod")
|
||||||
|
self.assertEqual(metadata["ip"], "127.0.0.1")
|
||||||
|
self.assertEqual(metadata["cpu_number"], 8)
|
||||||
|
self.assertEqual(metadata["memory_in_bytes"], 1024)
|
||||||
|
self.assertEqual(metadata["gpu_number"], 0)
|
||||||
|
|
||||||
|
|
||||||
|
if __name__ == "__main__":
|
||||||
|
unittest.main()
|
||||||
@@ -9,21 +9,21 @@ RUN apt-get update && \
|
|||||||
apt-get install -y wget supervisor net-tools inetutils-ping vim ca-certificates passwd && \
|
apt-get install -y wget supervisor net-tools inetutils-ping vim ca-certificates passwd && \
|
||||||
apt-get clean && rm -rf /var/lib/apt/lists/*
|
apt-get clean && rm -rf /var/lib/apt/lists/*
|
||||||
|
|
||||||
# 设置 Alertmanager 版本
|
# 设置 Alertmanager 版本(与本地离线包保持一致)
|
||||||
ARG ALERTMANAGER_VERSION=0.28.1
|
ARG ALERTMANAGER_VERSION=0.28.1
|
||||||
|
|
||||||
# 下载并解压 Alertmanager 二进制
|
# 使用仓库内预置的离线包构建(无需联网)
|
||||||
RUN wget https://github.com/prometheus/alertmanager/releases/download/v${ALERTMANAGER_VERSION}/alertmanager-${ALERTMANAGER_VERSION}.linux-amd64.tar.gz && \
|
COPY src/alert/alertmanager/build/alertmanager-${ALERTMANAGER_VERSION}.linux-amd64.tar.gz /tmp/
|
||||||
tar xvf alertmanager-${ALERTMANAGER_VERSION}.linux-amd64.tar.gz && \
|
RUN tar xvf /tmp/alertmanager-${ALERTMANAGER_VERSION}.linux-amd64.tar.gz -C /tmp && \
|
||||||
mv alertmanager-${ALERTMANAGER_VERSION}.linux-amd64 /usr/local/alertmanager && \
|
mv /tmp/alertmanager-${ALERTMANAGER_VERSION}.linux-amd64 /usr/local/alertmanager && \
|
||||||
rm alertmanager-${ALERTMANAGER_VERSION}.linux-amd64.tar.gz
|
rm -f /tmp/alertmanager-${ALERTMANAGER_VERSION}.linux-amd64.tar.gz
|
||||||
|
|
||||||
ENV ALERTMANAGER_BASE_PATH=/private/argus/alert/alertmanager
|
ENV ALERTMANAGER_BASE_PATH=/private/argus/alert/alertmanager
|
||||||
|
|
||||||
ARG ARGUS_UID=2133
|
ARG ARGUS_BUILD_UID=2133
|
||||||
ARG ARGUS_GID=2015
|
ARG ARGUS_BUILD_GID=2015
|
||||||
ENV ARGUS_UID=${ARGUS_UID}
|
ENV ARGUS_BUILD_UID=${ARGUS_BUILD_UID}
|
||||||
ENV ARGUS_GID=${ARGUS_GID}
|
ENV ARGUS_BUILD_GID=${ARGUS_BUILD_GID}
|
||||||
|
|
||||||
RUN mkdir -p /usr/share/alertmanager && \
|
RUN mkdir -p /usr/share/alertmanager && \
|
||||||
mkdir -p ${ALERTMANAGER_BASE_PATH} && \
|
mkdir -p ${ALERTMANAGER_BASE_PATH} && \
|
||||||
@@ -33,16 +33,24 @@ RUN mkdir -p /usr/share/alertmanager && \
|
|||||||
|
|
||||||
# 创建 alertmanager 用户(可自定义 UID/GID)
|
# 创建 alertmanager 用户(可自定义 UID/GID)
|
||||||
# 创建 alertmanager 用户组
|
# 创建 alertmanager 用户组
|
||||||
RUN groupadd -g ${ARGUS_GID} alertmanager
|
RUN set -eux; \
|
||||||
|
# 确保目标 GID 存在;若已被占用,直接使用该 GID(组名不限)\
|
||||||
|
if ! getent group "${ARGUS_BUILD_GID}" >/dev/null; then \
|
||||||
|
groupadd -g "${ARGUS_BUILD_GID}" alertmanager || true; \
|
||||||
|
fi; \
|
||||||
|
# 确保存在 alertmanager 用户;若 UID 已被占用,跳过并继续使用现有 UID 的用户
|
||||||
|
if ! id alertmanager >/dev/null 2>&1; then \
|
||||||
|
if getent passwd "${ARGUS_BUILD_UID}" >/dev/null; then \
|
||||||
|
# UID 已占用,则创建同名用户但不指定 UID(避免冲突),仅保证 user 存在
|
||||||
|
useradd -M -s /usr/sbin/nologin -g "${ARGUS_BUILD_GID}" alertmanager || true; \
|
||||||
|
else \
|
||||||
|
useradd -M -s /usr/sbin/nologin -u "${ARGUS_BUILD_UID}" -g "${ARGUS_BUILD_GID}" alertmanager || true; \
|
||||||
|
fi; \
|
||||||
|
else \
|
||||||
|
usermod -g "${ARGUS_BUILD_GID}" alertmanager || true; \
|
||||||
|
fi
|
||||||
|
|
||||||
# 创建 alertmanager 用户并指定组
|
RUN chown -R "${ARGUS_BUILD_UID}:${ARGUS_BUILD_GID}" /usr/share/alertmanager /alertmanager ${ALERTMANAGER_BASE_PATH} /private/argus/etc /usr/local/bin || true
|
||||||
RUN useradd -M -s /usr/sbin/nologin -u ${ARGUS_UID} -g ${ARGUS_GID} alertmanager
|
|
||||||
|
|
||||||
RUN chown -R alertmanager:alertmanager /usr/share/alertmanager && \
|
|
||||||
chown -R alertmanager:alertmanager /alertmanager && \
|
|
||||||
chown -R alertmanager:alertmanager ${ALERTMANAGER_BASE_PATH} && \
|
|
||||||
chown -R alertmanager:alertmanager /private/argus/etc && \
|
|
||||||
chown -R alertmanager:alertmanager /usr/local/bin
|
|
||||||
|
|
||||||
# 配置内网 apt 源 (如果指定了内网选项)
|
# 配置内网 apt 源 (如果指定了内网选项)
|
||||||
RUN if [ "$USE_INTRANET" = "true" ]; then \
|
RUN if [ "$USE_INTRANET" = "true" ]; then \
|
||||||
@@ -86,4 +94,3 @@ EXPOSE 9093
|
|||||||
|
|
||||||
# 使用 supervisor 作为入口点
|
# 使用 supervisor 作为入口点
|
||||||
CMD ["/usr/bin/supervisord", "-c", "/etc/supervisor/conf.d/supervisord.conf"]
|
CMD ["/usr/bin/supervisord", "-c", "/etc/supervisor/conf.d/supervisord.conf"]
|
||||||
|
|
||||||
|
|||||||
Binary file not shown.
@@ -5,8 +5,8 @@ docker pull ubuntu:24.04
|
|||||||
source src/alert/tests/.env
|
source src/alert/tests/.env
|
||||||
|
|
||||||
docker build \
|
docker build \
|
||||||
--build-arg ARGUS_UID=${ARGUS_UID} \
|
--build-arg ARGUS_BUILD_UID=${ARGUS_BUILD_UID} \
|
||||||
--build-arg ARGUS_GID=${ARGUS_GID} \
|
--build-arg ARGUS_BUILD_GID=${ARGUS_BUILD_GID} \
|
||||||
-f src/alert/alertmanager/build/Dockerfile \
|
-f src/alert/alertmanager/build/Dockerfile \
|
||||||
-t argus-alertmanager:latest .
|
-t argus-alertmanager:latest .
|
||||||
|
|
||||||
|
|||||||
22
src/alert/alertmanager/build/fetch-dist.sh
Normal file
22
src/alert/alertmanager/build/fetch-dist.sh
Normal file
@@ -0,0 +1,22 @@
|
|||||||
|
#!/usr/bin/env bash
|
||||||
|
set -euo pipefail
|
||||||
|
|
||||||
|
# 下载 Alertmanager 离线安装包到本目录,用于 Docker 构建时 COPY
|
||||||
|
# 用法:
|
||||||
|
# ./fetch-dist.sh [version]
|
||||||
|
# 示例:
|
||||||
|
# ./fetch-dist.sh 0.28.1
|
||||||
|
|
||||||
|
VER="${1:-0.28.1}"
|
||||||
|
OUT="alertmanager-${VER}.linux-amd64.tar.gz"
|
||||||
|
URL="https://github.com/prometheus/alertmanager/releases/download/v${VER}/${OUT}"
|
||||||
|
|
||||||
|
if [[ -f "$OUT" ]]; then
|
||||||
|
echo "[INFO] $OUT already exists, skip download"
|
||||||
|
exit 0
|
||||||
|
fi
|
||||||
|
|
||||||
|
echo "[INFO] Downloading $URL"
|
||||||
|
curl -fL --retry 3 --connect-timeout 10 -o "$OUT" "$URL"
|
||||||
|
echo "[OK] Saved to $(pwd)/$OUT"
|
||||||
|
|
||||||
@@ -7,10 +7,8 @@ ALERTMANAGER_BASE_PATH=${ALERTMANAGER_BASE_PATH:-/private/argus/alert/alertmanag
|
|||||||
|
|
||||||
echo "[INFO] Alertmanager base path: ${ALERTMANAGER_BASE_PATH}"
|
echo "[INFO] Alertmanager base path: ${ALERTMANAGER_BASE_PATH}"
|
||||||
|
|
||||||
# 生成配置文件
|
# 使用容器内的 /etc/alertmanager/alertmanager.yml 作为配置文件,避免写入挂载卷导致的权限问题
|
||||||
echo "[INFO] Generating Alertmanager configuration file..."
|
echo "[INFO] Using /etc/alertmanager/alertmanager.yml as configuration"
|
||||||
sed "s|\${ALERTMANAGER_BASE_PATH}|${ALERTMANAGER_BASE_PATH}|g" \
|
|
||||||
/etc/alertmanager/alertmanager.yml > ${ALERTMANAGER_BASE_PATH}/alertmanager.yml
|
|
||||||
|
|
||||||
|
|
||||||
# 记录容器 IP 地址
|
# 记录容器 IP 地址
|
||||||
|
|||||||
@@ -1,5 +1,5 @@
|
|||||||
DATA_ROOT=/home/argus/tmp/private/argus
|
DATA_ROOT=/home/argus/tmp/private/argus
|
||||||
ARGUS_UID=1048
|
ARGUS_BUILD_UID=1048
|
||||||
ARGUS_GID=1048
|
ARGUS_BUILD_GID=1048
|
||||||
|
|
||||||
USE_INTRANET=false
|
USE_INTRANET=false
|
||||||
|
|||||||
19
src/alert/tests/data/alertmanager/alertmanager.yml
Normal file
19
src/alert/tests/data/alertmanager/alertmanager.yml
Normal file
@@ -0,0 +1,19 @@
|
|||||||
|
global:
|
||||||
|
resolve_timeout: 5m
|
||||||
|
|
||||||
|
route:
|
||||||
|
group_by: ['alertname', 'instance'] # 分组:相同 alertname + instance 的告警合并
|
||||||
|
group_wait: 30s # 第一个告警后,等 30s 看是否有同组告警一起发
|
||||||
|
group_interval: 5m # 同组告警变化后,至少 5 分钟再发一次
|
||||||
|
repeat_interval: 3h # 相同告警,3 小时重复提醒一次
|
||||||
|
receiver: 'null'
|
||||||
|
|
||||||
|
receivers:
|
||||||
|
- name: 'null'
|
||||||
|
|
||||||
|
inhibit_rules:
|
||||||
|
- source_match:
|
||||||
|
severity: 'critical' # critical 告警存在时
|
||||||
|
target_match:
|
||||||
|
severity: 'warning' # 抑制相同 instance 的 warning 告警
|
||||||
|
equal: ['instance']
|
||||||
0
src/alert/tests/data/alertmanager/nflog
Normal file
0
src/alert/tests/data/alertmanager/nflog
Normal file
0
src/alert/tests/data/alertmanager/silences
Normal file
0
src/alert/tests/data/alertmanager/silences
Normal file
1
src/alert/tests/data/etc/alertmanager.alert.argus.com
Normal file
1
src/alert/tests/data/etc/alertmanager.alert.argus.com
Normal file
@@ -0,0 +1 @@
|
|||||||
|
172.18.0.2
|
||||||
@@ -4,15 +4,15 @@ services:
|
|||||||
context: ../../../
|
context: ../../../
|
||||||
dockerfile: src/alert/alertmanager/build/Dockerfile
|
dockerfile: src/alert/alertmanager/build/Dockerfile
|
||||||
args:
|
args:
|
||||||
ARGUS_UID: ${ARGUS_UID:-2133}
|
ARGUS_BUILD_UID: ${ARGUS_BUILD_UID:-2133}
|
||||||
ARGUS_GID: ${ARGUS_GID:-2015}
|
ARGUS_BUILD_GID: ${ARGUS_BUILD_GID:-2015}
|
||||||
USE_INTRANET: ${USE_INTRANET:-false}
|
USE_INTRANET: ${USE_INTRANET:-false}
|
||||||
image: argus-alertmanager:latest
|
image: argus-alertmanager:latest
|
||||||
container_name: argus-alertmanager
|
container_name: argus-alertmanager
|
||||||
environment:
|
environment:
|
||||||
- ALERTMANAGER_BASE_PATH=/private/argus/alert/alertmanager
|
- ALERTMANAGER_BASE_PATH=/private/argus/alert/alertmanager
|
||||||
- ARGUS_UID=${ARGUS_UID:-2133}
|
- ARGUS_BUILD_UID=${ARGUS_BUILD_UID:-2133}
|
||||||
- ARGUS_GID=${ARGUS_GID:-2015}
|
- ARGUS_BUILD_GID=${ARGUS_BUILD_GID:-2015}
|
||||||
ports:
|
ports:
|
||||||
- "${ARGUS_PORT:-9093}:9093"
|
- "${ARGUS_PORT:-9093}:9093"
|
||||||
volumes:
|
volumes:
|
||||||
|
|||||||
@@ -26,6 +26,7 @@ RUN apt-get update && \
|
|||||||
apt-get install -y \
|
apt-get install -y \
|
||||||
bind9 \
|
bind9 \
|
||||||
bind9utils \
|
bind9utils \
|
||||||
|
dnsutils \
|
||||||
bind9-doc \
|
bind9-doc \
|
||||||
supervisor \
|
supervisor \
|
||||||
net-tools \
|
net-tools \
|
||||||
|
|||||||
@@ -17,6 +17,9 @@ log_message() {
|
|||||||
|
|
||||||
log_message "DNS监控脚本启动"
|
log_message "DNS监控脚本启动"
|
||||||
|
|
||||||
|
log_message "删除DNS备份文件(如果存在)"
|
||||||
|
rm -f $DNS_BACKUP
|
||||||
|
|
||||||
while true; do
|
while true; do
|
||||||
if [ -f "$DNS_CONF" ]; then
|
if [ -f "$DNS_CONF" ]; then
|
||||||
if [ -f "$DNS_BACKUP" ]; then
|
if [ -f "$DNS_BACKUP" ]; then
|
||||||
|
|||||||
@@ -2,7 +2,7 @@
|
|||||||
set -euo pipefail
|
set -euo pipefail
|
||||||
|
|
||||||
ES_HOST="${ELASTICSEARCH_HOSTS:-http://es:9200}"
|
ES_HOST="${ELASTICSEARCH_HOSTS:-http://es:9200}"
|
||||||
KB_HOST="http://localhost:5601"
|
KB_HOST="${KB_HOST:-http://127.0.0.1:5601}"
|
||||||
|
|
||||||
echo "[INFO] Starting Kibana post-start configuration..."
|
echo "[INFO] Starting Kibana post-start configuration..."
|
||||||
|
|
||||||
@@ -83,50 +83,37 @@ fix_replicas_idempotent() {
|
|||||||
}
|
}
|
||||||
|
|
||||||
# 幂等创建数据视图
|
# 幂等创建数据视图
|
||||||
|
create_or_ensure_data_view() {
|
||||||
|
local name="$1"
|
||||||
|
local title="$2"
|
||||||
|
|
||||||
|
local list_response
|
||||||
|
list_response=$(curl -fsS "$KB_HOST/api/data_views" -H 'kbn-xsrf: true' 2>/dev/null || echo "")
|
||||||
|
|
||||||
|
if [ -z "$list_response" ]; then
|
||||||
|
echo "[WARN] Failed to list data views, skipping creation check for $title"
|
||||||
|
return
|
||||||
|
fi
|
||||||
|
|
||||||
|
if echo "$list_response" | grep -Fq "\"title\":\"$title\""; then
|
||||||
|
echo "[INFO] Data view $title already exists, skipping"
|
||||||
|
return
|
||||||
|
fi
|
||||||
|
|
||||||
|
echo "[INFO] Creating data view for $title indices (allowNoIndex)"
|
||||||
|
|
||||||
|
curl -fsS -X POST "$KB_HOST/api/data_views/data_view?allowNoIndex=true" \
|
||||||
|
-H 'kbn-xsrf: true' \
|
||||||
|
-H 'Content-Type: application/json' \
|
||||||
|
-d "{\"data_view\":{\"name\":\"$name\",\"title\":\"$title\",\"timeFieldName\":\"@timestamp\",\"allowNoIndex\":true}}" \
|
||||||
|
>/dev/null && echo "[OK] Created $name data view" || echo "[WARN] Failed to create $name data view"
|
||||||
|
}
|
||||||
|
|
||||||
create_data_views_idempotent() {
|
create_data_views_idempotent() {
|
||||||
echo "[INFO] Checking and creating data views..."
|
echo "[INFO] Checking and creating data views..."
|
||||||
|
|
||||||
# 检查是否存在匹配的索引
|
create_or_ensure_data_view "train" "train-*"
|
||||||
local train_indices=$(curl -s "$ES_HOST/_cat/indices/train-*?h=index" 2>/dev/null | wc -l || echo "0")
|
create_or_ensure_data_view "infer" "infer-*"
|
||||||
local infer_indices=$(curl -s "$ES_HOST/_cat/indices/infer-*?h=index" 2>/dev/null | wc -l || echo "0")
|
|
||||||
|
|
||||||
# 创建 train 数据视图
|
|
||||||
if [ "$train_indices" -gt 0 ]; then
|
|
||||||
# 检查数据视图是否已存在
|
|
||||||
local train_exists=$(curl -s "$KB_HOST/api/data_views" -H 'kbn-xsrf: true' 2>/dev/null | grep '"title":"train-\*"' | wc -l )
|
|
||||||
|
|
||||||
if [ "$train_exists" -eq 0 ]; then
|
|
||||||
echo "[INFO] Creating data view for train-* indices"
|
|
||||||
curl -fsS -X POST "$KB_HOST/api/data_views/data_view" \
|
|
||||||
-H 'kbn-xsrf: true' \
|
|
||||||
-H 'Content-Type: application/json' \
|
|
||||||
-d '{"data_view":{"name":"train","title":"train-*","timeFieldName":"@timestamp"}}' \
|
|
||||||
>/dev/null && echo "[OK] Created train data view" || echo "[WARN] Failed to create train data view"
|
|
||||||
else
|
|
||||||
echo "[INFO] Train data view already exists, skipping"
|
|
||||||
fi
|
|
||||||
else
|
|
||||||
echo "[INFO] No train-* indices found, skipping train data view creation"
|
|
||||||
fi
|
|
||||||
|
|
||||||
# 创建 infer 数据视图
|
|
||||||
if [ "$infer_indices" -gt 0 ]; then
|
|
||||||
# 检查数据视图是否已存在
|
|
||||||
local infer_exists=$(curl -s "$KB_HOST/api/data_views" -H 'kbn-xsrf: true' 2>/dev/null | grep '"title":"infer-\*"' | wc -l )
|
|
||||||
|
|
||||||
if [ "$infer_exists" -eq 0 ]; then
|
|
||||||
echo "[INFO] Creating data view for infer-* indices"
|
|
||||||
curl -fsS -X POST "$KB_HOST/api/data_views/data_view" \
|
|
||||||
-H 'kbn-xsrf: true' \
|
|
||||||
-H 'Content-Type: application/json' \
|
|
||||||
-d '{"data_view":{"name":"infer","title":"infer-*","timeFieldName":"@timestamp"}}' \
|
|
||||||
>/dev/null && echo "[OK] Created infer data view" || echo "[WARN] Failed to create infer data view"
|
|
||||||
else
|
|
||||||
echo "[INFO] Infer data view already exists, skipping"
|
|
||||||
fi
|
|
||||||
else
|
|
||||||
echo "[INFO] No infer-* indices found, skipping infer data view creation"
|
|
||||||
fi
|
|
||||||
}
|
}
|
||||||
|
|
||||||
# 主逻辑
|
# 主逻辑
|
||||||
|
|||||||
@@ -115,20 +115,32 @@ show_step "Health" "Check service health"
|
|||||||
echo "[INFO] Checking service health..."
|
echo "[INFO] Checking service health..."
|
||||||
|
|
||||||
# 检查 Elasticsearch 健康状态
|
# 检查 Elasticsearch 健康状态
|
||||||
|
health_check_ok=1
|
||||||
es_health=$(curl -s "http://localhost:9200/_cluster/health" | grep -o '"status":"[^"]*"' | cut -d'"' -f4)
|
es_health=$(curl -s "http://localhost:9200/_cluster/health" | grep -o '"status":"[^"]*"' | cut -d'"' -f4)
|
||||||
if [ "$es_health" = "green" ] || [ "$es_health" = "yellow" ]; then
|
if [ "$es_health" = "green" ] || [ "$es_health" = "yellow" ]; then
|
||||||
echo "✅ Elasticsearch health: $es_health"
|
echo "✅ Elasticsearch health: $es_health"
|
||||||
else
|
else
|
||||||
echo "❌ Elasticsearch health: $es_health"
|
echo "❌ Elasticsearch health: $es_health"
|
||||||
|
health_check_ok=0
|
||||||
fi
|
fi
|
||||||
|
|
||||||
# 检查 Kibana 状态
|
# 检查 Kibana 状态
|
||||||
if curl -fs "http://localhost:5601/api/status" >/dev/null 2>&1; then
|
if curl -fs "http://localhost:5601/api/status" >/dev/null 2>&1; then
|
||||||
kb_status="available"
|
kb_status="available"
|
||||||
echo "✅ Kibana status: $kb_status"
|
echo "✅ Kibana status: $kb_status"
|
||||||
|
|
||||||
|
data_views_json=$(curl -fs "http://localhost:5601/api/data_views" -H 'kbn-xsrf: true' 2>/dev/null || true)
|
||||||
|
if echo "$data_views_json" | grep -F '"title":"train-*"' >/dev/null 2>&1 && \
|
||||||
|
echo "$data_views_json" | grep -F '"title":"infer-*"' >/dev/null 2>&1; then
|
||||||
|
echo "✅ Kibana data views: train-* and infer-* present"
|
||||||
|
else
|
||||||
|
echo "❌ Kibana data views missing: train-* or infer-*"
|
||||||
|
health_check_ok=0
|
||||||
|
fi
|
||||||
else
|
else
|
||||||
kb_status="unavailable"
|
kb_status="unavailable"
|
||||||
echo "⚠️ Kibana status: $kb_status"
|
echo "⚠️ Kibana status: $kb_status"
|
||||||
|
health_check_ok=0
|
||||||
fi
|
fi
|
||||||
|
|
||||||
# 检查 Fluent-Bit 指标
|
# 检查 Fluent-Bit 指标
|
||||||
@@ -139,6 +151,13 @@ if [ "$fb_host01_uptime" -gt 0 ] && [ "$fb_host02_uptime" -gt 0 ]; then
|
|||||||
echo "✅ Fluent-Bit services: host01 uptime=${fb_host01_uptime}s, host02 uptime=${fb_host02_uptime}s"
|
echo "✅ Fluent-Bit services: host01 uptime=${fb_host01_uptime}s, host02 uptime=${fb_host02_uptime}s"
|
||||||
else
|
else
|
||||||
echo "⚠️ Fluent-Bit services: host01 uptime=${fb_host01_uptime}s, host02 uptime=${fb_host02_uptime}s"
|
echo "⚠️ Fluent-Bit services: host01 uptime=${fb_host01_uptime}s, host02 uptime=${fb_host02_uptime}s"
|
||||||
|
health_check_ok=0
|
||||||
|
fi
|
||||||
|
|
||||||
|
if [ "$health_check_ok" -eq 1 ]; then
|
||||||
|
true
|
||||||
|
else
|
||||||
|
false
|
||||||
fi
|
fi
|
||||||
|
|
||||||
verify_step "Service health check"
|
verify_step "Service health check"
|
||||||
|
|||||||
@@ -3,12 +3,13 @@ set -euo pipefail
|
|||||||
|
|
||||||
usage() {
|
usage() {
|
||||||
cat >&2 <<'USAGE'
|
cat >&2 <<'USAGE'
|
||||||
Usage: $0 [--intranet] [--offline] [--tag <image_tag>]
|
Usage: $0 [--intranet] [--offline] [--tag <image_tag>] [--no-cache]
|
||||||
|
|
||||||
Options:
|
Options:
|
||||||
--intranet 使用指定的 PyPI 镜像源(默认清华镜像)。
|
--intranet 使用指定的 PyPI 镜像源(默认清华镜像)。
|
||||||
--offline 完全离线构建,依赖 offline_wheels/ 目录中的离线依赖包。
|
--offline 完全离线构建,依赖 offline_wheels/ 目录中的离线依赖包。
|
||||||
--tag <image_tag> 自定义镜像标签,默认 argus-master:latest。
|
--tag <image_tag> 自定义镜像标签,默认 argus-master:latest。
|
||||||
|
--no-cache 不使用 Docker 构建缓存。
|
||||||
USAGE
|
USAGE
|
||||||
}
|
}
|
||||||
|
|
||||||
@@ -19,6 +20,7 @@ IMAGE_TAG="${IMAGE_TAG:-argus-master:latest}"
|
|||||||
DOCKERFILE="src/master/Dockerfile"
|
DOCKERFILE="src/master/Dockerfile"
|
||||||
BUILD_ARGS=()
|
BUILD_ARGS=()
|
||||||
OFFLINE_MODE=0
|
OFFLINE_MODE=0
|
||||||
|
NO_CACHE=0
|
||||||
|
|
||||||
source "$PROJECT_ROOT/scripts/common/build_user.sh"
|
source "$PROJECT_ROOT/scripts/common/build_user.sh"
|
||||||
load_build_user
|
load_build_user
|
||||||
@@ -45,6 +47,11 @@ while [[ "$#" -gt 0 ]]; do
|
|||||||
IMAGE_TAG="$2"
|
IMAGE_TAG="$2"
|
||||||
shift 2
|
shift 2
|
||||||
;;
|
;;
|
||||||
|
--no-cache)
|
||||||
|
NO_CACHE=1
|
||||||
|
BUILD_ARGS+=("--no-cache")
|
||||||
|
shift
|
||||||
|
;;
|
||||||
-h|--help)
|
-h|--help)
|
||||||
usage
|
usage
|
||||||
exit 0
|
exit 0
|
||||||
|
|||||||
@@ -4,6 +4,15 @@
|
|||||||
|
|
||||||
set -e
|
set -e
|
||||||
|
|
||||||
|
# PID 文件检测,防止重复执行
|
||||||
|
PIDFILE="/var/run/check_health.pid"
|
||||||
|
if [ -f "$PIDFILE" ] && kill -0 $(cat "$PIDFILE") 2>/dev/null; then
|
||||||
|
echo "健康检查脚本已在运行中,跳过本次执行" >&2
|
||||||
|
exit 0
|
||||||
|
fi
|
||||||
|
echo $$ > "$PIDFILE"
|
||||||
|
trap "rm -f $PIDFILE" EXIT
|
||||||
|
|
||||||
# 获取脚本所在目录
|
# 获取脚本所在目录
|
||||||
SCRIPT_DIR="$(cd "$(dirname "${BASH_SOURCE[0]}")" && pwd)"
|
SCRIPT_DIR="$(cd "$(dirname "${BASH_SOURCE[0]}")" && pwd)"
|
||||||
HEALTH_LOG_FILE="$SCRIPT_DIR/.health_log"
|
HEALTH_LOG_FILE="$SCRIPT_DIR/.health_log"
|
||||||
|
|||||||
@@ -200,22 +200,22 @@ parse_version_info() {
|
|||||||
VERSION=$(grep '"version"' "$VERSION_FILE_PATH" | sed 's/.*"version": *"\([^"]*\)".*/\1/')
|
VERSION=$(grep '"version"' "$VERSION_FILE_PATH" | sed 's/.*"version": *"\([^"]*\)".*/\1/')
|
||||||
BUILD_TIME=$(grep '"build_time"' "$VERSION_FILE_PATH" | sed 's/.*"build_time": *"\([^"]*\)".*/\1/')
|
BUILD_TIME=$(grep '"build_time"' "$VERSION_FILE_PATH" | sed 's/.*"build_time": *"\([^"]*\)".*/\1/')
|
||||||
|
|
||||||
# 解析 artifact_list
|
# 解析 artifact_list(跳过字段名本身)
|
||||||
grep -A 100 '"artifact_list"' "$VERSION_FILE_PATH" | grep -E '^\s*"[^"]+":\s*"[^"]+"' | while read line; do
|
grep -A 100 '"artifact_list"' "$VERSION_FILE_PATH" | grep -v '"artifact_list"' | grep -E '^\s*"[^"]+":\s*"[^"]+"' | while read line; do
|
||||||
component=$(echo "$line" | sed 's/.*"\([^"]*\)":\s*"[^"]*".*/\1/')
|
component=$(echo "$line" | sed 's/.*"\([^"]*\)":\s*"[^"]*".*/\1/')
|
||||||
version=$(echo "$line" | sed 's/.*"[^"]*":\s*"\([^"]*\)".*/\1/')
|
version=$(echo "$line" | sed 's/.*"[^"]*":\s*"\([^"]*\)".*/\1/')
|
||||||
echo "$component:$version" >> "$TEMP_DIR/components.txt"
|
echo "$component:$version" >> "$TEMP_DIR/components.txt"
|
||||||
done
|
done
|
||||||
|
|
||||||
# 解析 checksums
|
# 解析 checksums(跳过字段名本身)
|
||||||
grep -A 100 '"checksums"' "$VERSION_FILE_PATH" | grep -E '^\s*"[^"]+":\s*"[^"]+"' | while read line; do
|
grep -A 100 '"checksums"' "$VERSION_FILE_PATH" | grep -v '"checksums"' | grep -E '^\s*"[^"]+":\s*"[^"]+"' | while read line; do
|
||||||
component=$(echo "$line" | sed 's/.*"\([^"]*\)":\s*"[^"]*".*/\1/')
|
component=$(echo "$line" | sed 's/.*"\([^"]*\)":\s*"[^"]*".*/\1/')
|
||||||
checksum=$(echo "$line" | sed 's/.*"[^"]*":\s*"\([^"]*\)".*/\1/')
|
checksum=$(echo "$line" | sed 's/.*"[^"]*":\s*"\([^"]*\)".*/\1/')
|
||||||
echo "$component:$checksum" >> "$TEMP_DIR/checksums.txt"
|
echo "$component:$checksum" >> "$TEMP_DIR/checksums.txt"
|
||||||
done
|
done
|
||||||
|
|
||||||
# 解析 install_order
|
# 解析 install_order(跳过字段名本身,只取数组元素)
|
||||||
grep -A 100 '"install_order"' "$VERSION_FILE_PATH" | grep -E '^\s*"[^"]+"' | while read line; do
|
grep -A 100 '"install_order"' "$VERSION_FILE_PATH" | grep -v '"install_order"' | grep -E '^\s*"[^"]+"' | while read line; do
|
||||||
component=$(echo "$line" | sed 's/.*"\([^"]*\)".*/\1/')
|
component=$(echo "$line" | sed 's/.*"\([^"]*\)".*/\1/')
|
||||||
echo "$component" >> "$TEMP_DIR/install_order.txt"
|
echo "$component" >> "$TEMP_DIR/install_order.txt"
|
||||||
done
|
done
|
||||||
@@ -317,6 +317,30 @@ create_install_dirs() {
|
|||||||
log_success "安装目录创建完成: $INSTALL_DIR"
|
log_success "安装目录创建完成: $INSTALL_DIR"
|
||||||
}
|
}
|
||||||
|
|
||||||
|
# 获取系统版本
|
||||||
|
get_system_version() {
|
||||||
|
if [[ ! -f /etc/os-release ]]; then
|
||||||
|
log_error "无法检测操作系统版本"
|
||||||
|
return 1
|
||||||
|
fi
|
||||||
|
|
||||||
|
source /etc/os-release
|
||||||
|
|
||||||
|
# 提取主版本号
|
||||||
|
case "$VERSION_ID" in
|
||||||
|
"20.04")
|
||||||
|
echo "ubuntu20"
|
||||||
|
;;
|
||||||
|
"22.04")
|
||||||
|
echo "ubuntu22"
|
||||||
|
;;
|
||||||
|
*)
|
||||||
|
log_warning "未识别的Ubuntu版本: $VERSION_ID,尝试使用ubuntu22"
|
||||||
|
echo "ubuntu22"
|
||||||
|
;;
|
||||||
|
esac
|
||||||
|
}
|
||||||
|
|
||||||
# 安装系统依赖包
|
# 安装系统依赖包
|
||||||
install_system_deps() {
|
install_system_deps() {
|
||||||
log_info "检查系统依赖包..."
|
log_info "检查系统依赖包..."
|
||||||
@@ -330,21 +354,43 @@ install_system_deps() {
|
|||||||
return 0
|
return 0
|
||||||
fi
|
fi
|
||||||
|
|
||||||
# 检查是否有tar.gz文件
|
# 获取系统版本对应的依赖目录
|
||||||
|
local system_version=$(get_system_version)
|
||||||
|
local version_deps_dir="$deps_dir/$system_version"
|
||||||
|
|
||||||
|
log_info "检测到系统版本: $system_version"
|
||||||
|
|
||||||
|
# 检查版本特定的依赖目录是否存在
|
||||||
|
if [[ ! -d "$version_deps_dir" ]]; then
|
||||||
|
log_warning "未找到 $system_version 版本的依赖目录: $version_deps_dir"
|
||||||
|
# 回退到旧的逻辑,检查根deps目录
|
||||||
local deps_count=$(find "$deps_dir" -name "*.tar.gz" | wc -l)
|
local deps_count=$(find "$deps_dir" -name "*.tar.gz" | wc -l)
|
||||||
if [[ $deps_count -eq 0 ]]; then
|
if [[ $deps_count -eq 0 ]]; then
|
||||||
log_info "deps 目录中没有 tar.gz 文件,跳过系统依赖包安装"
|
log_info "deps 目录中没有 tar.gz 文件,跳过系统依赖包安装"
|
||||||
return 0
|
return 0
|
||||||
fi
|
fi
|
||||||
|
version_deps_dir="$deps_dir"
|
||||||
|
else
|
||||||
|
# 检查版本目录中是否有tar.gz文件
|
||||||
|
local deps_count=$(find "$version_deps_dir" -name "*.tar.gz" | wc -l)
|
||||||
|
if [[ $deps_count -eq 0 ]]; then
|
||||||
|
log_info "$system_version 版本目录中没有 tar.gz 文件,跳过系统依赖包安装"
|
||||||
|
return 0
|
||||||
|
fi
|
||||||
|
fi
|
||||||
|
|
||||||
log_info "找到 $deps_count 个系统依赖包,开始安装..."
|
log_info "找到 $system_version 版本的依赖包,开始安装..."
|
||||||
|
|
||||||
# 创建临时目录用于解压依赖包
|
# 创建临时目录用于解压依赖包
|
||||||
local deps_temp_dir="$TEMP_DIR/deps"
|
local deps_temp_dir="${TEMP_DIR:-/tmp}/deps"
|
||||||
mkdir -p "$deps_temp_dir"
|
mkdir -p "$deps_temp_dir"
|
||||||
|
|
||||||
|
# 定义要检查的核心依赖
|
||||||
|
local CORE_DEPS=(jq cron curl)
|
||||||
|
local FAILED_DEPS=()
|
||||||
|
|
||||||
# 处理每个tar.gz文件
|
# 处理每个tar.gz文件
|
||||||
find "$deps_dir" -name "*.tar.gz" | while read tar_file; do
|
find "$version_deps_dir" -name "*.tar.gz" | while read tar_file; do
|
||||||
local tar_basename=$(basename "$tar_file")
|
local tar_basename=$(basename "$tar_file")
|
||||||
local extract_name="${tar_basename%.tar.gz}"
|
local extract_name="${tar_basename%.tar.gz}"
|
||||||
|
|
||||||
@@ -362,40 +408,61 @@ install_system_deps() {
|
|||||||
fi
|
fi
|
||||||
|
|
||||||
# 进入解压目录,查找deb包
|
# 进入解压目录,查找deb包
|
||||||
cd "$extract_dir"
|
cd "$extract_dir" || continue
|
||||||
local deb_count=$(find . -name "*.deb" | wc -l)
|
local deb_files=(*.deb)
|
||||||
|
if [[ ${#deb_files[@]} -gt 0 ]]; then
|
||||||
|
log_info " 找到 ${#deb_files[@]} 个 deb 包,开始安装..."
|
||||||
|
|
||||||
if [[ $deb_count -gt 0 ]]; then
|
for deb in "${deb_files[@]}"; do
|
||||||
log_info " 找到 $deb_count 个 deb 包,开始安装..."
|
local pkg_name
|
||||||
|
pkg_name=$(dpkg-deb -f "$deb" Package 2>/dev/null)
|
||||||
|
|
||||||
# 1. 先尝试安装所有deb包
|
# 如果已安装,则跳过
|
||||||
log_info " 第1步:批量安装deb包..."
|
if dpkg -s "$pkg_name" &>/dev/null; then
|
||||||
if dpkg -i *.deb 2>/dev/null; then
|
log_success " $pkg_name 已安装,跳过"
|
||||||
log_success " 所有deb包安装成功"
|
continue
|
||||||
|
fi
|
||||||
|
|
||||||
|
# 尝试安装
|
||||||
|
log_info " 安装 $pkg_name..."
|
||||||
|
if DEBIAN_FRONTEND=noninteractive dpkg -i "$deb" &>/dev/null; then
|
||||||
|
log_success " $pkg_name 安装成功"
|
||||||
else
|
else
|
||||||
log_warning " 部分deb包安装失败,可能存在依赖问题"
|
log_warning " $pkg_name 安装失败,尝试修复依赖..."
|
||||||
|
if DEBIAN_FRONTEND=noninteractive apt-get install -f -y &>/dev/null; then
|
||||||
# 2. 使用apt-get修复依赖
|
if dpkg -s "$pkg_name" &>/dev/null; then
|
||||||
log_info " 第2步:修复依赖关系..."
|
log_success " $pkg_name 修复安装成功"
|
||||||
if apt-get install -f -y; then
|
|
||||||
log_success " 依赖关系修复完成"
|
|
||||||
else
|
else
|
||||||
log_error " 依赖关系修复失败"
|
log_error " $pkg_name 仍未安装成功"
|
||||||
# 继续处理其他包,不退出
|
FAILED_DEPS+=("$pkg_name")
|
||||||
|
fi
|
||||||
|
else
|
||||||
|
log_error " $pkg_name 自动修复失败"
|
||||||
|
FAILED_DEPS+=("$pkg_name")
|
||||||
fi
|
fi
|
||||||
fi
|
fi
|
||||||
|
done
|
||||||
else
|
else
|
||||||
log_info " $tar_basename 中没有找到deb包,跳过"
|
log_info " $tar_basename 中没有找到deb包,跳过"
|
||||||
fi
|
fi
|
||||||
|
|
||||||
# 返回到依赖临时目录
|
# 返回到依赖临时目录
|
||||||
cd "$deps_temp_dir"
|
cd "$deps_temp_dir" || continue
|
||||||
done
|
done
|
||||||
|
|
||||||
# 检查并启动 cron 服务
|
# 检查并启动 cron 服务
|
||||||
start_cron_service
|
start_cron_service
|
||||||
|
|
||||||
log_success "系统依赖包安装完成"
|
# 总结安装结果
|
||||||
|
if [[ ${#FAILED_DEPS[@]} -gt 0 ]]; then
|
||||||
|
log_error "以下系统依赖未能成功安装,安装终止,请手动安装后重试:"
|
||||||
|
for f in "${FAILED_DEPS[@]}"; do
|
||||||
|
echo " - $f"
|
||||||
|
done
|
||||||
|
exit 1
|
||||||
|
else
|
||||||
|
log_success "系统依赖包安装完成,全部就绪"
|
||||||
|
fi
|
||||||
}
|
}
|
||||||
|
|
||||||
# 启动 cron 服务
|
# 启动 cron 服务
|
||||||
@@ -637,6 +704,18 @@ EOF
|
|||||||
log_success "安装记录已创建: $install_record_file"
|
log_success "安装记录已创建: $install_record_file"
|
||||||
}
|
}
|
||||||
|
|
||||||
|
# 检查cron任务是否已存在
|
||||||
|
check_cron_task_exists() {
|
||||||
|
local task_pattern="$1"
|
||||||
|
local temp_cron="$2"
|
||||||
|
|
||||||
|
if grep -q "$task_pattern" "$temp_cron"; then
|
||||||
|
return 0 # 任务已存在
|
||||||
|
else
|
||||||
|
return 1 # 任务不存在
|
||||||
|
fi
|
||||||
|
}
|
||||||
|
|
||||||
# 设置健康检查定时任务
|
# 设置健康检查定时任务
|
||||||
setup_health_check_cron() {
|
setup_health_check_cron() {
|
||||||
log_info "设置健康检查定时任务..."
|
log_info "设置健康检查定时任务..."
|
||||||
@@ -661,7 +740,7 @@ setup_health_check_cron() {
|
|||||||
crontab -l 2>/dev/null > "$temp_cron" || touch "$temp_cron"
|
crontab -l 2>/dev/null > "$temp_cron" || touch "$temp_cron"
|
||||||
|
|
||||||
# 检查并删除旧的健康检查任务
|
# 检查并删除旧的健康检查任务
|
||||||
if grep -q "check_health.sh" "$temp_cron"; then
|
if check_cron_task_exists "check_health.sh" "$temp_cron"; then
|
||||||
log_info "发现旧的健康检查定时任务,正在更新..."
|
log_info "发现旧的健康检查定时任务,正在更新..."
|
||||||
# 删除所有包含check_health.sh的行
|
# 删除所有包含check_health.sh的行
|
||||||
grep -v "check_health.sh" "$temp_cron" > "$temp_cron.new"
|
grep -v "check_health.sh" "$temp_cron" > "$temp_cron.new"
|
||||||
@@ -716,7 +795,7 @@ setup_dns_sync_cron() {
|
|||||||
crontab -l 2>/dev/null > "$temp_cron" || touch "$temp_cron"
|
crontab -l 2>/dev/null > "$temp_cron" || touch "$temp_cron"
|
||||||
|
|
||||||
# 检查并删除旧的 DNS 同步任务
|
# 检查并删除旧的 DNS 同步任务
|
||||||
if grep -q "sync_dns.sh" "$temp_cron"; then
|
if check_cron_task_exists "sync_dns.sh" "$temp_cron"; then
|
||||||
log_info "发现旧的 DNS 同步定时任务,正在更新..."
|
log_info "发现旧的 DNS 同步定时任务,正在更新..."
|
||||||
# 删除所有包含sync_dns.sh的行
|
# 删除所有包含sync_dns.sh的行
|
||||||
grep -v "sync_dns.sh" "$temp_cron" > "$temp_cron.new"
|
grep -v "sync_dns.sh" "$temp_cron" > "$temp_cron.new"
|
||||||
@@ -724,16 +803,15 @@ setup_dns_sync_cron() {
|
|||||||
log_info "旧的 DNS 同步定时任务已删除"
|
log_info "旧的 DNS 同步定时任务已删除"
|
||||||
fi
|
fi
|
||||||
|
|
||||||
# 添加新的定时任务(每30秒执行一次)
|
# 添加新的定时任务(每1分钟执行一次)
|
||||||
# 直接使用版本目录中的 DNS 同步脚本
|
# 直接使用版本目录中的 DNS 同步脚本
|
||||||
echo "# Argus-Metrics DNS 同步定时任务" >> "$temp_cron"
|
echo "# Argus-Metrics DNS 同步定时任务" >> "$temp_cron"
|
||||||
echo "* * * * * $sync_dns_script >> $INSTALL_DIR/.dns_sync.log 2>&1" >> "$temp_cron"
|
echo "* * * * * $sync_dns_script >> $INSTALL_DIR/.dns_sync.log 2>&1" >> "$temp_cron"
|
||||||
echo "* * * * * sleep 30; $sync_dns_script >> $INSTALL_DIR/.dns_sync.log 2>&1" >> "$temp_cron"
|
|
||||||
|
|
||||||
# 安装新的crontab
|
# 安装新的crontab
|
||||||
if crontab "$temp_cron"; then
|
if crontab "$temp_cron"; then
|
||||||
log_success "DNS 同步定时任务设置成功"
|
log_success "DNS 同步定时任务设置成功"
|
||||||
log_info " 执行频率: 每30秒"
|
log_info " 执行频率: 每1分钟"
|
||||||
log_info " 日志文件: $INSTALL_DIR/.dns_sync.log"
|
log_info " 日志文件: $INSTALL_DIR/.dns_sync.log"
|
||||||
log_info " 查看定时任务: crontab -l"
|
log_info " 查看定时任务: crontab -l"
|
||||||
log_info " 删除定时任务: crontab -e"
|
log_info " 删除定时任务: crontab -e"
|
||||||
@@ -771,7 +849,7 @@ setup_version_check_cron() {
|
|||||||
crontab -l > "$temp_cron" 2>/dev/null || touch "$temp_cron"
|
crontab -l > "$temp_cron" 2>/dev/null || touch "$temp_cron"
|
||||||
|
|
||||||
# 检查是否已存在版本校验定时任务
|
# 检查是否已存在版本校验定时任务
|
||||||
if grep -q "check_version.sh" "$temp_cron"; then
|
if check_cron_task_exists "check_version.sh" "$temp_cron"; then
|
||||||
log_info "发现旧的版本校验定时任务,正在更新..."
|
log_info "发现旧的版本校验定时任务,正在更新..."
|
||||||
# 删除所有包含check_version.sh的行
|
# 删除所有包含check_version.sh的行
|
||||||
grep -v "check_version.sh" "$temp_cron" > "$temp_cron.new"
|
grep -v "check_version.sh" "$temp_cron" > "$temp_cron.new"
|
||||||
@@ -824,7 +902,7 @@ setup_restart_cron() {
|
|||||||
crontab -l > "$temp_cron" 2>/dev/null || touch "$temp_cron"
|
crontab -l > "$temp_cron" 2>/dev/null || touch "$temp_cron"
|
||||||
|
|
||||||
# 检查是否已存在自动重启定时任务
|
# 检查是否已存在自动重启定时任务
|
||||||
if grep -q "restart_unhealthy.sh" "$temp_cron"; then
|
if check_cron_task_exists "restart_unhealthy.sh" "$temp_cron"; then
|
||||||
log_info "发现旧的自动重启定时任务,正在更新..."
|
log_info "发现旧的自动重启定时任务,正在更新..."
|
||||||
# 删除所有包含restart_unhealthy.sh的行
|
# 删除所有包含restart_unhealthy.sh的行
|
||||||
grep -v "restart_unhealthy.sh" "$temp_cron" > "$temp_cron.new"
|
grep -v "restart_unhealthy.sh" "$temp_cron" > "$temp_cron.new"
|
||||||
@@ -885,9 +963,9 @@ main() {
|
|||||||
check_system
|
check_system
|
||||||
find_version_file
|
find_version_file
|
||||||
create_install_dirs
|
create_install_dirs
|
||||||
|
install_system_deps
|
||||||
parse_version_info
|
parse_version_info
|
||||||
verify_checksums
|
verify_checksums
|
||||||
install_system_deps
|
|
||||||
install_components
|
install_components
|
||||||
copy_config_files
|
copy_config_files
|
||||||
create_install_record
|
create_install_record
|
||||||
@@ -895,6 +973,20 @@ main() {
|
|||||||
setup_dns_sync_cron
|
setup_dns_sync_cron
|
||||||
setup_version_check_cron
|
setup_version_check_cron
|
||||||
setup_restart_cron
|
setup_restart_cron
|
||||||
|
|
||||||
|
# 注释掉立即执行健康检查,避免与cron任务重复执行
|
||||||
|
# log_info "立即执行一次健康检查..."
|
||||||
|
# local check_health_script="$INSTALL_DIR/check_health.sh"
|
||||||
|
# if [[ -f "$check_health_script" ]]; then
|
||||||
|
# if "$check_health_script" >> "$INSTALL_DIR/.health_check.log" 2>&1; then
|
||||||
|
# log_success "健康检查执行完成"
|
||||||
|
# else
|
||||||
|
# log_warning "健康检查执行失败,请检查日志: $INSTALL_DIR/.health_check.log"
|
||||||
|
# fi
|
||||||
|
# else
|
||||||
|
# log_warning "健康检查脚本不存在: $check_health_script"
|
||||||
|
# fi
|
||||||
|
|
||||||
show_install_info
|
show_install_info
|
||||||
}
|
}
|
||||||
|
|
||||||
|
|||||||
@@ -29,26 +29,68 @@ log_error() {
|
|||||||
show_help() {
|
show_help() {
|
||||||
echo "Argus-Metric Artifact 发布脚本"
|
echo "Argus-Metric Artifact 发布脚本"
|
||||||
echo
|
echo
|
||||||
echo "用法: $0 <版本号>"
|
echo "用法: $0 <版本号> [选项]"
|
||||||
echo
|
echo
|
||||||
echo "参数:"
|
echo "参数:"
|
||||||
echo " <版本号> 要发布的版本号,对应 artifact 目录中的版本"
|
echo " <版本号> 要发布的版本号,对应 artifact 目录中的版本"
|
||||||
echo
|
echo
|
||||||
|
echo "选项:"
|
||||||
|
echo " --output-dir <路径> 指定输出目录 (默认: /private/argus/ftp/share/)"
|
||||||
|
echo " --owner <uid:gid> 指定文件所有者 (默认: 2133:2015)"
|
||||||
|
echo " -h, --help 显示此帮助信息"
|
||||||
|
echo
|
||||||
echo "示例:"
|
echo "示例:"
|
||||||
echo " $0 1.20.0 # 发布 1.20.0 版本"
|
echo " $0 1.20.0 # 使用默认配置发布"
|
||||||
|
echo " $0 1.20.0 --output-dir /tmp/publish # 指定输出目录"
|
||||||
|
echo " $0 1.20.0 --owner 1000:1000 # 指定文件所有者"
|
||||||
|
echo " $0 1.20.0 --output-dir /srv/ftp --owner root:root # 同时指定两者"
|
||||||
echo
|
echo
|
||||||
}
|
}
|
||||||
|
|
||||||
# 检查参数
|
# 默认配置
|
||||||
if [[ $# -ne 1 ]]; then
|
DEFAULT_PUBLISH_DIR="/private/argus/ftp/share/"
|
||||||
|
DEFAULT_OWNER="2133:2015"
|
||||||
|
|
||||||
|
# 解析参数
|
||||||
|
VERSION=""
|
||||||
|
PUBLISH_DIR="$DEFAULT_PUBLISH_DIR"
|
||||||
|
OWNER="$DEFAULT_OWNER"
|
||||||
|
|
||||||
|
while [[ $# -gt 0 ]]; do
|
||||||
|
case $1 in
|
||||||
|
-h|--help)
|
||||||
|
show_help
|
||||||
|
exit 0
|
||||||
|
;;
|
||||||
|
--output-dir)
|
||||||
|
PUBLISH_DIR="$2"
|
||||||
|
shift 2
|
||||||
|
;;
|
||||||
|
--owner)
|
||||||
|
OWNER="$2"
|
||||||
|
shift 2
|
||||||
|
;;
|
||||||
|
*)
|
||||||
|
if [[ -z "$VERSION" ]]; then
|
||||||
|
VERSION="$1"
|
||||||
|
shift
|
||||||
|
else
|
||||||
|
log_error "未知参数: $1"
|
||||||
|
show_help
|
||||||
|
exit 1
|
||||||
|
fi
|
||||||
|
;;
|
||||||
|
esac
|
||||||
|
done
|
||||||
|
|
||||||
|
# 检查版本号是否提供
|
||||||
|
if [[ -z "$VERSION" ]]; then
|
||||||
log_error "请提供版本号参数"
|
log_error "请提供版本号参数"
|
||||||
show_help
|
show_help
|
||||||
exit 1
|
exit 1
|
||||||
fi
|
fi
|
||||||
|
|
||||||
VERSION="$1"
|
|
||||||
ARTIFACT_DIR="artifact/$VERSION"
|
ARTIFACT_DIR="artifact/$VERSION"
|
||||||
PUBLISH_DIR="/Users/sundapeng/Project/nlp/aiops/client-plugins/all-in-one/publish/"
|
|
||||||
|
|
||||||
# 检查版本目录是否存在
|
# 检查版本目录是否存在
|
||||||
if [[ ! -d "$ARTIFACT_DIR" ]]; then
|
if [[ ! -d "$ARTIFACT_DIR" ]]; then
|
||||||
@@ -57,11 +99,32 @@ if [[ ! -d "$ARTIFACT_DIR" ]]; then
|
|||||||
fi
|
fi
|
||||||
|
|
||||||
log_info "开始发布版本: $VERSION"
|
log_info "开始发布版本: $VERSION"
|
||||||
|
log_info "输出目录: $PUBLISH_DIR"
|
||||||
|
log_info "文件所有者: $OWNER"
|
||||||
|
|
||||||
# 确保发布目录存在
|
# 确保发布目录存在
|
||||||
log_info "确保发布目录存在: $PUBLISH_DIR"
|
log_info "确保发布目录存在: $PUBLISH_DIR"
|
||||||
mkdir -p "$PUBLISH_DIR"
|
mkdir -p "$PUBLISH_DIR"
|
||||||
|
|
||||||
|
IFS=':' read -r OWNER_UID OWNER_GID <<< "$OWNER"
|
||||||
|
if [[ -z "$OWNER_UID" || -z "$OWNER_GID" ]]; then
|
||||||
|
log_error "--owner 格式不正确,应为 uid:gid"
|
||||||
|
exit 1
|
||||||
|
fi
|
||||||
|
|
||||||
|
CURRENT_UID=$(id -u)
|
||||||
|
CURRENT_GID=$(id -g)
|
||||||
|
if [[ "$OWNER_UID" != "$CURRENT_UID" || "$OWNER_GID" != "$CURRENT_GID" ]]; then
|
||||||
|
if [[ "$CURRENT_UID" -ne 0 ]]; then
|
||||||
|
log_error "当前用户 (${CURRENT_UID}:${CURRENT_GID}) 无法设置所有者为 ${OWNER_UID}:${OWNER_GID}"
|
||||||
|
log_error "请以目标用户运行脚本或预先调整目录权限"
|
||||||
|
exit 1
|
||||||
|
fi
|
||||||
|
NEED_CHOWN=true
|
||||||
|
else
|
||||||
|
NEED_CHOWN=false
|
||||||
|
fi
|
||||||
|
|
||||||
# 创建临时目录用于打包
|
# 创建临时目录用于打包
|
||||||
TEMP_PACKAGE_DIR="/tmp/argus-metric-package-$$"
|
TEMP_PACKAGE_DIR="/tmp/argus-metric-package-$$"
|
||||||
mkdir -p "$TEMP_PACKAGE_DIR"
|
mkdir -p "$TEMP_PACKAGE_DIR"
|
||||||
@@ -167,17 +230,28 @@ cd "$TEMP_PACKAGE_DIR"
|
|||||||
tar -czf "$PUBLISH_DIR/$TAR_NAME" *
|
tar -czf "$PUBLISH_DIR/$TAR_NAME" *
|
||||||
cd - > /dev/null
|
cd - > /dev/null
|
||||||
|
|
||||||
|
if [[ "$NEED_CHOWN" == true ]]; then
|
||||||
|
log_info "设置文件所有者为: $OWNER"
|
||||||
|
chown "$OWNER" "$PUBLISH_DIR/$TAR_NAME"
|
||||||
|
fi
|
||||||
|
|
||||||
# 清理临时目录
|
# 清理临时目录
|
||||||
rm -rf "$TEMP_PACKAGE_DIR"
|
rm -rf "$TEMP_PACKAGE_DIR"
|
||||||
|
|
||||||
# 更新 LATEST_VERSION 文件
|
# 更新 LATEST_VERSION 文件
|
||||||
log_info "更新 LATEST_VERSION 文件..."
|
log_info "更新 LATEST_VERSION 文件..."
|
||||||
echo "$VERSION" > "$PUBLISH_DIR/LATEST_VERSION"
|
echo "$VERSION" > "$PUBLISH_DIR/LATEST_VERSION"
|
||||||
|
if [[ "$NEED_CHOWN" == true ]]; then
|
||||||
|
chown "$OWNER" "$PUBLISH_DIR/LATEST_VERSION"
|
||||||
|
fi
|
||||||
|
|
||||||
# 复制 DNS 配置文件到发布目录根目录(直接从 config 目录复制)
|
# 复制 DNS 配置文件到发布目录根目录(直接从 config 目录复制)
|
||||||
if [[ -f "config/dns.conf" ]]; then
|
if [[ -f "config/dns.conf" ]]; then
|
||||||
log_info "复制 DNS 配置文件到发布目录根目录..."
|
log_info "复制 DNS 配置文件到发布目录根目录..."
|
||||||
cp "config/dns.conf" "$PUBLISH_DIR/"
|
cp "config/dns.conf" "$PUBLISH_DIR/"
|
||||||
|
if [[ "$NEED_CHOWN" == true ]]; then
|
||||||
|
chown "$OWNER" "$PUBLISH_DIR/dns.conf"
|
||||||
|
fi
|
||||||
log_success "DNS 配置文件复制完成: $PUBLISH_DIR/dns.conf"
|
log_success "DNS 配置文件复制完成: $PUBLISH_DIR/dns.conf"
|
||||||
else
|
else
|
||||||
log_warning "未找到 config/dns.conf 文件,跳过 DNS 配置文件复制"
|
log_warning "未找到 config/dns.conf 文件,跳过 DNS 配置文件复制"
|
||||||
@@ -187,6 +261,9 @@ fi
|
|||||||
if [[ -f "scripts/setup.sh" ]]; then
|
if [[ -f "scripts/setup.sh" ]]; then
|
||||||
log_info "复制 setup.sh 到发布目录..."
|
log_info "复制 setup.sh 到发布目录..."
|
||||||
cp "scripts/setup.sh" "$PUBLISH_DIR/"
|
cp "scripts/setup.sh" "$PUBLISH_DIR/"
|
||||||
|
if [[ "$NEED_CHOWN" == true ]]; then
|
||||||
|
chown "$OWNER" "$PUBLISH_DIR/setup.sh"
|
||||||
|
fi
|
||||||
fi
|
fi
|
||||||
|
|
||||||
# 显示发布结果
|
# 显示发布结果
|
||||||
|
|||||||
@@ -2,6 +2,15 @@
|
|||||||
|
|
||||||
# 此脚本会检查各组件的健康状态,并重启不健康的组件
|
# 此脚本会检查各组件的健康状态,并重启不健康的组件
|
||||||
|
|
||||||
|
# PID 文件检测,防止重复执行
|
||||||
|
PIDFILE="/var/run/restart_unhealthy.pid"
|
||||||
|
if [ -f "$PIDFILE" ] && kill -0 $(cat "$PIDFILE") 2>/dev/null; then
|
||||||
|
echo "自动重启脚本已在运行中,跳过本次执行" >&2
|
||||||
|
exit 0
|
||||||
|
fi
|
||||||
|
echo $$ > "$PIDFILE"
|
||||||
|
trap "rm -f $PIDFILE" EXIT
|
||||||
|
|
||||||
# 获取脚本所在目录
|
# 获取脚本所在目录
|
||||||
SCRIPT_DIR="$(cd "$(dirname "${BASH_SOURCE[0]}")" && pwd)"
|
SCRIPT_DIR="$(cd "$(dirname "${BASH_SOURCE[0]}")" && pwd)"
|
||||||
INSTALL_RECORD_FILE="$SCRIPT_DIR/.install_record"
|
INSTALL_RECORD_FILE="$SCRIPT_DIR/.install_record"
|
||||||
|
|||||||
@@ -1,244 +1,143 @@
|
|||||||
#!/bin/bash
|
#!/bin/bash
|
||||||
|
|
||||||
# DNS 同步脚本
|
|
||||||
# 比较 FTP 根目录的 dns.conf 和本地的 dns.conf,如果有变化则同步到 /etc/resolv.conf
|
|
||||||
|
|
||||||
set -e
|
set -e
|
||||||
|
|
||||||
# 颜色定义
|
# 颜色
|
||||||
RED='\033[0;31m'
|
RED='\033[0;31m'; GREEN='\033[0;32m'; YELLOW='\033[1;33m'; BLUE='\033[0;34m'; NC='\033[0m'
|
||||||
GREEN='\033[0;32m'
|
|
||||||
YELLOW='\033[1;33m'
|
|
||||||
BLUE='\033[0;34m'
|
|
||||||
NC='\033[0m' # No Color
|
|
||||||
|
|
||||||
# 日志函数 - 输出到 stderr 避免影响函数返回值
|
# 日志函数
|
||||||
log_info() {
|
log_info() { echo -e "${BLUE}[INFO]${NC} $1" >&2; }
|
||||||
echo -e "${BLUE}[INFO]${NC} $1" >&2
|
log_success() { echo -e "${GREEN}[SUCCESS]${NC} $1" >&2; }
|
||||||
}
|
log_warning() { echo -e "${YELLOW}[WARNING]${NC} $1" >&2; }
|
||||||
|
log_error() { echo -e "${RED}[ERROR]${NC} $1" >&2; }
|
||||||
|
|
||||||
log_success() {
|
|
||||||
echo -e "${GREEN}[SUCCESS]${NC} $1" >&2
|
|
||||||
}
|
|
||||||
|
|
||||||
log_warning() {
|
|
||||||
echo -e "${YELLOW}[WARNING]${NC} $1" >&2
|
|
||||||
}
|
|
||||||
|
|
||||||
log_error() {
|
|
||||||
echo -e "${RED}[ERROR]${NC} $1" >&2
|
|
||||||
}
|
|
||||||
|
|
||||||
# 获取脚本所在目录
|
|
||||||
SCRIPT_DIR="$(cd "$(dirname "${BASH_SOURCE[0]}")" && pwd)"
|
SCRIPT_DIR="$(cd "$(dirname "${BASH_SOURCE[0]}")" && pwd)"
|
||||||
LOCAL_DNS_CONF="/opt/argus-metric/dns.conf"
|
LOCAL_DNS_CONF="/opt/argus-metric/dns.conf"
|
||||||
REMOTE_DNS_CONF_URL=""
|
|
||||||
RESOLV_CONF="/etc/resolv.conf"
|
RESOLV_CONF="/etc/resolv.conf"
|
||||||
|
ALT_RESOLV_CONF="/run/resolv.conf"
|
||||||
LOG_FILE="/opt/argus-metric/.dns_sync.log"
|
LOG_FILE="/opt/argus-metric/.dns_sync.log"
|
||||||
|
REMOTE_DNS_CONF_URL=""
|
||||||
|
|
||||||
# 从环境变量或配置文件获取 FTP 服务器信息
|
# 获取 FTP 配置
|
||||||
get_ftp_config() {
|
get_ftp_config() {
|
||||||
# 优先从环境变量获取配置
|
|
||||||
log_info "获取 FTP 配置信息..."
|
log_info "获取 FTP 配置信息..."
|
||||||
|
|
||||||
# 如果环境变量中没有设置,则尝试从配置文件读取
|
|
||||||
if [[ -z "$FTP_SERVER" || -z "$FTP_USER" || -z "$FTP_PASSWORD" ]]; then
|
if [[ -z "$FTP_SERVER" || -z "$FTP_USER" || -z "$FTP_PASSWORD" ]]; then
|
||||||
local config_file="$SCRIPT_DIR/config.env"
|
[[ -f "$SCRIPT_DIR/config.env" ]] && source "$SCRIPT_DIR/config.env"
|
||||||
if [[ -f "$config_file" ]]; then
|
|
||||||
log_info "从配置文件读取 FTP 配置: $config_file"
|
|
||||||
source "$config_file"
|
|
||||||
fi
|
fi
|
||||||
else
|
|
||||||
log_info "使用环境变量中的 FTP 配置"
|
|
||||||
fi
|
|
||||||
|
|
||||||
# 设置默认值(如果环境变量和配置文件都没有设置)
|
|
||||||
FTP_SERVER="${FTP_SERVER:-localhost}"
|
FTP_SERVER="${FTP_SERVER:-localhost}"
|
||||||
FTP_USER="${FTP_USER:-ftpuser}"
|
FTP_USER="${FTP_USER:-ftpuser}"
|
||||||
FTP_PASSWORD="${FTP_PASSWORD:-ZGClab1234!}"
|
FTP_PASSWORD="${FTP_PASSWORD:-ZGClab1234!}"
|
||||||
|
|
||||||
# 构建远程 DNS 配置文件 URL
|
|
||||||
REMOTE_DNS_CONF_URL="ftp://${FTP_USER}:${FTP_PASSWORD}@${FTP_SERVER}/dns.conf"
|
REMOTE_DNS_CONF_URL="ftp://${FTP_USER}:${FTP_PASSWORD}@${FTP_SERVER}/dns.conf"
|
||||||
|
|
||||||
log_info "FTP 配置来源: ${FTP_CONFIG_SOURCE:-环境变量/配置文件}"
|
|
||||||
}
|
}
|
||||||
|
|
||||||
# 下载远程 DNS 配置文件
|
# 下载远程 dns.conf
|
||||||
download_remote_dns_conf() {
|
download_remote_dns_conf() {
|
||||||
local temp_file="/tmp/dns.conf.remote.$$"
|
local tmp="/tmp/dns.remote.$$"
|
||||||
|
|
||||||
log_info "从 FTP 服务器下载 DNS 配置文件..."
|
|
||||||
log_info "远程地址: $REMOTE_DNS_CONF_URL"
|
|
||||||
log_info "FTP 服务器: $FTP_SERVER"
|
|
||||||
log_info "FTP 用户: $FTP_USER"
|
|
||||||
|
|
||||||
# 先测试 FTP 连接
|
|
||||||
log_info "测试 FTP 连接..."
|
log_info "测试 FTP 连接..."
|
||||||
if curl -u "${FTP_USER}:${FTP_PASSWORD}" -sfI "ftp://${FTP_SERVER}/" >/dev/null 2>&1; then
|
if ! curl -u "${FTP_USER}:${FTP_PASSWORD}" -sfI "ftp://${FTP_SERVER}/" >/dev/null; then
|
||||||
log_success "FTP 服务器连接成功"
|
log_error "无法连接到 FTP 服务器: $FTP_SERVER"; return 1
|
||||||
else
|
|
||||||
log_error "无法连接到 FTP 服务器: $FTP_SERVER"
|
|
||||||
log_error "请检查:"
|
|
||||||
log_error " 1. FTP 服务器是否运行"
|
|
||||||
log_error " 2. 网络连接是否正常"
|
|
||||||
log_error " 3. 服务器地址是否正确"
|
|
||||||
return 1
|
|
||||||
fi
|
fi
|
||||||
|
if ! curl -u "${FTP_USER}:${FTP_PASSWORD}" -sf "ftp://${FTP_SERVER}/dns.conf" -o "$tmp" 2>/dev/null; then
|
||||||
# 测试 dns.conf 文件是否存在
|
log_error "下载 dns.conf 失败"; rm -f "$tmp"; return 1
|
||||||
log_info "检查远程 dns.conf 文件是否存在..."
|
|
||||||
if curl -u "${FTP_USER}:${FTP_PASSWORD}" -sfI "ftp://${FTP_SERVER}/dns.conf" >/dev/null 2>&1; then
|
|
||||||
log_success "远程 dns.conf 文件存在"
|
|
||||||
else
|
|
||||||
log_error "远程 dns.conf 文件不存在或无法访问"
|
|
||||||
log_error "请检查 FTP 服务器根目录下是否有 dns.conf 文件"
|
|
||||||
return 1
|
|
||||||
fi
|
|
||||||
|
|
||||||
# 尝试下载文件
|
|
||||||
log_info "开始下载 dns.conf 文件..."
|
|
||||||
if curl -u "${FTP_USER}:${FTP_PASSWORD}" -sf "ftp://${FTP_SERVER}/dns.conf" -o "$temp_file" 2>/dev/null; then
|
|
||||||
log_success "远程 DNS 配置文件下载成功"
|
|
||||||
echo "$temp_file"
|
|
||||||
else
|
|
||||||
log_error "下载 dns.conf 文件失败"
|
|
||||||
log_error "尝试手动测试命令:"
|
|
||||||
log_error " curl -u ${FTP_USER}:${FTP_PASSWORD} ftp://${FTP_SERVER}/dns.conf"
|
|
||||||
rm -f "$temp_file"
|
|
||||||
return 1
|
|
||||||
fi
|
fi
|
||||||
|
echo "$tmp"
|
||||||
}
|
}
|
||||||
|
|
||||||
# 比较两个文件是否相同
|
# 文件比较
|
||||||
compare_files() {
|
compare_files() { diff -q "$1" "$2" >/dev/null 2>&1; }
|
||||||
local file1="$1"
|
|
||||||
local file2="$2"
|
|
||||||
|
|
||||||
if [[ ! -f "$file1" || ! -f "$file2" ]]; then
|
# 从 dns.conf 提取有效 IP
|
||||||
return 1
|
get_dns_ips() {
|
||||||
fi
|
grep -Eo '^[0-9]{1,3}(\.[0-9]{1,3}){3}$' "$1" | sort -u
|
||||||
|
|
||||||
# 使用 diff 比较文件内容
|
|
||||||
if diff -q "$file1" "$file2" >/dev/null 2>&1; then
|
|
||||||
return 0 # 文件相同
|
|
||||||
else
|
|
||||||
return 1 # 文件不同
|
|
||||||
fi
|
|
||||||
}
|
}
|
||||||
|
|
||||||
# 将 DNS 配置追加到 /etc/resolv.conf
|
# 安全更新 resolv.conf(保留符号链接)
|
||||||
update_resolv_conf() {
|
update_resolv_conf() {
|
||||||
local dns_conf_file="$1"
|
local dns_conf="$1"
|
||||||
|
local dns_ips
|
||||||
|
mapfile -t dns_ips < <(get_dns_ips "$dns_conf")
|
||||||
|
[[ ${#dns_ips[@]} -eq 0 ]] && { log_warning "未检测到有效 DNS"; return; }
|
||||||
|
|
||||||
log_info "更新 /etc/resolv.conf 文件..."
|
local target_file="$RESOLV_CONF"
|
||||||
|
if [[ ! -w "$RESOLV_CONF" ]]; then
|
||||||
# 备份原始文件
|
log_warning "/etc/resolv.conf 不可写,使用兜底路径 $ALT_RESOLV_CONF"
|
||||||
if [[ -f "$RESOLV_CONF" ]]; then
|
target_file="$ALT_RESOLV_CONF"
|
||||||
cp "$RESOLV_CONF" "${RESOLV_CONF}.backup.$(date +%Y%m%d_%H%M%S)"
|
|
||||||
log_info "已备份原始 resolv.conf 文件"
|
|
||||||
fi
|
fi
|
||||||
|
|
||||||
# 读取 DNS 配置文件并追加到 resolv.conf
|
local temp="/tmp/resolv.new.$$"
|
||||||
while IFS= read -r line; do
|
cp "$target_file" "${target_file}.backup.$(date +%Y%m%d_%H%M%S)" 2>/dev/null || true
|
||||||
# 跳过空行和注释行
|
log_info "更新 DNS 配置文件: $target_file"
|
||||||
[[ -z "$line" || "$line" =~ ^[[:space:]]*# ]] && continue
|
|
||||||
|
|
||||||
# 验证是否为有效的 IP 地址
|
# 写入新的 nameserver 行
|
||||||
if [[ "$line" =~ ^[0-9]{1,3}\.[0-9]{1,3}\.[0-9]{1,3}\.[0-9]{1,3}$ ]]; then
|
for ip in "${dns_ips[@]}"; do
|
||||||
# 检查是否已存在相同的 nameserver 行
|
echo "nameserver $ip"
|
||||||
if ! grep -q "nameserver $line" "$RESOLV_CONF" 2>/dev/null; then
|
done >"$temp"
|
||||||
echo "nameserver $line" >> "$RESOLV_CONF"
|
|
||||||
log_info "添加 DNS 服务器: $line"
|
# 追加原内容(去掉重复 nameserver)
|
||||||
|
grep -v '^nameserver' "$target_file" >>"$temp" 2>/dev/null || true
|
||||||
|
awk '!a[$0]++' "$temp" >"${temp}.uniq"
|
||||||
|
|
||||||
|
# ⚙️ 使用 cat 原地覆盖,避免 mv 引发 “设备忙”
|
||||||
|
if cat "${temp}.uniq" >"$target_file" 2>/dev/null; then
|
||||||
|
chmod 644 "$target_file"
|
||||||
|
log_success "DNS 更新完成: ${dns_ips[*]}"
|
||||||
else
|
else
|
||||||
log_info "DNS 服务器已存在,跳过: $line"
|
log_error "无法写入 $target_file,可能被系统锁定"
|
||||||
fi
|
fi
|
||||||
else
|
|
||||||
log_warning "跳过无效的 DNS 地址: $line"
|
|
||||||
fi
|
|
||||||
done < "$dns_conf_file"
|
|
||||||
|
|
||||||
# 设置文件权限
|
rm -f "$temp" "${temp}.uniq"
|
||||||
chmod 644 "$RESOLV_CONF"
|
|
||||||
|
|
||||||
log_success "/etc/resolv.conf 文件更新完成"
|
|
||||||
}
|
}
|
||||||
|
|
||||||
# 记录同步日志
|
# 检查 resolv.conf 是否包含 dns.conf 内容
|
||||||
log_sync() {
|
ensure_dns_in_resolv() {
|
||||||
local message="$1"
|
local dns_conf="$1"
|
||||||
local timestamp=$(date '+%Y-%m-%d %H:%M:%S')
|
local dns_ips
|
||||||
echo "[$timestamp] $message" >> "$LOG_FILE"
|
mapfile -t dns_ips < <(get_dns_ips "$dns_conf")
|
||||||
|
[[ ${#dns_ips[@]} -eq 0 ]] && return
|
||||||
|
|
||||||
|
for ip in "${dns_ips[@]}"; do
|
||||||
|
if ! grep -q "nameserver $ip" "$RESOLV_CONF" 2>/dev/null; then
|
||||||
|
log_warning "检测到 /etc/resolv.conf 缺少 $ip,执行兜底修复"
|
||||||
|
update_resolv_conf "$dns_conf"
|
||||||
|
return
|
||||||
|
fi
|
||||||
|
done
|
||||||
|
log_info "/etc/resolv.conf 已包含所有 DNS"
|
||||||
}
|
}
|
||||||
|
|
||||||
# 主函数
|
log_sync() { echo "[$(date '+%F %T')] $1" >>"$LOG_FILE"; }
|
||||||
|
|
||||||
main() {
|
main() {
|
||||||
log_info "开始 DNS 同步检查..."
|
log_info "开始 DNS 同步检查..."
|
||||||
log_sync "DNS 同步检查开始"
|
mkdir -p /opt/argus-metric
|
||||||
|
|
||||||
# 确保系统目录存在
|
|
||||||
mkdir -p "/opt/argus-metric"
|
|
||||||
|
|
||||||
# 获取 FTP 配置
|
|
||||||
get_ftp_config
|
get_ftp_config
|
||||||
|
local remote_file
|
||||||
# 检查本地 DNS 配置文件是否存在
|
if ! remote_file=$(download_remote_dns_conf); then
|
||||||
if [[ ! -f "$LOCAL_DNS_CONF" ]]; then
|
log_error "下载失败"; log_sync "同步失败"; exit 1
|
||||||
log_warning "本地 DNS 配置文件不存在: $LOCAL_DNS_CONF"
|
|
||||||
log_warning "将下载远程配置文件并更新系统 DNS 设置"
|
|
||||||
|
|
||||||
# 下载远程配置文件
|
|
||||||
if remote_file=$(download_remote_dns_conf); then
|
|
||||||
# 复制到本地
|
|
||||||
cp "$remote_file" "$LOCAL_DNS_CONF"
|
|
||||||
log_success "远程 DNS 配置文件已保存到本地"
|
|
||||||
|
|
||||||
# 更新 resolv.conf
|
|
||||||
update_resolv_conf "$LOCAL_DNS_CONF"
|
|
||||||
log_sync "首次同步完成,DNS 配置已更新"
|
|
||||||
|
|
||||||
# 清理临时文件
|
|
||||||
rm -f "$remote_file"
|
|
||||||
else
|
|
||||||
log_error "无法下载远程 DNS 配置文件,同步失败"
|
|
||||||
log_sync "同步失败:无法下载远程配置文件"
|
|
||||||
exit 1
|
|
||||||
fi
|
fi
|
||||||
else
|
|
||||||
log_info "本地 DNS 配置文件存在: $LOCAL_DNS_CONF"
|
|
||||||
|
|
||||||
# 下载远程配置文件进行比较
|
if [[ ! -f "$LOCAL_DNS_CONF" ]]; then
|
||||||
if remote_file=$(download_remote_dns_conf); then
|
log_info "本地 dns.conf 不存在,初始化..."
|
||||||
# 比较文件
|
cp "$remote_file" "$LOCAL_DNS_CONF"
|
||||||
if compare_files "$LOCAL_DNS_CONF" "$remote_file"; then
|
update_resolv_conf "$LOCAL_DNS_CONF"
|
||||||
log_info "DNS 配置文件无变化,无需更新"
|
log_sync "首次同步完成"
|
||||||
log_sync "DNS 配置文件无变化"
|
else
|
||||||
else
|
if compare_files "$LOCAL_DNS_CONF" "$remote_file"; then
|
||||||
log_info "检测到 DNS 配置文件有变化,开始同步..."
|
log_info "dns.conf 无变化"
|
||||||
log_sync "检测到 DNS 配置文件变化,开始同步"
|
ensure_dns_in_resolv "$LOCAL_DNS_CONF"
|
||||||
|
log_sync "dns.conf 无变化,执行兜底检查"
|
||||||
# 更新本地配置文件
|
else
|
||||||
|
log_info "检测到 DNS 配置更新"
|
||||||
cp "$remote_file" "$LOCAL_DNS_CONF"
|
cp "$remote_file" "$LOCAL_DNS_CONF"
|
||||||
log_success "本地 DNS 配置文件已更新"
|
|
||||||
|
|
||||||
# 更新 resolv.conf
|
|
||||||
update_resolv_conf "$LOCAL_DNS_CONF"
|
update_resolv_conf "$LOCAL_DNS_CONF"
|
||||||
log_sync "DNS 配置同步完成"
|
log_sync "DNS 配置同步完成"
|
||||||
fi
|
fi
|
||||||
|
fi
|
||||||
|
|
||||||
# 清理临时文件
|
|
||||||
rm -f "$remote_file"
|
rm -f "$remote_file"
|
||||||
else
|
log_success "DNS 同步流程完成"
|
||||||
log_error "无法下载远程 DNS 配置文件,跳过本次同步"
|
|
||||||
log_sync "同步失败:无法下载远程配置文件"
|
|
||||||
exit 1
|
|
||||||
fi
|
|
||||||
fi
|
|
||||||
|
|
||||||
log_success "DNS 同步检查完成"
|
|
||||||
log_sync "DNS 同步检查完成"
|
|
||||||
}
|
}
|
||||||
|
|
||||||
# 脚本入口
|
|
||||||
if [[ "${BASH_SOURCE[0]}" == "${0}" ]]; then
|
if [[ "${BASH_SOURCE[0]}" == "${0}" ]]; then
|
||||||
main "$@"
|
main "$@"
|
||||||
fi
|
fi
|
||||||
|
|||||||
@@ -31,25 +31,26 @@ RUN mkdir -p /var/log/supervisor
|
|||||||
ENV FTP_BASE_PATH=/private/argus/ftp
|
ENV FTP_BASE_PATH=/private/argus/ftp
|
||||||
|
|
||||||
# 设置域名环境变量
|
# 设置域名环境变量
|
||||||
ENV DOMAIN=prom.ftp.argus.com
|
ENV DOMAIN=ftp.metric.argus.com
|
||||||
|
|
||||||
# 设置FTP用户密码环境变量
|
# 设置FTP用户密码环境变量
|
||||||
ENV FTP_PASSWORD=ZGClab1234!
|
ENV FTP_PASSWORD=ZGClab1234!
|
||||||
|
|
||||||
# 设置用户和组ID环境变量
|
# 设置用户和组ID环境变量
|
||||||
ARG FTP_UID=2133
|
ARG ARGUS_BUILD_UID=2133
|
||||||
ARG FTP_GID=2015
|
ARG ARGUS_BUILD_GID=2015
|
||||||
ENV FTP_UID=${FTP_UID}
|
|
||||||
ENV FTP_GID=${FTP_GID}
|
ENV ARGUS_BUILD_UID=${ARGUS_BUILD_UID} \
|
||||||
|
ARGUS_BUILD_GID=${ARGUS_BUILD_GID}
|
||||||
|
|
||||||
# 创建FTP用户和目录结构
|
# 创建FTP用户和目录结构
|
||||||
RUN groupadd -g ${FTP_GID} ftpuser && \
|
RUN groupadd -g ${ARGUS_BUILD_GID} ftpuser && \
|
||||||
useradd -u ${FTP_UID} -g ${FTP_GID} -d ${FTP_BASE_PATH}/share -s /bin/bash ftpuser && \
|
useradd -u ${ARGUS_BUILD_UID} -g ${ARGUS_BUILD_GID} -d ${FTP_BASE_PATH}/share -s /bin/bash ftpuser && \
|
||||||
mkdir -p ${FTP_BASE_PATH}/share \
|
mkdir -p ${FTP_BASE_PATH}/share \
|
||||||
&& mkdir -p /private/argus/etc \
|
&& mkdir -p /private/argus/etc \
|
||||||
&& mkdir -p /var/log/vsftpd \
|
&& mkdir -p /var/log/vsftpd \
|
||||||
&& mkdir -p /var/run/vsftpd/empty \
|
&& chown -R ftpuser:ftpuser ${FTP_BASE_PATH} \
|
||||||
&& chown -R ftpuser:ftpuser ${FTP_BASE_PATH}
|
&& mkdir -p /var/run/vsftpd/empty
|
||||||
|
|
||||||
# 创建vsftpd配置目录和用户列表文件
|
# 创建vsftpd配置目录和用户列表文件
|
||||||
RUN mkdir -p /etc/vsftpd && \
|
RUN mkdir -p /etc/vsftpd && \
|
||||||
|
|||||||
@@ -32,6 +32,9 @@ IP=$(ifconfig eth0 | awk '/inet /{print $2}' || hostname -i)
|
|||||||
echo "current IP: ${IP}"
|
echo "current IP: ${IP}"
|
||||||
echo "${IP}" > /private/argus/etc/${DOMAIN}
|
echo "${IP}" > /private/argus/etc/${DOMAIN}
|
||||||
|
|
||||||
|
chown ${ARGUS_BUILD_UID}:${ARGUS_BUILD_GID} /private/argus/etc/${DOMAIN}
|
||||||
|
chmod +x /private/argus/etc/${DOMAIN}
|
||||||
|
|
||||||
# 启动vsftpd
|
# 启动vsftpd
|
||||||
echo "[INFO] Starting vsftpd..."
|
echo "[INFO] Starting vsftpd..."
|
||||||
exec /usr/sbin/vsftpd /tmp/vsftpd.conf
|
exec /usr/sbin/vsftpd /tmp/vsftpd.conf
|
||||||
|
|||||||
@@ -2,6 +2,18 @@ FROM grafana/grafana:11.1.0
|
|||||||
|
|
||||||
USER root
|
USER root
|
||||||
|
|
||||||
|
# 构建参数:是否使用内网镜像
|
||||||
|
ARG USE_INTRANET=false
|
||||||
|
|
||||||
|
# 根据是否为内网构建切换 apk 源
|
||||||
|
RUN if [ "$USE_INTRANET" = "true" ]; then \
|
||||||
|
echo "Configuring intranet apk repositories..." && \
|
||||||
|
sed -i 's#https\?://[^/]\+#http://10.68.64.1#g' /etc/apk/repositories; \
|
||||||
|
else \
|
||||||
|
echo "Configuring public apk repositories..." && \
|
||||||
|
sed -i 's#https\?://[^/]\+#https://mirrors.aliyun.com#g' /etc/apk/repositories; \
|
||||||
|
fi
|
||||||
|
|
||||||
# 安装必要的工具
|
# 安装必要的工具
|
||||||
RUN apk add --no-cache \
|
RUN apk add --no-cache \
|
||||||
supervisor \
|
supervisor \
|
||||||
@@ -10,6 +22,11 @@ RUN apk add --no-cache \
|
|||||||
vim \
|
vim \
|
||||||
bash
|
bash
|
||||||
|
|
||||||
|
# 部署镜像时恢复到部署侧使用的内网镜像源
|
||||||
|
RUN if [ "$USE_INTRANET" = "true" ]; then \
|
||||||
|
sed -i 's#https\?://[^/]\+#https://10.92.132.52/mirrors#g' /etc/apk/repositories; \
|
||||||
|
fi
|
||||||
|
|
||||||
# supervisor 日志目录
|
# supervisor 日志目录
|
||||||
RUN mkdir -p /var/log/supervisor
|
RUN mkdir -p /var/log/supervisor
|
||||||
|
|
||||||
@@ -17,36 +34,39 @@ RUN mkdir -p /var/log/supervisor
|
|||||||
ENV GRAFANA_BASE_PATH=/private/argus/metric/grafana
|
ENV GRAFANA_BASE_PATH=/private/argus/metric/grafana
|
||||||
|
|
||||||
# 设置用户和组ID环境变量
|
# 设置用户和组ID环境变量
|
||||||
ARG GRAFANA_UID=2133
|
ARG ARGUS_BUILD_UID=2133
|
||||||
ARG GRAFANA_GID=2015
|
ARG ARGUS_BUILD_GID=2015
|
||||||
ENV GRAFANA_UID=${GRAFANA_UID}
|
|
||||||
ENV GRAFANA_GID=${GRAFANA_GID}
|
ENV ARGUS_BUILD_UID=${ARGUS_BUILD_UID} \
|
||||||
|
ARGUS_BUILD_GID=${ARGUS_BUILD_GID}
|
||||||
|
|
||||||
# 创建基本目录结构
|
# 创建基本目录结构
|
||||||
RUN mkdir -p /private/argus/etc \
|
RUN mkdir -p /private/argus/etc \
|
||||||
&& mkdir -p /private/argus/metric/grafana/data \
|
&& mkdir -p ${GRAFANA_BASE_PATH}/data \
|
||||||
&& mkdir -p /private/argus/metric/grafana/logs \
|
&& mkdir -p ${GRAFANA_BASE_PATH}/logs \
|
||||||
&& mkdir -p /private/argus/metric/grafana/plugins \
|
&& mkdir -p ${GRAFANA_BASE_PATH}/plugins \
|
||||||
&& mkdir -p /private/argus/metric/grafana/provisioning/datasources \
|
&& mkdir -p ${GRAFANA_BASE_PATH}/provisioning/datasources \
|
||||||
&& mkdir -p /private/argus/metric/grafana/provisioning/dashboards \
|
&& mkdir -p ${GRAFANA_BASE_PATH}/provisioning/dashboards \
|
||||||
&& mkdir -p /private/argus/metric/grafana/data/sessions \
|
&& mkdir -p ${GRAFANA_BASE_PATH}/data/sessions \
|
||||||
&& mkdir -p /private/argus/metric/grafana/data/dashboards \
|
&& mkdir -p ${GRAFANA_BASE_PATH}/data/dashboards \
|
||||||
&& mkdir -p /private/argus/metric/grafana/config \
|
&& mkdir -p ${GRAFANA_BASE_PATH}/config \
|
||||||
&& mkdir -p /etc/grafana \
|
&& mkdir -p /etc/grafana \
|
||||||
&& mkdir -p /var/lib/grafana \
|
&& mkdir -p /var/lib/grafana \
|
||||||
&& mkdir -p /var/log/grafana
|
&& mkdir -p /var/log/grafana
|
||||||
|
|
||||||
# 修改 Grafana 用户 UID/GID 并授权
|
# 修改 Grafana 用户 UID/GID 并授权
|
||||||
RUN deluser grafana && \
|
RUN deluser grafana && \
|
||||||
addgroup -g ${GRAFANA_GID} grafana && \
|
addgroup -g ${ARGUS_BUILD_GID} grafana && \
|
||||||
adduser -u ${GRAFANA_UID} -G grafana -s /bin/sh -D grafana && \
|
adduser -u ${ARGUS_BUILD_UID} -G grafana -s /bin/sh -D grafana && \
|
||||||
chown -R grafana:grafana /var/lib/grafana /etc/grafana /var/log/grafana /private/argus
|
chown -R grafana:grafana /var/lib/grafana /etc/grafana /var/log/grafana ${GRAFANA_BASE_PATH}
|
||||||
|
|
||||||
# 复制配置文件到容器内临时位置
|
# 复制配置文件到容器内临时位置
|
||||||
COPY grafana.ini /tmp/grafana.ini
|
COPY grafana.ini /tmp/grafana.ini
|
||||||
COPY datasources/datasources.yml /tmp/datasources.yml
|
COPY datasources/datasources.yml /tmp/datasources.yml
|
||||||
COPY dashboards/dashboards.yml /tmp/dashboards.yml
|
COPY dashboards/dashboards.yml /tmp/dashboards.yml
|
||||||
COPY dashboards/default_dashboard_by_hostname.json /tmp/default_dashboard.json
|
COPY dashboards/default_dashboard_by_hostname.json /tmp/default_dashboard.json
|
||||||
|
COPY dashboards/default_cluster_dashboard.json /tmp/default_cluster_dashboard.json
|
||||||
|
COPY dashboards/default_dashboard_by_instance.json /tmp/default_dashboard_by_instance.json
|
||||||
|
|
||||||
# supervisor 配置
|
# supervisor 配置
|
||||||
COPY supervisord.conf /etc/supervisor/conf.d/supervisord.conf
|
COPY supervisord.conf /etc/supervisor/conf.d/supervisord.conf
|
||||||
|
|||||||
@@ -0,0 +1,570 @@
|
|||||||
|
{
|
||||||
|
"annotations": {
|
||||||
|
"list": [
|
||||||
|
{
|
||||||
|
"builtIn": 1,
|
||||||
|
"datasource": {
|
||||||
|
"type": "grafana",
|
||||||
|
"uid": "-- Grafana --"
|
||||||
|
},
|
||||||
|
"enable": true,
|
||||||
|
"hide": true,
|
||||||
|
"iconColor": "rgba(0, 211, 255, 1)",
|
||||||
|
"name": "Annotations & Alerts",
|
||||||
|
"type": "dashboard"
|
||||||
|
}
|
||||||
|
]
|
||||||
|
},
|
||||||
|
"editable": true,
|
||||||
|
"fiscalYearStartMonth": 0,
|
||||||
|
"graphTooltip": 0,
|
||||||
|
"id": 3,
|
||||||
|
"links": [],
|
||||||
|
"panels": [
|
||||||
|
{
|
||||||
|
"datasource": "prometheus",
|
||||||
|
"fieldConfig": {
|
||||||
|
"defaults": {
|
||||||
|
"mappings": [],
|
||||||
|
"thresholds": {
|
||||||
|
"mode": "absolute",
|
||||||
|
"steps": [
|
||||||
|
{
|
||||||
|
"color": "green"
|
||||||
|
},
|
||||||
|
{
|
||||||
|
"color": "red",
|
||||||
|
"value": 80
|
||||||
|
}
|
||||||
|
]
|
||||||
|
}
|
||||||
|
},
|
||||||
|
"overrides": []
|
||||||
|
},
|
||||||
|
"gridPos": {
|
||||||
|
"h": 5,
|
||||||
|
"w": 8,
|
||||||
|
"x": 0,
|
||||||
|
"y": 0
|
||||||
|
},
|
||||||
|
"id": 1,
|
||||||
|
"options": {
|
||||||
|
"colorMode": "value",
|
||||||
|
"graphMode": "none",
|
||||||
|
"justifyMode": "auto",
|
||||||
|
"orientation": "auto",
|
||||||
|
"percentChangeColorMode": "standard",
|
||||||
|
"reduceOptions": {
|
||||||
|
"calcs": [
|
||||||
|
"mean"
|
||||||
|
],
|
||||||
|
"fields": "",
|
||||||
|
"values": false
|
||||||
|
},
|
||||||
|
"showPercentChange": false,
|
||||||
|
"textMode": "auto",
|
||||||
|
"wideLayout": true
|
||||||
|
},
|
||||||
|
"pluginVersion": "11.1.0",
|
||||||
|
"targets": [
|
||||||
|
{
|
||||||
|
"expr": "100 * (1 - avg(rate(node_cpu_seconds_total{mode='idle'}[5m])))",
|
||||||
|
"refId": "A"
|
||||||
|
}
|
||||||
|
],
|
||||||
|
"title": "CPU 平均利用率(%)",
|
||||||
|
"type": "stat"
|
||||||
|
},
|
||||||
|
{
|
||||||
|
"datasource": "prometheus",
|
||||||
|
"fieldConfig": {
|
||||||
|
"defaults": {
|
||||||
|
"mappings": [],
|
||||||
|
"thresholds": {
|
||||||
|
"mode": "absolute",
|
||||||
|
"steps": [
|
||||||
|
{
|
||||||
|
"color": "green"
|
||||||
|
},
|
||||||
|
{
|
||||||
|
"color": "red",
|
||||||
|
"value": 80
|
||||||
|
}
|
||||||
|
]
|
||||||
|
}
|
||||||
|
},
|
||||||
|
"overrides": []
|
||||||
|
},
|
||||||
|
"gridPos": {
|
||||||
|
"h": 5,
|
||||||
|
"w": 8,
|
||||||
|
"x": 8,
|
||||||
|
"y": 0
|
||||||
|
},
|
||||||
|
"id": 2,
|
||||||
|
"options": {
|
||||||
|
"colorMode": "value",
|
||||||
|
"graphMode": "none",
|
||||||
|
"justifyMode": "auto",
|
||||||
|
"orientation": "auto",
|
||||||
|
"percentChangeColorMode": "standard",
|
||||||
|
"reduceOptions": {
|
||||||
|
"calcs": [
|
||||||
|
"mean"
|
||||||
|
],
|
||||||
|
"fields": "",
|
||||||
|
"values": false
|
||||||
|
},
|
||||||
|
"showPercentChange": false,
|
||||||
|
"textMode": "auto",
|
||||||
|
"wideLayout": true
|
||||||
|
},
|
||||||
|
"pluginVersion": "11.1.0",
|
||||||
|
"targets": [
|
||||||
|
{
|
||||||
|
"expr": "avg(1 - ((node_memory_MemFree_bytes + node_memory_Buffers_bytes + node_memory_Cached_bytes) / node_memory_MemTotal_bytes)) * 100",
|
||||||
|
"refId": "A"
|
||||||
|
}
|
||||||
|
],
|
||||||
|
"title": "内存平均利用率(%)",
|
||||||
|
"type": "stat"
|
||||||
|
},
|
||||||
|
{
|
||||||
|
"datasource": "prometheus",
|
||||||
|
"fieldConfig": {
|
||||||
|
"defaults": {
|
||||||
|
"mappings": [],
|
||||||
|
"thresholds": {
|
||||||
|
"mode": "absolute",
|
||||||
|
"steps": [
|
||||||
|
{
|
||||||
|
"color": "green"
|
||||||
|
},
|
||||||
|
{
|
||||||
|
"color": "red",
|
||||||
|
"value": 80
|
||||||
|
}
|
||||||
|
]
|
||||||
|
}
|
||||||
|
},
|
||||||
|
"overrides": []
|
||||||
|
},
|
||||||
|
"gridPos": {
|
||||||
|
"h": 5,
|
||||||
|
"w": 8,
|
||||||
|
"x": 16,
|
||||||
|
"y": 0
|
||||||
|
},
|
||||||
|
"id": 4,
|
||||||
|
"options": {
|
||||||
|
"colorMode": "value",
|
||||||
|
"graphMode": "none",
|
||||||
|
"justifyMode": "auto",
|
||||||
|
"orientation": "auto",
|
||||||
|
"percentChangeColorMode": "standard",
|
||||||
|
"reduceOptions": {
|
||||||
|
"calcs": [
|
||||||
|
"last"
|
||||||
|
],
|
||||||
|
"fields": "",
|
||||||
|
"values": false
|
||||||
|
},
|
||||||
|
"showPercentChange": false,
|
||||||
|
"textMode": "auto",
|
||||||
|
"wideLayout": true
|
||||||
|
},
|
||||||
|
"pluginVersion": "11.1.0",
|
||||||
|
"targets": [
|
||||||
|
{
|
||||||
|
"expr": "count(count by(hostname) (up{job='node'} == 1))",
|
||||||
|
"refId": "A"
|
||||||
|
}
|
||||||
|
],
|
||||||
|
"title": "节点在线数",
|
||||||
|
"type": "stat"
|
||||||
|
},
|
||||||
|
{
|
||||||
|
"datasource": "prometheus",
|
||||||
|
"fieldConfig": {
|
||||||
|
"defaults": {
|
||||||
|
"mappings": [],
|
||||||
|
"max": 100,
|
||||||
|
"min": 0,
|
||||||
|
"thresholds": {
|
||||||
|
"mode": "absolute",
|
||||||
|
"steps": [
|
||||||
|
{
|
||||||
|
"color": "green"
|
||||||
|
},
|
||||||
|
{
|
||||||
|
"color": "red",
|
||||||
|
"value": 80
|
||||||
|
}
|
||||||
|
]
|
||||||
|
},
|
||||||
|
"unit": "percent"
|
||||||
|
},
|
||||||
|
"overrides": []
|
||||||
|
},
|
||||||
|
"gridPos": {
|
||||||
|
"h": 6,
|
||||||
|
"w": 6,
|
||||||
|
"x": 0,
|
||||||
|
"y": 5
|
||||||
|
},
|
||||||
|
"id": 6,
|
||||||
|
"options": {
|
||||||
|
"minVizHeight": 75,
|
||||||
|
"minVizWidth": 75,
|
||||||
|
"orientation": "auto",
|
||||||
|
"reduceOptions": {
|
||||||
|
"calcs": [
|
||||||
|
"lastNotNull"
|
||||||
|
],
|
||||||
|
"fields": "",
|
||||||
|
"values": false
|
||||||
|
},
|
||||||
|
"showThresholdLabels": false,
|
||||||
|
"showThresholdMarkers": true,
|
||||||
|
"sizing": "auto"
|
||||||
|
},
|
||||||
|
"pluginVersion": "11.1.0",
|
||||||
|
"targets": [
|
||||||
|
{
|
||||||
|
"expr": "avg by (hostname) (DCGM_FI_DEV_GPU_UTIL)",
|
||||||
|
"refId": "A"
|
||||||
|
}
|
||||||
|
],
|
||||||
|
"title": "GPU 平均利用率 (%)",
|
||||||
|
"type": "gauge"
|
||||||
|
},
|
||||||
|
{
|
||||||
|
"datasource": "prometheus",
|
||||||
|
"fieldConfig": {
|
||||||
|
"defaults": {
|
||||||
|
"mappings": [],
|
||||||
|
"max": 100,
|
||||||
|
"min": 0,
|
||||||
|
"thresholds": {
|
||||||
|
"mode": "absolute",
|
||||||
|
"steps": [
|
||||||
|
{
|
||||||
|
"color": "green"
|
||||||
|
},
|
||||||
|
{
|
||||||
|
"color": "red",
|
||||||
|
"value": 80
|
||||||
|
}
|
||||||
|
]
|
||||||
|
},
|
||||||
|
"unit": "percent"
|
||||||
|
},
|
||||||
|
"overrides": []
|
||||||
|
},
|
||||||
|
"gridPos": {
|
||||||
|
"h": 6,
|
||||||
|
"w": 6,
|
||||||
|
"x": 6,
|
||||||
|
"y": 5
|
||||||
|
},
|
||||||
|
"id": 12,
|
||||||
|
"options": {
|
||||||
|
"minVizHeight": 75,
|
||||||
|
"minVizWidth": 75,
|
||||||
|
"orientation": "auto",
|
||||||
|
"reduceOptions": {
|
||||||
|
"calcs": [
|
||||||
|
"lastNotNull"
|
||||||
|
],
|
||||||
|
"fields": "",
|
||||||
|
"values": false
|
||||||
|
},
|
||||||
|
"showThresholdLabels": false,
|
||||||
|
"showThresholdMarkers": true,
|
||||||
|
"sizing": "auto"
|
||||||
|
},
|
||||||
|
"pluginVersion": "11.1.0",
|
||||||
|
"targets": [
|
||||||
|
{
|
||||||
|
"expr": "round(avg(DCGM_FI_DEV_FB_USED{job='dcgm'}/(DCGM_FI_DEV_FB_USED{job='dcgm'} + DCGM_FI_DEV_FB_FREE{job='dcgm'})) * 100)",
|
||||||
|
"refId": "A"
|
||||||
|
}
|
||||||
|
],
|
||||||
|
"title": "显存平均利用率 (%)",
|
||||||
|
"type": "gauge"
|
||||||
|
},
|
||||||
|
{
|
||||||
|
"datasource": "prometheus",
|
||||||
|
"fieldConfig": {
|
||||||
|
"defaults": {
|
||||||
|
"mappings": [],
|
||||||
|
"max": 100,
|
||||||
|
"min": 0,
|
||||||
|
"thresholds": {
|
||||||
|
"mode": "absolute",
|
||||||
|
"steps": [
|
||||||
|
{
|
||||||
|
"color": "green"
|
||||||
|
},
|
||||||
|
{
|
||||||
|
"color": "red",
|
||||||
|
"value": 80
|
||||||
|
}
|
||||||
|
]
|
||||||
|
},
|
||||||
|
"unit": "percent"
|
||||||
|
},
|
||||||
|
"overrides": []
|
||||||
|
},
|
||||||
|
"gridPos": {
|
||||||
|
"h": 6,
|
||||||
|
"w": 6,
|
||||||
|
"x": 12,
|
||||||
|
"y": 5
|
||||||
|
},
|
||||||
|
"id": 7,
|
||||||
|
"options": {
|
||||||
|
"minVizHeight": 75,
|
||||||
|
"minVizWidth": 75,
|
||||||
|
"orientation": "auto",
|
||||||
|
"reduceOptions": {
|
||||||
|
"calcs": [
|
||||||
|
"lastNotNull"
|
||||||
|
],
|
||||||
|
"fields": "",
|
||||||
|
"values": false
|
||||||
|
},
|
||||||
|
"showThresholdLabels": false,
|
||||||
|
"showThresholdMarkers": true,
|
||||||
|
"sizing": "auto"
|
||||||
|
},
|
||||||
|
"pluginVersion": "11.1.0",
|
||||||
|
"targets": [
|
||||||
|
{
|
||||||
|
"expr": "avg by (hostname) (DCGM_FI_DEV_GPU_TEMP)",
|
||||||
|
"refId": "A"
|
||||||
|
}
|
||||||
|
],
|
||||||
|
"title": "GPU 温度 (℃)",
|
||||||
|
"type": "gauge"
|
||||||
|
},
|
||||||
|
{
|
||||||
|
"datasource": "prometheus",
|
||||||
|
"fieldConfig": {
|
||||||
|
"defaults": {
|
||||||
|
"mappings": [],
|
||||||
|
"max": 300,
|
||||||
|
"min": 0,
|
||||||
|
"thresholds": {
|
||||||
|
"mode": "absolute",
|
||||||
|
"steps": [
|
||||||
|
{
|
||||||
|
"color": "green"
|
||||||
|
},
|
||||||
|
{
|
||||||
|
"color": "orange",
|
||||||
|
"value": 200
|
||||||
|
},
|
||||||
|
{
|
||||||
|
"color": "red",
|
||||||
|
"value": 300
|
||||||
|
}
|
||||||
|
]
|
||||||
|
},
|
||||||
|
"unit": "watt"
|
||||||
|
},
|
||||||
|
"overrides": []
|
||||||
|
},
|
||||||
|
"gridPos": {
|
||||||
|
"h": 6,
|
||||||
|
"w": 6,
|
||||||
|
"x": 18,
|
||||||
|
"y": 5
|
||||||
|
},
|
||||||
|
"id": 8,
|
||||||
|
"options": {
|
||||||
|
"minVizHeight": 75,
|
||||||
|
"minVizWidth": 75,
|
||||||
|
"orientation": "auto",
|
||||||
|
"reduceOptions": {
|
||||||
|
"calcs": [
|
||||||
|
"lastNotNull"
|
||||||
|
],
|
||||||
|
"fields": "",
|
||||||
|
"values": false
|
||||||
|
},
|
||||||
|
"showThresholdLabels": true,
|
||||||
|
"showThresholdMarkers": true,
|
||||||
|
"sizing": "auto"
|
||||||
|
},
|
||||||
|
"pluginVersion": "11.1.0",
|
||||||
|
"targets": [
|
||||||
|
{
|
||||||
|
"expr": "avg by (hostname) (DCGM_FI_DEV_POWER_USAGE)",
|
||||||
|
"refId": "A"
|
||||||
|
}
|
||||||
|
],
|
||||||
|
"title": "GPU 平均实时功耗 (W)",
|
||||||
|
"type": "gauge"
|
||||||
|
},
|
||||||
|
{
|
||||||
|
"datasource": "prometheus",
|
||||||
|
"fieldConfig": {
|
||||||
|
"defaults": {
|
||||||
|
"custom": {
|
||||||
|
"align": "center",
|
||||||
|
"cellOptions": {
|
||||||
|
"type": "auto"
|
||||||
|
},
|
||||||
|
"inspect": false
|
||||||
|
},
|
||||||
|
"decimals": 1,
|
||||||
|
"mappings": [],
|
||||||
|
"thresholds": {
|
||||||
|
"mode": "absolute",
|
||||||
|
"steps": [
|
||||||
|
{
|
||||||
|
"color": "green"
|
||||||
|
},
|
||||||
|
{
|
||||||
|
"color": "red",
|
||||||
|
"value": 80
|
||||||
|
}
|
||||||
|
]
|
||||||
|
},
|
||||||
|
"unit": "percent"
|
||||||
|
},
|
||||||
|
"overrides": []
|
||||||
|
},
|
||||||
|
"gridPos": {
|
||||||
|
"h": 12,
|
||||||
|
"w": 24,
|
||||||
|
"x": 0,
|
||||||
|
"y": 11
|
||||||
|
},
|
||||||
|
"id": 11,
|
||||||
|
"options": {
|
||||||
|
"cellHeight": "sm",
|
||||||
|
"cellLinks": [
|
||||||
|
{
|
||||||
|
"title": "跳转至节点详情",
|
||||||
|
"url": "http://127.0.0.1:3000/d/node_gpu_metrics/node-and-gpu-metrics?orgId=1&refresh=15s&var-hostname=${__data.fields.hostname}"
|
||||||
|
}
|
||||||
|
],
|
||||||
|
"footer": {
|
||||||
|
"countRows": false,
|
||||||
|
"fields": "",
|
||||||
|
"reducer": [
|
||||||
|
"sum"
|
||||||
|
],
|
||||||
|
"show": false
|
||||||
|
},
|
||||||
|
"showHeader": true,
|
||||||
|
"sortBy": [
|
||||||
|
{
|
||||||
|
"desc": true,
|
||||||
|
"displayName": "GPU 使用率"
|
||||||
|
}
|
||||||
|
]
|
||||||
|
},
|
||||||
|
"pluginVersion": "11.1.0",
|
||||||
|
"targets": [
|
||||||
|
{
|
||||||
|
"expr": "up{job=\"dcgm\"} + on(hostname) group_left(ip, node_id) up{job=\"dcgm\"}*0",
|
||||||
|
"format": "table",
|
||||||
|
"instant": true,
|
||||||
|
"refId": "node_info"
|
||||||
|
},
|
||||||
|
{
|
||||||
|
"expr": "round(100 - avg by(hostname)(rate(node_cpu_seconds_total{job=\"node\",mode=\"idle\"}[5m])) * 100, 0.1)",
|
||||||
|
"format": "table",
|
||||||
|
"instant": true,
|
||||||
|
"refId": "CPU"
|
||||||
|
},
|
||||||
|
{
|
||||||
|
"expr": "round((1 - avg by(hostname)(node_memory_MemAvailable_bytes{job=\"node\"} / node_memory_MemTotal_bytes{job=\"node\"})) * 100, 0.1)",
|
||||||
|
"format": "table",
|
||||||
|
"instant": true,
|
||||||
|
"refId": "MEM"
|
||||||
|
},
|
||||||
|
{
|
||||||
|
"expr": "round(avg by(hostname)(DCGM_FI_DEV_GPU_UTIL{job=\"dcgm\"}), 0.1)",
|
||||||
|
"format": "table",
|
||||||
|
"instant": true,
|
||||||
|
"refId": "GPU_UTIL"
|
||||||
|
},
|
||||||
|
{
|
||||||
|
"expr": "round(avg by(hostname)(DCGM_FI_DEV_FB_USED{job=\"dcgm\"} / (DCGM_FI_DEV_FB_USED{job=\"dcgm\"} + DCGM_FI_DEV_FB_FREE{job=\"dcgm\"}) * 100), 0.1)",
|
||||||
|
"format": "table",
|
||||||
|
"instant": true,
|
||||||
|
"refId": "GPU_MEM"
|
||||||
|
}
|
||||||
|
],
|
||||||
|
"title": "节点列表(CPU / 内存 / GPU)",
|
||||||
|
"transformations": [
|
||||||
|
{
|
||||||
|
"id": "seriesToColumns",
|
||||||
|
"options": {
|
||||||
|
"byField": "hostname"
|
||||||
|
}
|
||||||
|
},
|
||||||
|
{
|
||||||
|
"id": "organize",
|
||||||
|
"options": {
|
||||||
|
"excludeByName": {
|
||||||
|
"Time": true,
|
||||||
|
"Value #node_info": true,
|
||||||
|
"hostname_1": true,
|
||||||
|
"hostname_2": true,
|
||||||
|
"hostname_3": true,
|
||||||
|
"instance": true,
|
||||||
|
"ip_1": true,
|
||||||
|
"job": true,
|
||||||
|
"node_id_1": true
|
||||||
|
},
|
||||||
|
"indexByName": {
|
||||||
|
"CPU 使用率": 3,
|
||||||
|
"GPU 使用率": 5,
|
||||||
|
"GPU 显存占用": 6,
|
||||||
|
"IP 地址": 1,
|
||||||
|
"主机名": 0,
|
||||||
|
"内存使用率": 4,
|
||||||
|
"节点 ID": 2
|
||||||
|
},
|
||||||
|
"renameByName": {
|
||||||
|
"Value #CPU": "CPU 使用率",
|
||||||
|
"Value #GPU_MEM": "GPU 显存占用",
|
||||||
|
"Value #GPU_UTIL": "GPU 使用率",
|
||||||
|
"Value #MEM": "内存使用率",
|
||||||
|
"hostname": "主机名",
|
||||||
|
"ip": "IP 地址",
|
||||||
|
"node_id": "节点 ID",
|
||||||
|
"user_id": "用户ID"
|
||||||
|
}
|
||||||
|
}
|
||||||
|
}
|
||||||
|
],
|
||||||
|
"type": "table"
|
||||||
|
}
|
||||||
|
],
|
||||||
|
"refresh": "5s",
|
||||||
|
"schemaVersion": 39,
|
||||||
|
"tags": [
|
||||||
|
"cluster",
|
||||||
|
"gpu",
|
||||||
|
"system"
|
||||||
|
],
|
||||||
|
"templating": {
|
||||||
|
"list": []
|
||||||
|
},
|
||||||
|
"time": {
|
||||||
|
"from": "now-1h",
|
||||||
|
"to": "now"
|
||||||
|
},
|
||||||
|
"timepicker": {},
|
||||||
|
"timezone": "browser",
|
||||||
|
"title": "Cluster Dashboard",
|
||||||
|
"uid": "cluster-dashboard",
|
||||||
|
"version": 34,
|
||||||
|
"weekStart": ""
|
||||||
|
}
|
||||||
@@ -8,7 +8,7 @@ datasources:
|
|||||||
type: prometheus
|
type: prometheus
|
||||||
access: proxy
|
access: proxy
|
||||||
uid: eezk1zvkie4g0a
|
uid: eezk1zvkie4g0a
|
||||||
url: http://10.211.55.5:9090
|
url: http://prom.metric.argus.com:9090
|
||||||
isDefault: true
|
isDefault: true
|
||||||
editable: true
|
editable: true
|
||||||
jsonData:
|
jsonData:
|
||||||
|
|||||||
@@ -9,6 +9,7 @@ DOMAIN=grafana.metric.argus.com
|
|||||||
IP=$(ifconfig | awk '/inet / && $2 != "127.0.0.1" {print $2; exit}')
|
IP=$(ifconfig | awk '/inet / && $2 != "127.0.0.1" {print $2; exit}')
|
||||||
echo "current IP: ${IP}"
|
echo "current IP: ${IP}"
|
||||||
echo "${IP}" > /private/argus/etc/${DOMAIN}
|
echo "${IP}" > /private/argus/etc/${DOMAIN}
|
||||||
|
chmod +x /private/argus/etc/${DOMAIN}
|
||||||
|
|
||||||
# 确保必要目录存在(权限已在 Dockerfile 中设置)
|
# 确保必要目录存在(权限已在 Dockerfile 中设置)
|
||||||
mkdir -p /private/argus/metric/grafana/data
|
mkdir -p /private/argus/metric/grafana/data
|
||||||
@@ -27,7 +28,6 @@ mkdir -p /var/lib/grafana
|
|||||||
if [ -f "/tmp/grafana.ini" ]; then
|
if [ -f "/tmp/grafana.ini" ]; then
|
||||||
echo "[INFO] Copying grafana.ini to /private/argus/metric/grafana/config/"
|
echo "[INFO] Copying grafana.ini to /private/argus/metric/grafana/config/"
|
||||||
cp /tmp/grafana.ini /private/argus/metric/grafana/config/grafana.ini
|
cp /tmp/grafana.ini /private/argus/metric/grafana/config/grafana.ini
|
||||||
chown grafana:grafana /private/argus/metric/grafana/config/grafana.ini
|
|
||||||
echo "[INFO] Grafana configuration copied successfully"
|
echo "[INFO] Grafana configuration copied successfully"
|
||||||
fi
|
fi
|
||||||
|
|
||||||
@@ -44,15 +44,18 @@ else
|
|||||||
fi
|
fi
|
||||||
|
|
||||||
# 复制数据源配置文件到挂载目录
|
# 复制数据源配置文件到挂载目录
|
||||||
if [ -f "/tmp/datasources.yml" ]; then
|
DS_OUT="/private/argus/metric/grafana/provisioning/datasources/datasources.yml"
|
||||||
echo "[INFO] Copying datasource configuration to /private/argus/metric/grafana/provisioning/datasources/"
|
PROM_DOMAIN="prom.metric.argus.com:9090"
|
||||||
cp /tmp/datasources.yml /private/argus/metric/grafana/provisioning/datasources/datasources.yml
|
|
||||||
chown grafana:grafana /private/argus/metric/grafana/provisioning/datasources/datasources.yml
|
if [ -f "/tmp/datasources.yml" ] && [ ! -f "$DS_OUT" ]; then
|
||||||
echo "[INFO] Datasource configuration copied successfully"
|
echo "[INFO] Initializing datasource provisioning file from /tmp"
|
||||||
elif [ -d "/private/argus/metric/grafana/provisioning/datasources" ] && [ "$(ls -A /private/argus/metric/grafana/provisioning/datasources)" ]; then
|
cp /tmp/datasources.yml "$DS_OUT"
|
||||||
echo "[INFO] Found existing datasource provisioning files in /private/argus/metric/grafana/provisioning/datasources"
|
fi
|
||||||
# 确保数据源配置目录权限正确
|
|
||||||
chown -R grafana:grafana /private/argus/metric/grafana/provisioning/datasources
|
# 统一将数据源 URL 规范为 prom.metric.argus.com:9090
|
||||||
|
if [ -f "$DS_OUT" ]; then
|
||||||
|
sed -i -E "s#^\s*url:\s*http://[^[:space:]]+# url: http://$PROM_DOMAIN#g" "$DS_OUT" || true
|
||||||
|
echo "[INFO] Datasource URL normalized to http://$PROM_DOMAIN"
|
||||||
elif [ -d "/etc/grafana/provisioning/datasources" ] && [ "$(ls -A /etc/grafana/provisioning/datasources)" ]; then
|
elif [ -d "/etc/grafana/provisioning/datasources" ] && [ "$(ls -A /etc/grafana/provisioning/datasources)" ]; then
|
||||||
echo "[INFO] Found datasource provisioning files in /etc/grafana/provisioning/datasources"
|
echo "[INFO] Found datasource provisioning files in /etc/grafana/provisioning/datasources"
|
||||||
# 确保数据源配置目录权限正确
|
# 确保数据源配置目录权限正确
|
||||||
@@ -65,20 +68,37 @@ fi
|
|||||||
if [ -f "/tmp/dashboards.yml" ]; then
|
if [ -f "/tmp/dashboards.yml" ]; then
|
||||||
echo "[INFO] Copying dashboard configuration to /private/argus/metric/grafana/provisioning/dashboards/"
|
echo "[INFO] Copying dashboard configuration to /private/argus/metric/grafana/provisioning/dashboards/"
|
||||||
cp /tmp/dashboards.yml /private/argus/metric/grafana/provisioning/dashboards/dashboards.yml
|
cp /tmp/dashboards.yml /private/argus/metric/grafana/provisioning/dashboards/dashboards.yml
|
||||||
chown grafana:grafana /private/argus/metric/grafana/provisioning/dashboards/dashboards.yml
|
|
||||||
echo "[INFO] Dashboard configuration copied successfully"
|
echo "[INFO] Dashboard configuration copied successfully"
|
||||||
fi
|
fi
|
||||||
|
|
||||||
# 复制默认仪表板到挂载目录
|
# 复制默认仪表板到挂载目录(按需,不覆盖已存在文件)
|
||||||
if [ -f "/tmp/default_dashboard.json" ]; then
|
copy_dashboard_if_missing() {
|
||||||
echo "[INFO] Copying default dashboard to /private/argus/metric/grafana/provisioning/dashboards/"
|
local src="$1"; local dst_name="$2"
|
||||||
cp /tmp/default_dashboard.json /private/argus/metric/grafana/provisioning/dashboards/default_dashboard.json
|
local dst_dir="/private/argus/metric/grafana/provisioning/dashboards"
|
||||||
chown grafana:grafana /private/argus/metric/grafana/provisioning/dashboards/default_dashboard.json
|
local dst="$dst_dir/$dst_name"
|
||||||
echo "[INFO] Default dashboard copied successfully"
|
if [ -f "$src" ]; then
|
||||||
fi
|
if [ ! -f "$dst" ]; then
|
||||||
|
echo "[INFO] Installing dashboard: $dst_name"
|
||||||
|
cp "$src" "$dst"
|
||||||
|
else
|
||||||
|
echo "[INFO] Dashboard exists, skip: $dst_name"
|
||||||
|
fi
|
||||||
|
fi
|
||||||
|
}
|
||||||
|
|
||||||
# 确保所有配置目录权限正确
|
copy_dashboard_if_missing "/tmp/default_dashboard.json" "default_dashboard.json"
|
||||||
chown -R grafana:grafana /private/argus/metric/grafana/provisioning/
|
copy_dashboard_if_missing "/tmp/default_cluster_dashboard.json" "default_cluster_dashboard.json"
|
||||||
|
copy_dashboard_if_missing "/tmp/default_dashboard_by_instance.json" "default_dashboard_by_instance.json"
|
||||||
|
|
||||||
|
# 规范面板中的数据源字段:将字符串 "prometheus" 替换为 null(使用默认数据源)
|
||||||
|
DB_DIR="/private/argus/metric/grafana/provisioning/dashboards"
|
||||||
|
if [ -d "$DB_DIR" ]; then
|
||||||
|
for f in "$DB_DIR"/*.json; do
|
||||||
|
[ -f "$f" ] || continue
|
||||||
|
sed -i -E 's/"datasource"\s*:\s*"prometheus"/"datasource": null/g' "$f" || true
|
||||||
|
done
|
||||||
|
echo "[INFO] Normalized dashboard datasource to default (null)"
|
||||||
|
fi
|
||||||
|
|
||||||
# 启动 Grafana
|
# 启动 Grafana
|
||||||
if [ -n "$CONFIG_FILE" ]; then
|
if [ -n "$CONFIG_FILE" ]; then
|
||||||
|
|||||||
@@ -48,11 +48,11 @@ RUN mkdir -p /var/log/supervisor
|
|||||||
ENV PROMETHEUS_BASE_PATH=/private/argus/metric/prometheus
|
ENV PROMETHEUS_BASE_PATH=/private/argus/metric/prometheus
|
||||||
|
|
||||||
# 设置用户和组ID环境变量
|
# 设置用户和组ID环境变量
|
||||||
ARG PROMETHEUS_UID=2133
|
ARG ARGUS_BUILD_UID=2133
|
||||||
ARG PROMETHEUS_GID=2015
|
ARG ARGUS_BUILD_GID=2015
|
||||||
ENV PROMETHEUS_UID=${PROMETHEUS_UID}
|
|
||||||
ENV PROMETHEUS_GID=${PROMETHEUS_GID}
|
|
||||||
|
|
||||||
|
ENV ARGUS_BUILD_UID=${ARGUS_BUILD_UID} \
|
||||||
|
ARGUS_BUILD_GID=${ARGUS_BUILD_GID}
|
||||||
# 创建目录结构
|
# 创建目录结构
|
||||||
RUN mkdir -p ${PROMETHEUS_BASE_PATH}/rules \
|
RUN mkdir -p ${PROMETHEUS_BASE_PATH}/rules \
|
||||||
&& mkdir -p ${PROMETHEUS_BASE_PATH}/targets \
|
&& mkdir -p ${PROMETHEUS_BASE_PATH}/targets \
|
||||||
@@ -61,11 +61,26 @@ RUN mkdir -p ${PROMETHEUS_BASE_PATH}/rules \
|
|||||||
&& ln -s ${PROMETHEUS_BASE_PATH} /prometheus
|
&& ln -s ${PROMETHEUS_BASE_PATH} /prometheus
|
||||||
|
|
||||||
# 修改 Prometheus 用户 UID/GID 并授权
|
# 修改 Prometheus 用户 UID/GID 并授权
|
||||||
RUN usermod -u ${PROMETHEUS_UID} nobody && \
|
RUN set -eux; \
|
||||||
groupmod -g ${PROMETHEUS_GID} nogroup && \
|
existing_user=""; \
|
||||||
chown -h nobody:nogroup /prometheus && \
|
if getent passwd "${ARGUS_BUILD_UID}" >/dev/null 2>&1; then \
|
||||||
chown -R nobody:nogroup /private/argus/metric /etc/prometheus && \
|
existing_user="$(getent passwd "${ARGUS_BUILD_UID}" | cut -d: -f1)"; \
|
||||||
chown -R nobody:nogroup ${PROMETHEUS_BASE_PATH}
|
fi; \
|
||||||
|
if [ -n "$existing_user" ] && [ "$existing_user" != "nobody" ]; then \
|
||||||
|
userdel -r "$existing_user" || true; \
|
||||||
|
fi; \
|
||||||
|
existing_group=""; \
|
||||||
|
if getent group "${ARGUS_BUILD_GID}" >/dev/null 2>&1; then \
|
||||||
|
existing_group="$(getent group "${ARGUS_BUILD_GID}" | cut -d: -f1)"; \
|
||||||
|
fi; \
|
||||||
|
if [ -n "$existing_group" ] && [ "$existing_group" != "nogroup" ]; then \
|
||||||
|
groupdel "$existing_group" || true; \
|
||||||
|
fi; \
|
||||||
|
usermod -u ${ARGUS_BUILD_UID} nobody; \
|
||||||
|
groupmod -g ${ARGUS_BUILD_GID} nogroup; \
|
||||||
|
chown -h nobody:nogroup /prometheus; \
|
||||||
|
chown -R nobody:nogroup ${PROMETHEUS_BASE_PATH}; \
|
||||||
|
chown -R nobody:nogroup /etc/prometheus
|
||||||
|
|
||||||
# supervisor 配置
|
# supervisor 配置
|
||||||
COPY supervisord.conf /etc/supervisor/conf.d/supervisord.conf
|
COPY supervisord.conf /etc/supervisor/conf.d/supervisord.conf
|
||||||
|
|||||||
@@ -7,7 +7,7 @@ global:
|
|||||||
alerting:
|
alerting:
|
||||||
alertmanagers:
|
alertmanagers:
|
||||||
- static_configs:
|
- static_configs:
|
||||||
- targets: []
|
- targets: ["alertmanager.alert.argus.com:9093"]
|
||||||
|
|
||||||
# 规则目录
|
# 规则目录
|
||||||
rule_files:
|
rule_files:
|
||||||
|
|||||||
@@ -17,6 +17,7 @@ sed "s|\${PROMETHEUS_BASE_PATH}|${PROMETHEUS_BASE_PATH}|g" \
|
|||||||
IP=$(ifconfig eth0 | awk '/inet /{print $2}')
|
IP=$(ifconfig eth0 | awk '/inet /{print $2}')
|
||||||
echo "current IP: ${IP}"
|
echo "current IP: ${IP}"
|
||||||
echo "${IP}" > /private/argus/etc/${DOMAIN}
|
echo "${IP}" > /private/argus/etc/${DOMAIN}
|
||||||
|
chmod +x /private/argus/etc/${DOMAIN}
|
||||||
|
|
||||||
exec /bin/prometheus \
|
exec /bin/prometheus \
|
||||||
--config.file=${PROMETHEUS_BASE_PATH}/prometheus.yml \
|
--config.file=${PROMETHEUS_BASE_PATH}/prometheus.yml \
|
||||||
|
|||||||
2
src/metric/tests/.gitignore
vendored
2
src/metric/tests/.gitignore
vendored
@@ -1,7 +1,7 @@
|
|||||||
.env
|
.env
|
||||||
data/
|
data/
|
||||||
images-cache/
|
images-cache/
|
||||||
|
private-test-node/
|
||||||
*.tar
|
*.tar
|
||||||
*.log
|
*.log
|
||||||
.DS_Store
|
.DS_Store
|
||||||
|
|
||||||
|
|||||||
@@ -1,171 +1,97 @@
|
|||||||
# E2E Test - Argus Metric 部署测试
|
# E2E Test - Argus Metric 部署测试
|
||||||
## 概述
|
## 1. 概述
|
||||||
|
|
||||||
本项目用于对 Argus Metric 模块进行端到端(E2E)部署测试。
|
本项目用于对 Argus Metric 模块进行端到端(E2E)部署测试。
|
||||||
通过一键脚本可快速搭建 Prometheus、FTP、Grafana 等服务,验证 Metric 模块的完整部署与运行流程。
|
通过一键脚本可快速搭建 Prometheus、FTP、Grafana 等服务,验证 Metric 模块的完整部署与运行流程。
|
||||||
|
|
||||||
## 拉取完整项目,进入 metric.tests 目录
|
功能包括:
|
||||||
|
|
||||||
``` bash
|
- 自动启动所需服务和测试节点
|
||||||
git clone https://git.nasp.fit/NASP/argus.git
|
- 发布安装包到 FTP
|
||||||
|
- CPU/GPU 节点客户端安装测试
|
||||||
|
- 验证安装结果与服务可用性
|
||||||
|
- 支持环境清理和分步调试
|
||||||
|
|
||||||
cd {$PROJECT_ROOT}/argus/src/metric/tests
|
## 2. 前置条件
|
||||||
|
|
||||||
|
在开始部署和测试之前,请确保完成以下准备工作:
|
||||||
|
|
||||||
|
### 2.1 检查 all-in-one-full 客户端安装包
|
||||||
|
确认客户端安装包目录是否存在:
|
||||||
|
```bash
|
||||||
|
{$PROJECT_ROOT}/argus/src/metric/client-plugins/all-in-one-full
|
||||||
```
|
```
|
||||||
|
本项目依赖完整的 all-in-one-full 安装包,其中包含大量二进制文件、依赖包和测试制品,由于体积较大,无法直接上传到 Git 仓库。**请联系项目管理员获取最新版本的完整框架。**
|
||||||
|
|
||||||
## 一键构建与部署 Prometheus / FTP / Grafana
|
### 2.2 配置环境变量
|
||||||
### 1. 修改环境变量文件
|
查看配置文件是否存在,如不存在,则复制示例配置文件并根据实际环境修改:
|
||||||
|
```bash
|
||||||
将示例配置文件复制为 .env 并根据实际情况修改:
|
cd {$PROJECT_ROOT}/argus/src/metric/tests
|
||||||
|
|
||||||
``` bash
|
|
||||||
cp env.example .env
|
cp env.example .env
|
||||||
```
|
```
|
||||||
|
.env 文件用于指定构建UID:GID、FTP 配置、版本号等信息,确保各脚本运行时可以正确访问资源。
|
||||||
|
|
||||||
### 2. 一键启动服务
|
### 2.3 离线镜像准备
|
||||||
|
- 步骤1:在**在线服务器**执行以下脚本,会拉取和构建所需的 Docker 镜像:
|
||||||
|
``` bash
|
||||||
|
cd {$PROJECT_ROOT}/argus/src/metric/tests
|
||||||
|
bash scripts/01_start_services.sh
|
||||||
|
bash scripts/save-images.sh
|
||||||
|
```
|
||||||
|
- 步骤2:镜像将被保存到 metric.tests.images-cache 目录中,用于离线迁移和后续导入。
|
||||||
|
- 步骤3:若目标服务器无法联网,可将该目录拷贝到离线服务器,并执行:
|
||||||
|
``` bash
|
||||||
|
cd {$PROJECT_ROOT}/argus/src/metric/tests
|
||||||
|
bash scripts/load-images.sh
|
||||||
|
```
|
||||||
|
- 即可导入镜像并执行下面的QuickStart或分步操作。
|
||||||
|
|
||||||
执行以下命令完成环境初始化、镜像构建与服务启动:
|
## 3. QuickStart
|
||||||
|
|
||||||
``` bash
|
执行完整的端到端测试流程:
|
||||||
sudo bash start-all.sh
|
|
||||||
|
```bash
|
||||||
|
bash scripts/00_e2e_test.sh
|
||||||
```
|
```
|
||||||
|
|
||||||
该脚本将自动完成:
|
该脚本将自动执行以下步骤:
|
||||||
- 初始化目录结构(如 /private/argus/metric)
|
1. 启动所有服务(Prometheus、FTP、Grafana、测试节点)
|
||||||
- 构建各服务 Docker 镜像
|
2. 发布安装包到 FTP 服务
|
||||||
- 启动 Prometheus、FTP、Grafana 容器
|
3. 在 CPU 测试节点上安装客户端
|
||||||
|
4. 在 GPU 测试节点上安装客户端
|
||||||
|
5. 验证安装结果
|
||||||
|
6. 清理测试环境
|
||||||
|
|
||||||
### 3. 检查容器日志
|
## 4. 分步执行
|
||||||
|
|
||||||
可手动验证容器运行状态:
|
| 步骤 | 脚本 | 功能描述 |
|
||||||
|
|--------------|-------------------------------------------|--------------------------------------------------------|
|
||||||
|
| 启动基础服务 | bash scripts/01_start_services.sh | 构建 Docker 镜像、创建持久化目录、启动容器服务 |
|
||||||
|
| 发布安装包 | bash scripts/02_publish_artifact.sh | 自动递增版本号、打包安装制品、发布到 FTP |
|
||||||
|
| CPU 节点安装 | bash scripts/03_test_node_install.sh | 在 CPU 节点下载安装程序并执行安装 |
|
||||||
|
| GPU 节点安装 | bash scripts/04_test_gpu_node_install.sh | 在 GPU 节点下载安装程序并执行安装 |
|
||||||
|
| 验证安装 | bash scripts/05_verify_install.sh | 检查监控端口、端口连通性及服务可用性 |
|
||||||
|
| 清理环境 | bash scripts/06_cleanup.sh | 停止并清理所有测试容器及环境 |
|
||||||
|
|
||||||
``` bash
|
## 5. 查看监控采集数据及展示面板
|
||||||
docker logs argus-metric-ftp
|
|
||||||
docker logs argus-metric-grafana
|
|
||||||
docker logs argus-metric-prometheus
|
|
||||||
```
|
|
||||||
|
|
||||||
如日志输出中无 ERROR 或 supervisor 报错信息,则表示服务启动正常。
|
|
||||||
|
|
||||||
## 客户端安装包打包与分发
|
|
||||||
|
|
||||||
> **前置说明**:完整的 `all-in-one` 安装包打包分发框架因包含大量二进制文件和依赖包,无法上传至 Git 仓库。请先联系项目管理员获取最新的 `all-in-one` 完整框架,再执行后续操作。
|
|
||||||
|
|
||||||
打包后服务端会将安装包发布至 FTP 共享目录,默认路径为:
|
|
||||||
|
|
||||||
``` bash
|
|
||||||
$DATA_ROOT/ftp/share
|
|
||||||
```
|
|
||||||
|
|
||||||
发布后的文件权限与 FTP 目录账户保持一致。
|
|
||||||
|
|
||||||
### 1. 递增版本号
|
|
||||||
``` bash
|
|
||||||
bash scripts/version-manager.sh bump minor
|
|
||||||
```
|
|
||||||
该脚本会自动更新版本号(如 1.101.0 → 1.102.0)。
|
|
||||||
|
|
||||||
### 2. 打包安装制品
|
|
||||||
``` bash
|
|
||||||
bash scripts/package_artifact.sh
|
|
||||||
```
|
|
||||||
执行后会在输出目录中生成压缩包或安装脚本。
|
|
||||||
|
|
||||||
### 3. 发布制品至 FTP
|
|
||||||
``` bash
|
|
||||||
sudo bash scripts/publish_artifact.sh $VERSION --output-dir $OUTPUT_DIR --owner $UID:$GID
|
|
||||||
```
|
|
||||||
|
|
||||||
参数说明:
|
|
||||||
|
|
||||||
参数 说明
|
|
||||||
$VERSION 发布版本号(如 1.102.0)
|
|
||||||
$OUTPUT_DIR 输出目录(默认 /private/argus/ftp/share)
|
|
||||||
$UID:$GID 文件属主(用户ID:组ID)
|
|
||||||
|
|
||||||
示例:
|
|
||||||
|
|
||||||
``` bash
|
|
||||||
sudo bash scripts/publish_artifact.sh 1.102.0 --output-dir /private/argus/ftp/share --owner 2133:2015
|
|
||||||
```
|
|
||||||
|
|
||||||
更多详情可参考 client-plugins/all-in-one/README.md。
|
|
||||||
|
|
||||||
## 客户端安装(通过 FTP)
|
|
||||||
|
|
||||||
客户端下载与安装步骤如下:
|
|
||||||
|
|
||||||
``` bash
|
|
||||||
curl -u ${USER}:${PASSWD} ftp://${FTP_SERVER}/setup.sh -o setup.sh
|
|
||||||
chmod +x setup.sh
|
|
||||||
sudo bash setup.sh --server ${FTP_SERVER} --user ${USER} --password ${PASSWD} --port ${PORT}
|
|
||||||
```
|
|
||||||
|
|
||||||
参数说明:
|
|
||||||
|
|
||||||
参数 说明
|
|
||||||
$FTP_SERVER 服务器 IP 地址
|
|
||||||
$USER 默认 ftpuser
|
|
||||||
$PASSWD 默认 ZGClab1234!
|
|
||||||
$PORT FTP 服务端口(需与 .env 保持一致)
|
|
||||||
|
|
||||||
示例:
|
|
||||||
|
|
||||||
``` bash
|
|
||||||
curl -u ftpuser:ZGClab1234! ftp://10.211.55.4/setup.sh -o setup.sh
|
|
||||||
chmod +x setup.sh
|
|
||||||
sudo bash setup.sh --server 10.211.55.4 --user ftpuser --password 'ZGClab1234!' --port 2122
|
|
||||||
```
|
|
||||||
|
|
||||||
更多细节可参考 client-plugins/all-in-one/README.md。
|
|
||||||
|
|
||||||
## 模拟 Argus-Master 配置下发
|
|
||||||
|
|
||||||
可通过手动写入 nodes.json 文件模拟 Argus-Master 对 Argus-Metric 的配置下发:
|
|
||||||
|
|
||||||
``` json
|
|
||||||
[
|
|
||||||
{
|
|
||||||
"node_id": "A1",
|
|
||||||
"user_id": "sundapeng",
|
|
||||||
"ip": "10.211.55.4",
|
|
||||||
"hostname": "dev-sundapeng-nsche-wohen-pod-0",
|
|
||||||
"labels": ["label-a", "label-b"]
|
|
||||||
}
|
|
||||||
]
|
|
||||||
```
|
|
||||||
|
|
||||||
路径:
|
|
||||||
|
|
||||||
``` bash
|
|
||||||
${DATA_ROOT}/prometheus/nodes.json
|
|
||||||
```
|
|
||||||
|
|
||||||
Argus-Metric 中的 prometheus 模块会自动解析该文件,并将其拆分生成目标配置:
|
|
||||||
|
|
||||||
``` bash
|
|
||||||
${DATA_ROOT}/prometheus/targets/
|
|
||||||
```
|
|
||||||
|
|
||||||
## Grafana 手动配置(如未自动接入 Prometheus)
|
|
||||||
|
|
||||||
如 Grafana 未自动导入 Prometheus 数据源,可手动执行以下操作:
|
|
||||||
|
|
||||||
1. 添加数据源
|
|
||||||
- 进入 Grafana → Data sources
|
|
||||||
- 选择 Add data source → Prometheus
|
|
||||||
- URL 填写:http://prometheus:9090(Docker 内部 DNS 地址)
|
|
||||||
|
|
||||||
2. 导入测试 Dashboard
|
|
||||||
- 打开 Grafana → Dashboards → Import
|
|
||||||
- 上传或粘贴 test_grafana_dashboard.json
|
|
||||||
|
|
||||||
## 查看监控数据
|
|
||||||
Prometheus 访问以下地址查看节点活性:
|
Prometheus 访问以下地址查看节点活性:
|
||||||
``` bash
|
``` bash
|
||||||
http://127.0.0.1:9091/targets
|
http://127.0.0.1:9090/targets
|
||||||
```
|
```
|
||||||
|
|
||||||
Grafana 访问以下地址查看监控大屏:
|
Grafana 访问以下地址查看监控大屏:
|
||||||
``` bash
|
``` bash
|
||||||
http://127.0.0.1:3000/d/node_gpu_metrics/node-and-gpu-metrics
|
http://127.0.0.1:3000/d/node_gpu_metrics/node-and-gpu-metrics
|
||||||
```
|
```
|
||||||
|
|
||||||
|
PS: 如果 Grafana 未自动导入 Prometheus 数据源,可手动执行以下操作:
|
||||||
|
|
||||||
|
1. 添加数据源
|
||||||
|
- 进入 Grafana → Data sources
|
||||||
|
- 选择 Add data source → Prometheus
|
||||||
|
- URL 填写:http://prom.metric.argus.com:9090
|
||||||
|
|
||||||
|
2. 导入测试 Dashboard
|
||||||
|
- 打开 Grafana → Dashboards → Import
|
||||||
|
- 上传或粘贴 test_grafana_dashboard.json
|
||||||
39
src/metric/tests/client-test-gpu-node/build/Dockerfile
Normal file
39
src/metric/tests/client-test-gpu-node/build/Dockerfile
Normal file
@@ -0,0 +1,39 @@
|
|||||||
|
# 使用NVIDIA官方CUDA基础镜像
|
||||||
|
FROM nvidia/cuda:12.2.2-runtime-ubuntu22.04
|
||||||
|
|
||||||
|
ENV DEBIAN_FRONTEND=noninteractive
|
||||||
|
|
||||||
|
# 设置时区
|
||||||
|
ENV TZ=Asia/Shanghai
|
||||||
|
|
||||||
|
RUN apt-get update -qq && \
|
||||||
|
apt-get install -y -qq \
|
||||||
|
tzdata \
|
||||||
|
curl \
|
||||||
|
wget \
|
||||||
|
gnupg2 \
|
||||||
|
software-properties-common \
|
||||||
|
ca-certificates \
|
||||||
|
&& rm -rf /var/lib/apt/lists/*
|
||||||
|
|
||||||
|
# 配置时区
|
||||||
|
RUN ln -snf /usr/share/zoneinfo/$TZ /etc/localtime && echo $TZ > /etc/timezone
|
||||||
|
|
||||||
|
WORKDIR /app
|
||||||
|
|
||||||
|
# 创建启动脚本,在运行时验证GPU
|
||||||
|
COPY <<EOF /app/start.sh
|
||||||
|
#!/bin/bash
|
||||||
|
echo "检查GPU环境..."
|
||||||
|
if command -v nvidia-smi &> /dev/null; then
|
||||||
|
nvidia-smi
|
||||||
|
echo "GPU环境正常"
|
||||||
|
else
|
||||||
|
echo "警告: nvidia-smi 命令不可用,请确保容器运行时启用了GPU支持"
|
||||||
|
fi
|
||||||
|
exec "\$@"
|
||||||
|
EOF
|
||||||
|
|
||||||
|
RUN chmod +x /app/start.sh
|
||||||
|
|
||||||
|
CMD ["/app/start.sh", "/bin/bash"]
|
||||||
6
src/metric/tests/client-test-node/build/Dockerfile
Normal file
6
src/metric/tests/client-test-node/build/Dockerfile
Normal file
@@ -0,0 +1,6 @@
|
|||||||
|
FROM ubuntu:22.04
|
||||||
|
RUN apt-get update -qq && \
|
||||||
|
DEBIAN_FRONTEND=noninteractive apt-get install -y -qq tzdata && \
|
||||||
|
rm -rf /var/lib/apt/lists/*
|
||||||
|
ENV TZ=Asia/Shanghai
|
||||||
|
|
||||||
@@ -1,29 +1,32 @@
|
|||||||
|
networks:
|
||||||
|
default:
|
||||||
|
name: argus-debug-net
|
||||||
|
external: true
|
||||||
|
|
||||||
services:
|
services:
|
||||||
ftp:
|
ftp:
|
||||||
build:
|
|
||||||
context: ../ftp/build
|
|
||||||
dockerfile: Dockerfile
|
|
||||||
args:
|
|
||||||
FTP_UID: ${FTP_UID:-2133}
|
|
||||||
FTP_GID: ${FTP_GID:-2015}
|
|
||||||
image: argus-metric-ftp:latest
|
image: argus-metric-ftp:latest
|
||||||
container_name: argus-ftp
|
container_name: argus-ftp
|
||||||
restart: unless-stopped
|
restart: unless-stopped
|
||||||
environment:
|
environment:
|
||||||
|
- TZ=Asia/Shanghai
|
||||||
- FTP_BASE_PATH=/private/argus/ftp
|
- FTP_BASE_PATH=/private/argus/ftp
|
||||||
- FTP_PASSWORD=${FTP_PASSWORD:-ZGClab1234!}
|
- FTP_PASSWORD=${FTP_PASSWORD:-ZGClab1234!}
|
||||||
- DOMAIN=${FTP_DOMAIN:-prom.ftp.argus.com}
|
- DOMAIN=${FTP_DOMAIN:-ftp.metric.argus.com}
|
||||||
- FTP_UID=${FTP_UID:-2133}
|
- ARGUS_BUILD_UID=${ARGUS_BUILD_UID:-2133}
|
||||||
- FTP_GID=${FTP_GID:-2015}
|
- ARGUS_BUILD_GID=${ARGUS_BUILD_GID:-2015}
|
||||||
ports:
|
ports:
|
||||||
- "${FTP_PORT:-21}:21"
|
- "${FTP_PORT:-21}:21"
|
||||||
- "${FTP_DATA_PORT:-20}:20"
|
- "${FTP_DATA_PORT:-20}:20"
|
||||||
- "21100-21110:21100-21110"
|
- "21100-21110:21100-21110"
|
||||||
volumes:
|
volumes:
|
||||||
- ${DATA_ROOT:-./data}/ftp:/private/argus/ftp
|
- ${DATA_ROOT:-/private}/argus/metric/ftp:/private/argus/ftp
|
||||||
- ${DATA_ROOT:-./data}/etc:/private/argus/etc
|
- ${DATA_ROOT:-/private}/argus/etc:/private/argus/etc
|
||||||
|
- /etc/localtime:/etc/localtime:ro
|
||||||
|
- /etc/timezone:/etc/timezone:ro
|
||||||
networks:
|
networks:
|
||||||
- argus-network
|
default:
|
||||||
|
ipv4_address: 172.30.0.40
|
||||||
logging:
|
logging:
|
||||||
driver: "json-file"
|
driver: "json-file"
|
||||||
options:
|
options:
|
||||||
@@ -31,27 +34,24 @@ services:
|
|||||||
max-file: "3"
|
max-file: "3"
|
||||||
|
|
||||||
prometheus:
|
prometheus:
|
||||||
build:
|
|
||||||
context: ../prometheus/build
|
|
||||||
dockerfile: Dockerfile
|
|
||||||
args:
|
|
||||||
PROMETHEUS_UID: ${PROMETHEUS_UID:-2133}
|
|
||||||
PROMETHEUS_GID: ${PROMETHEUS_GID:-2015}
|
|
||||||
USE_INTRANET: ${USE_INTRANET:-false}
|
|
||||||
image: argus-metric-prometheus:latest
|
image: argus-metric-prometheus:latest
|
||||||
container_name: argus-prometheus
|
container_name: argus-prometheus
|
||||||
restart: unless-stopped
|
restart: unless-stopped
|
||||||
environment:
|
environment:
|
||||||
|
- TZ=Asia/Shanghai
|
||||||
- PROMETHEUS_BASE_PATH=/private/argus/metric/prometheus
|
- PROMETHEUS_BASE_PATH=/private/argus/metric/prometheus
|
||||||
- PROMETHEUS_UID=${PROMETHEUS_UID:-2133}
|
- ARGUS_BUILD_UID=${ARGUS_BUILD_UID:-2133}
|
||||||
- PROMETHEUS_GID=${PROMETHEUS_GID:-2015}
|
- ARGUS_BUILD_GID=${ARGUS_BUILD_GID:-2015}
|
||||||
ports:
|
ports:
|
||||||
- "${PROMETHEUS_PORT:-9090}:9090"
|
- "${PROMETHEUS_PORT:-9090}:9090"
|
||||||
volumes:
|
volumes:
|
||||||
- ${DATA_ROOT:-./data}/prometheus:/private/argus/metric/prometheus
|
- ${DATA_ROOT:-/private}/argus/metric/prometheus:/private/argus/metric/prometheus
|
||||||
- ${DATA_ROOT:-./data}/etc:/private/argus/etc
|
- ${DATA_ROOT:-/private}/argus/etc:/private/argus/etc
|
||||||
|
- /etc/localtime:/etc/localtime:ro
|
||||||
|
- /etc/timezone:/etc/timezone:ro
|
||||||
networks:
|
networks:
|
||||||
- argus-network
|
default:
|
||||||
|
ipv4_address: 172.30.0.41
|
||||||
logging:
|
logging:
|
||||||
driver: "json-file"
|
driver: "json-file"
|
||||||
options:
|
options:
|
||||||
@@ -59,29 +59,27 @@ services:
|
|||||||
max-file: "3"
|
max-file: "3"
|
||||||
|
|
||||||
grafana:
|
grafana:
|
||||||
build:
|
|
||||||
context: ../grafana/build
|
|
||||||
dockerfile: Dockerfile
|
|
||||||
args:
|
|
||||||
GRAFANA_UID: ${GRAFANA_UID:-2133}
|
|
||||||
GRAFANA_GID: ${GRAFANA_GID:-2015}
|
|
||||||
image: argus-metric-grafana:latest
|
image: argus-metric-grafana:latest
|
||||||
container_name: argus-grafana
|
container_name: argus-grafana
|
||||||
restart: unless-stopped
|
restart: unless-stopped
|
||||||
environment:
|
environment:
|
||||||
|
- TZ=Asia/Shanghai
|
||||||
- GRAFANA_BASE_PATH=/private/argus/metric/grafana
|
- GRAFANA_BASE_PATH=/private/argus/metric/grafana
|
||||||
- GRAFANA_UID=${GRAFANA_UID:-2133}
|
- ARGUS_BUILD_UID=${ARGUS_BUILD_UID:-2133}
|
||||||
- GRAFANA_GID=${GRAFANA_GID:-2015}
|
- ARGUS_BUILD_GID=${ARGUS_BUILD_GID:-2015}
|
||||||
- GF_SERVER_HTTP_PORT=3000
|
- GF_SERVER_HTTP_PORT=3000
|
||||||
- GF_LOG_LEVEL=warn
|
- GF_LOG_LEVEL=warn
|
||||||
- GF_LOG_MODE=console
|
- GF_LOG_MODE=console
|
||||||
ports:
|
ports:
|
||||||
- "${GRAFANA_PORT:-3000}:3000"
|
- "${GRAFANA_PORT:-3000}:3000"
|
||||||
volumes:
|
volumes:
|
||||||
- ${DATA_ROOT:-./data}/grafana:/private/argus/metric/grafana
|
- ${DATA_ROOT:-/private}/argus/metric/grafana:/private/argus/metric/grafana
|
||||||
- ${DATA_ROOT:-./data}/etc:/private/argus/etc
|
- ${DATA_ROOT:-/private}/argus/etc:/private/argus/etc
|
||||||
|
- /etc/localtime:/etc/localtime:ro
|
||||||
|
- /etc/timezone:/etc/timezone:ro
|
||||||
networks:
|
networks:
|
||||||
- argus-network
|
default:
|
||||||
|
ipv4_address: 172.30.0.42
|
||||||
depends_on:
|
depends_on:
|
||||||
- prometheus
|
- prometheus
|
||||||
logging:
|
logging:
|
||||||
@@ -90,16 +88,72 @@ services:
|
|||||||
max-size: "10m"
|
max-size: "10m"
|
||||||
max-file: "3"
|
max-file: "3"
|
||||||
|
|
||||||
networks:
|
test-node:
|
||||||
argus-network:
|
image: argus-metric-test-node:latest
|
||||||
driver: bridge
|
container_name: argus-metric-test-node
|
||||||
name: argus-network
|
hostname: test-metric-node-001
|
||||||
|
restart: unless-stopped
|
||||||
|
privileged: true
|
||||||
|
depends_on:
|
||||||
|
- ftp
|
||||||
|
- prometheus
|
||||||
|
environment:
|
||||||
|
- TZ=Asia/Shanghai
|
||||||
|
- DEBIAN_FRONTEND=noninteractive
|
||||||
|
- FTP_DOMAIN=${FTP_DOMAIN:-ftp.metric.argus.com}
|
||||||
|
- FTP_SERVER=${FTP_SERVER:-172.30.0.40}
|
||||||
|
- FTP_USER=${FTP_USER:-ftpuser}
|
||||||
|
- FTP_PASSWORD=${FTP_PASSWORD:-ZGClab1234!}
|
||||||
|
- FTP_PORT=${FTP_PORT:-21}
|
||||||
|
volumes:
|
||||||
|
- ${DATA_ROOT:-/private}/argus/agent:/private/argus/agent
|
||||||
|
- /etc/localtime:/etc/localtime:ro
|
||||||
|
- /etc/timezone:/etc/timezone:ro
|
||||||
|
command: sleep infinity
|
||||||
|
networks:
|
||||||
|
default:
|
||||||
|
ipv4_address: 172.30.0.50
|
||||||
|
logging:
|
||||||
|
driver: "json-file"
|
||||||
|
options:
|
||||||
|
max-size: "10m"
|
||||||
|
max-file: "3"
|
||||||
|
|
||||||
volumes:
|
test-gpu-node:
|
||||||
ftp_data:
|
image: argus-metric-test-gpu-node:latest
|
||||||
driver: local
|
container_name: argus-metric-test-gpu-node
|
||||||
prometheus_data:
|
hostname: test-metric-gpu-node-001
|
||||||
driver: local
|
restart: unless-stopped
|
||||||
grafana_data:
|
privileged: true
|
||||||
driver: local
|
runtime: nvidia
|
||||||
|
deploy:
|
||||||
|
resources:
|
||||||
|
reservations:
|
||||||
|
devices:
|
||||||
|
- driver: nvidia
|
||||||
|
count: all
|
||||||
|
capabilities:
|
||||||
|
- gpu
|
||||||
|
depends_on:
|
||||||
|
- ftp
|
||||||
|
- prometheus
|
||||||
|
environment:
|
||||||
|
- TZ=Asia/Shanghai
|
||||||
|
- DEBIAN_FRONTEND=noninteractive
|
||||||
|
- NVIDIA_VISIBLE_DEVICES=all
|
||||||
|
- NVIDIA_DRIVER_CAPABILITIES=compute,utility
|
||||||
|
- GPU_MODE=gpu
|
||||||
|
volumes:
|
||||||
|
- ${DATA_ROOT:-/private}/argus/agent:/private/argus/agent
|
||||||
|
- /etc/localtime:/etc/localtime:ro
|
||||||
|
- /etc/timezone:/etc/timezone:ro
|
||||||
|
command: sleep infinity
|
||||||
|
networks:
|
||||||
|
default:
|
||||||
|
ipv4_address: 172.30.0.51
|
||||||
|
logging:
|
||||||
|
driver: "json-file"
|
||||||
|
options:
|
||||||
|
max-size: "10m"
|
||||||
|
max-file: "3"
|
||||||
|
|
||||||
|
|||||||
@@ -1,19 +1,15 @@
|
|||||||
# 用户和组配置
|
# 统一用户和组配置
|
||||||
FTP_UID=2133
|
ARGUS_BUILD_UID=1048
|
||||||
FTP_GID=2015
|
ARGUS_BUILD_GID=1048
|
||||||
PROMETHEUS_UID=2133
|
|
||||||
PROMETHEUS_GID=2015
|
|
||||||
GRAFANA_UID=2133
|
|
||||||
GRAFANA_GID=2015
|
|
||||||
|
|
||||||
# 数据根目录
|
# 数据根目录
|
||||||
DATA_ROOT=/private/argus
|
DATA_ROOT=/private
|
||||||
|
|
||||||
# FTP 配置
|
# FTP 配置
|
||||||
FTP_PORT=2122
|
FTP_PORT=21
|
||||||
FTP_DATA_PORT=2022
|
FTP_DATA_PORT=20
|
||||||
FTP_PASSWORD=ZGClab1234!
|
FTP_PASSWORD=ZGClab1234!
|
||||||
FTP_DOMAIN=prom.ftp.argus.com
|
FTP_DOMAIN=ftp.metric.argus.com
|
||||||
|
|
||||||
# Prometheus 配置
|
# Prometheus 配置
|
||||||
PROMETHEUS_PORT=9090
|
PROMETHEUS_PORT=9090
|
||||||
|
|||||||
@@ -1,90 +0,0 @@
|
|||||||
#!/bin/bash
|
|
||||||
|
|
||||||
# 初始化目录脚本
|
|
||||||
# 用于创建所有必要的数据目录并设置正确的权限
|
|
||||||
|
|
||||||
set -e
|
|
||||||
|
|
||||||
SCRIPT_DIR="$(cd "$(dirname "${BASH_SOURCE[0]}")" && pwd)"
|
|
||||||
cd "$SCRIPT_DIR"
|
|
||||||
|
|
||||||
# 加载 .env 文件(如果存在)
|
|
||||||
if [ -f .env ]; then
|
|
||||||
echo "加载 .env 配置文件..."
|
|
||||||
source .env
|
|
||||||
fi
|
|
||||||
|
|
||||||
# 默认配置
|
|
||||||
FTP_UID=${FTP_UID:-2133}
|
|
||||||
FTP_GID=${FTP_GID:-2015}
|
|
||||||
PROMETHEUS_UID=${PROMETHEUS_UID:-2133}
|
|
||||||
PROMETHEUS_GID=${PROMETHEUS_GID:-2015}
|
|
||||||
GRAFANA_UID=${GRAFANA_UID:-2133}
|
|
||||||
GRAFANA_GID=${GRAFANA_GID:-2015}
|
|
||||||
DATA_ROOT=${DATA_ROOT:-./data}
|
|
||||||
|
|
||||||
echo "开始初始化目录结构..."
|
|
||||||
echo "数据目录: ${DATA_ROOT}"
|
|
||||||
echo ""
|
|
||||||
|
|
||||||
# 创建 FTP 目录
|
|
||||||
echo "创建 FTP 目录..."
|
|
||||||
sudo mkdir -p ${DATA_ROOT}/ftp/share
|
|
||||||
sudo chown -R ${FTP_UID}:${FTP_GID} ${DATA_ROOT}/ftp
|
|
||||||
sudo chmod -R 755 ${DATA_ROOT}/ftp
|
|
||||||
|
|
||||||
# 创建 Prometheus 目录
|
|
||||||
echo "创建 Prometheus 目录..."
|
|
||||||
sudo mkdir -p ${DATA_ROOT}/prometheus/{data,rules,targets}
|
|
||||||
|
|
||||||
# 创建默认的 targets 文件(先创建文件再改权限)
|
|
||||||
if [ ! -f "${DATA_ROOT}/prometheus/targets/node_exporter.json" ]; then
|
|
||||||
echo "创建默认 node_exporter targets..."
|
|
||||||
echo '[
|
|
||||||
{
|
|
||||||
"targets": [],
|
|
||||||
"labels": {
|
|
||||||
"job": "node"
|
|
||||||
}
|
|
||||||
}
|
|
||||||
]' | sudo tee ${DATA_ROOT}/prometheus/targets/node_exporter.json > /dev/null
|
|
||||||
fi
|
|
||||||
|
|
||||||
if [ ! -f "${DATA_ROOT}/prometheus/targets/dcgm_exporter.json" ]; then
|
|
||||||
echo "创建默认 dcgm_exporter targets..."
|
|
||||||
echo '[
|
|
||||||
{
|
|
||||||
"targets": [],
|
|
||||||
"labels": {
|
|
||||||
"job": "dcgm"
|
|
||||||
}
|
|
||||||
}
|
|
||||||
]' | sudo tee ${DATA_ROOT}/prometheus/targets/dcgm_exporter.json > /dev/null
|
|
||||||
fi
|
|
||||||
|
|
||||||
# 统一设置 Prometheus 目录权限
|
|
||||||
sudo chown -R ${PROMETHEUS_UID}:${PROMETHEUS_GID} ${DATA_ROOT}/prometheus
|
|
||||||
sudo chmod -R 755 ${DATA_ROOT}/prometheus
|
|
||||||
|
|
||||||
# 创建 Grafana 目录
|
|
||||||
echo "创建 Grafana 目录..."
|
|
||||||
sudo mkdir -p ${DATA_ROOT}/grafana/{data,logs,plugins,provisioning/datasources,provisioning/dashboards,data/sessions,data/dashboards,config}
|
|
||||||
sudo chown -R ${GRAFANA_UID}:${GRAFANA_GID} ${DATA_ROOT}/grafana
|
|
||||||
sudo chmod -R 755 ${DATA_ROOT}/grafana
|
|
||||||
|
|
||||||
# 创建公共配置目录
|
|
||||||
sudo mkdir -p ${DATA_ROOT}/etc
|
|
||||||
sudo chown -R ${FTP_UID}:${FTP_GID} ${DATA_ROOT}/etc
|
|
||||||
sudo chmod -R 755 ${DATA_ROOT}/etc
|
|
||||||
|
|
||||||
echo "目录初始化完成!"
|
|
||||||
echo ""
|
|
||||||
echo "目录结构:"
|
|
||||||
echo " ${DATA_ROOT}/"
|
|
||||||
echo " ├── ftp/ (UID:${FTP_UID}, GID:${FTP_GID})"
|
|
||||||
echo " ├── prometheus/ (UID:${PROMETHEUS_UID}, GID:${PROMETHEUS_GID})"
|
|
||||||
echo " ├── grafana/ (UID:${GRAFANA_UID}, GID:${GRAFANA_GID})"
|
|
||||||
echo " └── etc/ (UID:${FTP_UID}, GID:${FTP_GID})"
|
|
||||||
echo ""
|
|
||||||
echo "您现在可以运行 'docker-compose up -d' 来启动所有服务"
|
|
||||||
|
|
||||||
20
src/metric/tests/scripts/00_e2e_test.sh
Executable file
20
src/metric/tests/scripts/00_e2e_test.sh
Executable file
@@ -0,0 +1,20 @@
|
|||||||
|
#!/bin/bash
|
||||||
|
set -e
|
||||||
|
|
||||||
|
SCRIPT_DIR="$(dirname "$0")"
|
||||||
|
|
||||||
|
echo "=========================================="
|
||||||
|
echo "Argus Metric E2E Test"
|
||||||
|
echo "=========================================="
|
||||||
|
|
||||||
|
bash "$SCRIPT_DIR/01_start_services.sh"
|
||||||
|
bash "$SCRIPT_DIR/02_publish_artifact.sh"
|
||||||
|
bash "$SCRIPT_DIR/03_test_node_install.sh"
|
||||||
|
bash "$SCRIPT_DIR/04_test_gpu_node_install.sh"
|
||||||
|
bash "$SCRIPT_DIR/05_verify_install.sh"
|
||||||
|
bash "$SCRIPT_DIR/06_cleanup.sh"
|
||||||
|
|
||||||
|
echo "=========================================="
|
||||||
|
echo "E2E 测试完成"
|
||||||
|
echo "=========================================="
|
||||||
|
|
||||||
20
src/metric/tests/scripts/01_start_services.sh
Executable file
20
src/metric/tests/scripts/01_start_services.sh
Executable file
@@ -0,0 +1,20 @@
|
|||||||
|
#!/bin/bash
|
||||||
|
set -e
|
||||||
|
|
||||||
|
SCRIPT_DIR="$(cd "$(dirname "$0")" && pwd)"
|
||||||
|
|
||||||
|
echo "[01] 启动所有服务..."
|
||||||
|
bash "$SCRIPT_DIR/common/start-all.sh"
|
||||||
|
|
||||||
|
echo "[01] 等待服务就绪..."
|
||||||
|
sleep 5
|
||||||
|
|
||||||
|
echo "[01] 检查服务状态..."
|
||||||
|
docker ps | grep argus-ftp
|
||||||
|
docker ps | grep argus-prometheus
|
||||||
|
docker ps | grep argus-grafana
|
||||||
|
docker ps | grep argus-metric-test-node
|
||||||
|
docker ps | grep argus-metric-test-gpu-node
|
||||||
|
|
||||||
|
echo "[01] 基础服务已启动"
|
||||||
|
|
||||||
60
src/metric/tests/scripts/02_publish_artifact.sh
Executable file
60
src/metric/tests/scripts/02_publish_artifact.sh
Executable file
@@ -0,0 +1,60 @@
|
|||||||
|
#!/bin/bash
|
||||||
|
set -e
|
||||||
|
|
||||||
|
SCRIPT_DIR="$(cd "$(dirname "$0")" && pwd)"
|
||||||
|
TEST_DIR="$(cd "$SCRIPT_DIR/.." && pwd)"
|
||||||
|
PLUGIN_DIR="$(cd "$SCRIPT_DIR/../../client-plugins/all-in-one-full" && pwd)"
|
||||||
|
|
||||||
|
# 加载 .env
|
||||||
|
if [ -f "$TEST_DIR/.env" ]; then
|
||||||
|
source "$TEST_DIR/.env"
|
||||||
|
fi
|
||||||
|
|
||||||
|
# 检测容器挂载目录
|
||||||
|
if docker ps --format '{{.Names}}' | grep -q '^argus-ftp$'; then
|
||||||
|
FTP_MOUNT=$(docker inspect argus-ftp --format '{{range .Mounts}}{{if eq .Destination "/private/argus/ftp"}}{{.Source}}{{end}}{{end}}')
|
||||||
|
OUTPUT_DIR="${FTP_MOUNT}/share"
|
||||||
|
echo "[02] 容器挂载: $OUTPUT_DIR"
|
||||||
|
else
|
||||||
|
OUTPUT_DIR="${DATA_ROOT:-$TEST_DIR/data}/ftp/share"
|
||||||
|
echo "[02] 默认路径: $OUTPUT_DIR"
|
||||||
|
fi
|
||||||
|
|
||||||
|
OWNER="${ARGUS_BUILD_UID:-2133}:${ARGUS_BUILD_GID:-2015}"
|
||||||
|
|
||||||
|
cd "$PLUGIN_DIR"
|
||||||
|
|
||||||
|
echo "[02] 递增版本号..."
|
||||||
|
bash scripts/version-manager.sh bump minor
|
||||||
|
|
||||||
|
VERSION_FILE="config/VERSION"
|
||||||
|
if [ ! -f "$VERSION_FILE" ]; then
|
||||||
|
echo "[02] 错误: 未找到 $VERSION_FILE"
|
||||||
|
exit 1
|
||||||
|
fi
|
||||||
|
|
||||||
|
VERSION=$(cat "$VERSION_FILE" | tr -d '[:space:]')
|
||||||
|
echo "[02] 新版本: $VERSION"
|
||||||
|
|
||||||
|
echo "[02] 构建安装包..."
|
||||||
|
bash scripts/package_artifact.sh --force
|
||||||
|
|
||||||
|
echo "[02] 发布到 FTP: $OUTPUT_DIR"
|
||||||
|
sudo bash scripts/publish_artifact.sh "$VERSION" --output-dir "$OUTPUT_DIR" --owner "$OWNER"
|
||||||
|
|
||||||
|
echo "[02] 设置文件权限..."
|
||||||
|
# 设置所有者
|
||||||
|
sudo chown -R "$OWNER" "$OUTPUT_DIR"
|
||||||
|
# 设置目录权限为 755 (rwxr-xr-x)
|
||||||
|
sudo find "$OUTPUT_DIR" -type d -exec chmod 755 {} \;
|
||||||
|
# 设置文件权限为 644 (rw-r--r--)
|
||||||
|
sudo find "$OUTPUT_DIR" -type f -exec chmod 644 {} \;
|
||||||
|
# 特别处理 .sh 文件,给予执行权限 755
|
||||||
|
sudo find "$OUTPUT_DIR" -type f -name "*.sh" -exec chmod 755 {} \;
|
||||||
|
echo "[02] 权限设置完成 (UID:GID=$OWNER, dirs=755, files=644, scripts=755)"
|
||||||
|
|
||||||
|
echo "[02] 发布完成,验证文件..."
|
||||||
|
ls -lh "$OUTPUT_DIR"
|
||||||
|
|
||||||
|
echo "[02] 完成"
|
||||||
|
|
||||||
33
src/metric/tests/scripts/03_test_node_install.sh
Executable file
33
src/metric/tests/scripts/03_test_node_install.sh
Executable file
@@ -0,0 +1,33 @@
|
|||||||
|
#!/bin/bash
|
||||||
|
set -e
|
||||||
|
|
||||||
|
FTP_SERVER="${FTP_SERVER:-172.30.0.40}"
|
||||||
|
FTP_USER="${FTP_USER:-ftpuser}"
|
||||||
|
FTP_PASSWORD="${FTP_PASSWORD:-ZGClab1234!}"
|
||||||
|
FTP_PORT="${FTP_PORT:-21}"
|
||||||
|
|
||||||
|
FTP_HOST="${FTP_SERVER}"
|
||||||
|
|
||||||
|
echo "[03] 进入测试节点执行安装..."
|
||||||
|
echo "[03] 使用 FTP 地址: ${FTP_HOST}:${FTP_PORT}"
|
||||||
|
|
||||||
|
docker exec argus-metric-test-node bash -c "
|
||||||
|
set -e
|
||||||
|
|
||||||
|
if ! command -v curl &>/dev/null; then
|
||||||
|
echo '[03] curl 未安装,正在安装...'
|
||||||
|
apt-get update && apt-get install -y curl
|
||||||
|
fi
|
||||||
|
|
||||||
|
cd /tmp
|
||||||
|
echo '[03] 下载 setup.sh...'
|
||||||
|
curl -u ${FTP_USER}:${FTP_PASSWORD} ftp://${FTP_HOST}:${FTP_PORT}/setup.sh -o setup.sh
|
||||||
|
|
||||||
|
echo '[03] 执行安装...'
|
||||||
|
chmod +x setup.sh
|
||||||
|
bash setup.sh --server ${FTP_HOST} --user ${FTP_USER} --password '${FTP_PASSWORD}' --port ${FTP_PORT}
|
||||||
|
|
||||||
|
echo '[03] 安装完成'
|
||||||
|
"
|
||||||
|
|
||||||
|
echo "[03] 完成"
|
||||||
47
src/metric/tests/scripts/04_test_gpu_node_install.sh
Executable file
47
src/metric/tests/scripts/04_test_gpu_node_install.sh
Executable file
@@ -0,0 +1,47 @@
|
|||||||
|
#!/bin/bash
|
||||||
|
set -e
|
||||||
|
|
||||||
|
SCRIPT_DIR="$(cd "$(dirname "${BASH_SOURCE[0]}")" && pwd)"
|
||||||
|
COMMON_DIR="$SCRIPT_DIR/common"
|
||||||
|
|
||||||
|
FTP_SERVER="${FTP_SERVER:-172.30.0.40}"
|
||||||
|
FTP_USER="${FTP_USER:-ftpuser}"
|
||||||
|
FTP_PASSWORD="${FTP_PASSWORD:-ZGClab1234!}"
|
||||||
|
FTP_PORT="${FTP_PORT:-21}"
|
||||||
|
|
||||||
|
FTP_HOST="${FTP_SERVER}"
|
||||||
|
|
||||||
|
echo "[04] 检测GPU环境..."
|
||||||
|
# 检测GPU环境
|
||||||
|
if bash "$COMMON_DIR/check-gpu.sh"; then
|
||||||
|
echo "[04] GPU环境可用,继续执行GPU节点安装"
|
||||||
|
GPU_AVAILABLE=true
|
||||||
|
else
|
||||||
|
echo "[04] GPU环境不可用,跳过GPU节点安装"
|
||||||
|
GPU_AVAILABLE=false
|
||||||
|
exit 0
|
||||||
|
fi
|
||||||
|
|
||||||
|
echo "[04] 进入测试节点执行安装..."
|
||||||
|
echo "[04] 使用 FTP 地址: ${FTP_HOST}:${FTP_PORT}"
|
||||||
|
|
||||||
|
docker exec argus-metric-test-gpu-node bash -c "
|
||||||
|
set -e
|
||||||
|
|
||||||
|
if ! command -v curl &>/dev/null; then
|
||||||
|
echo '[04] curl 未安装,正在安装...'
|
||||||
|
apt-get update && apt-get install -y curl
|
||||||
|
fi
|
||||||
|
|
||||||
|
cd /tmp
|
||||||
|
echo '[04] 下载 setup.sh...'
|
||||||
|
curl -u ${FTP_USER}:${FTP_PASSWORD} ftp://${FTP_HOST}:${FTP_PORT}/setup.sh -o setup.sh
|
||||||
|
|
||||||
|
echo '[04] 执行安装...'
|
||||||
|
chmod +x setup.sh
|
||||||
|
bash setup.sh --server ${FTP_HOST} --user ${FTP_USER} --password '${FTP_PASSWORD}' --port ${FTP_PORT}
|
||||||
|
|
||||||
|
echo '[04] 安装完成'
|
||||||
|
"
|
||||||
|
|
||||||
|
echo "[04] 完成"
|
||||||
96
src/metric/tests/scripts/05_verify_install.sh
Executable file
96
src/metric/tests/scripts/05_verify_install.sh
Executable file
@@ -0,0 +1,96 @@
|
|||||||
|
#!/bin/bash
|
||||||
|
set -e
|
||||||
|
|
||||||
|
echo "[04] 验证安装结果 - 检查监控端口..."
|
||||||
|
echo "=========================================="
|
||||||
|
|
||||||
|
# 检查容器是否运行
|
||||||
|
if ! docker ps --format '{{.Names}}' | grep -q '^argus-metric-test-node$'; then
|
||||||
|
echo "错误: 容器 argus-metric-test-node 未运行"
|
||||||
|
exit 1
|
||||||
|
fi
|
||||||
|
|
||||||
|
ERRORS=0
|
||||||
|
|
||||||
|
# ==================== 检查监听端口 ====================
|
||||||
|
echo ""
|
||||||
|
echo "[1] 检查监听端口..."
|
||||||
|
echo "----------------------------------------"
|
||||||
|
CHECK_RESULT=$(docker exec argus-metric-test-node bash -c '
|
||||||
|
if command -v netstat >/dev/null 2>&1; then
|
||||||
|
echo "使用 netstat 检查端口:"
|
||||||
|
if netstat -tlnp 2>/dev/null | grep -E ":(9100|9400|2020)"; then
|
||||||
|
echo "✓ 找到监控端口"
|
||||||
|
exit 0
|
||||||
|
else
|
||||||
|
echo "✗ 未找到监控端口 (9100/9400/2020)"
|
||||||
|
exit 1
|
||||||
|
fi
|
||||||
|
elif command -v ss >/dev/null 2>&1; then
|
||||||
|
echo "使用 ss 检查端口:"
|
||||||
|
if ss -tlnp 2>/dev/null | grep -E ":(9100|9400|2020)"; then
|
||||||
|
echo "✓ 找到监控端口"
|
||||||
|
exit 0
|
||||||
|
else
|
||||||
|
echo "✗ 未找到监控端口 (9100/9400/2020)"
|
||||||
|
exit 1
|
||||||
|
fi
|
||||||
|
elif command -v lsof >/dev/null 2>&1; then
|
||||||
|
echo "使用 lsof 检查端口:"
|
||||||
|
if lsof -i :9100 -i :9400 -i :2020 2>/dev/null | grep LISTEN; then
|
||||||
|
echo "✓ 找到监控端口"
|
||||||
|
exit 0
|
||||||
|
else
|
||||||
|
echo "✗ 未找到监控端口 (9100/9400/2020)"
|
||||||
|
exit 1
|
||||||
|
fi
|
||||||
|
else
|
||||||
|
echo "? 没有可用的端口检查工具 (netstat/ss/lsof),跳过此检查"
|
||||||
|
exit 0
|
||||||
|
fi
|
||||||
|
')
|
||||||
|
echo "$CHECK_RESULT"
|
||||||
|
# 只有在明确失败时才计入错误(exit 1),没有工具(exit 0)不算错误
|
||||||
|
if echo "$CHECK_RESULT" | grep -q "✗ 未找到监控端口"; then
|
||||||
|
ERRORS=$((ERRORS + 1))
|
||||||
|
fi
|
||||||
|
|
||||||
|
# ==================== 测试端口连通性 ====================
|
||||||
|
echo ""
|
||||||
|
echo "[2] 测试端口连通性..."
|
||||||
|
echo "----------------------------------------"
|
||||||
|
docker exec argus-metric-test-node bash -c '
|
||||||
|
if command -v curl >/dev/null 2>&1; then
|
||||||
|
FAILED=0
|
||||||
|
for port in 9100 9400 2020; do
|
||||||
|
echo -n "端口 $port: "
|
||||||
|
if curl -s --connect-timeout 2 "http://localhost:$port/metrics" > /dev/null 2>&1; then
|
||||||
|
echo "✓ 可访问 (/metrics)"
|
||||||
|
elif curl -s --connect-timeout 2 "http://localhost:$port/" > /dev/null 2>&1; then
|
||||||
|
echo "✓ 可访问 (根路径)"
|
||||||
|
else
|
||||||
|
echo "✗ 不可访问"
|
||||||
|
FAILED=$((FAILED + 1))
|
||||||
|
fi
|
||||||
|
done
|
||||||
|
exit $FAILED
|
||||||
|
else
|
||||||
|
echo "? curl 不可用,跳过连通性测试"
|
||||||
|
exit 0
|
||||||
|
fi
|
||||||
|
' || ERRORS=$((ERRORS + 1))
|
||||||
|
|
||||||
|
echo ""
|
||||||
|
echo "=========================================="
|
||||||
|
if [ $ERRORS -eq 0 ]; then
|
||||||
|
echo "✓ [04] 验证完成 - 所有端口检查通过"
|
||||||
|
else
|
||||||
|
echo "✗ [04] 验证失败 - 发现 $ERRORS 个问题"
|
||||||
|
echo ""
|
||||||
|
echo "调试建议:"
|
||||||
|
echo " 1. 进入容器检查: docker exec -it argus-metric-test-node bash"
|
||||||
|
echo " 2. 查看进程: docker exec argus-metric-test-node ps aux"
|
||||||
|
echo " 3. 查看日志: docker exec argus-metric-test-node cat /tmp/argus_install.log"
|
||||||
|
exit 1
|
||||||
|
fi
|
||||||
|
echo "=========================================="
|
||||||
11
src/metric/tests/scripts/06_cleanup.sh
Executable file
11
src/metric/tests/scripts/06_cleanup.sh
Executable file
@@ -0,0 +1,11 @@
|
|||||||
|
#!/bin/bash
|
||||||
|
set -e
|
||||||
|
|
||||||
|
SCRIPT_DIR="$(cd "$(dirname "$0")" && pwd)"
|
||||||
|
|
||||||
|
echo "[05] 清理环境..."
|
||||||
|
|
||||||
|
bash "$SCRIPT_DIR/common/stop-all.sh" || true
|
||||||
|
|
||||||
|
echo "[05] 清理完成"
|
||||||
|
|
||||||
59
src/metric/tests/scripts/common/check-gpu.sh
Executable file
59
src/metric/tests/scripts/common/check-gpu.sh
Executable file
@@ -0,0 +1,59 @@
|
|||||||
|
#!/bin/bash
|
||||||
|
|
||||||
|
# GPU环境检测脚本
|
||||||
|
# 检测系统是否有NVIDIA GPU硬件
|
||||||
|
|
||||||
|
set -e
|
||||||
|
|
||||||
|
# 检测函数
|
||||||
|
check_gpu_support() {
|
||||||
|
echo "检测GPU环境..."
|
||||||
|
|
||||||
|
# 方法1: 检测GPU设备文件
|
||||||
|
if ls /dev/nvidia* &>/dev/null; then
|
||||||
|
echo "✓ 检测到NVIDIA GPU设备文件"
|
||||||
|
return 0
|
||||||
|
fi
|
||||||
|
|
||||||
|
# 方法2: 检测lspci中的NVIDIA设备(Linux)
|
||||||
|
if command -v lspci &> /dev/null; then
|
||||||
|
if lspci | grep -i nvidia &> /dev/null; then
|
||||||
|
echo "✓ 检测到NVIDIA GPU硬件"
|
||||||
|
return 0
|
||||||
|
fi
|
||||||
|
fi
|
||||||
|
|
||||||
|
# 方法3: 检测nvidia-smi
|
||||||
|
if command -v nvidia-smi &> /dev/null; then
|
||||||
|
if nvidia-smi &> /dev/null; then
|
||||||
|
echo "✓ 检测到NVIDIA GPU硬件"
|
||||||
|
return 0
|
||||||
|
fi
|
||||||
|
fi
|
||||||
|
|
||||||
|
echo "✗ 未检测到NVIDIA GPU硬件"
|
||||||
|
return 1
|
||||||
|
}
|
||||||
|
|
||||||
|
# 主函数
|
||||||
|
main() {
|
||||||
|
echo "=========================================="
|
||||||
|
echo " GPU环境检测"
|
||||||
|
echo "=========================================="
|
||||||
|
echo ""
|
||||||
|
|
||||||
|
if check_gpu_support; then
|
||||||
|
echo ""
|
||||||
|
echo "结果: GPU环境可用"
|
||||||
|
exit 0
|
||||||
|
else
|
||||||
|
echo ""
|
||||||
|
echo "结果: GPU环境不可用,将跳过GPU相关服务"
|
||||||
|
exit 1
|
||||||
|
fi
|
||||||
|
}
|
||||||
|
|
||||||
|
# 如果直接运行此脚本
|
||||||
|
if [ "${BASH_SOURCE[0]}" = "${0}" ]; then
|
||||||
|
main "$@"
|
||||||
|
fi
|
||||||
@@ -6,7 +6,8 @@
|
|||||||
set -e
|
set -e
|
||||||
|
|
||||||
SCRIPT_DIR="$(cd "$(dirname "${BASH_SOURCE[0]}")" && pwd)"
|
SCRIPT_DIR="$(cd "$(dirname "${BASH_SOURCE[0]}")" && pwd)"
|
||||||
cd "$SCRIPT_DIR"
|
TEST_DIR="$(cd "$SCRIPT_DIR/../.." && pwd)"
|
||||||
|
cd "$TEST_DIR"
|
||||||
|
|
||||||
echo "=========================================="
|
echo "=========================================="
|
||||||
echo " 路径检查脚本"
|
echo " 路径检查脚本"
|
||||||
@@ -18,15 +19,15 @@ echo ""
|
|||||||
|
|
||||||
# 检查配置文件
|
# 检查配置文件
|
||||||
echo "检查配置文件..."
|
echo "检查配置文件..."
|
||||||
if [ -f "$SCRIPT_DIR/docker-compose.yml" ]; then
|
if [ -f "$TEST_DIR/docker-compose.yml" ]; then
|
||||||
echo " ✓ docker-compose.yml 存在"
|
echo " ✓ docker-compose.yml 存在"
|
||||||
else
|
else
|
||||||
echo " ✗ docker-compose.yml 不存在"
|
echo " ✗ docker-compose.yml 不存在"
|
||||||
fi
|
fi
|
||||||
|
|
||||||
if [ -f "$SCRIPT_DIR/.env" ]; then
|
if [ -f "$TEST_DIR/.env" ]; then
|
||||||
echo " ✓ .env 存在"
|
echo " ✓ .env 存在"
|
||||||
elif [ -f "$SCRIPT_DIR/env.example" ]; then
|
elif [ -f "$TEST_DIR/env.example" ]; then
|
||||||
echo " ⚠ .env 不存在,但 env.example 存在"
|
echo " ⚠ .env 不存在,但 env.example 存在"
|
||||||
else
|
else
|
||||||
echo " ✗ .env 和 env.example 都不存在"
|
echo " ✗ .env 和 env.example 都不存在"
|
||||||
61
src/metric/tests/scripts/common/init-directories.sh
Executable file
61
src/metric/tests/scripts/common/init-directories.sh
Executable file
@@ -0,0 +1,61 @@
|
|||||||
|
#!/bin/bash
|
||||||
|
|
||||||
|
# 初始化目录脚本
|
||||||
|
|
||||||
|
set -e
|
||||||
|
|
||||||
|
SCRIPT_DIR="$(cd "$(dirname "${BASH_SOURCE[0]}")" && pwd)"
|
||||||
|
TEST_DIR="$(cd "$SCRIPT_DIR/../.." && pwd)"
|
||||||
|
cd "$TEST_DIR"
|
||||||
|
|
||||||
|
# 加载 .env 文件(如果存在)
|
||||||
|
if [ -f .env ]; then
|
||||||
|
echo "加载 .env 配置文件..."
|
||||||
|
source .env
|
||||||
|
fi
|
||||||
|
|
||||||
|
# 默认配置
|
||||||
|
ARGUS_BUILD_UID=${ARGUS_BUILD_UID:-2133}
|
||||||
|
ARGUS_BUILD_GID=${ARGUS_BUILD_GID:-2015}
|
||||||
|
DATA_ROOT=${DATA_ROOT:-/private}
|
||||||
|
|
||||||
|
echo "开始初始化目录结构..."
|
||||||
|
echo "数据根目录: ${DATA_ROOT}"
|
||||||
|
echo "统一 UID: ${ARGUS_BUILD_UID}"
|
||||||
|
echo "统一 GID: ${ARGUS_BUILD_GID}"
|
||||||
|
|
||||||
|
# 创建基础目录结构
|
||||||
|
echo "创建基础目录结构..."
|
||||||
|
sudo mkdir -p ${DATA_ROOT}/argus/metric
|
||||||
|
sudo mkdir -p ${DATA_ROOT}/argus/etc
|
||||||
|
sudo mkdir -p ${DATA_ROOT}/argus/agent
|
||||||
|
|
||||||
|
# 创建 FTP 目录
|
||||||
|
echo "创建 FTP 目录..."
|
||||||
|
sudo mkdir -p ${DATA_ROOT}/argus/metric/ftp/share
|
||||||
|
|
||||||
|
# 创建 Prometheus 目录
|
||||||
|
echo "创建 Prometheus 目录..."
|
||||||
|
sudo mkdir -p ${DATA_ROOT}/argus/metric/prometheus/{data,rules,targets}
|
||||||
|
|
||||||
|
# 创建 Grafana 目录
|
||||||
|
echo "创建 Grafana 目录..."
|
||||||
|
sudo mkdir -p ${DATA_ROOT}/argus/metric/grafana/{data,logs,plugins,provisioning/datasources,provisioning/dashboards,data/sessions,data/dashboards,config}
|
||||||
|
|
||||||
|
# 统一设置所有目录权限
|
||||||
|
echo "设置目录权限..."
|
||||||
|
sudo chown -R ${ARGUS_BUILD_UID}:${ARGUS_BUILD_GID} ${DATA_ROOT}/argus/metric
|
||||||
|
sudo chmod -R 755 ${DATA_ROOT}/argus/metric
|
||||||
|
|
||||||
|
echo "目录初始化完成!"
|
||||||
|
echo ""
|
||||||
|
echo "目录结构:"
|
||||||
|
echo " ${DATA_ROOT}/"
|
||||||
|
echo " ├── argus/ (UID:${ARGUS_BUILD_UID}, GID:${ARGUS_BUILD_GID})"
|
||||||
|
echo " │ ├── metric/"
|
||||||
|
echo " │ │ ├── ftp/"
|
||||||
|
echo " │ │ ├── prometheus/"
|
||||||
|
echo " │ │ └── grafana/"
|
||||||
|
echo ""
|
||||||
|
echo "您现在可以运行 'docker-compose up -d' 来启动所有服务"
|
||||||
|
|
||||||
@@ -6,7 +6,8 @@
|
|||||||
set -e
|
set -e
|
||||||
|
|
||||||
SCRIPT_DIR="$(cd "$(dirname "${BASH_SOURCE[0]}")" && pwd)"
|
SCRIPT_DIR="$(cd "$(dirname "${BASH_SOURCE[0]}")" && pwd)"
|
||||||
cd "$SCRIPT_DIR"
|
TEST_DIR="$(cd "$SCRIPT_DIR/../.." && pwd)"
|
||||||
|
cd "$TEST_DIR"
|
||||||
|
|
||||||
# 检测 docker-compose 命令
|
# 检测 docker-compose 命令
|
||||||
if command -v docker-compose &> /dev/null; then
|
if command -v docker-compose &> /dev/null; then
|
||||||
@@ -19,7 +20,7 @@ else
|
|||||||
fi
|
fi
|
||||||
|
|
||||||
# 镜像缓存目录
|
# 镜像缓存目录
|
||||||
IMAGE_CACHE_DIR="./images-cache"
|
IMAGE_CACHE_DIR="$TEST_DIR/images-cache"
|
||||||
mkdir -p "$IMAGE_CACHE_DIR"
|
mkdir -p "$IMAGE_CACHE_DIR"
|
||||||
|
|
||||||
# 定义镜像列表
|
# 定义镜像列表
|
||||||
125
src/metric/tests/scripts/common/start-all.sh
Executable file
125
src/metric/tests/scripts/common/start-all.sh
Executable file
@@ -0,0 +1,125 @@
|
|||||||
|
#!/bin/bash
|
||||||
|
|
||||||
|
# 一键启动脚本
|
||||||
|
# 用于初始化目录并启动所有服务
|
||||||
|
# 镜像构建已移至 build/build_images.sh
|
||||||
|
|
||||||
|
set -e
|
||||||
|
|
||||||
|
SCRIPT_DIR="$(cd "$(dirname "${BASH_SOURCE[0]}")" && pwd)"
|
||||||
|
TEST_DIR="$(cd "$SCRIPT_DIR/../.." && pwd)"
|
||||||
|
cd "$TEST_DIR"
|
||||||
|
|
||||||
|
echo "=========================================="
|
||||||
|
echo " Argus Metrics 一键启动脚本"
|
||||||
|
echo "=========================================="
|
||||||
|
echo ""
|
||||||
|
echo "当前工作目录: $TEST_DIR"
|
||||||
|
echo ""
|
||||||
|
|
||||||
|
# 检查 Docker 和 Docker Compose
|
||||||
|
if ! command -v docker &> /dev/null; then
|
||||||
|
echo "错误: 未找到 docker 命令,请先安装 Docker"
|
||||||
|
exit 1
|
||||||
|
fi
|
||||||
|
|
||||||
|
# 检查 docker compose 命令
|
||||||
|
if ! docker compose version &> /dev/null 2>&1; then
|
||||||
|
echo "错误: 未找到 docker compose 命令,请确保 Docker Compose V2 已安装"
|
||||||
|
exit 1
|
||||||
|
fi
|
||||||
|
echo "使用: docker compose"
|
||||||
|
echo "Compose 文件: $TEST_DIR/docker-compose.yml"
|
||||||
|
echo ""
|
||||||
|
|
||||||
|
|
||||||
|
# 检查并创建 .env 文件
|
||||||
|
if [ ! -f .env ]; then
|
||||||
|
echo "未找到 .env 文件,从 env.example 创建..."
|
||||||
|
cp env.example .env
|
||||||
|
echo "已创建 .env 文件,请根据需要修改配置"
|
||||||
|
fi
|
||||||
|
|
||||||
|
# 加载环境变量
|
||||||
|
source .env
|
||||||
|
|
||||||
|
# 检查并创建 Docker 网络
|
||||||
|
echo "检查 Docker 网络..."
|
||||||
|
NETWORK_NAME="argus-debug-net"
|
||||||
|
if docker network inspect "$NETWORK_NAME" >/dev/null 2>&1; then
|
||||||
|
echo "网络 $NETWORK_NAME 已存在"
|
||||||
|
else
|
||||||
|
echo "创建网络 $NETWORK_NAME..."
|
||||||
|
docker network create --driver bridge --subnet 172.30.0.0/16 "$NETWORK_NAME"
|
||||||
|
echo "网络创建成功"
|
||||||
|
fi
|
||||||
|
echo ""
|
||||||
|
|
||||||
|
echo "1. 初始化目录结构..."
|
||||||
|
bash "$SCRIPT_DIR/init-directories.sh"
|
||||||
|
|
||||||
|
echo ""
|
||||||
|
echo "2. 检测GPU环境..."
|
||||||
|
# 检测GPU环境
|
||||||
|
if bash "$SCRIPT_DIR/check-gpu.sh"; then
|
||||||
|
echo "GPU环境可用,将启动GPU节点"
|
||||||
|
GPU_AVAILABLE=true
|
||||||
|
else
|
||||||
|
echo "GPU环境不可用,跳过GPU节点"
|
||||||
|
GPU_AVAILABLE=false
|
||||||
|
fi
|
||||||
|
|
||||||
|
echo ""
|
||||||
|
echo "3. 检查 Docker 镜像..."
|
||||||
|
|
||||||
|
# 检查必要的镜像是否存在
|
||||||
|
BASE_IMAGES=("argus-metric-ftp:latest" "argus-metric-prometheus:latest" "argus-metric-grafana:latest" "argus-metric-test-node:latest")
|
||||||
|
GPU_IMAGES=("argus-metric-test-gpu-node:latest")
|
||||||
|
|
||||||
|
# 先检查基础镜像
|
||||||
|
missing_images=()
|
||||||
|
for image in "${BASE_IMAGES[@]}"; do
|
||||||
|
if ! docker images --format "{{.Repository}}:{{.Tag}}" | grep -q "^${image}$"; then
|
||||||
|
missing_images+=("$image")
|
||||||
|
fi
|
||||||
|
done
|
||||||
|
|
||||||
|
# 检查GPU镜像(如果GPU环境可用)
|
||||||
|
if [ "$GPU_AVAILABLE" = true ]; then
|
||||||
|
for image in "${GPU_IMAGES[@]}"; do
|
||||||
|
if ! docker images --format "{{.Repository}}:{{.Tag}}" | grep -q "^${image}$"; then
|
||||||
|
missing_images+=("$image")
|
||||||
|
fi
|
||||||
|
done
|
||||||
|
fi
|
||||||
|
|
||||||
|
if [ ${#missing_images[@]} -gt 0 ]; then
|
||||||
|
echo "以下镜像缺失,请先运行 build/build_images.sh 构建镜像:"
|
||||||
|
for image in "${missing_images[@]}"; do
|
||||||
|
echo " • $image"
|
||||||
|
done
|
||||||
|
echo ""
|
||||||
|
echo "构建命令:"
|
||||||
|
echo " ./build/build_images.sh --metric"
|
||||||
|
exit 1
|
||||||
|
else
|
||||||
|
echo "所有必要镜像已存在"
|
||||||
|
fi
|
||||||
|
|
||||||
|
echo ""
|
||||||
|
echo "4. 启动基础服务..."
|
||||||
|
cd "$TEST_DIR"
|
||||||
|
|
||||||
|
# 根据GPU环境决定启动的服务
|
||||||
|
if [ "$GPU_AVAILABLE" = true ]; then
|
||||||
|
echo "启动所有服务(包括GPU节点)..."
|
||||||
|
docker compose up -d ftp prometheus grafana test-node test-gpu-node
|
||||||
|
else
|
||||||
|
echo "启动基础服务(跳过GPU节点)..."
|
||||||
|
docker compose up -d ftp prometheus grafana test-node
|
||||||
|
fi
|
||||||
|
|
||||||
|
echo ""
|
||||||
|
echo "4. 等待服务启动..."
|
||||||
|
sleep 5
|
||||||
|
|
||||||
50
src/metric/tests/scripts/common/stop-all.sh
Executable file
50
src/metric/tests/scripts/common/stop-all.sh
Executable file
@@ -0,0 +1,50 @@
|
|||||||
|
#!/bin/bash
|
||||||
|
|
||||||
|
# 停止所有服务脚本
|
||||||
|
|
||||||
|
set -e
|
||||||
|
|
||||||
|
SCRIPT_DIR="$(cd "$(dirname "${BASH_SOURCE[0]}")" && pwd)"
|
||||||
|
TEST_DIR="$(cd "$SCRIPT_DIR/../.." && pwd)"
|
||||||
|
cd "$TEST_DIR"
|
||||||
|
|
||||||
|
# 检查 docker compose 命令
|
||||||
|
if ! docker compose version &> /dev/null 2>&1; then
|
||||||
|
echo "错误: 未找到 docker compose 命令,请确保 Docker Compose V2 已安装"
|
||||||
|
exit 1
|
||||||
|
fi
|
||||||
|
|
||||||
|
echo "=========================================="
|
||||||
|
echo " 停止 Argus Metrics 服务"
|
||||||
|
echo "=========================================="
|
||||||
|
echo ""
|
||||||
|
echo "使用: docker compose"
|
||||||
|
echo "Compose 文件: $TEST_DIR/docker-compose.yml"
|
||||||
|
echo ""
|
||||||
|
|
||||||
|
# 检查是否有运行的容器
|
||||||
|
if [ "$(docker compose ps -q)" ]; then
|
||||||
|
echo "停止所有服务..."
|
||||||
|
docker compose stop
|
||||||
|
|
||||||
|
echo ""
|
||||||
|
read -p "是否要删除容器? (y/N): " -n 1 -r
|
||||||
|
echo
|
||||||
|
if [[ $REPLY =~ ^[Yy]$ ]]; then
|
||||||
|
docker compose down
|
||||||
|
echo "容器已删除"
|
||||||
|
|
||||||
|
read -p "是否要删除数据卷? (y/N): " -n 1 -r
|
||||||
|
echo
|
||||||
|
if [[ $REPLY =~ ^[Yy]$ ]]; then
|
||||||
|
docker compose down -v
|
||||||
|
echo "数据卷已删除"
|
||||||
|
fi
|
||||||
|
fi
|
||||||
|
else
|
||||||
|
echo "没有运行的服务"
|
||||||
|
fi
|
||||||
|
|
||||||
|
echo ""
|
||||||
|
echo "完成!"
|
||||||
|
|
||||||
85
src/metric/tests/scripts/load-images.sh
Executable file
85
src/metric/tests/scripts/load-images.sh
Executable file
@@ -0,0 +1,85 @@
|
|||||||
|
#!/bin/bash
|
||||||
|
|
||||||
|
# 镜像加载脚本
|
||||||
|
# 用于从 tar 文件加载 Docker 镜像
|
||||||
|
|
||||||
|
set -e
|
||||||
|
|
||||||
|
SCRIPT_DIR="$(cd "$(dirname "${BASH_SOURCE[0]}")" && pwd)"
|
||||||
|
TEST_DIR="$(cd "$SCRIPT_DIR/.." && pwd)"
|
||||||
|
INPUT_DIR="${1:-$TEST_DIR/images-cache}"
|
||||||
|
|
||||||
|
echo "=========================================="
|
||||||
|
echo " Docker 镜像加载脚本"
|
||||||
|
echo "=========================================="
|
||||||
|
echo ""
|
||||||
|
echo "输入目录: $INPUT_DIR"
|
||||||
|
echo ""
|
||||||
|
|
||||||
|
# 检查输入目录是否存在
|
||||||
|
if [ ! -d "$INPUT_DIR" ]; then
|
||||||
|
echo "错误: 目录不存在: $INPUT_DIR"
|
||||||
|
exit 1
|
||||||
|
fi
|
||||||
|
|
||||||
|
# 查找所有tar文件并加载
|
||||||
|
total=0
|
||||||
|
success=0
|
||||||
|
failed=0
|
||||||
|
|
||||||
|
# 查找目录下所有.tar文件
|
||||||
|
tar_files=($(find "$INPUT_DIR" -name "*.tar" -type f 2>/dev/null | sort))
|
||||||
|
|
||||||
|
if [ ${#tar_files[@]} -eq 0 ]; then
|
||||||
|
echo "错误: 在目录 $INPUT_DIR 中未找到任何 .tar 文件"
|
||||||
|
exit 1
|
||||||
|
fi
|
||||||
|
|
||||||
|
echo "找到 ${#tar_files[@]} 个镜像文件:"
|
||||||
|
for tar_file in "${tar_files[@]}"; do
|
||||||
|
echo " - $(basename "$tar_file")"
|
||||||
|
done
|
||||||
|
echo ""
|
||||||
|
|
||||||
|
for tar_file in "${tar_files[@]}"; do
|
||||||
|
total=$((total + 1))
|
||||||
|
tar_filename=$(basename "$tar_file")
|
||||||
|
|
||||||
|
echo "[$total] 处理: $tar_filename"
|
||||||
|
|
||||||
|
# 强制加载,不检查镜像是否已存在
|
||||||
|
echo " 加载镜像..."
|
||||||
|
if docker load -i "$tar_file"; then
|
||||||
|
echo " 加载成功: $tar_filename"
|
||||||
|
success=$((success + 1))
|
||||||
|
else
|
||||||
|
echo " 加载失败: $tar_filename"
|
||||||
|
failed=$((failed + 1))
|
||||||
|
fi
|
||||||
|
echo ""
|
||||||
|
done
|
||||||
|
|
||||||
|
echo "=========================================="
|
||||||
|
echo " 加载完成"
|
||||||
|
echo "=========================================="
|
||||||
|
echo ""
|
||||||
|
echo "统计:"
|
||||||
|
echo " 总计: $total"
|
||||||
|
echo " 成功: $success"
|
||||||
|
echo " 失败: $failed"
|
||||||
|
echo ""
|
||||||
|
|
||||||
|
# 显示当前所有镜像
|
||||||
|
echo "当前所有镜像:"
|
||||||
|
docker images
|
||||||
|
echo ""
|
||||||
|
|
||||||
|
if [ $failed -gt 0 ]; then
|
||||||
|
echo "部分镜像加载失败,请检查!"
|
||||||
|
exit 1
|
||||||
|
fi
|
||||||
|
|
||||||
|
if [ $success -gt 0 ]; then
|
||||||
|
echo "镜像加载成功!"
|
||||||
|
fi
|
||||||
|
|
||||||
94
src/metric/tests/scripts/save-images.sh
Executable file
94
src/metric/tests/scripts/save-images.sh
Executable file
@@ -0,0 +1,94 @@
|
|||||||
|
#!/bin/bash
|
||||||
|
|
||||||
|
# 镜像保存脚本
|
||||||
|
# 用于保存 Docker 镜像到 tar 文件,便于离线部署
|
||||||
|
|
||||||
|
set -e
|
||||||
|
|
||||||
|
SCRIPT_DIR="$(cd "$(dirname "${BASH_SOURCE[0]}")" && pwd)"
|
||||||
|
TEST_DIR="$(cd "$SCRIPT_DIR/.." && pwd)"
|
||||||
|
OUTPUT_DIR="${1:-$TEST_DIR/images-cache}"
|
||||||
|
|
||||||
|
echo "=========================================="
|
||||||
|
echo " Docker 镜像保存脚本"
|
||||||
|
echo "=========================================="
|
||||||
|
echo ""
|
||||||
|
echo "输出目录: $OUTPUT_DIR"
|
||||||
|
echo ""
|
||||||
|
|
||||||
|
# 创建输出目录
|
||||||
|
mkdir -p "$OUTPUT_DIR"
|
||||||
|
|
||||||
|
# 定义镜像名称(与 docker-compose.yml 保持一致)
|
||||||
|
declare -A IMAGES=(
|
||||||
|
["argus-metric-ftp:latest"]="argus-ftp.tar"
|
||||||
|
["argus-metric-prometheus:latest"]="argus-prometheus.tar"
|
||||||
|
["argus-metric-grafana:latest"]="argus-grafana.tar"
|
||||||
|
["argus-metric-test-node:latest"]="argus-test-node.tar"
|
||||||
|
["argus-metric-test-gpu-node:latest"]="argus-test-gpu-node.tar"
|
||||||
|
)
|
||||||
|
|
||||||
|
# 检查镜像是否存在并保存
|
||||||
|
total=0
|
||||||
|
success=0
|
||||||
|
failed=0
|
||||||
|
|
||||||
|
for image in "${!IMAGES[@]}"; do
|
||||||
|
total=$((total + 1))
|
||||||
|
output_file="${OUTPUT_DIR}/${IMAGES[$image]}"
|
||||||
|
|
||||||
|
echo "[$total] 检查镜像: $image"
|
||||||
|
|
||||||
|
if docker images --format "{{.Repository}}:{{.Tag}}" | grep -q "^${image}$"; then
|
||||||
|
echo " ✓ 镜像存在,开始保存..."
|
||||||
|
if docker save -o "$output_file" "$image"; then
|
||||||
|
file_size=$(ls -lh "$output_file" | awk '{print $5}')
|
||||||
|
echo " ✓ 保存成功: ${IMAGES[$image]} ($file_size)"
|
||||||
|
success=$((success + 1))
|
||||||
|
else
|
||||||
|
echo " ✗ 保存失败: $image"
|
||||||
|
failed=$((failed + 1))
|
||||||
|
fi
|
||||||
|
else
|
||||||
|
echo " ✗ 镜像不存在,请先构建镜像"
|
||||||
|
failed=$((failed + 1))
|
||||||
|
fi
|
||||||
|
echo ""
|
||||||
|
done
|
||||||
|
|
||||||
|
echo "=========================================="
|
||||||
|
echo " 保存完成"
|
||||||
|
echo "=========================================="
|
||||||
|
echo ""
|
||||||
|
echo "统计:"
|
||||||
|
echo " 总计: $total"
|
||||||
|
echo " 成功: $success"
|
||||||
|
echo " 失败: $failed"
|
||||||
|
echo ""
|
||||||
|
echo "输出目录: $OUTPUT_DIR"
|
||||||
|
echo ""
|
||||||
|
|
||||||
|
if [ $success -gt 0 ]; then
|
||||||
|
echo "已保存的文件:"
|
||||||
|
ls -lh "$OUTPUT_DIR"/*.tar 2>/dev/null || true
|
||||||
|
echo ""
|
||||||
|
echo "文件列表:"
|
||||||
|
for image in "${!IMAGES[@]}"; do
|
||||||
|
output_file="${OUTPUT_DIR}/${IMAGES[$image]}"
|
||||||
|
if [ -f "$output_file" ]; then
|
||||||
|
file_size=$(ls -lh "$output_file" | awk '{print $5}')
|
||||||
|
echo " - ${IMAGES[$image]} ($file_size)"
|
||||||
|
fi
|
||||||
|
done
|
||||||
|
fi
|
||||||
|
|
||||||
|
echo ""
|
||||||
|
echo "使用说明:"
|
||||||
|
echo "1. 将 images-cache 目录复制到目标服务器的 ~/argus/src/metric/tests/ 下"
|
||||||
|
echo "2. 在目标服务器运行: bash scripts/common/start-all.sh"
|
||||||
|
echo ""
|
||||||
|
|
||||||
|
if [ $failed -gt 0 ]; then
|
||||||
|
exit 1
|
||||||
|
fi
|
||||||
|
|
||||||
@@ -1,199 +0,0 @@
|
|||||||
#!/bin/bash
|
|
||||||
|
|
||||||
# 一键启动脚本
|
|
||||||
# 用于初始化目录、构建镜像并启动所有服务
|
|
||||||
|
|
||||||
set -e
|
|
||||||
|
|
||||||
SCRIPT_DIR="$(cd "$(dirname "${BASH_SOURCE[0]}")" && pwd)"
|
|
||||||
cd "$SCRIPT_DIR"
|
|
||||||
|
|
||||||
echo "=========================================="
|
|
||||||
echo " Argus Metrics 一键启动脚本"
|
|
||||||
echo "=========================================="
|
|
||||||
echo ""
|
|
||||||
echo "当前工作目录: $SCRIPT_DIR"
|
|
||||||
echo ""
|
|
||||||
|
|
||||||
# 检查 Docker 和 Docker Compose
|
|
||||||
if ! command -v docker &> /dev/null; then
|
|
||||||
echo "错误: 未找到 docker 命令,请先安装 Docker"
|
|
||||||
exit 1
|
|
||||||
fi
|
|
||||||
|
|
||||||
# 检测 docker-compose 命令(兼容新旧版本)
|
|
||||||
COMPOSE_FILE="$SCRIPT_DIR/docker-compose.yml"
|
|
||||||
if command -v docker-compose &> /dev/null; then
|
|
||||||
DOCKER_COMPOSE="docker-compose -f $COMPOSE_FILE"
|
|
||||||
echo "使用: docker-compose"
|
|
||||||
elif docker compose version &> /dev/null 2>&1; then
|
|
||||||
DOCKER_COMPOSE="docker compose -f $COMPOSE_FILE"
|
|
||||||
echo "使用: docker compose"
|
|
||||||
else
|
|
||||||
echo "错误: 未找到 docker-compose 或 docker compose 命令"
|
|
||||||
exit 1
|
|
||||||
fi
|
|
||||||
echo "Compose 文件: $COMPOSE_FILE"
|
|
||||||
echo ""
|
|
||||||
|
|
||||||
# 检查必要的构建目录
|
|
||||||
echo "检查构建目录..."
|
|
||||||
BUILD_DIRS=(
|
|
||||||
"../ftp/build"
|
|
||||||
"../prometheus/build"
|
|
||||||
"../grafana/build"
|
|
||||||
)
|
|
||||||
|
|
||||||
for dir in "${BUILD_DIRS[@]}"; do
|
|
||||||
if [ ! -d "$dir" ]; then
|
|
||||||
echo "错误: 构建目录不存在: $dir"
|
|
||||||
echo "完整路径: $(cd "$(dirname "$dir")" 2>/dev/null && pwd)/$(basename "$dir")"
|
|
||||||
exit 1
|
|
||||||
else
|
|
||||||
echo " ✓ 找到: $dir"
|
|
||||||
fi
|
|
||||||
done
|
|
||||||
echo ""
|
|
||||||
|
|
||||||
# 检查并创建 .env 文件
|
|
||||||
if [ ! -f .env ]; then
|
|
||||||
echo "未找到 .env 文件,从 env.example 创建..."
|
|
||||||
cp env.example .env
|
|
||||||
echo "已创建 .env 文件,请根据需要修改配置"
|
|
||||||
fi
|
|
||||||
|
|
||||||
# 加载环境变量
|
|
||||||
source .env
|
|
||||||
|
|
||||||
echo "1. 初始化目录结构..."
|
|
||||||
bash "$SCRIPT_DIR/init-directories.sh"
|
|
||||||
|
|
||||||
echo ""
|
|
||||||
echo "2. 准备 Docker 镜像..."
|
|
||||||
|
|
||||||
# 检查镜像是否存在
|
|
||||||
IMAGE_CACHE_DIR="./images-cache"
|
|
||||||
IMAGES=("argus-metric-ftp:latest" "argus-metric-prometheus:latest" "argus-metric-grafana:latest")
|
|
||||||
all_images_exist=true
|
|
||||||
|
|
||||||
for image in "${IMAGES[@]}"; do
|
|
||||||
if ! docker images --format "{{.Repository}}:{{.Tag}}" | grep -q "^${image}$"; then
|
|
||||||
all_images_exist=false
|
|
||||||
break
|
|
||||||
fi
|
|
||||||
done
|
|
||||||
|
|
||||||
if $all_images_exist; then
|
|
||||||
echo "所有镜像已存在,跳过构建"
|
|
||||||
else
|
|
||||||
echo "检测到缺失镜像,尝试从缓存加载..."
|
|
||||||
|
|
||||||
# 尝试从缓存加载
|
|
||||||
loaded_from_cache=false
|
|
||||||
if [ -d "$IMAGE_CACHE_DIR" ]; then
|
|
||||||
for image in "${IMAGES[@]}"; do
|
|
||||||
if ! docker images --format "{{.Repository}}:{{.Tag}}" | grep -q "^${image}$"; then
|
|
||||||
# 镜像不存在,尝试加载
|
|
||||||
case "$image" in
|
|
||||||
"argus-metric-ftp:latest")
|
|
||||||
cache_file="${IMAGE_CACHE_DIR}/argus-ftp.tar"
|
|
||||||
;;
|
|
||||||
"argus-metric-prometheus:latest")
|
|
||||||
cache_file="${IMAGE_CACHE_DIR}/argus-prometheus.tar"
|
|
||||||
;;
|
|
||||||
"argus-metric-grafana:latest")
|
|
||||||
cache_file="${IMAGE_CACHE_DIR}/argus-grafana.tar"
|
|
||||||
;;
|
|
||||||
esac
|
|
||||||
|
|
||||||
if [ -f "$cache_file" ]; then
|
|
||||||
echo " 从缓存加载: $image"
|
|
||||||
docker load -i "$cache_file"
|
|
||||||
loaded_from_cache=true
|
|
||||||
fi
|
|
||||||
fi
|
|
||||||
done
|
|
||||||
fi
|
|
||||||
|
|
||||||
# 检查加载后是否还有缺失的镜像
|
|
||||||
need_build=false
|
|
||||||
for image in "${IMAGES[@]}"; do
|
|
||||||
if ! docker images --format "{{.Repository}}:{{.Tag}}" | grep -q "^${image}$"; then
|
|
||||||
need_build=true
|
|
||||||
break
|
|
||||||
fi
|
|
||||||
done
|
|
||||||
|
|
||||||
if $need_build; then
|
|
||||||
echo ""
|
|
||||||
echo "部分镜像缺失,开始构建..."
|
|
||||||
echo "工作目录: $(pwd)"
|
|
||||||
cd "$SCRIPT_DIR"
|
|
||||||
$DOCKER_COMPOSE build
|
|
||||||
|
|
||||||
# 询问是否保存镜像
|
|
||||||
echo ""
|
|
||||||
read -p "是否保存镜像到缓存以便下次快速启动? (Y/n): " -n 1 -r
|
|
||||||
echo
|
|
||||||
if [[ ! $REPLY =~ ^[Nn]$ ]]; then
|
|
||||||
mkdir -p "$IMAGE_CACHE_DIR"
|
|
||||||
echo "保存镜像到缓存..."
|
|
||||||
for image in "${IMAGES[@]}"; do
|
|
||||||
case "$image" in
|
|
||||||
"argus-metric-ftp:latest")
|
|
||||||
docker save -o "${IMAGE_CACHE_DIR}/argus-ftp.tar" "$image" && echo " 已保存: argus-ftp.tar"
|
|
||||||
;;
|
|
||||||
"argus-metric-prometheus:latest")
|
|
||||||
docker save -o "${IMAGE_CACHE_DIR}/argus-prometheus.tar" "$image" && echo " 已保存: argus-prometheus.tar"
|
|
||||||
;;
|
|
||||||
"argus-metric-grafana:latest")
|
|
||||||
docker save -o "${IMAGE_CACHE_DIR}/argus-grafana.tar" "$image" && echo " 已保存: argus-grafana.tar"
|
|
||||||
;;
|
|
||||||
esac
|
|
||||||
done
|
|
||||||
echo "镜像已保存到: $IMAGE_CACHE_DIR/"
|
|
||||||
fi
|
|
||||||
elif $loaded_from_cache; then
|
|
||||||
echo ""
|
|
||||||
echo "所有镜像已从缓存加载完成!"
|
|
||||||
fi
|
|
||||||
fi
|
|
||||||
|
|
||||||
echo ""
|
|
||||||
echo "3. 启动服务..."
|
|
||||||
cd "$SCRIPT_DIR"
|
|
||||||
$DOCKER_COMPOSE up -d
|
|
||||||
|
|
||||||
echo ""
|
|
||||||
echo "4. 等待服务启动..."
|
|
||||||
sleep 5
|
|
||||||
|
|
||||||
echo ""
|
|
||||||
echo "5. 检查服务状态..."
|
|
||||||
cd "$SCRIPT_DIR"
|
|
||||||
$DOCKER_COMPOSE ps
|
|
||||||
|
|
||||||
echo ""
|
|
||||||
echo "=========================================="
|
|
||||||
echo " 服务启动完成!"
|
|
||||||
echo "=========================================="
|
|
||||||
echo ""
|
|
||||||
echo "服务访问地址:"
|
|
||||||
echo " - FTP: ftp://localhost:${FTP_PORT:-21}"
|
|
||||||
echo " 用户名: ftpuser"
|
|
||||||
echo " 密码: ${FTP_PASSWORD:-ZGClab1234!}"
|
|
||||||
echo ""
|
|
||||||
echo " - Prometheus: http://localhost:${PROMETHEUS_PORT:-9090}"
|
|
||||||
echo ""
|
|
||||||
echo " - Grafana: http://localhost:${GRAFANA_PORT:-3000}"
|
|
||||||
echo " 用户名: admin"
|
|
||||||
echo " 密码: admin"
|
|
||||||
echo ""
|
|
||||||
echo "常用命令:"
|
|
||||||
echo " 查看日志: $DOCKER_COMPOSE logs -f [service]"
|
|
||||||
echo " 停止服务: $DOCKER_COMPOSE stop"
|
|
||||||
echo " 重启服务: $DOCKER_COMPOSE restart"
|
|
||||||
echo " 停止并删除: $DOCKER_COMPOSE down"
|
|
||||||
echo " 停止并删除卷: $DOCKER_COMPOSE down -v"
|
|
||||||
echo ""
|
|
||||||
|
|
||||||
@@ -1,51 +0,0 @@
|
|||||||
#!/bin/bash
|
|
||||||
|
|
||||||
# 停止所有服务脚本
|
|
||||||
|
|
||||||
set -e
|
|
||||||
|
|
||||||
SCRIPT_DIR="$(cd "$(dirname "${BASH_SOURCE[0]}")" && pwd)"
|
|
||||||
cd "$SCRIPT_DIR"
|
|
||||||
|
|
||||||
# 检测 docker-compose 命令(兼容新旧版本)
|
|
||||||
COMPOSE_FILE="$SCRIPT_DIR/docker-compose.yml"
|
|
||||||
if command -v docker-compose &> /dev/null; then
|
|
||||||
DOCKER_COMPOSE="docker-compose -f $COMPOSE_FILE"
|
|
||||||
elif docker compose version &> /dev/null 2>&1; then
|
|
||||||
DOCKER_COMPOSE="docker compose -f $COMPOSE_FILE"
|
|
||||||
else
|
|
||||||
echo "错误: 未找到 docker-compose 或 docker compose 命令"
|
|
||||||
exit 1
|
|
||||||
fi
|
|
||||||
|
|
||||||
echo "=========================================="
|
|
||||||
echo " 停止 Argus Metrics 服务"
|
|
||||||
echo "=========================================="
|
|
||||||
echo ""
|
|
||||||
|
|
||||||
# 检查是否有运行的容器
|
|
||||||
if [ "$($DOCKER_COMPOSE ps -q)" ]; then
|
|
||||||
echo "停止所有服务..."
|
|
||||||
$DOCKER_COMPOSE stop
|
|
||||||
|
|
||||||
echo ""
|
|
||||||
read -p "是否要删除容器? (y/N): " -n 1 -r
|
|
||||||
echo
|
|
||||||
if [[ $REPLY =~ ^[Yy]$ ]]; then
|
|
||||||
$DOCKER_COMPOSE down
|
|
||||||
echo "容器已删除"
|
|
||||||
|
|
||||||
read -p "是否要删除数据卷? (y/N): " -n 1 -r
|
|
||||||
echo
|
|
||||||
if [[ $REPLY =~ ^[Yy]$ ]]; then
|
|
||||||
$DOCKER_COMPOSE down -v
|
|
||||||
echo "数据卷已删除"
|
|
||||||
fi
|
|
||||||
fi
|
|
||||||
else
|
|
||||||
echo "没有运行的服务"
|
|
||||||
fi
|
|
||||||
|
|
||||||
echo ""
|
|
||||||
echo "完成!"
|
|
||||||
|
|
||||||
12
src/sys/debug/.env.example
Normal file
12
src/sys/debug/.env.example
Normal file
@@ -0,0 +1,12 @@
|
|||||||
|
# Generated by 01_bootstrap.sh
|
||||||
|
SYS_DEBUG_PRIVATE_CORE=/absolute/path/to/private
|
||||||
|
SYS_DEBUG_PRIVATE_NODEA=/absolute/path/to/private-nodea
|
||||||
|
SYS_DEBUG_PRIVATE_NODEB=/absolute/path/to/private-nodeb
|
||||||
|
SYS_DEBUG_TMP_DIR=/absolute/path/to/tmp
|
||||||
|
SYS_DEBUG_NETWORK_NAME=argus-debug-net
|
||||||
|
SYS_DEBUG_NETWORK_SUBNET=172.30.0.0/16
|
||||||
|
SYS_DEBUG_NETWORK_GATEWAY=172.30.0.1
|
||||||
|
SYS_DEBUG_PROJECT_NAME=argus-debug
|
||||||
|
SYS_DEBUG_CONTAINER_PREFIX=argus-debug
|
||||||
|
ARGUS_BUILD_UID=2133
|
||||||
|
ARGUS_BUILD_GID=2015
|
||||||
68
src/sys/debug/README.md
Normal file
68
src/sys/debug/README.md
Normal file
@@ -0,0 +1,68 @@
|
|||||||
|
# ARGUS 系统调试部署模式
|
||||||
|
|
||||||
|
该目录提供基于系统级 E2E 测试构建的调试部署流程,便于本地快速复现与排查问题。核心特性:
|
||||||
|
|
||||||
|
- 独立 docker 网络 `argus-debug-net`(默认子网 `172.30.0.0/16`),避免与 `src/sys/tests` 冲突。
|
||||||
|
- 私有数据目录可通过参数自定义,例如 `--private-root /tmp/argus-debug`。
|
||||||
|
- 默认保留调试过程生成的文件,避免 `down`/`bootstrap` 自动删除。
|
||||||
|
|
||||||
|
## 快速开始
|
||||||
|
|
||||||
|
```bash
|
||||||
|
cd src/sys/debug
|
||||||
|
|
||||||
|
# 仅首次需要,创建 external 网络
|
||||||
|
./scripts/network-create.sh
|
||||||
|
|
||||||
|
# 初始化目录/构建 agent/写入 .env
|
||||||
|
./scripts/01_bootstrap.sh --private-root /tmp/argus-debug
|
||||||
|
|
||||||
|
# 启动调试栈
|
||||||
|
./scripts/02_up.sh
|
||||||
|
|
||||||
|
# 根据需要执行验证脚本(03~08)
|
||||||
|
./scripts/03_wait_ready.sh
|
||||||
|
...
|
||||||
|
|
||||||
|
# 调试结束停止服务
|
||||||
|
./scripts/09_down.sh
|
||||||
|
|
||||||
|
# 若需移除网络或数据
|
||||||
|
./scripts/network-destroy.sh
|
||||||
|
./scripts/clean-data.sh
|
||||||
|
```
|
||||||
|
|
||||||
|
> **提示**:调试与测试栈不能同时运行,应保持 `src/sys/tests` 中的 `argus-sys` 栈已停止。
|
||||||
|
|
||||||
|
## 参数与环境变量
|
||||||
|
|
||||||
|
- `--private-root <path>`:同时指定核心服务与两个节点的私有目录根,脚本自动派生 `private`、`private-nodea`、`private-nodeb`。
|
||||||
|
- `--private-core <path>`、`--private-nodea <path>`、`--private-nodeb <path>`:分别覆盖单独目录。
|
||||||
|
- 环境变量可覆盖 `.env` 中写入的值,例如 `export SYS_DEBUG_NETWORK_NAME=my-debug-net`。
|
||||||
|
- `.env` 文件字段:
|
||||||
|
- `SYS_DEBUG_PRIVATE_CORE`
|
||||||
|
- `SYS_DEBUG_PRIVATE_NODEA`
|
||||||
|
- `SYS_DEBUG_PRIVATE_NODEB`
|
||||||
|
- `SYS_DEBUG_TMP_DIR`
|
||||||
|
- `SYS_DEBUG_NETWORK_NAME`
|
||||||
|
- `SYS_DEBUG_NETWORK_SUBNET`
|
||||||
|
- `SYS_DEBUG_NETWORK_GATEWAY`
|
||||||
|
- `SYS_DEBUG_PROJECT_NAME`
|
||||||
|
- `SYS_DEBUG_CONTAINER_PREFIX`
|
||||||
|
- `ARGUS_BUILD_UID` / `ARGUS_BUILD_GID`
|
||||||
|
|
||||||
|
## 脚本说明
|
||||||
|
|
||||||
|
- `scripts/common.sh`:通用函数与环境加载。
|
||||||
|
- `scripts/network-create.sh` / `network-destroy.sh`:管理 external 网络。
|
||||||
|
- `scripts/00_debug_all.sh`:顺序执行 01~08(默认不执行 09)。
|
||||||
|
- `scripts/clean-data.sh`:选择性清理宿主机私有数据。
|
||||||
|
- `scripts/03_wait_ready.sh`:除了等待各服务就绪,还会在 Elasticsearch 就绪后自动将磁盘水位阈值放宽(97%/98%/99%),避免在磁盘紧张的调试环境中分片分配失败。
|
||||||
|
- `scripts/08_restart_agent_reregister.sh`:将 node-b 切换到 `SYS_DEBUG_NODEB_FIXED_IP`(默认 `172.30.0.200`),如果目标地址与当前 IP 相同脚本会报错提醒重新选择地址。
|
||||||
|
- 其它 `01~09` 与测试目录对应,但针对参数化路径及网络做了调整。
|
||||||
|
|
||||||
|
## 注意事项
|
||||||
|
|
||||||
|
- 若宿主机未安装 Docker,脚本将提示错误并退出。
|
||||||
|
- 当指定的私有目录已存在数据时,脚本不会清理,请确认内容安全后再复用。
|
||||||
|
- 与测试环境共用镜像:请提前执行仓库根目录的 `./build/build_images.sh`。
|
||||||
147
src/sys/debug/docker-compose.yml
Normal file
147
src/sys/debug/docker-compose.yml
Normal file
@@ -0,0 +1,147 @@
|
|||||||
|
version: "3.8"
|
||||||
|
|
||||||
|
networks:
|
||||||
|
argus-debug-net:
|
||||||
|
external: true
|
||||||
|
name: ${SYS_DEBUG_NETWORK_NAME:-argus-debug-net}
|
||||||
|
|
||||||
|
services:
|
||||||
|
bind:
|
||||||
|
image: ${BIND_IMAGE_TAG:-argus-bind9:latest}
|
||||||
|
container_name: ${SYS_DEBUG_CONTAINER_PREFIX:-argus-debug}-bind
|
||||||
|
networks:
|
||||||
|
argus-debug-net:
|
||||||
|
ipv4_address: ${SYS_DEBUG_BIND_IP:-172.30.0.2}
|
||||||
|
volumes:
|
||||||
|
- ${SYS_DEBUG_PRIVATE_CORE}:/private
|
||||||
|
restart: unless-stopped
|
||||||
|
|
||||||
|
master:
|
||||||
|
image: ${MASTER_IMAGE_TAG:-argus-master:latest}
|
||||||
|
container_name: ${SYS_DEBUG_CONTAINER_PREFIX:-argus-debug}-master
|
||||||
|
depends_on:
|
||||||
|
- bind
|
||||||
|
environment:
|
||||||
|
- OFFLINE_THRESHOLD_SECONDS=6
|
||||||
|
- ONLINE_THRESHOLD_SECONDS=2
|
||||||
|
- SCHEDULER_INTERVAL_SECONDS=1
|
||||||
|
- ARGUS_BUILD_UID=${ARGUS_BUILD_UID:-2133}
|
||||||
|
- ARGUS_BUILD_GID=${ARGUS_BUILD_GID:-2015}
|
||||||
|
ports:
|
||||||
|
- "32300:3000"
|
||||||
|
volumes:
|
||||||
|
- ${SYS_DEBUG_PRIVATE_CORE}/argus/master:/private/argus/master
|
||||||
|
- ${SYS_DEBUG_PRIVATE_CORE}/argus/metric/prometheus:/private/argus/metric/prometheus
|
||||||
|
- ${SYS_DEBUG_PRIVATE_CORE}/argus/etc:/private/argus/etc
|
||||||
|
networks:
|
||||||
|
argus-debug-net:
|
||||||
|
ipv4_address: ${SYS_DEBUG_MASTER_IP:-172.30.0.10}
|
||||||
|
restart: unless-stopped
|
||||||
|
|
||||||
|
es:
|
||||||
|
image: ${ES_IMAGE_TAG:-argus-elasticsearch:latest}
|
||||||
|
container_name: ${SYS_DEBUG_CONTAINER_PREFIX:-argus-debug}-es
|
||||||
|
environment:
|
||||||
|
- discovery.type=single-node
|
||||||
|
- xpack.security.enabled=false
|
||||||
|
- ES_JAVA_OPTS=-Xms512m -Xmx512m
|
||||||
|
- ARGUS_BUILD_UID=${ARGUS_BUILD_UID:-2133}
|
||||||
|
- ARGUS_BUILD_GID=${ARGUS_BUILD_GID:-2015}
|
||||||
|
volumes:
|
||||||
|
- ${SYS_DEBUG_PRIVATE_CORE}/argus/log/elasticsearch:/private/argus/log/elasticsearch
|
||||||
|
- ${SYS_DEBUG_PRIVATE_CORE}/argus/etc:/private/argus/etc
|
||||||
|
ports:
|
||||||
|
- "9200:9200"
|
||||||
|
networks:
|
||||||
|
argus-debug-net:
|
||||||
|
ipv4_address: ${SYS_DEBUG_ES_IP:-172.30.0.20}
|
||||||
|
restart: unless-stopped
|
||||||
|
|
||||||
|
kibana:
|
||||||
|
image: ${KIBANA_IMAGE_TAG:-argus-kibana:latest}
|
||||||
|
container_name: ${SYS_DEBUG_CONTAINER_PREFIX:-argus-debug}-kibana
|
||||||
|
environment:
|
||||||
|
- ELASTICSEARCH_HOSTS=http://es.log.argus.com:9200
|
||||||
|
- ARGUS_BUILD_UID=${ARGUS_BUILD_UID:-2133}
|
||||||
|
- ARGUS_BUILD_GID=${ARGUS_BUILD_GID:-2015}
|
||||||
|
volumes:
|
||||||
|
- ${SYS_DEBUG_PRIVATE_CORE}/argus/log/kibana:/private/argus/log/kibana
|
||||||
|
- ${SYS_DEBUG_PRIVATE_CORE}/argus/etc:/private/argus/etc
|
||||||
|
depends_on:
|
||||||
|
- es
|
||||||
|
ports:
|
||||||
|
- "5601:5601"
|
||||||
|
networks:
|
||||||
|
argus-debug-net:
|
||||||
|
ipv4_address: ${SYS_DEBUG_KIBANA_IP:-172.30.0.30}
|
||||||
|
restart: unless-stopped
|
||||||
|
|
||||||
|
node-a:
|
||||||
|
image: ubuntu:22.04
|
||||||
|
container_name: ${SYS_DEBUG_CONTAINER_PREFIX:-argus-debug}-node-a
|
||||||
|
hostname: ${SYS_DEBUG_NODEA_HOST:-dev-yyrshare-nbnyx10-cp2f-pod-0}
|
||||||
|
depends_on:
|
||||||
|
- master
|
||||||
|
- bind
|
||||||
|
- es
|
||||||
|
environment:
|
||||||
|
- MASTER_ENDPOINT=http://master.argus.com:3000
|
||||||
|
- REPORT_INTERVAL_SECONDS=2
|
||||||
|
- ARGUS_BUILD_UID=${ARGUS_BUILD_UID:-2133}
|
||||||
|
- ARGUS_BUILD_GID=${ARGUS_BUILD_GID:-2015}
|
||||||
|
- ES_HOST=es
|
||||||
|
- ES_PORT=9200
|
||||||
|
- CLUSTER=local
|
||||||
|
- RACK=dev
|
||||||
|
volumes:
|
||||||
|
- ${SYS_DEBUG_PRIVATE_NODEA}/argus/agent/${SYS_DEBUG_NODEA_HOST:-dev-yyrshare-nbnyx10-cp2f-pod-0}:/private/argus/agent/${SYS_DEBUG_NODEA_HOST:-dev-yyrshare-nbnyx10-cp2f-pod-0}
|
||||||
|
- ../../agent/dist/argus-agent:/usr/local/bin/argus-agent:ro
|
||||||
|
- ../tests/scripts/node_entrypoint.sh:/usr/local/bin/node-entrypoint.sh:ro
|
||||||
|
- ../../log/fluent-bit/build/start-fluent-bit.sh:/assets/start-fluent-bit.sh:ro
|
||||||
|
- ../../log/fluent-bit/build/etc:/assets/fluent-bit/etc:ro
|
||||||
|
- ../../log/fluent-bit/build/packages:/assets/fluent-bit/packages:ro
|
||||||
|
entrypoint:
|
||||||
|
- /usr/local/bin/node-entrypoint.sh
|
||||||
|
dns:
|
||||||
|
- ${SYS_DEBUG_BIND_IP:-172.30.0.2}
|
||||||
|
ports:
|
||||||
|
- "2020:2020"
|
||||||
|
networks:
|
||||||
|
argus-debug-net:
|
||||||
|
ipv4_address: ${SYS_DEBUG_NODEA_IP:-172.30.0.101}
|
||||||
|
restart: unless-stopped
|
||||||
|
|
||||||
|
node-b:
|
||||||
|
image: ubuntu:22.04
|
||||||
|
container_name: ${SYS_DEBUG_CONTAINER_PREFIX:-argus-debug}-node-b
|
||||||
|
hostname: ${SYS_DEBUG_NODEB_HOST:-dev-yyrshare-uuuu10-ep2f-pod-0}
|
||||||
|
depends_on:
|
||||||
|
- master
|
||||||
|
- bind
|
||||||
|
- es
|
||||||
|
environment:
|
||||||
|
- MASTER_ENDPOINT=http://master.argus.com:3000
|
||||||
|
- REPORT_INTERVAL_SECONDS=2
|
||||||
|
- ARGUS_BUILD_UID=${ARGUS_BUILD_UID:-2133}
|
||||||
|
- ARGUS_BUILD_GID=${ARGUS_BUILD_GID:-2015}
|
||||||
|
- ES_HOST=es
|
||||||
|
- ES_PORT=9200
|
||||||
|
- CLUSTER=local
|
||||||
|
- RACK=dev
|
||||||
|
volumes:
|
||||||
|
- ${SYS_DEBUG_PRIVATE_NODEB}/argus/agent/${SYS_DEBUG_NODEB_HOST:-dev-yyrshare-uuuu10-ep2f-pod-0}:/private/argus/agent/${SYS_DEBUG_NODEB_HOST:-dev-yyrshare-uuuu10-ep2f-pod-0}
|
||||||
|
- ../../agent/dist/argus-agent:/usr/local/bin/argus-agent:ro
|
||||||
|
- ../tests/scripts/node_entrypoint.sh:/usr/local/bin/node-entrypoint.sh:ro
|
||||||
|
- ../../log/fluent-bit/build/start-fluent-bit.sh:/assets/start-fluent-bit.sh:ro
|
||||||
|
- ../../log/fluent-bit/build/etc:/assets/fluent-bit/etc:ro
|
||||||
|
- ../../log/fluent-bit/build/packages:/assets/fluent-bit/packages:ro
|
||||||
|
entrypoint:
|
||||||
|
- /usr/local/bin/node-entrypoint.sh
|
||||||
|
dns:
|
||||||
|
- ${SYS_DEBUG_BIND_IP:-172.30.0.2}
|
||||||
|
ports:
|
||||||
|
- "2021:2020"
|
||||||
|
networks:
|
||||||
|
argus-debug-net:
|
||||||
|
ipv4_address: ${SYS_DEBUG_NODEB_IP:-172.30.0.102}
|
||||||
|
restart: unless-stopped
|
||||||
24
src/sys/debug/scripts/00_debug_all.sh
Executable file
24
src/sys/debug/scripts/00_debug_all.sh
Executable file
@@ -0,0 +1,24 @@
|
|||||||
|
#!/usr/bin/env bash
|
||||||
|
set -euo pipefail
|
||||||
|
|
||||||
|
SCRIPT_DIR="$(cd "$(dirname "${BASH_SOURCE[0]}")" && pwd)"
|
||||||
|
|
||||||
|
SCRIPTS=(
|
||||||
|
"01_bootstrap.sh"
|
||||||
|
"02_up.sh"
|
||||||
|
"03_wait_ready.sh"
|
||||||
|
"04_verify_dns_routing.sh"
|
||||||
|
"05_agent_register.sh"
|
||||||
|
"06_write_health_and_assert.sh"
|
||||||
|
"07_logs_send_and_assert.sh"
|
||||||
|
"08_restart_agent_reregister.sh"
|
||||||
|
)
|
||||||
|
|
||||||
|
for script in "${SCRIPTS[@]}"; do
|
||||||
|
echo "[SYS-DEBUG] Running $script"
|
||||||
|
"$SCRIPT_DIR/$script"
|
||||||
|
echo "[SYS-DEBUG] $script completed"
|
||||||
|
echo
|
||||||
|
done
|
||||||
|
|
||||||
|
echo "[SYS-DEBUG] Complete. Run scripts/09_down.sh when finished (data retained)."
|
||||||
210
src/sys/debug/scripts/01_bootstrap.sh
Executable file
210
src/sys/debug/scripts/01_bootstrap.sh
Executable file
@@ -0,0 +1,210 @@
|
|||||||
|
#!/usr/bin/env bash
|
||||||
|
set -euo pipefail
|
||||||
|
|
||||||
|
# shellcheck source=common.sh
|
||||||
|
source "$(cd "$(dirname "${BASH_SOURCE[0]}")" && pwd)/common.sh"
|
||||||
|
|
||||||
|
PRIVATE_ROOT=""
|
||||||
|
PRIVATE_CORE="$SYS_DEBUG_PRIVATE_CORE"
|
||||||
|
PRIVATE_NODEA="$SYS_DEBUG_PRIVATE_NODEA"
|
||||||
|
PRIVATE_NODEB="$SYS_DEBUG_PRIVATE_NODEB"
|
||||||
|
TMP_DIR_VAL="$SYS_DEBUG_TMP_DIR"
|
||||||
|
NETWORK_NAME="$SYS_DEBUG_NETWORK_NAME"
|
||||||
|
NETWORK_SUBNET="$SYS_DEBUG_NETWORK_SUBNET"
|
||||||
|
NETWORK_GATEWAY="$SYS_DEBUG_NETWORK_GATEWAY"
|
||||||
|
PROJECT_NAME="$SYS_DEBUG_PROJECT_NAME"
|
||||||
|
CONTAINER_PREFIX="$SYS_DEBUG_CONTAINER_PREFIX"
|
||||||
|
NODEB_FIXED_IP=${SYS_DEBUG_NODEB_FIXED_IP:-172.30.0.200}
|
||||||
|
|
||||||
|
usage() {
|
||||||
|
cat <<EOF
|
||||||
|
Usage: ${0##*/} [--private-root PATH] [--private-core PATH] \
|
||||||
|
[--private-nodea PATH] [--private-nodeb PATH] \
|
||||||
|
[--tmp-dir PATH] [--network-name NAME] \
|
||||||
|
[--network-subnet CIDR] [--network-gateway IP]
|
||||||
|
|
||||||
|
Prepare directories, build agent binary, and write .env for debug stack.
|
||||||
|
EOF
|
||||||
|
}
|
||||||
|
|
||||||
|
while [[ $# -gt 0 ]]; do
|
||||||
|
case "$1" in
|
||||||
|
--private-root)
|
||||||
|
shift; [[ $# -gt 0 ]] || { echo "--private-root requires value" >&2; exit 1; }
|
||||||
|
PRIVATE_ROOT="$1"
|
||||||
|
;;
|
||||||
|
--private-root=*)
|
||||||
|
PRIVATE_ROOT="${1#*=}"
|
||||||
|
;;
|
||||||
|
--private-core)
|
||||||
|
shift; [[ $# -gt 0 ]] || { echo "--private-core requires value" >&2; exit 1; }
|
||||||
|
PRIVATE_CORE="$1"
|
||||||
|
;;
|
||||||
|
--private-core=*)
|
||||||
|
PRIVATE_CORE="${1#*=}"
|
||||||
|
;;
|
||||||
|
--private-nodea)
|
||||||
|
shift; [[ $# -gt 0 ]] || { echo "--private-nodea requires value" >&2; exit 1; }
|
||||||
|
PRIVATE_NODEA="$1"
|
||||||
|
;;
|
||||||
|
--private-nodea=*)
|
||||||
|
PRIVATE_NODEA="${1#*=}"
|
||||||
|
;;
|
||||||
|
--private-nodeb)
|
||||||
|
shift; [[ $# -gt 0 ]] || { echo "--private-nodeb requires value" >&2; exit 1; }
|
||||||
|
PRIVATE_NODEB="$1"
|
||||||
|
;;
|
||||||
|
--private-nodeb=*)
|
||||||
|
PRIVATE_NODEB="${1#*=}"
|
||||||
|
;;
|
||||||
|
--tmp-dir)
|
||||||
|
shift; [[ $# -gt 0 ]] || { echo "--tmp-dir requires value" >&2; exit 1; }
|
||||||
|
TMP_DIR_VAL="$1"
|
||||||
|
;;
|
||||||
|
--tmp-dir=*)
|
||||||
|
TMP_DIR_VAL="${1#*=}"
|
||||||
|
;;
|
||||||
|
--network-name)
|
||||||
|
shift; [[ $# -gt 0 ]] || { echo "--network-name requires value" >&2; exit 1; }
|
||||||
|
NETWORK_NAME="$1"
|
||||||
|
;;
|
||||||
|
--network-name=*)
|
||||||
|
NETWORK_NAME="${1#*=}"
|
||||||
|
;;
|
||||||
|
--network-subnet)
|
||||||
|
shift; [[ $# -gt 0 ]] || { echo "--network-subnet requires value" >&2; exit 1; }
|
||||||
|
NETWORK_SUBNET="$1"
|
||||||
|
;;
|
||||||
|
--network-subnet=*)
|
||||||
|
NETWORK_SUBNET="${1#*=}"
|
||||||
|
;;
|
||||||
|
--network-gateway)
|
||||||
|
shift; [[ $# -gt 0 ]] || { echo "--network-gateway requires value" >&2; exit 1; }
|
||||||
|
NETWORK_GATEWAY="$1"
|
||||||
|
;;
|
||||||
|
--network-gateway=*)
|
||||||
|
NETWORK_GATEWAY="${1#*=}"
|
||||||
|
;;
|
||||||
|
-h|--help)
|
||||||
|
usage
|
||||||
|
exit 0
|
||||||
|
;;
|
||||||
|
*)
|
||||||
|
echo "Unknown argument: $1" >&2
|
||||||
|
usage >&2
|
||||||
|
exit 1
|
||||||
|
;;
|
||||||
|
esac
|
||||||
|
shift
|
||||||
|
done
|
||||||
|
|
||||||
|
if [[ -n "$PRIVATE_ROOT" ]]; then
|
||||||
|
PRIVATE_CORE="$PRIVATE_ROOT/private"
|
||||||
|
PRIVATE_NODEA="$PRIVATE_ROOT/private-nodea"
|
||||||
|
PRIVATE_NODEB="$PRIVATE_ROOT/private-nodeb"
|
||||||
|
fi
|
||||||
|
|
||||||
|
PRIVATE_CORE=$(abs_path "$PRIVATE_CORE")
|
||||||
|
PRIVATE_NODEA=$(abs_path "$PRIVATE_NODEA")
|
||||||
|
PRIVATE_NODEB=$(abs_path "$PRIVATE_NODEB")
|
||||||
|
TMP_DIR_VAL=$(abs_path "$TMP_DIR_VAL")
|
||||||
|
|
||||||
|
log "Preparing directories under $PRIVATE_CORE"
|
||||||
|
mkdir -p \
|
||||||
|
"$PRIVATE_CORE/argus/etc" \
|
||||||
|
"$PRIVATE_CORE/argus/bind" \
|
||||||
|
"$PRIVATE_CORE/argus/master" \
|
||||||
|
"$PRIVATE_CORE/argus/metric/prometheus" \
|
||||||
|
"$PRIVATE_CORE/argus/log/elasticsearch" \
|
||||||
|
"$PRIVATE_CORE/argus/log/kibana" \
|
||||||
|
"$PRIVATE_NODEA/argus/agent/$HOST_A/health" \
|
||||||
|
"$PRIVATE_NODEB/argus/agent/$HOST_B/health" \
|
||||||
|
"$TMP_DIR_VAL"
|
||||||
|
|
||||||
|
log "Aligning ownership for core directories"
|
||||||
|
chown -R "${ARGUS_BUILD_UID}:${ARGUS_BUILD_GID}" \
|
||||||
|
"$PRIVATE_CORE/argus/log/elasticsearch" \
|
||||||
|
"$PRIVATE_CORE/argus/log/kibana" \
|
||||||
|
"$PRIVATE_CORE/argus/etc" 2>/dev/null || true
|
||||||
|
|
||||||
|
log "Distributing update-dns.sh"
|
||||||
|
BIND_UPDATE_SRC="$REPO_ROOT/src/bind/build/update-dns.sh"
|
||||||
|
BIND_UPDATE_DEST="$PRIVATE_CORE/argus/etc/update-dns.sh"
|
||||||
|
if [[ -f "$BIND_UPDATE_SRC" ]]; then
|
||||||
|
cp "$BIND_UPDATE_SRC" "$BIND_UPDATE_DEST"
|
||||||
|
chmod +x "$BIND_UPDATE_DEST"
|
||||||
|
else
|
||||||
|
echo "[WARN] Missing $BIND_UPDATE_SRC" >&2
|
||||||
|
fi
|
||||||
|
|
||||||
|
require_docker
|
||||||
|
|
||||||
|
ensure_image() {
|
||||||
|
local image="$1"
|
||||||
|
if ! docker image inspect "$image" >/dev/null 2>&1; then
|
||||||
|
echo "[ERR] Missing image: $image. Run ./build/build_images.sh" >&2
|
||||||
|
exit 1
|
||||||
|
fi
|
||||||
|
}
|
||||||
|
|
||||||
|
log "Ensuring required images exist"
|
||||||
|
ensure_image "${ES_IMAGE_TAG:-argus-elasticsearch:latest}"
|
||||||
|
ensure_image "${KIBANA_IMAGE_TAG:-argus-kibana:latest}"
|
||||||
|
ensure_image "${BIND_IMAGE_TAG:-argus-bind9:latest}"
|
||||||
|
ensure_image "${MASTER_IMAGE_TAG:-argus-master:latest}"
|
||||||
|
|
||||||
|
log "Building agent binary"
|
||||||
|
pushd "$REPO_ROOT/src/agent" >/dev/null
|
||||||
|
./scripts/build_binary.sh
|
||||||
|
popd >/dev/null
|
||||||
|
|
||||||
|
AGENT_BIN="$REPO_ROOT/src/agent/dist/argus-agent"
|
||||||
|
if [[ ! -x "$AGENT_BIN" ]]; then
|
||||||
|
echo "[ERR] Agent binary not found at $AGENT_BIN" >&2
|
||||||
|
exit 1
|
||||||
|
fi
|
||||||
|
echo "$AGENT_BIN" > "$TMP_DIR_VAL/agent_binary_path"
|
||||||
|
|
||||||
|
log "Preparing environment file contents"
|
||||||
|
tmp_env="$(mktemp)"
|
||||||
|
cat > "$tmp_env" <<EOF
|
||||||
|
SYS_DEBUG_PRIVATE_CORE=$PRIVATE_CORE
|
||||||
|
SYS_DEBUG_PRIVATE_NODEA=$PRIVATE_NODEA
|
||||||
|
SYS_DEBUG_PRIVATE_NODEB=$PRIVATE_NODEB
|
||||||
|
SYS_DEBUG_TMP_DIR=$TMP_DIR_VAL
|
||||||
|
SYS_DEBUG_NETWORK_NAME=$NETWORK_NAME
|
||||||
|
SYS_DEBUG_NETWORK_SUBNET=$NETWORK_SUBNET
|
||||||
|
SYS_DEBUG_NETWORK_GATEWAY=$NETWORK_GATEWAY
|
||||||
|
SYS_DEBUG_PROJECT_NAME=$PROJECT_NAME
|
||||||
|
SYS_DEBUG_CONTAINER_PREFIX=$CONTAINER_PREFIX
|
||||||
|
SYS_DEBUG_NODEA_HOST=$HOST_A
|
||||||
|
SYS_DEBUG_NODEB_HOST=$HOST_B
|
||||||
|
SYS_DEBUG_BIND_IP=${SYS_DEBUG_BIND_IP:-172.30.0.2}
|
||||||
|
SYS_DEBUG_MASTER_IP=${SYS_DEBUG_MASTER_IP:-172.30.0.10}
|
||||||
|
SYS_DEBUG_ES_IP=${SYS_DEBUG_ES_IP:-172.30.0.20}
|
||||||
|
SYS_DEBUG_KIBANA_IP=${SYS_DEBUG_KIBANA_IP:-172.30.0.30}
|
||||||
|
SYS_DEBUG_NODEA_IP=${SYS_DEBUG_NODEA_IP:-172.30.0.101}
|
||||||
|
SYS_DEBUG_NODEB_IP=${SYS_DEBUG_NODEB_IP:-172.30.0.102}
|
||||||
|
SYS_DEBUG_NODEB_FIXED_IP=$NODEB_FIXED_IP
|
||||||
|
ARGUS_BUILD_UID=$ARGUS_BUILD_UID
|
||||||
|
ARGUS_BUILD_GID=$ARGUS_BUILD_GID
|
||||||
|
EOF
|
||||||
|
|
||||||
|
if [[ -f "$ENV_FILE" ]]; then
|
||||||
|
if cmp -s "$tmp_env" "$ENV_FILE"; then
|
||||||
|
log ".env already up-to-date"
|
||||||
|
rm -f "$tmp_env"
|
||||||
|
if [[ ! -f "$DEBUG_ROOT/.env.lock" ]]; then
|
||||||
|
cp "$ENV_FILE" "$DEBUG_ROOT/.env.lock"
|
||||||
|
fi
|
||||||
|
else
|
||||||
|
mv "$ENV_FILE" "$ENV_FILE.bak"
|
||||||
|
mv "$tmp_env" "$ENV_FILE"
|
||||||
|
cp "$ENV_FILE" "$DEBUG_ROOT/.env.lock"
|
||||||
|
log "Bootstrap updated .env (previous saved at ${ENV_FILE}.bak)"
|
||||||
|
fi
|
||||||
|
else
|
||||||
|
mv "$tmp_env" "$ENV_FILE"
|
||||||
|
cp "$ENV_FILE" "$DEBUG_ROOT/.env.lock"
|
||||||
|
log "Bootstrap created .env at $ENV_FILE"
|
||||||
|
fi
|
||||||
19
src/sys/debug/scripts/02_up.sh
Executable file
19
src/sys/debug/scripts/02_up.sh
Executable file
@@ -0,0 +1,19 @@
|
|||||||
|
#!/usr/bin/env bash
|
||||||
|
set -euo pipefail
|
||||||
|
|
||||||
|
# shellcheck source=common.sh
|
||||||
|
source "$(cd "$(dirname "${BASH_SOURCE[0]}")" && pwd)/common.sh"
|
||||||
|
|
||||||
|
ensure_env_file
|
||||||
|
ensure_paths_defined
|
||||||
|
require_docker
|
||||||
|
|
||||||
|
if ! docker network inspect "$SYS_DEBUG_NETWORK_NAME" >/dev/null 2>&1; then
|
||||||
|
echo "[ERR] Network $SYS_DEBUG_NETWORK_NAME not found. Run scripts/network-create.sh first." >&2
|
||||||
|
exit 1
|
||||||
|
fi
|
||||||
|
|
||||||
|
log "Starting debug stack on project $SYS_DEBUG_PROJECT_NAME"
|
||||||
|
compose up -d
|
||||||
|
|
||||||
|
log "Services started: master:32300 es:9200 kibana:5601 node-a:2020 node-b:2021"
|
||||||
84
src/sys/debug/scripts/03_wait_ready.sh
Executable file
84
src/sys/debug/scripts/03_wait_ready.sh
Executable file
@@ -0,0 +1,84 @@
|
|||||||
|
#!/usr/bin/env bash
|
||||||
|
set -euo pipefail
|
||||||
|
|
||||||
|
# shellcheck source=common.sh
|
||||||
|
source "$(cd "$(dirname "${BASH_SOURCE[0]}")" && pwd)/common.sh"
|
||||||
|
|
||||||
|
ensure_env_file
|
||||||
|
ensure_paths_defined
|
||||||
|
|
||||||
|
service_id() {
|
||||||
|
compose ps -q "$1"
|
||||||
|
}
|
||||||
|
|
||||||
|
wait_http() {
|
||||||
|
local url="$1"; local attempts="${2:-120}"; local i=1
|
||||||
|
while (( i <= attempts )); do
|
||||||
|
if curl -fsS "$url" >/dev/null 2>&1; then
|
||||||
|
return 0
|
||||||
|
fi
|
||||||
|
echo "[..] waiting $url ($i/$attempts)"
|
||||||
|
sleep 5
|
||||||
|
((i++))
|
||||||
|
done
|
||||||
|
echo "[ERR] Timeout waiting for $url" >&2
|
||||||
|
return 1
|
||||||
|
}
|
||||||
|
|
||||||
|
log "Waiting for ES/Kibana/Master/Fluent Bit/Bind"
|
||||||
|
|
||||||
|
attempt=1; max=120
|
||||||
|
while (( attempt <= max )); do
|
||||||
|
if curl -fsS "http://localhost:9200/_cluster/health?wait_for_status=yellow&timeout=1s" >/dev/null 2>&1; then
|
||||||
|
break
|
||||||
|
fi
|
||||||
|
echo "[..] waiting ES ($attempt/$max)"
|
||||||
|
sleep 5
|
||||||
|
((attempt++))
|
||||||
|
done
|
||||||
|
if (( attempt > max )); then
|
||||||
|
echo "[ERR] ES not ready" >&2
|
||||||
|
exit 1
|
||||||
|
fi
|
||||||
|
|
||||||
|
log "Applying relaxed ES disk watermarks for debug"
|
||||||
|
curl -fsS -XPUT "http://localhost:9200/_cluster/settings" \
|
||||||
|
-H 'Content-Type: application/json' \
|
||||||
|
-d '{
|
||||||
|
"transient": {
|
||||||
|
"cluster.routing.allocation.disk.watermark.low": "99%",
|
||||||
|
"cluster.routing.allocation.disk.watermark.high": "99%",
|
||||||
|
"cluster.routing.allocation.disk.watermark.flood_stage": "99%"
|
||||||
|
}
|
||||||
|
}' >/dev/null || echo "[WARN] Failed to adjust ES watermarks"
|
||||||
|
|
||||||
|
log "Waiting for Kibana to be available (HTTP 200)"
|
||||||
|
kb_attempt=1; kb_max=180
|
||||||
|
while (( kb_attempt <= kb_max )); do
|
||||||
|
body=$(curl -sS "http://localhost:5601/api/status" 2>/dev/null || true)
|
||||||
|
code=$(curl -s -o /dev/null -w "%{http_code}" "http://localhost:5601/api/status" || echo 000)
|
||||||
|
if [[ "$code" == "200" ]] && echo "$body" | grep -q '"level":"available"'; then
|
||||||
|
log "Kibana available"
|
||||||
|
break
|
||||||
|
fi
|
||||||
|
echo "[..] waiting kibana 200 ($kb_attempt/$kb_max), last_code=$code"
|
||||||
|
sleep 5
|
||||||
|
((kb_attempt++))
|
||||||
|
done
|
||||||
|
if (( kb_attempt > kb_max )); then
|
||||||
|
echo "[ERR] Kibana did not reach HTTP 200" >&2
|
||||||
|
exit 1
|
||||||
|
fi
|
||||||
|
|
||||||
|
wait_http "http://localhost:32300/readyz" 120
|
||||||
|
wait_http "http://localhost:2020/api/v2/metrics" 120
|
||||||
|
wait_http "http://localhost:2021/api/v2/metrics" 120
|
||||||
|
|
||||||
|
BIND_ID="$(service_id bind)"
|
||||||
|
if [[ -n "$BIND_ID" ]]; then
|
||||||
|
docker exec "$BIND_ID" named-checkconf >/dev/null
|
||||||
|
else
|
||||||
|
echo "[WARN] bind container id not found" >&2
|
||||||
|
fi
|
||||||
|
|
||||||
|
log "All services are ready"
|
||||||
51
src/sys/debug/scripts/04_verify_dns_routing.sh
Executable file
51
src/sys/debug/scripts/04_verify_dns_routing.sh
Executable file
@@ -0,0 +1,51 @@
|
|||||||
|
#!/usr/bin/env bash
|
||||||
|
set -euo pipefail
|
||||||
|
|
||||||
|
# shellcheck source=common.sh
|
||||||
|
source "$(cd "$(dirname "${BASH_SOURCE[0]}")" && pwd)/common.sh"
|
||||||
|
|
||||||
|
ensure_env_file
|
||||||
|
ensure_paths_defined
|
||||||
|
|
||||||
|
service_id() {
|
||||||
|
compose ps -q "$1"
|
||||||
|
}
|
||||||
|
|
||||||
|
log "Verifying DNS routing via bind"
|
||||||
|
|
||||||
|
MASTER_FILE="$SYS_DEBUG_PRIVATE_CORE/argus/etc/master.argus.com"
|
||||||
|
if [[ ! -f "$MASTER_FILE" ]]; then
|
||||||
|
echo "[ERR] master.argus.com file missing at $MASTER_FILE" >&2
|
||||||
|
exit 1
|
||||||
|
fi
|
||||||
|
MASTER_IP_HOST="$(tr -d '\r\n' < "$MASTER_FILE" || true)"
|
||||||
|
log "master.argus.com file content: $MASTER_IP_HOST"
|
||||||
|
|
||||||
|
BIN_ID="$(service_id bind)"
|
||||||
|
if [[ -n "$BIN_ID" ]]; then
|
||||||
|
DIG_IP="$(docker exec "$BIN_ID" dig +short master.argus.com A | tail -n1 || true)"
|
||||||
|
log "dig(master.argus.com) from bind container -> $DIG_IP"
|
||||||
|
if [[ -z "$DIG_IP" ]]; then
|
||||||
|
echo "[ERR] bind did not resolve master.argus.com" >&2
|
||||||
|
exit 1
|
||||||
|
fi
|
||||||
|
else
|
||||||
|
echo "[WARN] bind container not found; skip dig" >&2
|
||||||
|
fi
|
||||||
|
|
||||||
|
for node in node-a node-b; do
|
||||||
|
CID="$(service_id "$node")"
|
||||||
|
if [[ -z "$CID" ]]; then
|
||||||
|
echo "[ERR] Container for $node not found" >&2
|
||||||
|
exit 1
|
||||||
|
fi
|
||||||
|
log "Checking resolution inside $node"
|
||||||
|
if ! docker exec "$CID" getent hosts master.argus.com >/dev/null 2>&1; then
|
||||||
|
echo "[ERR] $node cannot resolve master.argus.com" >&2
|
||||||
|
exit 1
|
||||||
|
fi
|
||||||
|
RES="$(docker exec "$CID" getent hosts master.argus.com | awk '{print $1}' | head -n1)"
|
||||||
|
log "$node resolved master.argus.com -> $RES"
|
||||||
|
done
|
||||||
|
|
||||||
|
log "DNS routing verified"
|
||||||
84
src/sys/debug/scripts/05_agent_register.sh
Executable file
84
src/sys/debug/scripts/05_agent_register.sh
Executable file
@@ -0,0 +1,84 @@
|
|||||||
|
#!/usr/bin/env bash
|
||||||
|
set -euo pipefail
|
||||||
|
|
||||||
|
# shellcheck source=common.sh
|
||||||
|
source "$(cd "$(dirname "${BASH_SOURCE[0]}")" && pwd)/common.sh"
|
||||||
|
|
||||||
|
ensure_env_file
|
||||||
|
ensure_paths_defined
|
||||||
|
|
||||||
|
TMP_DIR_LOCAL="$TMP_DIR"
|
||||||
|
mkdir -p "$TMP_DIR_LOCAL"
|
||||||
|
|
||||||
|
API_BASE="http://localhost:32300/api/v1/master"
|
||||||
|
|
||||||
|
log "Waiting for agent nodes to register"
|
||||||
|
|
||||||
|
extract_node() {
|
||||||
|
local name="$1"; local output="$2"; local json_file="$3"
|
||||||
|
python3 - "$name" "$output" "$json_file" <<'PY'
|
||||||
|
import json, sys, pathlib
|
||||||
|
name = sys.argv[1]
|
||||||
|
out = pathlib.Path(sys.argv[2])
|
||||||
|
json_file = sys.argv[3]
|
||||||
|
with open(json_file, 'r') as fh:
|
||||||
|
data = json.load(fh)
|
||||||
|
node = next((n for n in data if n.get("name") == name), None)
|
||||||
|
if node:
|
||||||
|
out.write_text(node["id"])
|
||||||
|
print(node["id"])
|
||||||
|
PY
|
||||||
|
}
|
||||||
|
|
||||||
|
ID_A=""; ID_B=""
|
||||||
|
for _ in {1..60}; do
|
||||||
|
sleep 2
|
||||||
|
resp=$(curl -fsS "$API_BASE/nodes" 2>/dev/null || true)
|
||||||
|
[[ -z "$resp" ]] && continue
|
||||||
|
if ! echo "$resp" | head -c1 | grep -q '\['; then
|
||||||
|
continue
|
||||||
|
fi
|
||||||
|
echo "$resp" > "$TMP_DIR_LOCAL/nodes_list.json"
|
||||||
|
ID_A=$(extract_node "$HOST_A" "$TMP_DIR_LOCAL/node_id_a" "$TMP_DIR_LOCAL/nodes_list.json" 2>/dev/null || true)
|
||||||
|
ID_B=$(extract_node "$HOST_B" "$TMP_DIR_LOCAL/node_id_b" "$TMP_DIR_LOCAL/nodes_list.json" 2>/dev/null || true)
|
||||||
|
if [[ -s "$TMP_DIR_LOCAL/node_id_a" && -s "$TMP_DIR_LOCAL/node_id_b" ]]; then
|
||||||
|
break
|
||||||
|
fi
|
||||||
|
done
|
||||||
|
|
||||||
|
if [[ ! -s "$TMP_DIR_LOCAL/node_id_a" || ! -s "$TMP_DIR_LOCAL/node_id_b" ]]; then
|
||||||
|
echo "[ERR] Agents did not register in time" >&2
|
||||||
|
exit 1
|
||||||
|
fi
|
||||||
|
|
||||||
|
node_detail() {
|
||||||
|
local id="$1"; local out="$2"
|
||||||
|
curl -fsS "$API_BASE/nodes/$id" -o "$out"
|
||||||
|
}
|
||||||
|
|
||||||
|
node_detail "$(cat "$TMP_DIR_LOCAL/node_id_a")" "$TMP_DIR_LOCAL/detail_a.json"
|
||||||
|
node_detail "$(cat "$TMP_DIR_LOCAL/node_id_b")" "$TMP_DIR_LOCAL/detail_b.json"
|
||||||
|
|
||||||
|
python3 - "$TMP_DIR_LOCAL/detail_a.json" "$TMP_DIR_LOCAL/initial_ip_a" <<'PY'
|
||||||
|
import json, sys, pathlib
|
||||||
|
node=json.load(open(sys.argv[1]))
|
||||||
|
ip=node.get("meta_data",{}).get("ip")
|
||||||
|
assert ip, "missing ip"
|
||||||
|
pathlib.Path(sys.argv[2]).write_text(ip)
|
||||||
|
PY
|
||||||
|
|
||||||
|
python3 - "$TMP_DIR_LOCAL/detail_b.json" "$TMP_DIR_LOCAL/initial_ip_b" <<'PY'
|
||||||
|
import json, sys, pathlib
|
||||||
|
node=json.load(open(sys.argv[1]))
|
||||||
|
ip=node.get("meta_data",{}).get("ip")
|
||||||
|
assert ip, "missing ip"
|
||||||
|
pathlib.Path(sys.argv[2]).write_text(ip)
|
||||||
|
PY
|
||||||
|
|
||||||
|
NODE_JSON_A="$SYS_DEBUG_PRIVATE_NODEA/argus/agent/$HOST_A/node.json"
|
||||||
|
NODE_JSON_B="$SYS_DEBUG_PRIVATE_NODEB/argus/agent/$HOST_B/node.json"
|
||||||
|
|
||||||
|
[[ -f "$NODE_JSON_A" ]] || { echo "[ERR] node.json missing for $HOST_A" >&2; exit 1; }
|
||||||
|
[[ -f "$NODE_JSON_B" ]] || { echo "[ERR] node.json missing for $HOST_B" >&2; exit 1; }
|
||||||
|
|
||||||
|
log "Agents registered: $(cat "$TMP_DIR_LOCAL/node_id_a") , $(cat "$TMP_DIR_LOCAL/node_id_b")"
|
||||||
78
src/sys/debug/scripts/06_write_health_and_assert.sh
Executable file
78
src/sys/debug/scripts/06_write_health_and_assert.sh
Executable file
@@ -0,0 +1,78 @@
|
|||||||
|
#!/usr/bin/env bash
|
||||||
|
set -euo pipefail
|
||||||
|
|
||||||
|
# shellcheck source=common.sh
|
||||||
|
source "$(cd "$(dirname "${BASH_SOURCE[0]}")" && pwd)/common.sh"
|
||||||
|
|
||||||
|
ensure_env_file
|
||||||
|
ensure_paths_defined
|
||||||
|
|
||||||
|
API_BASE="http://localhost:32300/api/v1/master"
|
||||||
|
|
||||||
|
HEALTH_A="$SYS_DEBUG_PRIVATE_NODEA/argus/agent/$HOST_A/health"
|
||||||
|
HEALTH_B="$SYS_DEBUG_PRIVATE_NODEB/argus/agent/$HOST_B/health"
|
||||||
|
|
||||||
|
write_health() {
|
||||||
|
local dir="$1"; mkdir -p "$dir"
|
||||||
|
cat > "$dir/log-fluentbit.json" <<JSON
|
||||||
|
{ "status": "healthy", "timestamp": "2025-10-13T12:05:00Z" }
|
||||||
|
JSON
|
||||||
|
cat > "$dir/metric-node-exporter.json" <<JSON
|
||||||
|
{ "status": "healthy", "timestamp": "2025-10-13T12:05:00Z" }
|
||||||
|
JSON
|
||||||
|
}
|
||||||
|
|
||||||
|
log "Writing health files for both nodes"
|
||||||
|
write_health "$HEALTH_A"
|
||||||
|
write_health "$HEALTH_B"
|
||||||
|
|
||||||
|
ID_A="$TMP_DIR/node_id_a"
|
||||||
|
ID_B="$TMP_DIR/node_id_b"
|
||||||
|
|
||||||
|
[[ -f "$ID_A" && -f "$ID_B" ]] || { echo "[ERR] node id files missing in $TMP_DIR" >&2; exit 1; }
|
||||||
|
|
||||||
|
ID_A_VAL="$(cat "$ID_A")"
|
||||||
|
ID_B_VAL="$(cat "$ID_B")"
|
||||||
|
|
||||||
|
check_health() {
|
||||||
|
local id="$1"; local tries=40
|
||||||
|
for _ in $(seq 1 $tries); do
|
||||||
|
sleep 2
|
||||||
|
resp=$(curl -fsS "$API_BASE/nodes/$id" 2>/dev/null || true)
|
||||||
|
[[ -z "$resp" ]] && continue
|
||||||
|
echo "$resp" > "$TMP_DIR/node_${id}_detail.json"
|
||||||
|
if python3 - "$TMP_DIR/node_${id}_detail.json" <<'PY'
|
||||||
|
import json,sys
|
||||||
|
node=json.load(open(sys.argv[1]))
|
||||||
|
h=node.get("health",{})
|
||||||
|
if "log-fluentbit" in h and "metric-node-exporter" in h:
|
||||||
|
sys.exit(0)
|
||||||
|
sys.exit(1)
|
||||||
|
PY
|
||||||
|
then
|
||||||
|
return 0
|
||||||
|
fi
|
||||||
|
done
|
||||||
|
return 1
|
||||||
|
}
|
||||||
|
|
||||||
|
check_health "$ID_A_VAL" || { echo "[ERR] health keys not reported for node A" >&2; exit 1; }
|
||||||
|
check_health "$ID_B_VAL" || { echo "[ERR] health keys not reported for node B" >&2; exit 1; }
|
||||||
|
|
||||||
|
NODES_JSON="$SYS_DEBUG_PRIVATE_CORE/argus/metric/prometheus/nodes.json"
|
||||||
|
if [[ ! -f "$NODES_JSON" ]]; then
|
||||||
|
echo "[ERR] nodes.json missing at $NODES_JSON" >&2
|
||||||
|
exit 1
|
||||||
|
fi
|
||||||
|
|
||||||
|
python3 - "$NODES_JSON" <<'PY'
|
||||||
|
import json,sys
|
||||||
|
with open(sys.argv[1]) as h:
|
||||||
|
nodes=json.load(h)
|
||||||
|
if not isinstance(nodes, list):
|
||||||
|
raise SystemExit("nodes.json expected list")
|
||||||
|
if len(nodes) != 2:
|
||||||
|
raise SystemExit(f"expected 2 nodes online, got {len(nodes)}")
|
||||||
|
PY
|
||||||
|
|
||||||
|
log "Health reported and nodes.json has 2 online nodes"
|
||||||
73
src/sys/debug/scripts/07_logs_send_and_assert.sh
Executable file
73
src/sys/debug/scripts/07_logs_send_and_assert.sh
Executable file
@@ -0,0 +1,73 @@
|
|||||||
|
#!/usr/bin/env bash
|
||||||
|
set -euo pipefail
|
||||||
|
|
||||||
|
# shellcheck source=common.sh
|
||||||
|
source "$(cd "$(dirname "${BASH_SOURCE[0]}")" && pwd)/common.sh"
|
||||||
|
|
||||||
|
ensure_env_file
|
||||||
|
ensure_paths_defined
|
||||||
|
|
||||||
|
log "Sending logs and asserting ES counts"
|
||||||
|
|
||||||
|
get_count() {
|
||||||
|
local idx="$1"
|
||||||
|
curl -s "http://localhost:9200/${idx}/_count?ignore_unavailable=true&allow_no_indices=true" | sed -E 's/.*"count":([0-9]+).*/\1/' | awk 'NF{print $0;exit} END{if(NR==0)print 0}'
|
||||||
|
}
|
||||||
|
|
||||||
|
train0=$(get_count "train-*")
|
||||||
|
infer0=$(get_count "infer-*")
|
||||||
|
base=$((train0 + infer0))
|
||||||
|
log "initial counts: train=${train0} infer=${infer0} total=${base}"
|
||||||
|
|
||||||
|
service_id() {
|
||||||
|
compose ps -q "$1"
|
||||||
|
}
|
||||||
|
|
||||||
|
send_logs() {
|
||||||
|
local sid="$1"; local hosttag="$2"
|
||||||
|
docker exec "$sid" sh -lc 'mkdir -p /logs/train /logs/infer'
|
||||||
|
docker exec "$sid" sh -lc "ts=\
|
||||||
|
\$(date '+%F %T'); echo \"\$ts INFO [$hosttag] training step=1 loss=1.23 model=bert\" >> /logs/train/train-demo.log"
|
||||||
|
docker exec "$sid" sh -lc "ts=\
|
||||||
|
\$(date '+%F %T'); echo \"\$ts INFO [$hosttag] training step=2 loss=1.10 model=bert\" >> /logs/train/train-demo.log"
|
||||||
|
docker exec "$sid" sh -lc "ts=\
|
||||||
|
\$(date '+%F %T'); echo \"\$ts WARN [$hosttag] inference slow on batch=2 latency=1.9s\" >> /logs/infer/infer-demo.log"
|
||||||
|
}
|
||||||
|
|
||||||
|
CID_A="$(service_id node-a)"
|
||||||
|
CID_B="$(service_id node-b)"
|
||||||
|
|
||||||
|
[[ -n "$CID_A" && -n "$CID_B" ]] || { echo "[ERR] node containers not found" >&2; exit 1; }
|
||||||
|
|
||||||
|
send_logs "$CID_A" "host01"
|
||||||
|
send_logs "$CID_B" "host02"
|
||||||
|
|
||||||
|
log "Waiting for ES to ingest"
|
||||||
|
sleep 10
|
||||||
|
|
||||||
|
train1=$(get_count "train-*")
|
||||||
|
infer1=$(get_count "infer-*")
|
||||||
|
final=$((train1 + infer1))
|
||||||
|
log "final counts: train=${train1} infer=${infer1} total=${final}"
|
||||||
|
|
||||||
|
if (( final <= base )); then
|
||||||
|
echo "[ERR] ES total did not increase (${base} -> ${final})" >&2
|
||||||
|
exit 1
|
||||||
|
fi
|
||||||
|
|
||||||
|
if (( final < 4 )); then
|
||||||
|
echo "[ERR] ES total below expected threshold: ${final} < 4" >&2
|
||||||
|
exit 1
|
||||||
|
fi
|
||||||
|
|
||||||
|
es_health=$(curl -s "http://localhost:9200/_cluster/health" | grep -o '"status":"[^"]*"' | cut -d'"' -f4)
|
||||||
|
if [[ "$es_health" != "green" && "$es_health" != "yellow" ]]; then
|
||||||
|
echo "[ERR] ES health not green/yellow: $es_health" >&2
|
||||||
|
exit 1
|
||||||
|
fi
|
||||||
|
|
||||||
|
if ! curl -fs "http://localhost:5601/api/status" >/dev/null 2>&1; then
|
||||||
|
echo "[WARN] Kibana status endpoint not available"
|
||||||
|
fi
|
||||||
|
|
||||||
|
log "ES counts increased and services healthy"
|
||||||
110
src/sys/debug/scripts/08_restart_agent_reregister.sh
Executable file
110
src/sys/debug/scripts/08_restart_agent_reregister.sh
Executable file
@@ -0,0 +1,110 @@
|
|||||||
|
#!/usr/bin/env bash
|
||||||
|
set -euo pipefail
|
||||||
|
|
||||||
|
# shellcheck source=common.sh
|
||||||
|
source "$(cd "$(dirname "${BASH_SOURCE[0]}")" && pwd)/common.sh"
|
||||||
|
|
||||||
|
ensure_env_file
|
||||||
|
ensure_paths_defined
|
||||||
|
|
||||||
|
API_BASE="http://localhost:32300/api/v1/master"
|
||||||
|
NODE_ENTRYPOINT="$DEBUG_ROOT/../tests/scripts/node_entrypoint.sh"
|
||||||
|
[[ -f "$NODE_ENTRYPOINT" ]] || { echo "[ERR] node entrypoint script missing at $NODE_ENTRYPOINT" >&2; exit 1; }
|
||||||
|
|
||||||
|
TARGET_FIXED_IP="${SYS_DEBUG_NODEB_FIXED_IP:-172.30.0.200}"
|
||||||
|
|
||||||
|
ID_B_FILE="$TMP_DIR/node_id_b"
|
||||||
|
IP_INIT_FILE="$TMP_DIR/initial_ip_b"
|
||||||
|
[[ -f "$ID_B_FILE" && -f "$IP_INIT_FILE" ]] || { echo "[ERR] Required node id/ip files missing in $TMP_DIR" >&2; exit 1; }
|
||||||
|
|
||||||
|
ID_B="$(cat "$ID_B_FILE")"
|
||||||
|
IP0_B="$(cat "$IP_INIT_FILE")"
|
||||||
|
|
||||||
|
DETAIL_BEFORE="$TMP_DIR/node_b_before.json"
|
||||||
|
curl -fsS "$API_BASE/nodes/$ID_B" -o "$DETAIL_BEFORE"
|
||||||
|
LAST0=$(python3 - "$DETAIL_BEFORE" <<'PY'
|
||||||
|
import json,sys
|
||||||
|
node=json.load(open(sys.argv[1]))
|
||||||
|
print(node.get("last_updated",""))
|
||||||
|
PY
|
||||||
|
)
|
||||||
|
IP_BEFORE=$(python3 - "$DETAIL_BEFORE" <<'PY'
|
||||||
|
import json,sys
|
||||||
|
node=json.load(open(sys.argv[1]))
|
||||||
|
print(node.get("meta_data",{}).get("ip",""))
|
||||||
|
PY
|
||||||
|
)
|
||||||
|
|
||||||
|
if [[ "$IP_BEFORE" != "$IP0_B" ]]; then
|
||||||
|
echo "[ERR] Expected initial IP $IP0_B for node-b, got $IP_BEFORE" >&2
|
||||||
|
exit 1
|
||||||
|
fi
|
||||||
|
|
||||||
|
if [[ "$IP_BEFORE" == "$TARGET_FIXED_IP" ]]; then
|
||||||
|
echo "[ERR] node-b current IP $IP_BEFORE already matches target $TARGET_FIXED_IP. Configure SYS_DEBUG_NODEB_FIXED_IP to a different address before rerun." >&2
|
||||||
|
exit 1
|
||||||
|
fi
|
||||||
|
|
||||||
|
service_id() {
|
||||||
|
compose ps -q "$1"
|
||||||
|
}
|
||||||
|
|
||||||
|
log "Recreating node-b (old IP $IP_BEFORE) with static IP $TARGET_FIXED_IP"
|
||||||
|
compose rm -sf node-b >/dev/null 2>&1 || true
|
||||||
|
|
||||||
|
CONTAINER_NAME="${SYS_DEBUG_CONTAINER_PREFIX:-argus-debug}-node-b"
|
||||||
|
docker rm -f "$CONTAINER_NAME" >/dev/null 2>&1 || true
|
||||||
|
|
||||||
|
AGENT_BIN_PATH="$(cat "$TMP_DIR/agent_binary_path")"
|
||||||
|
[[ -f "$AGENT_BIN_PATH" ]] || { echo "[ERR] Agent binary path missing in $TMP_DIR" >&2; exit 1; }
|
||||||
|
|
||||||
|
require_docker
|
||||||
|
|
||||||
|
docker run -d \
|
||||||
|
--name "$CONTAINER_NAME" \
|
||||||
|
--hostname "$HOST_B" \
|
||||||
|
--network "$SYS_DEBUG_NETWORK_NAME" \
|
||||||
|
--ip "$TARGET_FIXED_IP" \
|
||||||
|
--dns "${SYS_DEBUG_BIND_IP:-172.30.0.2}" \
|
||||||
|
-e MASTER_ENDPOINT=http://master.argus.com:3000 \
|
||||||
|
-e REPORT_INTERVAL_SECONDS=2 \
|
||||||
|
-e ARGUS_BUILD_UID=$ARGUS_BUILD_UID \
|
||||||
|
-e ARGUS_BUILD_GID=$ARGUS_BUILD_GID \
|
||||||
|
-e ES_HOST=es \
|
||||||
|
-e ES_PORT=9200 \
|
||||||
|
-e CLUSTER=local \
|
||||||
|
-e RACK=dev \
|
||||||
|
-p 2021:2020 \
|
||||||
|
-v "$SYS_DEBUG_PRIVATE_NODEB/argus/agent/$HOST_B:/private/argus/agent/$HOST_B" \
|
||||||
|
-v "$AGENT_BIN_PATH:/usr/local/bin/argus-agent:ro" \
|
||||||
|
-v "$NODE_ENTRYPOINT:/usr/local/bin/node-entrypoint.sh:ro" \
|
||||||
|
-v "$REPO_ROOT/src/log/fluent-bit/build/start-fluent-bit.sh:/assets/start-fluent-bit.sh:ro" \
|
||||||
|
-v "$REPO_ROOT/src/log/fluent-bit/build/etc:/assets/fluent-bit/etc:ro" \
|
||||||
|
-v "$REPO_ROOT/src/log/fluent-bit/build/packages:/assets/fluent-bit/packages:ro" \
|
||||||
|
--entrypoint /usr/local/bin/node-entrypoint.sh \
|
||||||
|
ubuntu:22.04 >/dev/null
|
||||||
|
|
||||||
|
log "Waiting for node-b to re-register with new IP"
|
||||||
|
for _ in {1..40}; do
|
||||||
|
sleep 3
|
||||||
|
if curl -fsS "$API_BASE/nodes/$ID_B" -o "$TMP_DIR/node_b_after.json"; then
|
||||||
|
if python3 - "$TMP_DIR/node_b_after.json" "$LAST0" "$TARGET_FIXED_IP" <<'PY'
|
||||||
|
import json,sys
|
||||||
|
node=json.load(open(sys.argv[1]))
|
||||||
|
last0=sys.argv[2]
|
||||||
|
expected_ip=sys.argv[3]
|
||||||
|
ip=node.get("meta_data",{}).get("ip")
|
||||||
|
lu=node.get("last_updated")
|
||||||
|
if ip == expected_ip and lu and lu != last0:
|
||||||
|
sys.exit(0)
|
||||||
|
sys.exit(1)
|
||||||
|
PY
|
||||||
|
then
|
||||||
|
log "node-b IP updated: $IP_BEFORE -> $TARGET_FIXED_IP"
|
||||||
|
exit 0
|
||||||
|
fi
|
||||||
|
fi
|
||||||
|
done
|
||||||
|
|
||||||
|
echo "[ERR] node-b did not update to IP $TARGET_FIXED_IP in time" >&2
|
||||||
|
exit 1
|
||||||
13
src/sys/debug/scripts/09_down.sh
Executable file
13
src/sys/debug/scripts/09_down.sh
Executable file
@@ -0,0 +1,13 @@
|
|||||||
|
#!/usr/bin/env bash
|
||||||
|
set -euo pipefail
|
||||||
|
|
||||||
|
# shellcheck source=common.sh
|
||||||
|
source "$(cd "$(dirname "${BASH_SOURCE[0]}")" && pwd)/common.sh"
|
||||||
|
|
||||||
|
ensure_env_file
|
||||||
|
require_docker
|
||||||
|
|
||||||
|
log "Stopping debug stack (project $SYS_DEBUG_PROJECT_NAME)"
|
||||||
|
compose down --remove-orphans >/dev/null 2>&1 || true
|
||||||
|
|
||||||
|
log "Containers stopped. No host directories were removed."
|
||||||
66
src/sys/debug/scripts/clean-data.sh
Executable file
66
src/sys/debug/scripts/clean-data.sh
Executable file
@@ -0,0 +1,66 @@
|
|||||||
|
#!/usr/bin/env bash
|
||||||
|
set -euo pipefail
|
||||||
|
|
||||||
|
# shellcheck source=common.sh
|
||||||
|
source "$(cd "$(dirname "${BASH_SOURCE[0]}")" && pwd)/common.sh"
|
||||||
|
|
||||||
|
ensure_env_file
|
||||||
|
ensure_paths_defined
|
||||||
|
|
||||||
|
FORCE=false
|
||||||
|
while [[ $# -gt 0 ]]; do
|
||||||
|
case "$1" in
|
||||||
|
-y|--yes)
|
||||||
|
FORCE=true
|
||||||
|
;;
|
||||||
|
-h|--help)
|
||||||
|
cat <<USAGE
|
||||||
|
Usage: ${0##*/} [--yes]
|
||||||
|
|
||||||
|
Safely remove debug private directories after adjusting ownership.
|
||||||
|
USAGE
|
||||||
|
exit 0
|
||||||
|
;;
|
||||||
|
*)
|
||||||
|
echo "Unknown argument: $1" >&2
|
||||||
|
exit 1
|
||||||
|
;;
|
||||||
|
esac
|
||||||
|
shift
|
||||||
|
done
|
||||||
|
|
||||||
|
if [[ $FORCE == false ]]; then
|
||||||
|
read -r -p "This will delete debug private directories. Continue? [y/N] " reply
|
||||||
|
case "$reply" in
|
||||||
|
y|Y|yes|YES)
|
||||||
|
;;
|
||||||
|
*)
|
||||||
|
echo "Aborted"
|
||||||
|
exit 0
|
||||||
|
;;
|
||||||
|
esac
|
||||||
|
fi
|
||||||
|
|
||||||
|
paths=(
|
||||||
|
"$SYS_DEBUG_PRIVATE_CORE"
|
||||||
|
"$SYS_DEBUG_PRIVATE_NODEA"
|
||||||
|
"$SYS_DEBUG_PRIVATE_NODEB"
|
||||||
|
"$SYS_DEBUG_TMP_DIR"
|
||||||
|
)
|
||||||
|
|
||||||
|
require_docker
|
||||||
|
|
||||||
|
image="ubuntu:22.04"
|
||||||
|
|
||||||
|
for dir in "${paths[@]}"; do
|
||||||
|
[[ -d "$dir" ]] || continue
|
||||||
|
log "Fixing ownership for $dir"
|
||||||
|
if ! docker run --rm -v "$dir:/target" "$image" chown -R "$(id -u):$(id -g)" /target >/dev/null 2>&1; then
|
||||||
|
echo "[WARN] Failed to adjust ownership via $image, attempting local chown" >&2
|
||||||
|
chown -R "$(id -u):$(id -g)" "$dir" >/dev/null 2>&1 || true
|
||||||
|
fi
|
||||||
|
log "Removing $dir"
|
||||||
|
rm -rf "$dir"
|
||||||
|
done
|
||||||
|
|
||||||
|
log "Clean data completed"
|
||||||
96
src/sys/debug/scripts/common.sh
Executable file
96
src/sys/debug/scripts/common.sh
Executable file
@@ -0,0 +1,96 @@
|
|||||||
|
#!/usr/bin/env bash
|
||||||
|
set -euo pipefail
|
||||||
|
|
||||||
|
SCRIPT_DIR="$(cd "$(dirname "${BASH_SOURCE[0]}")" && pwd)"
|
||||||
|
DEBUG_ROOT="$(cd "$SCRIPT_DIR/.." && pwd)"
|
||||||
|
REPO_ROOT="$(cd "$DEBUG_ROOT/../../.." && pwd)"
|
||||||
|
ENV_FILE="$DEBUG_ROOT/.env"
|
||||||
|
|
||||||
|
source "$REPO_ROOT/scripts/common/build_user.sh"
|
||||||
|
load_build_user
|
||||||
|
|
||||||
|
if [[ -f "$ENV_FILE" ]]; then
|
||||||
|
set -a
|
||||||
|
# shellcheck disable=SC1090
|
||||||
|
source "$ENV_FILE"
|
||||||
|
set +a
|
||||||
|
fi
|
||||||
|
|
||||||
|
SYS_DEBUG_NETWORK_NAME=${SYS_DEBUG_NETWORK_NAME:-argus-debug-net}
|
||||||
|
SYS_DEBUG_NETWORK_SUBNET=${SYS_DEBUG_NETWORK_SUBNET:-172.30.0.0/16}
|
||||||
|
SYS_DEBUG_NETWORK_GATEWAY=${SYS_DEBUG_NETWORK_GATEWAY:-172.30.0.1}
|
||||||
|
SYS_DEBUG_PROJECT_NAME=${SYS_DEBUG_PROJECT_NAME:-argus-debug}
|
||||||
|
SYS_DEBUG_CONTAINER_PREFIX=${SYS_DEBUG_CONTAINER_PREFIX:-argus-debug}
|
||||||
|
SYS_DEBUG_PRIVATE_CORE=${SYS_DEBUG_PRIVATE_CORE:-$DEBUG_ROOT/private}
|
||||||
|
SYS_DEBUG_PRIVATE_NODEA=${SYS_DEBUG_PRIVATE_NODEA:-$DEBUG_ROOT/private-nodea}
|
||||||
|
SYS_DEBUG_PRIVATE_NODEB=${SYS_DEBUG_PRIVATE_NODEB:-$DEBUG_ROOT/private-nodeb}
|
||||||
|
SYS_DEBUG_TMP_DIR=${SYS_DEBUG_TMP_DIR:-$DEBUG_ROOT/tmp}
|
||||||
|
ARGUS_BUILD_UID=${ARGUS_BUILD_UID:-2133}
|
||||||
|
ARGUS_BUILD_GID=${ARGUS_BUILD_GID:-2015}
|
||||||
|
|
||||||
|
SYS_DEBUG_NODEA_HOST=${SYS_DEBUG_NODEA_HOST:-dev-yyrshare-nbnyx10-cp2f-pod-0}
|
||||||
|
SYS_DEBUG_NODEB_HOST=${SYS_DEBUG_NODEB_HOST:-dev-yyrshare-uuuu10-ep2f-pod-0}
|
||||||
|
|
||||||
|
HOST_A="$SYS_DEBUG_NODEA_HOST"
|
||||||
|
HOST_B="$SYS_DEBUG_NODEB_HOST"
|
||||||
|
|
||||||
|
COMPOSE_FILE="$DEBUG_ROOT/docker-compose.yml"
|
||||||
|
|
||||||
|
abs_path() {
|
||||||
|
python3 - "$1" <<'PY'
|
||||||
|
import os, sys
|
||||||
|
path = sys.argv[1]
|
||||||
|
print(os.path.abspath(path))
|
||||||
|
PY
|
||||||
|
}
|
||||||
|
|
||||||
|
ensure_command() {
|
||||||
|
local cmd="$1"
|
||||||
|
if ! command -v "$cmd" >/dev/null 2>&1; then
|
||||||
|
echo "[ERR] Required command '$cmd' not found" >&2
|
||||||
|
exit 1
|
||||||
|
fi
|
||||||
|
}
|
||||||
|
|
||||||
|
require_docker() {
|
||||||
|
ensure_command docker
|
||||||
|
}
|
||||||
|
|
||||||
|
compose() {
|
||||||
|
require_docker
|
||||||
|
local bin
|
||||||
|
if docker compose version >/dev/null 2>&1; then
|
||||||
|
bin=(docker compose)
|
||||||
|
else
|
||||||
|
bin=(docker-compose)
|
||||||
|
fi
|
||||||
|
"${bin[@]}" -p "$SYS_DEBUG_PROJECT_NAME" -f "$COMPOSE_FILE" "$@"
|
||||||
|
}
|
||||||
|
|
||||||
|
ensure_paths_defined() {
|
||||||
|
local missing=()
|
||||||
|
for name in SYS_DEBUG_PRIVATE_CORE SYS_DEBUG_PRIVATE_NODEA SYS_DEBUG_PRIVATE_NODEB SYS_DEBUG_TMP_DIR; do
|
||||||
|
if [[ -z "${!name:-}" ]]; then
|
||||||
|
missing+=("$name")
|
||||||
|
fi
|
||||||
|
done
|
||||||
|
if (( ${#missing[@]} > 0 )); then
|
||||||
|
echo "[ERR] Missing required environment variables: ${missing[*]}" >&2
|
||||||
|
echo " Run 01_bootstrap.sh first." >&2
|
||||||
|
exit 1
|
||||||
|
fi
|
||||||
|
}
|
||||||
|
|
||||||
|
ensure_env_file() {
|
||||||
|
if [[ ! -f "$ENV_FILE" ]]; then
|
||||||
|
echo "[ERR] Missing .env at $ENV_FILE. Run 01_bootstrap.sh first." >&2
|
||||||
|
exit 1
|
||||||
|
fi
|
||||||
|
}
|
||||||
|
|
||||||
|
log() {
|
||||||
|
echo "[INFO] $*"
|
||||||
|
}
|
||||||
|
|
||||||
|
TMP_DIR="$SYS_DEBUG_TMP_DIR"
|
||||||
|
mkdir -p "$TMP_DIR"
|
||||||
76
src/sys/debug/scripts/network-create.sh
Executable file
76
src/sys/debug/scripts/network-create.sh
Executable file
@@ -0,0 +1,76 @@
|
|||||||
|
#!/usr/bin/env bash
|
||||||
|
set -euo pipefail
|
||||||
|
|
||||||
|
# shellcheck source=common.sh
|
||||||
|
source "$(cd "$(dirname "${BASH_SOURCE[0]}")" && pwd)/common.sh"
|
||||||
|
|
||||||
|
NAME="$SYS_DEBUG_NETWORK_NAME"
|
||||||
|
SUBNET="$SYS_DEBUG_NETWORK_SUBNET"
|
||||||
|
GATEWAY="$SYS_DEBUG_NETWORK_GATEWAY"
|
||||||
|
|
||||||
|
usage() {
|
||||||
|
cat <<EOF
|
||||||
|
Usage: ${0##*/} [--name NAME] [--subnet CIDR] [--gateway IP]
|
||||||
|
|
||||||
|
Create (if missing) the external debug docker network.
|
||||||
|
|
||||||
|
Defaults derived from .env or:
|
||||||
|
name = $NAME
|
||||||
|
subnet = $SUBNET
|
||||||
|
gateway = $GATEWAY
|
||||||
|
EOF
|
||||||
|
}
|
||||||
|
|
||||||
|
while [[ $# -gt 0 ]]; do
|
||||||
|
case "$1" in
|
||||||
|
--name)
|
||||||
|
shift; [[ $# -gt 0 ]] || { echo "--name requires value" >&2; exit 1; }
|
||||||
|
NAME="$1"
|
||||||
|
;;
|
||||||
|
--name=*)
|
||||||
|
NAME="${1#*=}"
|
||||||
|
;;
|
||||||
|
--subnet)
|
||||||
|
shift; [[ $# -gt 0 ]] || { echo "--subnet requires value" >&2; exit 1; }
|
||||||
|
SUBNET="$1"
|
||||||
|
;;
|
||||||
|
--subnet=*)
|
||||||
|
SUBNET="${1#*=}"
|
||||||
|
;;
|
||||||
|
--gateway)
|
||||||
|
shift; [[ $# -gt 0 ]] || { echo "--gateway requires value" >&2; exit 1; }
|
||||||
|
GATEWAY="$1"
|
||||||
|
;;
|
||||||
|
--gateway=*)
|
||||||
|
GATEWAY="${1#*=}"
|
||||||
|
;;
|
||||||
|
-h|--help)
|
||||||
|
usage
|
||||||
|
exit 0
|
||||||
|
;;
|
||||||
|
*)
|
||||||
|
echo "Unknown argument: $1" >&2
|
||||||
|
usage >&2
|
||||||
|
exit 1
|
||||||
|
;;
|
||||||
|
esac
|
||||||
|
shift
|
||||||
|
done
|
||||||
|
|
||||||
|
require_docker
|
||||||
|
|
||||||
|
if docker network inspect "$NAME" >/dev/null 2>&1; then
|
||||||
|
log "Network $NAME already exists"
|
||||||
|
exit 0
|
||||||
|
fi
|
||||||
|
|
||||||
|
log "Creating network $NAME (subnet=$SUBNET gateway=$GATEWAY)"
|
||||||
|
docker network create \
|
||||||
|
--driver bridge \
|
||||||
|
--subnet "$SUBNET" \
|
||||||
|
--gateway "$GATEWAY" \
|
||||||
|
"$NAME"
|
||||||
|
|
||||||
|
mkdir -p "$TMP_DIR"
|
||||||
|
echo "$NAME" > "$TMP_DIR/network.created"
|
||||||
|
log "Network $NAME created"
|
||||||
55
src/sys/debug/scripts/network-destroy.sh
Executable file
55
src/sys/debug/scripts/network-destroy.sh
Executable file
@@ -0,0 +1,55 @@
|
|||||||
|
#!/usr/bin/env bash
|
||||||
|
set -euo pipefail
|
||||||
|
|
||||||
|
# shellcheck source=common.sh
|
||||||
|
source "$(cd "$(dirname "${BASH_SOURCE[0]}")" && pwd)/common.sh"
|
||||||
|
|
||||||
|
NAME="$SYS_DEBUG_NETWORK_NAME"
|
||||||
|
|
||||||
|
usage() {
|
||||||
|
cat <<EOF
|
||||||
|
Usage: ${0##*/} [--name NAME]
|
||||||
|
|
||||||
|
Destroy the debug docker network if no containers are attached.
|
||||||
|
EOF
|
||||||
|
}
|
||||||
|
|
||||||
|
while [[ $# -gt 0 ]]; do
|
||||||
|
case "$1" in
|
||||||
|
--name)
|
||||||
|
shift; [[ $# -gt 0 ]] || { echo "--name requires value" >&2; exit 1; }
|
||||||
|
NAME="$1"
|
||||||
|
;;
|
||||||
|
--name=*)
|
||||||
|
NAME="${1#*=}"
|
||||||
|
;;
|
||||||
|
-h|--help)
|
||||||
|
usage
|
||||||
|
exit 0
|
||||||
|
;;
|
||||||
|
*)
|
||||||
|
echo "Unknown argument: $1" >&2
|
||||||
|
usage >&2
|
||||||
|
exit 1
|
||||||
|
;;
|
||||||
|
esac
|
||||||
|
shift
|
||||||
|
done
|
||||||
|
|
||||||
|
require_docker
|
||||||
|
|
||||||
|
if ! docker network inspect "$NAME" >/dev/null 2>&1; then
|
||||||
|
log "Network $NAME not found; nothing to do"
|
||||||
|
exit 0
|
||||||
|
fi
|
||||||
|
|
||||||
|
attached=$(docker network inspect -f '{{range $id, $conf := .Containers}}{{printf "%s " $conf.Name}}{{end}}' "$NAME")
|
||||||
|
if [[ -n "${attached// }" ]]; then
|
||||||
|
echo "[ERR] Cannot remove network $NAME: still connected containers -> $attached" >&2
|
||||||
|
exit 1
|
||||||
|
fi
|
||||||
|
|
||||||
|
log "Deleting network $NAME"
|
||||||
|
docker network rm "$NAME" >/dev/null
|
||||||
|
rm -f "$TMP_DIR/network.created"
|
||||||
|
log "Network $NAME removed"
|
||||||
@@ -42,7 +42,7 @@
|
|||||||
- `./scripts/05_agent_register.sh` 获取两个节点的 `node_id` 与初始 IP,检查本地 `node.json`
|
- `./scripts/05_agent_register.sh` 获取两个节点的 `node_id` 与初始 IP,检查本地 `node.json`
|
||||||
- `./scripts/06_write_health_and_assert.sh` 写健康文件并断言 `nodes.json` 仅包含 2 个在线节点
|
- `./scripts/06_write_health_and_assert.sh` 写健康文件并断言 `nodes.json` 仅包含 2 个在线节点
|
||||||
- `./scripts/07_logs_send_and_assert.sh` 向两个节点写日志,断言 ES `train-*`/`infer-*` 计数增长
|
- `./scripts/07_logs_send_and_assert.sh` 向两个节点写日志,断言 ES `train-*`/`infer-*` 计数增长
|
||||||
- `./scripts/08_restart_agent_reregister.sh` `node-b` 改为固定 IP `172.29.0.200`,验证保持同一节点 ID 且 IP/时间戳更新
|
- `./scripts/08_restart_agent_reregister.sh` `node-b` 改为固定 IP `172.31.0.200`,验证保持同一节点 ID 且 IP/时间戳更新
|
||||||
- `./scripts/09_down.sh` 回收容器、网络并清理 `private*/`、`tmp/`
|
- `./scripts/09_down.sh` 回收容器、网络并清理 `private*/`、`tmp/`
|
||||||
|
|
||||||
- 重置环境
|
- 重置环境
|
||||||
@@ -53,8 +53,8 @@
|
|||||||
## 二、测试部署架构(docker-compose)
|
## 二、测试部署架构(docker-compose)
|
||||||
|
|
||||||
- 网络
|
- 网络
|
||||||
- 自定义 bridge:`argus-sys-net`,子网 `172.29.0.0/16`
|
- 自定义 bridge:`argus-sys-net`,子网 `172.31.0.0/16`
|
||||||
- 固定地址:bind=`172.29.0.2`,master=`172.29.0.10`
|
- 固定地址:bind=`172.31.0.2`,master=`172.31.0.10`
|
||||||
|
|
||||||
- 服务与端口
|
- 服务与端口
|
||||||
- `bind`(`argus-bind9:latest`):监听 53/tcp+udp;负责同步 `*.argus.com` 记录
|
- `bind`(`argus-bind9:latest`):监听 53/tcp+udp;负责同步 `*.argus.com` 记录
|
||||||
@@ -72,7 +72,7 @@
|
|||||||
- 节点容器的 Fluent Bit/agent 资产以只读方式挂载到 `/assets`/`/usr/local/bin/argus-agent`
|
- 节点容器的 Fluent Bit/agent 资产以只读方式挂载到 `/assets`/`/usr/local/bin/argus-agent`
|
||||||
|
|
||||||
- DNS 配置
|
- DNS 配置
|
||||||
- 节点容器通过 compose 配置 `dns: [172.29.0.2]` 指向 bind,不挂载 `/etc/resolv.conf`,也不依赖 `update-dns.sh`
|
- 节点容器通过 compose 配置 `dns: [172.31.0.2]` 指向 bind,不挂载 `/etc/resolv.conf`,也不依赖 `update-dns.sh`
|
||||||
- master/es/kibana 仍共享 `./private`,master 启动会写 `/private/argus/etc/master.argus.com` 供 bind 同步 A 记录
|
- master/es/kibana 仍共享 `./private`,master 启动会写 `/private/argus/etc/master.argus.com` 供 bind 同步 A 记录
|
||||||
|
|
||||||
- 节点入口
|
- 节点入口
|
||||||
|
|||||||
@@ -1,21 +1,18 @@
|
|||||||
version: "3.8"
|
|
||||||
|
|
||||||
networks:
|
networks:
|
||||||
default:
|
sysnet:
|
||||||
name: argus-sys-net
|
|
||||||
driver: bridge
|
driver: bridge
|
||||||
ipam:
|
ipam:
|
||||||
driver: default
|
driver: default
|
||||||
config:
|
config:
|
||||||
- subnet: 172.29.0.0/16
|
- subnet: 172.31.0.0/16
|
||||||
|
|
||||||
services:
|
services:
|
||||||
bind:
|
bind:
|
||||||
image: ${BIND_IMAGE_TAG:-argus-bind9:latest}
|
image: ${BIND_IMAGE_TAG:-argus-bind9:latest}
|
||||||
container_name: argus-bind-sys
|
container_name: argus-bind-sys
|
||||||
networks:
|
networks:
|
||||||
default:
|
sysnet:
|
||||||
ipv4_address: 172.29.0.2
|
ipv4_address: 172.31.0.2
|
||||||
volumes:
|
volumes:
|
||||||
- ./private:/private
|
- ./private:/private
|
||||||
restart: unless-stopped
|
restart: unless-stopped
|
||||||
@@ -38,8 +35,8 @@ services:
|
|||||||
- ./private/argus/metric/prometheus:/private/argus/metric/prometheus
|
- ./private/argus/metric/prometheus:/private/argus/metric/prometheus
|
||||||
- ./private/argus/etc:/private/argus/etc
|
- ./private/argus/etc:/private/argus/etc
|
||||||
networks:
|
networks:
|
||||||
default:
|
sysnet:
|
||||||
ipv4_address: 172.29.0.10
|
ipv4_address: 172.31.0.10
|
||||||
restart: unless-stopped
|
restart: unless-stopped
|
||||||
|
|
||||||
es:
|
es:
|
||||||
@@ -57,6 +54,9 @@ services:
|
|||||||
ports:
|
ports:
|
||||||
- "9200:9200"
|
- "9200:9200"
|
||||||
restart: unless-stopped
|
restart: unless-stopped
|
||||||
|
networks:
|
||||||
|
sysnet:
|
||||||
|
ipv4_address: 172.31.0.3
|
||||||
|
|
||||||
kibana:
|
kibana:
|
||||||
image: argus-kibana:latest
|
image: argus-kibana:latest
|
||||||
@@ -73,6 +73,9 @@ services:
|
|||||||
ports:
|
ports:
|
||||||
- "5601:5601"
|
- "5601:5601"
|
||||||
restart: unless-stopped
|
restart: unless-stopped
|
||||||
|
networks:
|
||||||
|
sysnet:
|
||||||
|
ipv4_address: 172.31.0.4
|
||||||
|
|
||||||
node-a:
|
node-a:
|
||||||
image: ubuntu:22.04
|
image: ubuntu:22.04
|
||||||
@@ -101,10 +104,12 @@ services:
|
|||||||
entrypoint:
|
entrypoint:
|
||||||
- /usr/local/bin/node-entrypoint.sh
|
- /usr/local/bin/node-entrypoint.sh
|
||||||
dns:
|
dns:
|
||||||
- 172.29.0.2
|
- 172.31.0.2
|
||||||
ports:
|
ports:
|
||||||
- "2020:2020"
|
- "2020:2020"
|
||||||
restart: unless-stopped
|
restart: unless-stopped
|
||||||
|
networks:
|
||||||
|
- sysnet
|
||||||
|
|
||||||
node-b:
|
node-b:
|
||||||
image: ubuntu:22.04
|
image: ubuntu:22.04
|
||||||
@@ -133,7 +138,256 @@ services:
|
|||||||
entrypoint:
|
entrypoint:
|
||||||
- /usr/local/bin/node-entrypoint.sh
|
- /usr/local/bin/node-entrypoint.sh
|
||||||
dns:
|
dns:
|
||||||
- 172.29.0.2
|
- 172.31.0.2
|
||||||
ports:
|
ports:
|
||||||
- "2021:2020"
|
- "2021:2020"
|
||||||
restart: unless-stopped
|
restart: unless-stopped
|
||||||
|
networks:
|
||||||
|
- sysnet
|
||||||
|
|
||||||
|
ftp:
|
||||||
|
image: argus-metric-ftp:latest
|
||||||
|
container_name: argus-ftp
|
||||||
|
restart: unless-stopped
|
||||||
|
environment:
|
||||||
|
- TZ=Asia/Shanghai
|
||||||
|
- FTP_BASE_PATH=/private/argus/ftp
|
||||||
|
- FTP_PASSWORD=${FTP_PASSWORD:-ZGClab1234!}
|
||||||
|
- DOMAIN=${FTP_DOMAIN:-ftp.metric.argus.com}
|
||||||
|
- ARGUS_BUILD_UID=${ARGUS_BUILD_UID:-2133}
|
||||||
|
- ARGUS_BUILD_GID=${ARGUS_BUILD_GID:-2015}
|
||||||
|
ports:
|
||||||
|
- "${FTP_PORT:-21}:21"
|
||||||
|
- "${FTP_DATA_PORT:-20}:20"
|
||||||
|
- "21100-21110:21100-21110"
|
||||||
|
volumes:
|
||||||
|
- ./private/argus/metric/ftp:/private/argus/ftp
|
||||||
|
- ./private/argus/etc:/private/argus/etc
|
||||||
|
- /etc/localtime:/etc/localtime:ro
|
||||||
|
- /etc/timezone:/etc/timezone:ro
|
||||||
|
networks:
|
||||||
|
sysnet:
|
||||||
|
ipv4_address: 172.31.0.40
|
||||||
|
logging:
|
||||||
|
driver: "json-file"
|
||||||
|
options:
|
||||||
|
max-size: "10m"
|
||||||
|
max-file: "3"
|
||||||
|
|
||||||
|
prometheus:
|
||||||
|
image: argus-metric-prometheus:latest
|
||||||
|
container_name: argus-prometheus
|
||||||
|
restart: unless-stopped
|
||||||
|
environment:
|
||||||
|
- TZ=Asia/Shanghai
|
||||||
|
- PROMETHEUS_BASE_PATH=/private/argus/metric/prometheus
|
||||||
|
- ARGUS_BUILD_UID=${ARGUS_BUILD_UID:-2133}
|
||||||
|
- ARGUS_BUILD_GID=${ARGUS_BUILD_GID:-2015}
|
||||||
|
ports:
|
||||||
|
- "${PROMETHEUS_PORT:-9090}:9090"
|
||||||
|
volumes:
|
||||||
|
- ./private/argus/metric/prometheus:/private/argus/metric/prometheus
|
||||||
|
- ./private/argus/etc:/private/argus/etc
|
||||||
|
- /etc/localtime:/etc/localtime:ro
|
||||||
|
- /etc/timezone:/etc/timezone:ro
|
||||||
|
networks:
|
||||||
|
sysnet:
|
||||||
|
ipv4_address: 172.31.0.41
|
||||||
|
logging:
|
||||||
|
driver: "json-file"
|
||||||
|
options:
|
||||||
|
max-size: "10m"
|
||||||
|
max-file: "3"
|
||||||
|
|
||||||
|
grafana:
|
||||||
|
image: argus-metric-grafana:latest
|
||||||
|
container_name: argus-grafana
|
||||||
|
restart: unless-stopped
|
||||||
|
environment:
|
||||||
|
- TZ=Asia/Shanghai
|
||||||
|
- GRAFANA_BASE_PATH=/private/argus/metric/grafana
|
||||||
|
- ARGUS_BUILD_UID=${ARGUS_BUILD_UID:-2133}
|
||||||
|
- ARGUS_BUILD_GID=${ARGUS_BUILD_GID:-2015}
|
||||||
|
- GF_SERVER_HTTP_PORT=3000
|
||||||
|
- GF_LOG_LEVEL=warn
|
||||||
|
- GF_LOG_MODE=console
|
||||||
|
- GF_PATHS_PROVISIONING=/private/argus/metric/grafana/provisioning
|
||||||
|
- GF_AUTH_ANONYMOUS_ENABLED=true
|
||||||
|
- GF_AUTH_ANONYMOUS_ORG_ROLE=Viewer
|
||||||
|
ports:
|
||||||
|
- "${GRAFANA_PORT:-3000}:3000"
|
||||||
|
volumes:
|
||||||
|
- ./private/argus/metric/grafana:/private/argus/metric/grafana
|
||||||
|
- ./private/argus/etc:/private/argus/etc
|
||||||
|
- /etc/localtime:/etc/localtime:ro
|
||||||
|
- /etc/timezone:/etc/timezone:ro
|
||||||
|
networks:
|
||||||
|
sysnet:
|
||||||
|
ipv4_address: 172.31.0.42
|
||||||
|
depends_on:
|
||||||
|
- prometheus
|
||||||
|
logging:
|
||||||
|
driver: "json-file"
|
||||||
|
options:
|
||||||
|
max-size: "10m"
|
||||||
|
max-file: "3"
|
||||||
|
|
||||||
|
# --- Added: Web Frontend (no host port; resolved by DNS as web.argus.com) ---
|
||||||
|
web-frontend:
|
||||||
|
image: argus-web-frontend:latest
|
||||||
|
container_name: argus-web-frontend
|
||||||
|
environment:
|
||||||
|
- ARGUS_BUILD_UID=${ARGUS_BUILD_UID:-2133}
|
||||||
|
- ARGUS_BUILD_GID=${ARGUS_BUILD_GID:-2015}
|
||||||
|
volumes:
|
||||||
|
- ./private/argus/etc:/private/argus/etc
|
||||||
|
networks:
|
||||||
|
sysnet:
|
||||||
|
ipv4_address: 172.31.0.80
|
||||||
|
restart: unless-stopped
|
||||||
|
logging:
|
||||||
|
driver: "json-file"
|
||||||
|
options:
|
||||||
|
max-size: "10m"
|
||||||
|
max-file: "3"
|
||||||
|
|
||||||
|
test-node:
|
||||||
|
image: ubuntu:22.04
|
||||||
|
container_name: argus-metric-test-node
|
||||||
|
hostname: test-metric-node-001
|
||||||
|
restart: unless-stopped
|
||||||
|
privileged: true
|
||||||
|
depends_on:
|
||||||
|
- ftp
|
||||||
|
- prometheus
|
||||||
|
environment:
|
||||||
|
- TZ=Asia/Shanghai
|
||||||
|
- DEBIAN_FRONTEND=noninteractive
|
||||||
|
- FTP_DOMAIN=${FTP_DOMAIN:-ftp.metric.argus.com}
|
||||||
|
- FTP_SERVER=${FTP_SERVER:-172.31.0.40}
|
||||||
|
- FTP_USER=${FTP_USER:-ftpuser}
|
||||||
|
- FTP_PASSWORD=${FTP_PASSWORD:-ZGClab1234!}
|
||||||
|
- FTP_PORT=${FTP_PORT:-21}
|
||||||
|
- ARGUS_BUILD_UID=${ARGUS_BUILD_UID:-2133}
|
||||||
|
- ARGUS_BUILD_GID=${ARGUS_BUILD_GID:-2015}
|
||||||
|
- METRIC_NODE_ROLE=cpu
|
||||||
|
volumes:
|
||||||
|
- ./private/argus/agent:/private/argus/agent
|
||||||
|
- ./scripts/metric/test-node-entrypoint.sh:/usr/local/bin/metric-test-node-entrypoint.sh:ro
|
||||||
|
- /etc/localtime:/etc/localtime:ro
|
||||||
|
- /etc/timezone:/etc/timezone:ro
|
||||||
|
entrypoint:
|
||||||
|
- /usr/local/bin/metric-test-node-entrypoint.sh
|
||||||
|
command:
|
||||||
|
- sleep
|
||||||
|
- infinity
|
||||||
|
networks:
|
||||||
|
sysnet:
|
||||||
|
ipv4_address: 172.31.0.50
|
||||||
|
logging:
|
||||||
|
driver: "json-file"
|
||||||
|
options:
|
||||||
|
max-size: "10m"
|
||||||
|
max-file: "3"
|
||||||
|
|
||||||
|
test-gpu-node:
|
||||||
|
profiles: ["gpu"]
|
||||||
|
image: nvidia/cuda:12.2.2-runtime-ubuntu22.04
|
||||||
|
container_name: argus-metric-test-gpu-node
|
||||||
|
hostname: test-metric-gpu-node-001
|
||||||
|
restart: unless-stopped
|
||||||
|
privileged: true
|
||||||
|
runtime: nvidia
|
||||||
|
deploy:
|
||||||
|
resources:
|
||||||
|
reservations:
|
||||||
|
devices:
|
||||||
|
- driver: nvidia
|
||||||
|
count: all
|
||||||
|
capabilities:
|
||||||
|
- gpu
|
||||||
|
depends_on:
|
||||||
|
- ftp
|
||||||
|
- prometheus
|
||||||
|
environment:
|
||||||
|
- TZ=Asia/Shanghai
|
||||||
|
- DEBIAN_FRONTEND=noninteractive
|
||||||
|
- NVIDIA_VISIBLE_DEVICES=all
|
||||||
|
- NVIDIA_DRIVER_CAPABILITIES=compute,utility
|
||||||
|
- GPU_MODE=gpu
|
||||||
|
- ARGUS_BUILD_UID=${ARGUS_BUILD_UID:-2133}
|
||||||
|
- ARGUS_BUILD_GID=${ARGUS_BUILD_GID:-2015}
|
||||||
|
- METRIC_NODE_ROLE=gpu
|
||||||
|
volumes:
|
||||||
|
- ./private/argus/agent:/private/argus/agent
|
||||||
|
- ./scripts/metric/test-node-entrypoint.sh:/usr/local/bin/metric-test-node-entrypoint.sh:ro
|
||||||
|
- /etc/localtime:/etc/localtime:ro
|
||||||
|
- /etc/timezone:/etc/timezone:ro
|
||||||
|
entrypoint:
|
||||||
|
- /usr/local/bin/metric-test-node-entrypoint.sh
|
||||||
|
command:
|
||||||
|
- sleep
|
||||||
|
- infinity
|
||||||
|
networks:
|
||||||
|
sysnet:
|
||||||
|
ipv4_address: 172.31.0.51
|
||||||
|
logging:
|
||||||
|
driver: "json-file"
|
||||||
|
options:
|
||||||
|
max-size: "10m"
|
||||||
|
max-file: "3"
|
||||||
|
|
||||||
|
# --- Added: Alertmanager ---
|
||||||
|
alertmanager:
|
||||||
|
image: argus-alertmanager:latest
|
||||||
|
container_name: argus-alertmanager
|
||||||
|
environment:
|
||||||
|
- ARGUS_BUILD_UID=${ARGUS_BUILD_UID:-2133}
|
||||||
|
- ARGUS_BUILD_GID=${ARGUS_BUILD_GID:-2015}
|
||||||
|
volumes:
|
||||||
|
- ./private/argus/etc:/private/argus/etc
|
||||||
|
- ./private/argus/alert/alertmanager:/private/argus/alert/alertmanager
|
||||||
|
networks:
|
||||||
|
sysnet:
|
||||||
|
ipv4_address: 172.31.0.82
|
||||||
|
ports:
|
||||||
|
- "9093:9093"
|
||||||
|
restart: unless-stopped
|
||||||
|
logging:
|
||||||
|
driver: "json-file"
|
||||||
|
options:
|
||||||
|
max-size: "10m"
|
||||||
|
max-file: "3"
|
||||||
|
|
||||||
|
# --- Added: Web Proxy (multi-port gateway) ---
|
||||||
|
web-proxy:
|
||||||
|
image: argus-web-proxy:latest
|
||||||
|
container_name: argus-web-proxy
|
||||||
|
depends_on:
|
||||||
|
- bind
|
||||||
|
- master
|
||||||
|
- grafana
|
||||||
|
- prometheus
|
||||||
|
- kibana
|
||||||
|
- alertmanager
|
||||||
|
environment:
|
||||||
|
- ARGUS_BUILD_UID=${ARGUS_BUILD_UID:-2133}
|
||||||
|
- ARGUS_BUILD_GID=${ARGUS_BUILD_GID:-2015}
|
||||||
|
volumes:
|
||||||
|
- ./private/argus/etc:/private/argus/etc
|
||||||
|
networks:
|
||||||
|
sysnet:
|
||||||
|
ipv4_address: 172.31.0.81
|
||||||
|
ports:
|
||||||
|
- "8080:8080"
|
||||||
|
- "8081:8081"
|
||||||
|
- "8082:8082"
|
||||||
|
- "8083:8083"
|
||||||
|
- "8084:8084"
|
||||||
|
- "8085:8085"
|
||||||
|
restart: unless-stopped
|
||||||
|
logging:
|
||||||
|
driver: "json-file"
|
||||||
|
options:
|
||||||
|
max-size: "10m"
|
||||||
|
max-file: "3"
|
||||||
|
|||||||
@@ -3,6 +3,38 @@ set -euo pipefail
|
|||||||
|
|
||||||
SCRIPT_DIR="$(cd "$(dirname "${BASH_SOURCE[0]}")" && pwd)"
|
SCRIPT_DIR="$(cd "$(dirname "${BASH_SOURCE[0]}")" && pwd)"
|
||||||
|
|
||||||
|
ENABLE_GPU=false
|
||||||
|
|
||||||
|
usage() {
|
||||||
|
cat <<'EOF'
|
||||||
|
Usage: 00_e2e_test.sh [options]
|
||||||
|
|
||||||
|
Options:
|
||||||
|
--enable-gpu 启用 GPU 相关拓扑与测试流程
|
||||||
|
-h, --help 显示帮助信息
|
||||||
|
EOF
|
||||||
|
}
|
||||||
|
|
||||||
|
while [[ $# -gt 0 ]]; do
|
||||||
|
case "$1" in
|
||||||
|
--enable-gpu)
|
||||||
|
ENABLE_GPU=true
|
||||||
|
shift
|
||||||
|
;;
|
||||||
|
-h|--help)
|
||||||
|
usage
|
||||||
|
exit 0
|
||||||
|
;;
|
||||||
|
*)
|
||||||
|
echo "Unknown argument: $1" >&2
|
||||||
|
usage
|
||||||
|
exit 1
|
||||||
|
;;
|
||||||
|
esac
|
||||||
|
done
|
||||||
|
|
||||||
|
export ARGUS_SYS_ENABLE_GPU=$ENABLE_GPU
|
||||||
|
|
||||||
SCRIPTS=(
|
SCRIPTS=(
|
||||||
"01_bootstrap.sh"
|
"01_bootstrap.sh"
|
||||||
"02_up.sh"
|
"02_up.sh"
|
||||||
@@ -12,6 +44,11 @@ SCRIPTS=(
|
|||||||
"06_write_health_and_assert.sh"
|
"06_write_health_and_assert.sh"
|
||||||
"07_logs_send_and_assert.sh"
|
"07_logs_send_and_assert.sh"
|
||||||
"08_restart_agent_reregister.sh"
|
"08_restart_agent_reregister.sh"
|
||||||
|
"10_metric_publish.sh"
|
||||||
|
"11_metric_node_install.sh"
|
||||||
|
"12_metric_gpu_install.sh"
|
||||||
|
"13_metric_verify.sh"
|
||||||
|
"14_metric_cleanup.sh"
|
||||||
"09_down.sh"
|
"09_down.sh"
|
||||||
)
|
)
|
||||||
|
|
||||||
@@ -23,4 +60,3 @@ for script in "${SCRIPTS[@]}"; do
|
|||||||
done
|
done
|
||||||
|
|
||||||
echo "[SYS-E2E] All tests completed"
|
echo "[SYS-E2E] All tests completed"
|
||||||
|
|
||||||
|
|||||||
@@ -22,24 +22,66 @@ ensure_image() {
|
|||||||
}
|
}
|
||||||
|
|
||||||
echo "[INFO] Preparing directories..."
|
echo "[INFO] Preparing directories..."
|
||||||
|
ensure_writable_dir() {
|
||||||
|
local path="$1"
|
||||||
|
local parent
|
||||||
|
parent="$(dirname "$path")"
|
||||||
|
mkdir -p "$parent" 2>/dev/null || true
|
||||||
|
mkdir -p "$path" 2>/dev/null || true
|
||||||
|
if [[ ! -w "$path" ]]; then
|
||||||
|
docker run --rm -v "$parent:/target" ubuntu:24.04 bash -lc "chown -R $(id -u):$(id -g) /target" >/dev/null 2>&1 || true
|
||||||
|
fi
|
||||||
|
mkdir -p "$path"
|
||||||
|
}
|
||||||
|
|
||||||
|
# preflight: make base dirs writable if inherited from root-owned mounts
|
||||||
|
ensure_writable_dir "$PRIVATE_CORE/argus"
|
||||||
|
ensure_writable_dir "$PRIVATE_CORE/argus/metric"
|
||||||
|
ensure_writable_dir "$PRIVATE_CORE/argus/metric/grafana"
|
||||||
|
ensure_writable_dir "$PRIVATE_CORE/argus/metric/prometheus"
|
||||||
|
|
||||||
mkdir -p \
|
mkdir -p \
|
||||||
"$PRIVATE_CORE/argus/etc" \
|
"$PRIVATE_CORE/argus/etc" \
|
||||||
"$PRIVATE_CORE/argus/bind" \
|
"$PRIVATE_CORE/argus/bind" \
|
||||||
"$PRIVATE_CORE/argus/master" \
|
"$PRIVATE_CORE/argus/master" \
|
||||||
"$PRIVATE_CORE/argus/metric/prometheus" \
|
"$PRIVATE_CORE/argus/metric/prometheus" \
|
||||||
|
"$PRIVATE_CORE/argus/alert/alertmanager" \
|
||||||
|
"$PRIVATE_CORE/argus/metric/ftp/share" \
|
||||||
|
"$PRIVATE_CORE/argus/metric/grafana/data" \
|
||||||
|
"$PRIVATE_CORE/argus/metric/grafana/logs" \
|
||||||
|
"$PRIVATE_CORE/argus/metric/grafana/plugins" \
|
||||||
|
"$PRIVATE_CORE/argus/metric/grafana/provisioning/datasources" \
|
||||||
|
"$PRIVATE_CORE/argus/metric/grafana/provisioning/dashboards" \
|
||||||
|
"$PRIVATE_CORE/argus/metric/grafana/data/sessions" \
|
||||||
|
"$PRIVATE_CORE/argus/metric/grafana/data/dashboards" \
|
||||||
|
"$PRIVATE_CORE/argus/metric/grafana/config" \
|
||||||
|
"$PRIVATE_CORE/argus/metric/prometheus/data" \
|
||||||
|
"$PRIVATE_CORE/argus/metric/prometheus/rules" \
|
||||||
|
"$PRIVATE_CORE/argus/metric/prometheus/targets" \
|
||||||
|
"$PRIVATE_CORE/argus/agent" \
|
||||||
"$PRIVATE_CORE/argus/log/elasticsearch" \
|
"$PRIVATE_CORE/argus/log/elasticsearch" \
|
||||||
"$PRIVATE_CORE/argus/log/kibana" \
|
"$PRIVATE_CORE/argus/log/kibana" \
|
||||||
"$PRIVATE_NODEA/argus/agent/dev-yyrshare-nbnyx10-cp2f-pod-0/health" \
|
"$PRIVATE_NODEA/argus/agent/dev-yyrshare-nbnyx10-cp2f-pod-0/health" \
|
||||||
"$PRIVATE_NODEB/argus/agent/dev-yyrshare-uuuu10-ep2f-pod-0/health" \
|
"$PRIVATE_NODEB/argus/agent/dev-yyrshare-uuuu10-ep2f-pod-0/health" \
|
||||||
"$TMP_DIR"
|
"$TMP_DIR"
|
||||||
|
|
||||||
# Align ownership for supervisor-managed services (ES/Kibana expect UID/GID inside container)
|
# Align ownership for supervisor-managed services (ES/Kibana/Grafana expect UID/GID inside container)
|
||||||
echo "[INFO] Fixing ownership for core private directories..."
|
echo "[INFO] Fixing ownership for core private directories..."
|
||||||
chown -R "${ARGUS_BUILD_UID}:${ARGUS_BUILD_GID}" \
|
chown -R "${ARGUS_BUILD_UID}:${ARGUS_BUILD_GID}" \
|
||||||
"$PRIVATE_CORE/argus/log/elasticsearch" \
|
"$PRIVATE_CORE/argus/log/elasticsearch" \
|
||||||
"$PRIVATE_CORE/argus/log/kibana" \
|
"$PRIVATE_CORE/argus/log/kibana" \
|
||||||
|
"$PRIVATE_CORE/argus/metric/grafana" \
|
||||||
|
"$PRIVATE_CORE/argus/metric/prometheus" \
|
||||||
|
"$PRIVATE_CORE/argus/alert" \
|
||||||
|
"$PRIVATE_CORE/argus/metric/ftp" \
|
||||||
|
"$PRIVATE_CORE/argus/agent" \
|
||||||
"$PRIVATE_CORE/argus/etc" 2>/dev/null || true
|
"$PRIVATE_CORE/argus/etc" 2>/dev/null || true
|
||||||
|
|
||||||
|
# 确保 alert 与 etc 目录组可写,便于非 root 且仅匹配 GID 的服务写入运行文件
|
||||||
|
chmod -R g+w "$PRIVATE_CORE/argus/alert" "$PRIVATE_CORE/argus/etc" 2>/dev/null || true
|
||||||
|
|
||||||
|
echo "[INFO] Using compose-managed network (auto-created by docker compose)"
|
||||||
|
|
||||||
echo "[INFO] Distributing update-dns.sh for core services (bind/master/es/kibana)"
|
echo "[INFO] Distributing update-dns.sh for core services (bind/master/es/kibana)"
|
||||||
BIND_UPDATE_SRC="$REPO_ROOT/src/bind/build/update-dns.sh"
|
BIND_UPDATE_SRC="$REPO_ROOT/src/bind/build/update-dns.sh"
|
||||||
BIND_UPDATE_DEST="$PRIVATE_CORE/argus/etc/update-dns.sh"
|
BIND_UPDATE_DEST="$PRIVATE_CORE/argus/etc/update-dns.sh"
|
||||||
@@ -55,6 +97,12 @@ ensure_image "argus-elasticsearch:latest"
|
|||||||
ensure_image "argus-kibana:latest"
|
ensure_image "argus-kibana:latest"
|
||||||
ensure_image "argus-bind9:latest"
|
ensure_image "argus-bind9:latest"
|
||||||
ensure_image "argus-master:latest"
|
ensure_image "argus-master:latest"
|
||||||
|
ensure_image "argus-metric-ftp:latest"
|
||||||
|
ensure_image "argus-metric-prometheus:latest"
|
||||||
|
ensure_image "argus-metric-grafana:latest"
|
||||||
|
ensure_image "argus-web-frontend:latest"
|
||||||
|
ensure_image "argus-web-proxy:latest"
|
||||||
|
ensure_image "argus-alertmanager:latest"
|
||||||
|
|
||||||
echo "[INFO] Building agent binary..."
|
echo "[INFO] Building agent binary..."
|
||||||
pushd "$REPO_ROOT/src/agent" >/dev/null
|
pushd "$REPO_ROOT/src/agent" >/dev/null
|
||||||
@@ -68,10 +116,50 @@ if [[ ! -x "$AGENT_BIN" ]]; then
|
|||||||
fi
|
fi
|
||||||
echo "$AGENT_BIN" > "$TMP_DIR/agent_binary_path"
|
echo "$AGENT_BIN" > "$TMP_DIR/agent_binary_path"
|
||||||
|
|
||||||
echo "[INFO] Writing .env with UID/GID"
|
# 检测GPU环境
|
||||||
|
REQUEST_GPU=${ARGUS_SYS_ENABLE_GPU:-false}
|
||||||
|
GPU_CHECK_SCRIPT="$REPO_ROOT/src/metric/tests/scripts/common/check-gpu.sh"
|
||||||
|
if [[ "$REQUEST_GPU" == "true" ]]; then
|
||||||
|
echo "[INFO] --enable-gpu 已启用,开始检测GPU环境..."
|
||||||
|
if [[ -f "$GPU_CHECK_SCRIPT" ]]; then
|
||||||
|
if bash "$GPU_CHECK_SCRIPT" >/dev/null 2>&1; then
|
||||||
|
echo "[INFO] GPU环境可用,将在 compose 中启用 test-gpu-node"
|
||||||
|
GPU_AVAILABLE=true
|
||||||
|
else
|
||||||
|
echo "[ERROR] 未检测到可用 GPU,但指定了 --enable-gpu" >&2
|
||||||
|
exit 1
|
||||||
|
fi
|
||||||
|
else
|
||||||
|
echo "[ERROR] 未找到 GPU 检测脚本: $GPU_CHECK_SCRIPT" >&2
|
||||||
|
exit 1
|
||||||
|
fi
|
||||||
|
else
|
||||||
|
GPU_AVAILABLE=false
|
||||||
|
echo "[INFO] GPU 支持未启用,跳过 GPU 检测"
|
||||||
|
fi
|
||||||
|
|
||||||
|
echo "[INFO] Writing .env with UID/GID and metric configuration"
|
||||||
cat > "$TEST_ROOT/.env" <<EOF
|
cat > "$TEST_ROOT/.env" <<EOF
|
||||||
ARGUS_BUILD_UID=$ARGUS_BUILD_UID
|
ARGUS_BUILD_UID=$ARGUS_BUILD_UID
|
||||||
ARGUS_BUILD_GID=$ARGUS_BUILD_GID
|
ARGUS_BUILD_GID=$ARGUS_BUILD_GID
|
||||||
|
|
||||||
|
# GPU 配置
|
||||||
|
ENABLE_GPU=$GPU_AVAILABLE
|
||||||
|
|
||||||
|
# FTP 配置
|
||||||
|
FTP_PORT=21
|
||||||
|
FTP_DATA_PORT=20
|
||||||
|
FTP_PASSWORD=ZGClab1234!
|
||||||
|
FTP_DOMAIN=ftp.metric.argus.com
|
||||||
|
|
||||||
|
# Prometheus 配置
|
||||||
|
PROMETHEUS_PORT=9090
|
||||||
|
|
||||||
|
# Grafana 配置
|
||||||
|
GRAFANA_PORT=3000
|
||||||
|
|
||||||
|
# 网络配置
|
||||||
|
USE_INTRANET=false
|
||||||
EOF
|
EOF
|
||||||
|
|
||||||
echo "[OK] Bootstrap completed"
|
echo "[OK] Bootstrap completed"
|
||||||
|
|||||||
@@ -3,6 +3,7 @@ set -euo pipefail
|
|||||||
|
|
||||||
SCRIPT_DIR="$(cd "$(dirname "${BASH_SOURCE[0]}")" && pwd)"
|
SCRIPT_DIR="$(cd "$(dirname "${BASH_SOURCE[0]}")" && pwd)"
|
||||||
TEST_ROOT="$(cd "$SCRIPT_DIR/.." && pwd)"
|
TEST_ROOT="$(cd "$SCRIPT_DIR/.." && pwd)"
|
||||||
|
REPO_ROOT="$(cd "$TEST_ROOT/../../.." && pwd)"
|
||||||
|
|
||||||
compose() {
|
compose() {
|
||||||
if docker compose version >/dev/null 2>&1; then
|
if docker compose version >/dev/null 2>&1; then
|
||||||
@@ -13,10 +14,73 @@ compose() {
|
|||||||
}
|
}
|
||||||
|
|
||||||
echo "[INFO] Bringing up system stack..."
|
echo "[INFO] Bringing up system stack..."
|
||||||
|
|
||||||
|
REQUEST_GPU=${ARGUS_SYS_ENABLE_GPU:-false}
|
||||||
|
GPU_AVAILABLE=false
|
||||||
|
GPU_CHECK_SCRIPT="$REPO_ROOT/src/metric/tests/scripts/common/check-gpu.sh"
|
||||||
|
|
||||||
|
if [[ "$REQUEST_GPU" == "true" ]]; then
|
||||||
|
echo "[INFO] --enable-gpu 生效,验证主机 GPU..."
|
||||||
|
if [[ -f "$GPU_CHECK_SCRIPT" ]]; then
|
||||||
|
if bash "$GPU_CHECK_SCRIPT" >/dev/null 2>&1; then
|
||||||
|
GPU_AVAILABLE=true
|
||||||
|
echo "[INFO] GPU 检测通过,将启动 gpu profile"
|
||||||
|
else
|
||||||
|
echo "[ERROR] 主机缺少可用 GPU,无法继续 --enable-gpu 流程" >&2
|
||||||
|
exit 1
|
||||||
|
fi
|
||||||
|
else
|
||||||
|
echo "[ERROR] 未找到 GPU 检测脚本: $GPU_CHECK_SCRIPT" >&2
|
||||||
|
exit 1
|
||||||
|
fi
|
||||||
|
else
|
||||||
|
echo "[INFO] 未启用 GPU 流程"
|
||||||
|
fi
|
||||||
|
|
||||||
pushd "$TEST_ROOT" >/dev/null
|
pushd "$TEST_ROOT" >/dev/null
|
||||||
compose -p argus-sys down --remove-orphans || true
|
compose -p argus-sys down --remove-orphans || true
|
||||||
compose -p argus-sys up -d
|
|
||||||
|
# 清理可能由 08 脚本创建的同名容器,避免 compose up 冲突
|
||||||
|
for name in argus-node-b; do
|
||||||
|
if docker ps -aqf "name=^${name}$" >/dev/null 2>&1 && [[ -n "$(docker ps -aqf "name=^${name}$")" ]]; then
|
||||||
|
docker rm -f "$name" >/dev/null 2>&1 || true
|
||||||
|
fi
|
||||||
|
done
|
||||||
|
|
||||||
|
# 预检:检查多端口网关所需宿主端口是否空闲
|
||||||
|
check_port_free() {
|
||||||
|
local p="$1"
|
||||||
|
if ss -ltnp 2>/dev/null | grep -q ":${p} "; then
|
||||||
|
echo "[ERR] Host port ${p} is already in use. Please free it before running 02_up.sh" >&2
|
||||||
|
ss -ltnp | awk -v p=":${p} " '$0 ~ p {print " " $0}' || true
|
||||||
|
return 1
|
||||||
|
fi
|
||||||
|
return 0
|
||||||
|
}
|
||||||
|
|
||||||
|
for port in 8080 8081 8082 8083 8084 8085; do
|
||||||
|
check_port_free "$port" || { echo "[ERR] Required port busy: $port"; exit 1; }
|
||||||
|
done
|
||||||
|
|
||||||
|
# 根据GPU可用性决定启动的服务
|
||||||
|
if [[ "$GPU_AVAILABLE" == true ]]; then
|
||||||
|
echo "[INFO] 启动所有服务(包含 gpu profile)..."
|
||||||
|
compose -p argus-sys --profile gpu up -d || true
|
||||||
|
else
|
||||||
|
echo "[INFO] 启动基础服务(不含 gpu profile)..."
|
||||||
|
compose -p argus-sys up -d || true
|
||||||
|
fi
|
||||||
|
|
||||||
|
# 若 web-proxy 处于 Created 状态,尝试单独启动一次(处理偶发 Address already in use 后端已释放的场景)
|
||||||
|
if docker ps -a --format '{{.Names}}\t{{.Status}}' | grep -q '^argus-web-proxy\s\+Created'; then
|
||||||
|
echo "[WARN] web-proxy in Created state; retry starting it..."
|
||||||
|
docker start argus-web-proxy || true
|
||||||
|
fi
|
||||||
|
|
||||||
popd >/dev/null
|
popd >/dev/null
|
||||||
|
|
||||||
echo "[OK] Services started: master:32300 es:9200 kibana:5601 node-a:2020 node-b:2021"
|
if [[ "$GPU_AVAILABLE" == true ]]; then
|
||||||
|
echo "[OK] Services started: master:32300 es:9200 kibana:5601 node-a:2020 node-b:2021 test-gpu-node:172.31.0.51"
|
||||||
|
else
|
||||||
|
echo "[OK] Services started: master:32300 es:9200 kibana:5601 node-a:2020 node-b:2021 (gpu skipped)"
|
||||||
|
fi
|
||||||
|
|||||||
@@ -29,6 +29,7 @@ echo "[INFO] Waiting for ES/Kibana/Master/Fluent Bit/Bind..."
|
|||||||
|
|
||||||
# ES (>= yellow)
|
# ES (>= yellow)
|
||||||
attempt=1; max=120
|
attempt=1; max=120
|
||||||
|
ES_T0=$(date +%s)
|
||||||
while (( attempt <= max )); do
|
while (( attempt <= max )); do
|
||||||
if curl -fsS "http://localhost:9200/_cluster/health?wait_for_status=yellow&timeout=1s" >/dev/null 2>&1; then
|
if curl -fsS "http://localhost:9200/_cluster/health?wait_for_status=yellow&timeout=1s" >/dev/null 2>&1; then
|
||||||
break
|
break
|
||||||
@@ -36,16 +37,19 @@ while (( attempt <= max )); do
|
|||||||
echo "[..] waiting ES ($attempt/$max)"; sleep 5; ((attempt++))
|
echo "[..] waiting ES ($attempt/$max)"; sleep 5; ((attempt++))
|
||||||
done
|
done
|
||||||
[[ $attempt -le $max ]] || { echo "[ERR] ES not ready" >&2; exit 1; }
|
[[ $attempt -le $max ]] || { echo "[ERR] ES not ready" >&2; exit 1; }
|
||||||
|
ES_T1=$(date +%s); echo "[TIME] ES ready in $((ES_T1-ES_T0))s"
|
||||||
|
|
||||||
# Kibana: must be HTTP 200 and overall.level=available
|
# Kibana: must be HTTP 200 and overall.level=available
|
||||||
echo "[INFO] Waiting for Kibana to be available (HTTP 200)..."
|
echo "[INFO] Waiting for Kibana to be available (HTTP 200)..."
|
||||||
kb_attempt=1; kb_max=180
|
kb_attempt=1; kb_max=180
|
||||||
|
KB_T0=$(date +%s)
|
||||||
while (( kb_attempt <= kb_max )); do
|
while (( kb_attempt <= kb_max )); do
|
||||||
body=$(curl -sS "http://localhost:5601/api/status" 2>/dev/null || true)
|
body=$(curl -sS "http://localhost:5601/api/status" 2>/dev/null || true)
|
||||||
code=$(curl -s -o /dev/null -w "%{http_code}" "http://localhost:5601/api/status" || echo 000)
|
code=$(curl -s -o /dev/null -w "%{http_code}" "http://localhost:5601/api/status" || echo 000)
|
||||||
if [[ "$code" == "200" ]]; then
|
if [[ "$code" == "200" ]]; then
|
||||||
if echo "$body" | grep -q '"level":"available"'; then
|
if echo "$body" | grep -q '"level":"available"'; then
|
||||||
echo "[OK] Kibana available (HTTP 200)"
|
KB_T1=$(date +%s)
|
||||||
|
echo "[OK] Kibana available (HTTP 200) in $((KB_T1-KB_T0))s"
|
||||||
break
|
break
|
||||||
fi
|
fi
|
||||||
fi
|
fi
|
||||||
@@ -58,11 +62,13 @@ if (( kb_attempt > kb_max )); then
|
|||||||
fi
|
fi
|
||||||
|
|
||||||
# Master
|
# Master
|
||||||
|
MASTER_T0=$(date +%s)
|
||||||
wait_http "http://localhost:32300/readyz" 120
|
wait_http "http://localhost:32300/readyz" 120
|
||||||
|
MASTER_T1=$(date +%s); echo "[TIME] Master readyz in $((MASTER_T1-MASTER_T0))s"
|
||||||
|
|
||||||
# Fluent Bit (host metrics on host ports)
|
# Fluent Bit (host metrics on host ports)
|
||||||
wait_http "http://localhost:2020/api/v2/metrics" 120
|
FB1_T0=$(date +%s); wait_http "http://localhost:2020/api/v2/metrics" 120; FB1_T1=$(date +%s); echo "[TIME] FluentBit:2020 in $((FB1_T1-FB1_T0))s"
|
||||||
wait_http "http://localhost:2021/api/v2/metrics" 120
|
FB2_T0=$(date +%s); wait_http "http://localhost:2021/api/v2/metrics" 120; FB2_T1=$(date +%s); echo "[TIME] FluentBit:2021 in $((FB2_T1-FB2_T0))s"
|
||||||
|
|
||||||
# Bind config check
|
# Bind config check
|
||||||
BIND_ID="$(service_id bind)"
|
BIND_ID="$(service_id bind)"
|
||||||
@@ -72,4 +78,63 @@ else
|
|||||||
echo "[WARN] bind container id not found"
|
echo "[WARN] bind container id not found"
|
||||||
fi
|
fi
|
||||||
|
|
||||||
|
# ========== Additional module readiness checks ==========
|
||||||
|
|
||||||
|
# Prometheus
|
||||||
|
PROM_T0=$(date +%s); wait_http "http://localhost:9090/-/ready" 120; PROM_T1=$(date +%s); echo "[TIME] Prometheus ready in $((PROM_T1-PROM_T0))s"
|
||||||
|
|
||||||
|
# Grafana health (database: ok)
|
||||||
|
echo "[INFO] Waiting for Grafana health..."
|
||||||
|
gf_attempt=1; gf_max=120
|
||||||
|
while (( gf_attempt <= gf_max )); do
|
||||||
|
gf_body=$(curl -sS "http://localhost:3000/api/health" 2>/dev/null || true)
|
||||||
|
gf_code=$(curl -s -o /dev/null -w "%{http_code}" "http://localhost:3000/api/health" || echo 000)
|
||||||
|
if [[ "$gf_code" == "200" ]] && echo "$gf_body" | grep -q '"database"\s*:\s*"ok"'; then
|
||||||
|
echo "[OK] Grafana health database=ok"
|
||||||
|
break
|
||||||
|
fi
|
||||||
|
echo "[..] waiting grafana health ($gf_attempt/$gf_max), last_code=$gf_code"
|
||||||
|
sleep 3; ((gf_attempt++))
|
||||||
|
done
|
||||||
|
if (( gf_attempt > gf_max )); then
|
||||||
|
echo "[ERR] Grafana /api/health not ready" >&2; exit 1
|
||||||
|
fi
|
||||||
|
|
||||||
|
# Alertmanager
|
||||||
|
wait_http "http://localhost:9093/api/v2/status" 120
|
||||||
|
|
||||||
|
# Web proxy checks(按端口细化)
|
||||||
|
code_for() { curl -s -o /dev/null -w "%{http_code}" "$1" || echo 000; }
|
||||||
|
header_val() { curl -s -D - -o /dev/null "$@" | awk -F': ' 'BEGIN{IGNORECASE=1}$1=="Access-Control-Allow-Origin"{gsub("\r","",$2);print $2}'; }
|
||||||
|
|
||||||
|
echo "[INFO] Checking web-proxy ports..."
|
||||||
|
|
||||||
|
# 8080 首页必须 200
|
||||||
|
tries=1; max=60; P8080_T0=$(date +%s)
|
||||||
|
while (( tries <= max )); do
|
||||||
|
c=$(code_for "http://localhost:8080/")
|
||||||
|
if [[ "$c" == "200" ]]; then P8080_T1=$(date +%s); echo "[OK] 8080 / ($c) in $((P8080_T1-P8080_T0))s"; break; fi
|
||||||
|
echo "[..] waiting 8080/ ($tries/$max), code=$c"; sleep 3; ((tries++))
|
||||||
|
done
|
||||||
|
(( tries <= max )) || { echo "[ERR] 8080/ not ready" >&2; exit 1; }
|
||||||
|
|
||||||
|
# 8083 Kibana 允许 200/302(上面已就绪,端口侧再快速确认)
|
||||||
|
tries=1; max=40; P8083_T0=$(date +%s)
|
||||||
|
while (( tries <= max )); do
|
||||||
|
c=$(code_for "http://localhost:8083/")
|
||||||
|
if [[ "$c" == "200" || "$c" == "302" ]]; then P8083_T1=$(date +%s); echo "[OK] 8083 / ($c) in $((P8083_T1-P8083_T0))s"; break; fi
|
||||||
|
echo "[..] waiting 8083/ ($tries/$max), code=$c"; sleep 3; ((tries++))
|
||||||
|
done
|
||||||
|
(( tries <= max )) || { echo "[ERR] 8083/ not ready" >&2; exit 1; }
|
||||||
|
|
||||||
|
# 8084 Alertmanager + CORS
|
||||||
|
P8084_T0=$(date +%s); wait_http "http://localhost:8084/api/v2/status" 60; P8084_T1=$(date +%s)
|
||||||
|
cors=$(header_val -H "Origin: http://localhost:8080" "http://localhost:8084/api/v2/status" || true)
|
||||||
|
if [[ -z "$cors" ]]; then echo "[ERR] 8084 CORS missing" >&2; exit 1; else echo "[OK] 8084 CORS: $cors in $((P8084_T1-P8084_T0))s"; fi
|
||||||
|
|
||||||
|
# 8085 Master /readyz + CORS(API 走 8085 才需跨域)
|
||||||
|
P8085_T0=$(date +%s); wait_http "http://localhost:8085/readyz" 60; P8085_T1=$(date +%s)
|
||||||
|
cors=$(header_val -H "Origin: http://localhost:8080" "http://localhost:8085/api/v1/master/nodes" || true)
|
||||||
|
if [[ -z "$cors" ]]; then echo "[ERR] 8085 CORS missing" >&2; exit 1; else echo "[OK] 8085 CORS: $cors in $((P8085_T1-P8085_T0))s"; fi
|
||||||
|
|
||||||
echo "[OK] All services are ready"
|
echo "[OK] All services are ready"
|
||||||
|
|||||||
@@ -4,20 +4,15 @@ set -euo pipefail
|
|||||||
SCRIPT_DIR="$(cd "$(dirname "${BASH_SOURCE[0]}")" && pwd)"
|
SCRIPT_DIR="$(cd "$(dirname "${BASH_SOURCE[0]}")" && pwd)"
|
||||||
TEST_ROOT="$(cd "$SCRIPT_DIR/.." && pwd)"
|
TEST_ROOT="$(cd "$SCRIPT_DIR/.." && pwd)"
|
||||||
|
|
||||||
compose() {
|
# 直接根据 container_name 获取容器ID,避免 compose project 名称不一致导致查找失败
|
||||||
if docker compose version >/dev/null 2>&1; then
|
cid_by_name() {
|
||||||
docker compose "$@"
|
docker ps -aqf "name=^$1$"
|
||||||
else
|
|
||||||
docker-compose "$@"
|
|
||||||
fi
|
|
||||||
}
|
|
||||||
|
|
||||||
service_id() {
|
|
||||||
compose -p argus-sys ps -q "$1"
|
|
||||||
}
|
}
|
||||||
|
|
||||||
echo "[INFO] Verifying DNS routing via bind..."
|
echo "[INFO] Verifying DNS routing via bind..."
|
||||||
|
|
||||||
|
pushd "$TEST_ROOT" >/dev/null
|
||||||
|
|
||||||
# Check master IP file exists in shared private
|
# Check master IP file exists in shared private
|
||||||
MASTER_FILE="$TEST_ROOT/private/argus/etc/master.argus.com"
|
MASTER_FILE="$TEST_ROOT/private/argus/etc/master.argus.com"
|
||||||
if [[ ! -f "$MASTER_FILE" ]]; then
|
if [[ ! -f "$MASTER_FILE" ]]; then
|
||||||
@@ -28,7 +23,7 @@ MASTER_IP_HOST="$(cat "$MASTER_FILE" | tr -d '\r\n' || true)"
|
|||||||
echo "[INFO] master.argus.com file content: ${MASTER_IP_HOST}"
|
echo "[INFO] master.argus.com file content: ${MASTER_IP_HOST}"
|
||||||
|
|
||||||
# dig inside bind container
|
# dig inside bind container
|
||||||
BIN_ID="$(service_id bind)"
|
BIN_ID="$(cid_by_name argus-bind-sys)"
|
||||||
if [[ -n "$BIN_ID" ]]; then
|
if [[ -n "$BIN_ID" ]]; then
|
||||||
DIG_IP="$(docker exec "$BIN_ID" dig +short master.argus.com A | tail -n1 || true)"
|
DIG_IP="$(docker exec "$BIN_ID" dig +short master.argus.com A | tail -n1 || true)"
|
||||||
echo "[INFO] dig(master.argus.com) from bind container -> $DIG_IP"
|
echo "[INFO] dig(master.argus.com) from bind container -> $DIG_IP"
|
||||||
@@ -39,8 +34,8 @@ else
|
|||||||
echo "[WARN] bind container not found; skip dig"
|
echo "[WARN] bind container not found; skip dig"
|
||||||
fi
|
fi
|
||||||
|
|
||||||
for node in node-a node-b; do
|
for node in argus-node-a argus-node-b; do
|
||||||
CID="$(service_id "$node")"
|
CID="$(cid_by_name "$node")"
|
||||||
echo "[INFO] Checking resolution inside $node..."
|
echo "[INFO] Checking resolution inside $node..."
|
||||||
if ! docker exec "$CID" getent hosts master.argus.com >/dev/null 2>&1; then
|
if ! docker exec "$CID" getent hosts master.argus.com >/dev/null 2>&1; then
|
||||||
echo "[ERR] $node cannot resolve master.argus.com" >&2
|
echo "[ERR] $node cannot resolve master.argus.com" >&2
|
||||||
@@ -50,5 +45,6 @@ for node in node-a node-b; do
|
|||||||
echo "[OK] $node resolved master.argus.com -> $RES"
|
echo "[OK] $node resolved master.argus.com -> $RES"
|
||||||
done
|
done
|
||||||
|
|
||||||
echo "[OK] DNS routing verified"
|
popd >/dev/null
|
||||||
|
|
||||||
|
echo "[OK] DNS routing verified"
|
||||||
|
|||||||
@@ -49,8 +49,35 @@ for _ in {1..60}; do
|
|||||||
fi
|
fi
|
||||||
done
|
done
|
||||||
|
|
||||||
|
# 若仍未全部注册,尝试重启 node-b 并再等待一轮(兼容 DNS/启动时序抖动)
|
||||||
if [[ ! -s "$TMP_DIR/node_id_a" || ! -s "$TMP_DIR/node_id_b" ]]; then
|
if [[ ! -s "$TMP_DIR/node_id_a" || ! -s "$TMP_DIR/node_id_b" ]]; then
|
||||||
echo "[ERR] Agents did not register in time" >&2
|
echo "[WARN] node-a or node-b not registered in first window; restarting node-b and retrying..." >&2
|
||||||
|
# 仅重启 node-b,避免影响 es/kibana/master
|
||||||
|
if docker ps --format '{{.Names}}' | grep -q '^argus-node-b$'; then
|
||||||
|
docker restart argus-node-b >/dev/null 2>&1 || true
|
||||||
|
fi
|
||||||
|
# 再等待一轮(最多 120 秒)
|
||||||
|
> "$TMP_DIR/node_id_b"
|
||||||
|
for _ in {1..60}; do
|
||||||
|
sleep 2
|
||||||
|
resp=$(curl -fsS "$API_BASE/nodes" 2>/dev/null || true)
|
||||||
|
[[ -z "$resp" ]] && continue
|
||||||
|
if ! echo "$resp" | head -c1 | grep -q '\['; then
|
||||||
|
continue
|
||||||
|
fi
|
||||||
|
echo "$resp" > "$TMP_DIR/nodes_list.json"
|
||||||
|
ID_A=$(extract_node "$HOST_A" "$TMP_DIR/node_id_a" "$TMP_DIR/nodes_list.json" 2>/dev/null || true)
|
||||||
|
ID_B=$(extract_node "$HOST_B" "$TMP_DIR/node_id_b" "$TMP_DIR/nodes_list.json" 2>/dev/null || true)
|
||||||
|
if [[ -s "$TMP_DIR/node_id_a" && -s "$TMP_DIR/node_id_b" ]]; then
|
||||||
|
break
|
||||||
|
fi
|
||||||
|
done
|
||||||
|
fi
|
||||||
|
|
||||||
|
if [[ ! -s "$TMP_DIR/node_id_a" || ! -s "$TMP_DIR/node_id_b" ]]; then
|
||||||
|
echo "[ERR] Agents did not register in time (after retry)" >&2
|
||||||
|
echo "[HINT] Current /nodes response:" >&2
|
||||||
|
sed -n '1,200p' "$TMP_DIR/nodes_list.json" >&2 || true
|
||||||
exit 1
|
exit 1
|
||||||
fi
|
fi
|
||||||
|
|
||||||
|
|||||||
@@ -3,9 +3,19 @@ set -euo pipefail
|
|||||||
|
|
||||||
echo "[INFO] Sending logs via node-a/node-b and asserting ES counts..."
|
echo "[INFO] Sending logs via node-a/node-b and asserting ES counts..."
|
||||||
|
|
||||||
|
# Robust count helper: tolerates 404/503 and non-JSON responses, returns integer >=0
|
||||||
get_count() {
|
get_count() {
|
||||||
local idx="$1"
|
local idx="$1"; local tmp; tmp=$(mktemp)
|
||||||
curl -s "http://localhost:9200/${idx}/_count?ignore_unavailable=true&allow_no_indices=true" | sed -E 's/.*"count":([0-9]+).*/\1/' | awk 'NF{print $0;exit} END{if(NR==0)print 0}'
|
local code
|
||||||
|
code=$(curl -s -o "$tmp" -w "%{http_code}" "http://localhost:9200/${idx}/_count?ignore_unavailable=true&allow_no_indices=true" || true)
|
||||||
|
if [[ "$code" == "200" ]]; then
|
||||||
|
local val
|
||||||
|
val=$(jq -r '(.count // 0) | tonumber? // 0' "$tmp" 2>/dev/null || echo 0)
|
||||||
|
echo "$val"
|
||||||
|
else
|
||||||
|
echo 0
|
||||||
|
fi
|
||||||
|
rm -f "$tmp"
|
||||||
}
|
}
|
||||||
|
|
||||||
train0=$(get_count "train-*")
|
train0=$(get_count "train-*")
|
||||||
@@ -32,11 +42,26 @@ send_logs "$node_a" "host01"
|
|||||||
send_logs "$node_b" "host02"
|
send_logs "$node_b" "host02"
|
||||||
|
|
||||||
echo "[INFO] Waiting for ES to ingest..."
|
echo "[INFO] Waiting for ES to ingest..."
|
||||||
sleep 10
|
# Proactively refresh indices (ignore errors if not created yet)
|
||||||
|
curl -s -X POST "http://localhost:9200/train-*/_refresh" >/dev/null 2>&1 || true
|
||||||
|
curl -s -X POST "http://localhost:9200/infer-*/_refresh" >/dev/null 2>&1 || true
|
||||||
|
|
||||||
train1=$(get_count "train-*")
|
# Retry up to 120s for counts to increase and reach threshold (>=4)
|
||||||
infer1=$(get_count "infer-*")
|
final=0
|
||||||
final=$((train1 + infer1))
|
threshold=4
|
||||||
|
for attempt in {1..60}; do
|
||||||
|
train1=$(get_count "train-*")
|
||||||
|
infer1=$(get_count "infer-*")
|
||||||
|
final=$((train1 + infer1))
|
||||||
|
if (( final > base && final >= threshold )); then
|
||||||
|
break
|
||||||
|
fi
|
||||||
|
echo "[..] waiting ES counts increase to >=${threshold} ($attempt/60) current=${final} base=${base}"
|
||||||
|
# refresh indices again to speed up visibility
|
||||||
|
curl -s -X POST "http://localhost:9200/train-*/_refresh" >/dev/null 2>&1 || true
|
||||||
|
curl -s -X POST "http://localhost:9200/infer-*/_refresh" >/dev/null 2>&1 || true
|
||||||
|
sleep 2
|
||||||
|
done
|
||||||
echo "[INFO] final counts: train=${train1} infer=${infer1} total=${final}"
|
echo "[INFO] final counts: train=${train1} infer=${infer1} total=${final}"
|
||||||
|
|
||||||
if (( final <= base )); then
|
if (( final <= base )); then
|
||||||
@@ -44,6 +69,7 @@ if (( final <= base )); then
|
|||||||
exit 1
|
exit 1
|
||||||
fi
|
fi
|
||||||
|
|
||||||
|
# Minimal threshold to be tolerant: expect at least 4 documents (2 train + 1 infer per node)
|
||||||
if (( final < 4 )); then
|
if (( final < 4 )); then
|
||||||
echo "[ERR] ES total below expected threshold: ${final} < 4" >&2
|
echo "[ERR] ES total below expected threshold: ${final} < 4" >&2
|
||||||
exit 1
|
exit 1
|
||||||
|
|||||||
@@ -8,6 +8,16 @@ REPO_ROOT="$(cd "$TEST_ROOT/../../.." && pwd)"
|
|||||||
|
|
||||||
API_BASE="http://localhost:32300/api/v1/master"
|
API_BASE="http://localhost:32300/api/v1/master"
|
||||||
|
|
||||||
|
if [[ -f "$TEST_ROOT/.env" ]]; then
|
||||||
|
set -a
|
||||||
|
# shellcheck disable=SC1090
|
||||||
|
source "$TEST_ROOT/.env"
|
||||||
|
set +a
|
||||||
|
else
|
||||||
|
source "$REPO_ROOT/scripts/common/build_user.sh"
|
||||||
|
load_build_user
|
||||||
|
fi
|
||||||
|
|
||||||
ID_B="$(cat "$TMP_DIR/node_id_b")"
|
ID_B="$(cat "$TMP_DIR/node_id_b")"
|
||||||
IP0_B="$(cat "$TMP_DIR/initial_ip_b")"
|
IP0_B="$(cat "$TMP_DIR/initial_ip_b")"
|
||||||
|
|
||||||
@@ -39,7 +49,7 @@ compose() {
|
|||||||
fi
|
fi
|
||||||
}
|
}
|
||||||
|
|
||||||
echo "[INFO] Recreating node-b with static IP 172.29.0.200..."
|
echo "[INFO] Recreating node-b with static IP 172.31.0.200..."
|
||||||
pushd "$TEST_ROOT" >/dev/null
|
pushd "$TEST_ROOT" >/dev/null
|
||||||
compose -p argus-sys rm -sf node-b || true
|
compose -p argus-sys rm -sf node-b || true
|
||||||
popd >/dev/null
|
popd >/dev/null
|
||||||
@@ -48,12 +58,27 @@ docker rm -f argus-node-b >/dev/null 2>&1 || true
|
|||||||
|
|
||||||
AGENT_BIN_PATH="$(cat "$TMP_DIR/agent_binary_path")"
|
AGENT_BIN_PATH="$(cat "$TMP_DIR/agent_binary_path")"
|
||||||
|
|
||||||
|
# 选择 compose 管理的网络名(默认 argus-sys_sysnet)。
|
||||||
|
detect_sysnet() {
|
||||||
|
if docker network inspect argus-sys_sysnet >/dev/null 2>&1; then
|
||||||
|
echo argus-sys_sysnet; return
|
||||||
|
fi
|
||||||
|
# 回退:从 master 容器推断所连网络(取第一个)
|
||||||
|
local n
|
||||||
|
n=$(docker inspect -f '{{range $k, $_ := .NetworkSettings.Networks}}{{println $k}}{{end}}' argus-master-sys 2>/dev/null | head -n1 || true)
|
||||||
|
if [[ -n "$n" ]]; then echo "$n"; return; fi
|
||||||
|
# 最后兜底:尝试项目默认网络(不保证有 IPAM)
|
||||||
|
echo argus-sys_default
|
||||||
|
}
|
||||||
|
SYSNET_NAME=$(detect_sysnet)
|
||||||
|
echo "[INFO] Using docker network: $SYSNET_NAME"
|
||||||
|
|
||||||
docker run -d \
|
docker run -d \
|
||||||
--name argus-node-b \
|
--name argus-node-b \
|
||||||
--hostname dev-yyrshare-uuuu10-ep2f-pod-0 \
|
--hostname dev-yyrshare-uuuu10-ep2f-pod-0 \
|
||||||
--network argus-sys-net \
|
--network "$SYSNET_NAME" \
|
||||||
--ip 172.29.0.200 \
|
--ip 172.31.0.200 \
|
||||||
--dns 172.29.0.2 \
|
--dns 172.31.0.2 \
|
||||||
-e MASTER_ENDPOINT=http://master.argus.com:3000 \
|
-e MASTER_ENDPOINT=http://master.argus.com:3000 \
|
||||||
-e REPORT_INTERVAL_SECONDS=2 \
|
-e REPORT_INTERVAL_SECONDS=2 \
|
||||||
-e ARGUS_BUILD_UID=${ARGUS_BUILD_UID:-2133} \
|
-e ARGUS_BUILD_UID=${ARGUS_BUILD_UID:-2133} \
|
||||||
@@ -80,15 +105,15 @@ node=json.load(open(sys.argv[1]))
|
|||||||
last0=sys.argv[2]
|
last0=sys.argv[2]
|
||||||
ip=node.get("meta_data",{}).get("ip")
|
ip=node.get("meta_data",{}).get("ip")
|
||||||
lu=node.get("last_updated")
|
lu=node.get("last_updated")
|
||||||
assert ip=="172.29.0.200"
|
assert ip=="172.31.0.200"
|
||||||
assert lu and lu!=last0
|
assert lu and lu!=last0
|
||||||
PY
|
PY
|
||||||
then
|
then
|
||||||
echo "[OK] node-b re-registered with new IP 172.29.0.200"
|
echo "[OK] node-b re-registered with new IP 172.31.0.200"
|
||||||
exit 0
|
exit 0
|
||||||
fi
|
fi
|
||||||
fi
|
fi
|
||||||
done
|
done
|
||||||
|
|
||||||
echo "[ERR] node-b did not update to IP 172.29.0.200 in time" >&2
|
echo "[ERR] node-b did not update to IP 172.31.0.200 in time" >&2
|
||||||
exit 1
|
exit 1
|
||||||
|
|||||||
Some files were not shown because too many files have changed in this diff Show More
Reference in New Issue
Block a user