93 lines
2.6 KiB
Bash
93 lines
2.6 KiB
Bash
|
#!/bin/bash
|
||
|
|
# Methods library for loki — sourced by install-loki.sh and generated run-*.sh scripts.
|
||
|
|
# SCRIPT_DIR must be set before sourcing.
|
||
|
|
|
||
|
|
_kubectl() {
|
||
|
|
if command -v kubectl >/dev/null 2>&1; then
|
||
|
|
command kubectl "$@"
|
||
|
|
elif command -v k0s >/dev/null 2>&1; then
|
||
|
|
k0s kubectl "$@"
|
||
|
|
else
|
||
|
|
echo "ERROR: no kubectl or k0s binary found" >&2; exit 127
|
||
|
|
fi
|
||
|
|
}
|
||
|
|
|
||
|
|
_require_env() {
|
||
|
|
local var="$1"
|
||
|
|
if [ -z "${!var:-}" ]; then
|
||
|
|
echo "ERROR: required variable $var is not set" >&2; exit 1
|
||
|
|
fi
|
||
|
|
}
|
||
|
|
|
||
|
|
_pod_name() {
|
||
|
|
_kubectl get pod \
|
||
|
|
--namespace "$LOKI_NAMESPACE" \
|
||
|
|
--selector "app=loki" \
|
||
|
|
-o jsonpath='{.items[0].metadata.name}' 2>/dev/null
|
||
|
|
}
|
||
|
|
|
||
|
|
_wait_for_pod() {
|
||
|
|
local timeout="${1:-180}"
|
||
|
|
_kubectl wait pod \
|
||
|
|
--namespace "$LOKI_NAMESPACE" \
|
||
|
|
--selector "app=loki" \
|
||
|
|
--for=condition=Ready \
|
||
|
|
--timeout="${timeout}s"
|
||
|
|
}
|
||
|
|
|
||
|
|
_plan_apply() {
|
||
|
|
local file="$1"
|
||
|
|
local path="$SCRIPT_DIR/manifests/${file}.yaml"
|
||
|
|
[ -f "$path" ] || { echo " [skip] ${file}.yaml not in manifests/"; return 0; }
|
||
|
|
_kubectl apply -f "$path"
|
||
|
|
}
|
||
|
|
|
||
|
|
_plan_apply_skip() {
|
||
|
|
local file="$1"
|
||
|
|
local path="$SCRIPT_DIR/manifests/${file}.yaml"
|
||
|
|
[ -f "$path" ] || { echo " [skip] ${file}.yaml not in manifests/"; return 0; }
|
||
|
|
if _kubectl get -f "$path" >/dev/null 2>&1; then
|
||
|
|
echo " [skip] ${file} already exists in cluster"
|
||
|
|
return 0
|
||
|
|
fi
|
||
|
|
_kubectl apply -f "$path"
|
||
|
|
}
|
||
|
|
|
||
|
|
_plan_rollout_restart() {
|
||
|
|
local file="$1"
|
||
|
|
local path="$SCRIPT_DIR/manifests/${file}.yaml"
|
||
|
|
[ -f "$path" ] || { echo " [skip] ${file}.yaml not in manifests/"; return 0; }
|
||
|
|
_kubectl rollout restart -f "$path"
|
||
|
|
}
|
||
|
|
|
||
|
|
_plan_delete() {
|
||
|
|
local file="$1"
|
||
|
|
local path="$SCRIPT_DIR/manifests/${file}.yaml"
|
||
|
|
[ -f "$path" ] || { echo " [skip] ${file}.yaml not in manifests/"; return 0; }
|
||
|
|
_kubectl delete -f "$path" --ignore-not-found
|
||
|
|
}
|
||
|
|
|
||
|
|
_method_wait-ready() {
|
||
|
|
_wait_for_pod 180
|
||
|
|
}
|
||
|
|
|
||
|
|
_method_create-credentials() {
|
||
|
|
[ -f "${SCRIPT_DIR}/_credentials.env" ] && set -a && source "${SCRIPT_DIR}/_credentials.env" && set +a
|
||
|
|
[ -f "${SCRIPT_DIR}/env-loki" ] && set -a && source "${SCRIPT_DIR}/env-loki" && set +a
|
||
|
|
|
||
|
|
if [ "${LOKI_STORAGE_BACKEND:-filesystem}" != "s3" ]; then
|
||
|
|
echo " [skip] storage_backend=${LOKI_STORAGE_BACKEND:-filesystem} — no S3 credentials secret needed"
|
||
|
|
return 0
|
||
|
|
fi
|
||
|
|
|
||
|
|
_require_env LOKI_S3_ACCESS_KEY
|
||
|
|
_require_env LOKI_S3_SECRET_KEY
|
||
|
|
|
||
|
|
_kubectl create secret generic loki-s3-credentials \
|
||
|
|
--namespace "$LOKI_NAMESPACE" \
|
||
|
|
--from-literal=LOKI_S3_ACCESS_KEY="$LOKI_S3_ACCESS_KEY" \
|
||
|
|
--from-literal=LOKI_S3_SECRET_KEY="$LOKI_S3_SECRET_KEY" \
|
||
|
|
--dry-run=client -o yaml | _kubectl apply -f -
|
||
|
|
echo " [ok] credentials secret 'loki-s3-credentials' created in ${LOKI_NAMESPACE}"
|
||
|
|
}
|