| 1 | #!/usr/bin/env bash |
| 2 | me="$(basename "${0}")" |
| 3 | |
| 4 | # Checks if netdata is running in a kubernetes pod and fetches: |
| 5 | # - pod's labels |
| 6 | # - kubernetes cluster name (GKE only) |
| 7 | |
| 8 | if [ -z "${KUBERNETES_SERVICE_HOST}" ] || [ -z "${KUBERNETES_PORT_443_TCP_PORT}" ] || [ -z "${MY_POD_NAMESPACE}" ] || [ -z "${MY_POD_NAME}" ]; then |
| 9 | exit 0 |
| 10 | fi |
| 11 | |
| 12 | if ! command -v jq >/dev/null 2>&1; then |
| 13 | echo >&2 "${me}: jq command not available. Please install jq to get host labels for kubernetes pods." |
| 14 | exit 1 |
| 15 | fi |
| 16 | |
| 17 | TOKEN="$(< /var/run/secrets/kubernetes.io/serviceaccount/token)" |
| 18 | HEADER="Authorization: Bearer $TOKEN" |
| 19 | HOST="$KUBERNETES_SERVICE_HOST:$KUBERNETES_PORT_443_TCP_PORT" |
| 20 | |
| 21 | URL="https://$HOST/api/v1/namespaces/$MY_POD_NAMESPACE/pods/$MY_POD_NAME" |
| 22 | if ! POD_DATA=$(curl --fail -sSk -H "$HEADER" "$URL" 2>&1); then |
| 23 | echo >&2 "${me}: error on curl '${URL}': ${POD_DATA}." |
| 24 | exit 1 |
| 25 | fi |
| 26 | |
| 27 | URL="https://$HOST/api/v1/namespaces/kube-system" |
| 28 | if ! KUBE_SYSTEM_NS_DATA=$(curl --fail -sSk -H "$HEADER" "$URL" 2>&1); then |
| 29 | echo >&2 "${me}: error on curl '${URL}': ${KUBE_SYSTEM_NS_DATA}." |
| 30 | exit 1 |
| 31 | fi |
| 32 | |
| 33 | if ! POD_LABELS=$(jq -r '.metadata.labels' <<< "$POD_DATA" | grep ':' | tr -d '," ' 2>&1); then |
| 34 | echo >&2 "${me}: error on 'jq' parse pod data: ${POD_LABELS}." |
| 35 | exit 1 |
| 36 | fi |
| 37 | |
| 38 | if ! KUBE_SYSTEM_NS_UID=$(jq -r '.metadata.uid' <<< "$KUBE_SYSTEM_NS_DATA" 2>&1); then |
| 39 | echo >&2 "${me}: error on 'jq' parse kube_system_ns: ${KUBE_SYSTEM_NS_UID}." |
| 40 | exit 1 |
| 41 | fi |
| 42 | |
| 43 | LABELS="$POD_LABELS\nk8s_cluster_id:$KUBE_SYSTEM_NS_UID" |
| 44 | |
| 45 | GCP_META_HEADER="Metadata-Flavor: Google" |
| 46 | GCP_META_URL="http://metadata/computeMetadata/v1" |
| 47 | GKE_CLUSTER_NAME="" |
| 48 | |
| 49 | if id=$(curl --fail -s -m 5 --noproxy "*" -H "$GCP_META_HEADER" "$GCP_META_URL/project/project-id"); then |
| 50 | loc=$(curl --fail -s -m 5 --noproxy "*" -H "$GCP_META_HEADER" "$GCP_META_URL/instance/attributes/cluster-location") |
| 51 | name=$(curl --fail -s -m 5 --noproxy "*" -H "$GCP_META_HEADER" "$GCP_META_URL/instance/attributes/cluster-name") |
| 52 | [ -n "$id" ] && [ -n "$loc" ] && [ -n "$name" ] && GKE_CLUSTER_NAME="gke_${id}_${loc}_${name}" |
| 53 | fi |
| 54 | |
| 55 | [ -n "$GKE_CLUSTER_NAME" ] && LABELS+="\nk8s_cluster_name:$GKE_CLUSTER_NAME" |
| 56 | |
| 57 | echo -e "$LABELS" |
| 58 | |
| 59 | exit 0 |