From c787742105644ab1924f79f6f34bb88223c2870b Mon Sep 17 00:00:00 2001
From: Dandelion <49650772+aroundabout@users.noreply.github.com>
Date: Tue, 24 Oct 2023 17:24:43 +0800
Subject: [PATCH] fix: always wait for storage if rocksdb is selected (#2333)

---
 .../src/assembly/static/bin/wait-storage.sh               | 8 +++++---
 1 file changed, 5 insertions(+), 3 deletions(-)

diff --git a/hugegraph-server/hugegraph-dist/src/assembly/static/bin/wait-storage.sh b/hugegraph-server/hugegraph-dist/src/assembly/static/bin/wait-storage.sh
index 3a98e8c56e..bdadeab234 100644
--- a/hugegraph-server/hugegraph-dist/src/assembly/static/bin/wait-storage.sh
+++ b/hugegraph-server/hugegraph-dist/src/assembly/static/bin/wait-storage.sh
@@ -48,7 +48,9 @@ while IFS=' ' read -r envvar_key envvar_val; do
 done < <(env | sort -r | awk -F= '{ st = index($0, "="); print $1 " " substr($0, st+1) }')
 
 # wait for storage
-if ! [ -z "${WAIT_STORAGE_TIMEOUT_S:-}" ]; then
-    timeout "${WAIT_STORAGE_TIMEOUT_S}s" bash -c \
-    "until bin/gremlin-console.sh -- -e $DETECT_STORAGE > /dev/null 2>&1; do echo \"waiting for storage...\"; sleep 5; done"
+if env | grep '^hugegraph\.' > /dev/null; then
+    if ! [ -z "${WAIT_STORAGE_TIMEOUT_S:-}" ]; then
+        timeout "${WAIT_STORAGE_TIMEOUT_S}s" bash -c \
+        "until bin/gremlin-console.sh -- -e $DETECT_STORAGE > /dev/null 2>&1; do echo \"waiting for storage...\"; sleep 5; done"
+    fi
 fi