From fa663be02988713dbad98732194512a7623f7b72 Mon Sep 17 00:00:00 2001
From: =?UTF-8?q?=C8=98tefan=20Talpalaru?= <stefantalpalaru@yahoo.com>
Date: Wed, 6 May 2020 00:02:39 +0200
Subject: [PATCH] launch_local_testnet.sh: get the metrics out with "--grafana"

---
 scripts/launch_local_testnet.sh | 41 ++++++++++++++++++++++++++++++++-
 tests/simulation/start.sh       |  4 +---
 2 files changed, 41 insertions(+), 4 deletions(-)

diff --git a/scripts/launch_local_testnet.sh b/scripts/launch_local_testnet.sh
index 636725c46..362e575b1 100755
--- a/scripts/launch_local_testnet.sh
+++ b/scripts/launch_local_testnet.sh
@@ -24,7 +24,7 @@ if [ ${PIPESTATUS[0]} != 4 ]; then
 fi
 
 OPTS="ht:n:d:"
-LONGOPTS="help,testnet:,nodes:,data-dir:,disable-htop,log-level:"
+LONGOPTS="help,testnet:,nodes:,data-dir:,disable-htop,log-level:,grafana"
 
 # default values
 TESTNET="1"
@@ -32,6 +32,7 @@ NUM_NODES="10"
 DATA_DIR="local_testnet_data"
 USE_HTOP="1"
 LOG_LEVEL="DEBUG"
+ENABLE_GRAFANA="0"
 
 print_help() {
 	cat <<EOF
@@ -46,6 +47,7 @@ CI run: $(basename $0) --disable-htop -- --verify-finalization --stop-at-epoch=5
 			(default: "${DATA_DIR}")
       --disable-htop	don't use "htop" to see the beacon_node processes
       --log-level	set the log level (default: ${LOG_LEVEL})
+      --grafana		generate Grafana dashboards (and Prometheus config file)
 EOF
 }
 
@@ -83,6 +85,10 @@ while true; do
 			LOG_LEVEL="$2"
 			shift 2
 			;;
+		--grafana)
+			ENABLE_GRAFANA="1"
+			shift
+			;;
 		--)
 			shift
 			break
@@ -100,6 +106,7 @@ if [[ $# != 0 ]]; then
 	shift $#
 fi
 NETWORK="testnet${TESTNET}"
+BASE_METRICS_PORT=8008
 
 rm -rf "${DATA_DIR}"
 DEPOSITS_DIR="${DATA_DIR}/deposits_dir"
@@ -140,6 +147,35 @@ BOOTSTRAP_IP="127.0.0.1"
 	--bootstrap-port=${BOOTSTRAP_PORT} \
 	--genesis-offset=5 # Delay in seconds
 
+if [[ "$ENABLE_GRAFANA" == "1" ]]; then
+	# Prometheus config
+	cat > "${DATA_DIR}/prometheus.yml" <<EOF
+global:
+  scrape_interval: 1s
+
+scrape_configs:
+  - job_name: "nimbus"
+    static_configs:
+EOF
+	for NUM_NODE in $(seq 0 $(( ${NUM_NODES} - 1 ))); do
+		cat >> "${DATA_DIR}/prometheus.yml" <<EOF
+      - targets: ['127.0.0.1:$(( BASE_METRICS_PORT + NUM_NODE ))']
+        labels:
+          node: '$NUM_NODE'
+EOF
+	done
+
+	# use the exported Grafana dashboard for a single node to create one for all nodes
+	PROCESS_DASHBOARD_BIN="build/process_dashboard"
+	if [[ ! -f "$PROCESS_DASHBOARD_BIN" ]]; then
+	  $MAKE process_dashboard
+	fi
+	"${PROCESS_DASHBOARD_BIN}" \
+	  --nodes=${NUM_NODES} \
+	  --in="tests/simulation/beacon-chain-sim-node0-Grafana-dashboard.json" \
+	  --out="${DATA_DIR}/local-testnet-all-nodes-Grafana-dashboard.json"
+fi
+
 # Kill child processes on Ctrl-C/SIGTERM/exit, passing the PID of this shell
 # instance as the parent and the target process name as a pattern to the
 # "pkill" command.
@@ -200,6 +236,9 @@ for NUM_NODE in $(seq 0 $(( ${NUM_NODES} - 1 ))); do
 		--data-dir="${NODE_DATA_DIR}" \
 		${BOOTSTRAP_ARG} \
 		--state-snapshot="${NETWORK_DIR}/genesis.ssz" \
+		--metrics \
+		--metrics-address="127.0.0.1" \
+		--metrics-port="$(( BASE_METRICS_PORT + NUM_NODE ))" \
 		${EXTRA_ARGS} \
 		> "${DATA_DIR}/log${NUM_NODE}.txt" 2>&1 &
 
diff --git a/tests/simulation/start.sh b/tests/simulation/start.sh
index 5b81fb739..2a7239584 100755
--- a/tests/simulation/start.sh
+++ b/tests/simulation/start.sh
@@ -26,10 +26,8 @@ LAST_VALIDATOR="$VALIDATORS_DIR/v$(printf '%07d' $LAST_VALIDATOR_NUM).deposit.js
 # Windows detection
 if uname | grep -qiE "mingw|msys"; then
   MAKE="mingw32-make"
-  EXE_SUFFIX=".exe"
 else
   MAKE="make"
-  EXE_SUFFIX=""
 fi
 
 # to allow overriding the program names
@@ -176,7 +174,7 @@ if [ -f "${MASTER_NODE_ADDRESS_FILE}" ]; then
   rm "${MASTER_NODE_ADDRESS_FILE}"
 fi
 
-PROCESS_DASHBOARD_BIN="build/process_dashboard${EXE_SUFFIX}"
+PROCESS_DASHBOARD_BIN="build/process_dashboard"
 
 if [[ ! -f "$PROCESS_DASHBOARD_BIN" ]]; then
   $MAKE NIMFLAGS="$CUSTOM_NIMFLAGS" process_dashboard