fix(bao-secrets): retry login+fetch with backoff on 429/504/non-JSON (#545/#590-adjacent)
Under a CI burst (a batch merge fires ~13 workflows x ~4 jobs, each doing an AppRole login = a raft write), the vault nodes return nginx 504s and the rate-limit quota returns 429s. Both are non-JSON, which crashed jq and hard-failed the job with a misleading "parse error". Add a bao_req helper that retries connect-failures/429/5xx with backoff+jitter (~60s over 6 tries) and surfaces real 4xx bodies unchanged. Transient throttling no longer fails CI. Co-Authored-By: Claude Opus 4.8 (1M context) <noreply@anthropic.com>
This commit is contained in:
co-authored by
Claude Opus 4.8
parent
db71896903
commit
a67d0cf3f3
+39
-12
@@ -35,14 +35,42 @@ runs:
|
|||||||
command -v curl >/dev/null || { echo "::error::bao-secrets needs 'curl' on the runner"; exit 1; }
|
command -v curl >/dev/null || { echo "::error::bao-secrets needs 'curl' on the runner"; exit 1; }
|
||||||
command -v jq >/dev/null || { echo "::error::bao-secrets needs 'jq' on the runner"; exit 1; }
|
command -v jq >/dev/null || { echo "::error::bao-secrets needs 'jq' on the runner"; exit 1; }
|
||||||
|
|
||||||
# 1) AppRole login -> short-lived (15m) token. No -f so we can read the
|
TOKEN=""
|
||||||
# error body; the token check below is the real gate.
|
# HTTP with retry + backoff. When many CI jobs log in at once the vault
|
||||||
login=$(curl -s --max-time 15 -X POST \
|
# nodes can return an nginx 504 (login is a raft write) or a 429 from the
|
||||||
--data "{\"role_id\":\"${BAO_ROLE_ID}\",\"secret_id\":\"${BAO_SECRET_ID}\"}" \
|
# rate-limit quota — both are transient and non-JSON, which used to crash
|
||||||
"${BAO_ADDR}/v1/auth/approle/login") || true
|
# jq and hard-fail the job. Retry those; surface real 4xx bodies as-is.
|
||||||
TOKEN=$(printf '%s' "$login" | jq -r '.auth.client_token // empty')
|
# Prints the response BODY on stdout.
|
||||||
|
bao_req() { # METHOD URL [DATA]
|
||||||
|
local method="$1" url="$2" data="${3:-}" attempt=1 max=6 code tmp secs
|
||||||
|
tmp="$(mktemp)"
|
||||||
|
local -a args=(-s -o "$tmp" -w '%{http_code}' --max-time 15 -X "$method")
|
||||||
|
[ -n "${TOKEN:-}" ] && args+=(-H "X-Vault-Token: ${TOKEN}")
|
||||||
|
[ -n "$data" ] && args+=(--data "$data")
|
||||||
|
args+=("$url")
|
||||||
|
while :; do
|
||||||
|
code=$(curl "${args[@]}" 2>/dev/null || echo 000)
|
||||||
|
case "$code" in
|
||||||
|
200|204) cat "$tmp"; rm -f "$tmp"; return 0 ;;
|
||||||
|
000|429|500|502|503|504)
|
||||||
|
if [ "$attempt" -ge "$max" ]; then
|
||||||
|
echo "::warning::bao $method failed after ${attempt} tries (last HTTP ${code})" >&2
|
||||||
|
cat "$tmp"; rm -f "$tmp"; return 1
|
||||||
|
fi
|
||||||
|
secs=$(( attempt * 4 + RANDOM % 4 )) # ~4,8,12,16,20s + jitter
|
||||||
|
echo " bao ${method} -> HTTP ${code}; retry ${attempt}/${max} in ${secs}s (cluster busy/rate-limited)" >&2
|
||||||
|
sleep "$secs"; attempt=$((attempt+1)) ;;
|
||||||
|
*) cat "$tmp"; rm -f "$tmp"; return 1 ;; # 4xx etc — surface to caller
|
||||||
|
esac
|
||||||
|
done
|
||||||
|
}
|
||||||
|
|
||||||
|
# 1) AppRole login -> short-lived token (batch token if the role is set so).
|
||||||
|
login="$(bao_req POST "${BAO_ADDR}/v1/auth/approle/login" \
|
||||||
|
"{\"role_id\":\"${BAO_ROLE_ID}\",\"secret_id\":\"${BAO_SECRET_ID}\"}")" || true
|
||||||
|
TOKEN=$(printf '%s' "$login" | jq -r '.auth.client_token // empty' 2>/dev/null || true)
|
||||||
if [ -z "$TOKEN" ]; then
|
if [ -z "$TOKEN" ]; then
|
||||||
echo "::error::OpenBao AppRole login failed: $(printf '%s' "$login" | jq -rc '.errors // "no response / network error"')"
|
echo "::error::OpenBao AppRole login failed (after retries): $(printf '%s' "$login" | jq -rc '.errors // "non-JSON/timeout — cluster busy or rate-limited"' 2>/dev/null || echo 'non-JSON response')"
|
||||||
exit 1
|
exit 1
|
||||||
fi
|
fi
|
||||||
echo "::add-mask::$TOKEN"
|
echo "::add-mask::$TOKEN"
|
||||||
@@ -59,11 +87,10 @@ runs:
|
|||||||
fi
|
fi
|
||||||
env_name="$1"; path="$2"; field="$3"
|
env_name="$1"; path="$2"; field="$3"
|
||||||
mount="${path%%/*}"; rest="${path#*/}" # KV v2: <mount>/data/<rest>
|
mount="${path%%/*}"; rest="${path#*/}" # KV v2: <mount>/data/<rest>
|
||||||
resp=$(curl -s --max-time 15 -H "X-Vault-Token: $TOKEN" \
|
resp="$(bao_req GET "${BAO_ADDR}/v1/${mount}/data/${rest}")" || true
|
||||||
"${BAO_ADDR}/v1/${mount}/data/${rest}") || true
|
val=$(printf '%s' "$resp" | jq -r --arg f "$field" '.data.data[$f] // empty' 2>/dev/null || true)
|
||||||
val=$(printf '%s' "$resp" | jq -r --arg f "$field" '.data.data[$f] // empty')
|
|
||||||
if [ -z "$val" ]; then
|
if [ -z "$val" ]; then
|
||||||
echo "::error::no value at '${path}' field '${field}' (wrong path/field, or this role's policy denies it): $(printf '%s' "$resp" | jq -rc '.errors // "empty"')"
|
echo "::error::no value at '${path}' field '${field}' (wrong path/field, this role's policy denies it, or cluster busy): $(printf '%s' "$resp" | jq -rc '.errors // "empty/non-JSON"' 2>/dev/null || echo 'non-JSON')"
|
||||||
exit 1
|
exit 1
|
||||||
fi
|
fi
|
||||||
# mask every line of the value (handles multiline secrets like keys)
|
# mask every line of the value (handles multiline secrets like keys)
|
||||||
@@ -76,6 +103,6 @@ runs:
|
|||||||
echo " ✓ ${env_name} <- ${path}#${field}"
|
echo " ✓ ${env_name} <- ${path}#${field}"
|
||||||
done <<< "$BAO_SECRETS"
|
done <<< "$BAO_SECRETS"
|
||||||
|
|
||||||
# 3) drop the token (short-lived anyway — just tidy)
|
# 3) drop the token (batch tokens can't be revoked; ignore errors)
|
||||||
curl -s --max-time 10 -H "X-Vault-Token: $TOKEN" \
|
curl -s --max-time 10 -H "X-Vault-Token: $TOKEN" \
|
||||||
-X POST "${BAO_ADDR}/v1/auth/token/revoke-self" >/dev/null 2>&1 || true
|
-X POST "${BAO_ADDR}/v1/auth/token/revoke-self" >/dev/null 2>&1 || true
|
||||||
|
|||||||
Reference in New Issue
Block a user