metrics.sh/lib/main.sh

#!/bin/sh

# load utils
for util in ./lib/utils/*.sh; do
  . $util
done

# init
__AVAILABLE_METRICS=
__AVAILABLE_REPORTERS=

main_load () {
  # load reporter
  for file in ./reporters/*.sh; do
    local filename=$(basename $file)
    local reporter=${filename%.*}

    load_reporter_with_prefix __r_${reporter}_ $file

    __AVAILABLE_REPORTERS=$(trim "$__AVAILABLE_REPORTERS $reporter")
  done

  # load metrics
  for file in ./metrics/*.sh; do
    local filename=$(basename $file)
    local metric=${filename%.*}

    load_metric_with_prefix __m_${metric}_ $file

    # register metric
    __AVAILABLE_METRICS=$(trim "$__AVAILABLE_METRICS $metric")
  done
}

main_init () {
  # handle args
  __METRICS=$(echo $1 | sed 's/,/ /g')
  __REPORTER=$2

  # create temp dir
  TEMP_DIR=$(make_temp_dir)

  # register trap
  trap '
    main_terminate
    trap - TERM && kill -- -$$ INT TERM EXIT
  ' INT TERM EXIT

  # check if reporter exists
  if ! in_array $__REPORTER "$__AVAILABLE_REPORTERS"; then
    echo "Error: reporter '$__REPORTER' is not available"
    exit 1
  fi

  # check if metrics exist
  for metric in $__METRICS; do
    if ! in_array $metric "$__AVAILABLE_METRICS"; then
      echo "Error: metric '$metric' is not available"
      exit 1
    fi
  done

  # init reporter
  if is_function __r_${__REPORTER}_init; then
    __r_${__REPORTER}_init
  fi

  # init metrics
  for metric in $__METRICS; do
    if ! is_function __m_${metric}_init; then
      continue
    fi

    __m_${metric}_init
  done
}

main_collect () {
  # used by metrics to return results
  report () {
    local _r_label _r_result
    if [ -z $2 ]; then
      _r_label=$metric
      _r_result="$1"
    else
      _r_label="$metric.$1"
      _r_result="$2"
    fi
    if is_number $_r_result; then
      __r_${__REPORTER}_report $_r_label $_r_result
    fi
  }

  # collect metrics
  for metric in $__METRICS; do
    if ! is_function __m_${metric}_collect; then
      continue
    fi

    # fork
    (while true; do
        __m_${metric}_collect
        sleep $INTERVAL
    done) &
  done

  # run forever
  sleep 2147483647 # `sleep infinity` is not portable
}

main_terminate () {
  # terminate metrics
  for metric in $__METRICS; do
    if ! is_function __m_${metric}_terminate; then
      continue
    fi
    __m_${metric}_terminate
  done

  # terminate reporter
  if is_function __r_${__REPORTER}_terminate; then
    __r_${__REPORTER}_terminate
  fi

  # delete temporary directory
  if [ -d $TEMP_DIR ]; then
    rmdir $TEMP_DIR
  fi
}

main_docs () {
  echo "# Metrics"
  for metric in $__AVAILABLE_METRICS; do
    if ! is_function __m_${metric}_docs; then
      continue
    fi

    echo
    echo "[$metric]"
    __m_${metric}_docs
  done

  echo
  echo "# REPORTERS"
  for reporter in $__AVAILABLE_REPORTERS; do
    if ! is_function __r_${reporter}_docs; then
      continue
    fi

    echo
    echo "[$reporter]"
    __r_${reporter}_docs
  done
}