92 lines
3.2 KiB
Bash
92 lines
3.2 KiB
Bash
|
#!/bin/bash
|
||
|
|
# Tier-1 dispatch for vector component operations.
|
||
|
|
set -euo pipefail
|
||
|
|
|
||
|
|
SCRIPT_DIR="$(cd "$(dirname "${BASH_SOURCE[0]}")" && pwd)"
|
||
|
|
cd "$SCRIPT_DIR"
|
||
|
|
|
||
|
|
[ -f "${SCRIPT_DIR}/env-vector" ] && source "${SCRIPT_DIR}/env-vector"
|
||
|
|
|
||
|
|
VECTOR_NAME="${VECTOR_NAME:-vector}"
|
||
|
|
VECTOR_NAMESPACE="${VECTOR_NAMESPACE:-observability}"
|
||
|
|
VECTOR_PORT="${VECTOR_PORT:-8686}"
|
||
|
|
|
||
|
|
CMD_TSK="${CMD_TSK:-${1:-install}}"
|
||
|
|
|
||
|
|
source "${SCRIPT_DIR}/vector-lib.sh"
|
||
|
|
|
||
|
|
_resolve_kubeconfig() {
|
||
|
|
if [ -n "${KUBECONFIG:-}" ] && [ -f "$KUBECONFIG" ]; then
|
||
|
|
return
|
||
|
|
fi
|
||
|
|
for candidate in /var/lib/k0s/pki/admin.conf /etc/kubernetes/admin.conf /root/.kube/config; do
|
||
|
|
if [ -f "$candidate" ]; then
|
||
|
|
export KUBECONFIG="$candidate"
|
||
|
|
return
|
||
|
|
fi
|
||
|
|
done
|
||
|
|
echo "ERROR: kubeconfig not found — set KUBECONFIG explicitly" >&2; exit 1
|
||
|
|
}
|
||
|
|
|
||
|
|
_do_install() {
|
||
|
|
for manifest in namespace.yaml serviceaccount.yaml clusterrole.yaml clusterrolebinding.yaml \
|
||
|
|
configmap.yaml daemonset.yaml service.yaml; do
|
||
|
|
local path="$SCRIPT_DIR/manifests/${manifest}"
|
||
|
|
[ -f "$path" ] && _kubectl apply -f "$path"
|
||
|
|
done
|
||
|
|
echo "Waiting for ${VECTOR_NAME} DaemonSet to be ready..."
|
||
|
|
_wait_for_pod 180
|
||
|
|
echo "${VECTOR_NAME} installed in namespace ${VECTOR_NAMESPACE}"
|
||
|
|
}
|
||
|
|
|
||
|
|
_do_update() {
|
||
|
|
for manifest in configmap.yaml daemonset.yaml; do
|
||
|
|
local path="$SCRIPT_DIR/manifests/${manifest}"
|
||
|
|
[ -f "$path" ] && _kubectl apply -f "$path"
|
||
|
|
done
|
||
|
|
_kubectl rollout restart "daemonset/${VECTOR_NAME}" --namespace "$VECTOR_NAMESPACE"
|
||
|
|
_wait_for_pod 180
|
||
|
|
echo "${VECTOR_NAME} updated"
|
||
|
|
}
|
||
|
|
|
||
|
|
_do_delete() {
|
||
|
|
for manifest in daemonset.yaml service.yaml clusterrolebinding.yaml clusterrole.yaml serviceaccount.yaml; do
|
||
|
|
local path="$SCRIPT_DIR/manifests/${manifest}"
|
||
|
|
[ -f "$path" ] && _kubectl delete -f "$path" --ignore-not-found
|
||
|
|
done
|
||
|
|
echo "${VECTOR_NAME} deleted from namespace ${VECTOR_NAMESPACE}"
|
||
|
|
}
|
||
|
|
|
||
|
|
_do_restart() {
|
||
|
|
_kubectl rollout restart "daemonset/${VECTOR_NAME}" --namespace "$VECTOR_NAMESPACE"
|
||
|
|
_wait_for_pod 180
|
||
|
|
echo "${VECTOR_NAME} restarted"
|
||
|
|
}
|
||
|
|
|
||
|
|
_do_health() {
|
||
|
|
local ready
|
||
|
|
ready=$(_kubectl get daemonset "${VECTOR_NAME}" \
|
||
|
|
--namespace "$VECTOR_NAMESPACE" \
|
||
|
|
-o jsonpath='{.status.numberReady}' 2>/dev/null || echo "0")
|
||
|
|
local desired
|
||
|
|
desired=$(_kubectl get daemonset "${VECTOR_NAME}" \
|
||
|
|
--namespace "$VECTOR_NAMESPACE" \
|
||
|
|
-o jsonpath='{.status.desiredNumberScheduled}' 2>/dev/null || echo "-1")
|
||
|
|
if [ "$ready" -eq "$desired" ] && [ "$desired" -gt 0 ]; then
|
||
|
|
echo "HEALTHY: ${VECTOR_NAME} ${ready}/${desired} pods ready"
|
||
|
|
else
|
||
|
|
echo "UNHEALTHY: ${VECTOR_NAME} ${ready}/${desired} pods ready" >&2; exit 1
|
||
|
|
fi
|
||
|
|
}
|
||
|
|
|
||
|
|
_resolve_kubeconfig
|
||
|
|
|
||
|
|
case "$CMD_TSK" in
|
||
|
|
install) [ -f "$SCRIPT_DIR/run-init.sh" ] && exec bash "$SCRIPT_DIR/run-init.sh"; _do_install ;;
|
||
|
|
update) [ -f "$SCRIPT_DIR/run-update.sh" ] && exec bash "$SCRIPT_DIR/run-update.sh"; _do_update ;;
|
||
|
|
delete) [ -f "$SCRIPT_DIR/run-delete.sh" ] && exec bash "$SCRIPT_DIR/run-delete.sh"; _do_delete ;;
|
||
|
|
restart) [ -f "$SCRIPT_DIR/run-restart.sh" ] && exec bash "$SCRIPT_DIR/run-restart.sh"; _do_restart ;;
|
||
|
|
health) _do_health ;;
|
||
|
|
*) echo "ERROR: unknown CMD_TSK='${CMD_TSK}'. Valid: install|update|delete|restart|health" >&2; exit 1 ;;
|
||
|
|
esac
|