plugins.d/cgroup-name.sh

   1 #!/usr/bin/env bash
   2
   3 # netdata
   4 # real-time performance and health monitoring, done right!
   5 # (C) 2016 Costa Tsaousis <costa@tsaousis.gr>
   6 # GPL v3+
   7 #
   8 # Script to find a better name for cgroups
   9 #
  10
  11 export PATH="${PATH}:/sbin:/usr/sbin:/usr/local/sbin"
  12 export LC_ALL=C
  13
  14 # -----------------------------------------------------------------------------
  15
  16 PROGRAM_NAME="$(basename "${0}")"
  17
  18 logdate() {
  19     date "+%Y-%m-%d %H:%M:%S"
  20 }
  21
  22 log() {
  23     local status="${1}"
  24     shift
  25
  26     echo >&2 "$(logdate): ${PROGRAM_NAME}: ${status}: ${*}"
  27
  28 }
  29
  30 warning() {
  31     log WARNING "${@}"
  32 }
  33
  34 error() {
  35     log ERROR "${@}"
  36 }
  37
  38 info() {
  39     log INFO "${@}"
  40 }
  41
  42 fatal() {
  43     log FATAL "${@}"
  44     exit 1
  45 }
  46
  47 debug=0
  48 debug() {
  49     [ $debug -eq 1 ] && log DEBUG "${@}"
  50 }
  51
  52 # -----------------------------------------------------------------------------
  53
  54 NETDATA_CONFIG_DIR="${NETDATA_CONFIG_DIR-/etc/netdata}"
  55 CONFIG="${NETDATA_CONFIG_DIR}/cgroups-names.conf"
  56 CGROUP="${1}"
  57 NAME=
  58
  59 # -----------------------------------------------------------------------------
  60
  61 if [ -z "${CGROUP}" ]
  62     then
  63     fatal "called without a cgroup name. Nothing to do."
  64 fi
  65
  66 if [ -f "${CONFIG}" ]
  67     then
  68     NAME="$(grep "^${CGROUP} " "${CONFIG}" | sed "s/[[:space:]]\+/ /g" | cut -d ' ' -f 2)"
  69     if [ -z "${NAME}" ]
  70         then
  71         info "cannot find cgroup '${CGROUP}' in '${CONFIG}'."
  72     fi
  73 #else
  74 #   info "configuration file '${CONFIG}' is not available."
  75 fi
  76
  77 function get_name_classic {
  78     local DOCKERID="$1"
  79     info "Running command: docker ps --filter=id=\"${DOCKERID}\" --format=\"{{.Names}}\""
  80     NAME="$( docker ps --filter=id="${DOCKERID}" --format="{{.Names}}" )"
  81     return 0
  82 }
  83
  84 function get_name_api {
  85     local DOCKERID="$1"
  86     if [ ! -S "/var/run/docker.sock" ]
  87         then
  88         warning "Can't find /var/run/docker.sock"
  89         return 1
  90     fi
  91     info "Running API command: /containers/${DOCKERID}/json"
  92     JSON=$(echo -e "GET /containers/${DOCKERID}/json HTTP/1.0\r\n" | nc -U /var/run/docker.sock | egrep '^{.*')
  93     NAME=$(echo $JSON | jq -r .Name,.Config.Hostname | grep -v null | head -n1 | sed 's|^/||')
  94     return 0
  95 }
  96
  97 if [ -z "${NAME}" ]
  98     then
  99     if [[ "${CGROUP}" =~ ^.*docker[-_/\.][a-fA-F0-9]+[-_\.]?.*$ ]]
 100         then
 101         DOCKERID="$( echo "${CGROUP}" | sed "s|^.*docker[-_/]\([a-fA-F0-9]\+\)[-_\.]\?.*$|\1|" )"
 102         # echo "DOCKERID=${DOCKERID}"
 103
 104         if [ ! -z "${DOCKERID}" -a \( ${#DOCKERID} -eq 64 -o ${#DOCKERID} -eq 12 \) ]
 105             then
 106             if hash docker 2>/dev/null
 107                 then
 108                 get_name_classic $DOCKERID
 109             else
 110                 get_name_api $DOCKERID || get_name_classic $DOCKERID
 111             fi
 112             if [ -z "${NAME}" ]
 113                 then
 114                 warning "cannot find the name of docker container '${DOCKERID}'"
 115                 NAME="${DOCKERID:0:12}"
 116             else
 117                 info "docker container '${DOCKERID}' is named '${NAME}'"
 118             fi
 119         fi
 120     fi
 121
 122     [ -z "${NAME}" ] && NAME="${CGROUP}"
 123     [ ${#NAME} -gt 100 ] && NAME="${NAME:0:100}"
 124 fi
 125
 126 info "cgroup '${CGROUP}' is called '${NAME}'"
 127 echo "${NAME}"