From 2feb3b35a7b3122715556752c46cddde410f89a1 Mon Sep 17 00:00:00 2001
From: nugaon <50576770+nugaon@users.noreply.github.com>
Date: Wed, 5 May 2021 14:02:26 +0200
Subject: [PATCH] feat: queen waits for worker connections (#12)

---
 scripts/bee.sh | 14 ++++++++++++++
 1 file changed, 14 insertions(+)

diff --git a/scripts/bee.sh b/scripts/bee.sh
index 20a7ab9..ebaa1c8 100755
--- a/scripts/bee.sh
+++ b/scripts/bee.sh
@@ -56,6 +56,12 @@ log_queen() {
     docker logs --tail 25 -f "$QUEEN_CONTAINER_NAME"
 }
 
+count_connected_peers() {
+    COUNT=$( curl -s http://localhost:1635/peers | python -c 'import json,sys; obj=json.load(sys.stdin); print len(obj["peers"]);' )
+    
+    echo "$COUNT"
+}
+
 MY_PATH=$(dirname "$0")              # relative
 MY_PATH=$( cd "$MY_PATH" && pwd )  # absolutized and normalized
 # Check used system variable set
@@ -222,6 +228,14 @@ for i in $(seq 1 1 "$WORKERS"); do
   fi
 done
 
+echo "Check whether the queen node has been connected to every worker..."
+while : ; do
+    COUNT=$(count_connected_peers)
+    [[ $COUNT < $WORKERS ]] || break
+    echo "Only $COUNT peers have been connected to the Queen Bee node yet. Waiting until $WORKERS"
+    sleep 2
+done
+
 # log Bee Queen
 if $LOG ; then
     log_queen