Initial commit of the O1 simulator framework.

[sim/o1-interface.git] / ntsimulator / scripts / docker_stats.sh

diff --git a/ntsimulator/scripts/docker_stats.sh b/ntsimulator/scripts/docker_stats.sh
new file mode 100755 (executable)
index 0000000..05d6309
--- /dev/null
+++ b/ntsimulator/scripts/docker_stats.sh
@@ -0,0 +1,39 @@
+#!/bin/bash
+
+# This script is used to complete the output of the docker stats command.
+# The docker stats command does not compute the total amount of resources (RAM or CPU)
+
+# Get the output of the docker stat command. Will be displayed at the end
+# Without modifying the special variable IFS the ouput of the docker stats command won't have
+CPU_CORES=`nproc`
+# the new lines thus resulting in a failure when using awk to process each line
+IFS=;
+mapfile -t DOCKER_PS_RESULT < <(/usr/bin/docker ps --all --format "{{.ID}}" --filter "label=NTS")
+
+CONTAINERS=""
+
+if [ ${#DOCKER_PS_RESULT[@]} -gt 0 ]
+then
+
+       for container in "${DOCKER_PS_RESULT[@]}"
+       do
+               CONTAINERS="$CONTAINERS $container"
+       done
+fi
+
+if [ -z "$CONTAINERS" ]
+then
+       CPU_SCALED=0
+       SUM_RAM=0
+else
+       DOCKER_STATS_COMMAND="/usr/bin/docker stats --no-stream --format \"table {{.CPUPerc}}\t{{.MemUsage}}\" ${CONTAINERS}"
+       DOCKER_STATS_COMMAND_RESULT=$(eval "$DOCKER_STATS_COMMAND")
+
+       SUM_CPU=`echo $DOCKER_STATS_COMMAND_RESULT | tail -n +2 | sed "s/%//g" | awk '{s+=$1} END {print s}'`
+       SUM_RAM=`echo $DOCKER_STATS_COMMAND_RESULT | tail -n +2 | sed "s/%//g" | awk '{s+=$2} END {print s}'`
+
+       CPU_SCALED=$(echo "scale=2; $SUM_CPU/$CPU_CORES" | bc)
+fi
+
+# Output the result
+echo -e "CPU=${CPU_SCALED}%;RAM=${SUM_RAM}MiB"
\ No newline at end of file