From 5307437a32c0029d46f5727a197f38f6223d2b24 Mon Sep 17 00:00:00 2001
From: Siyuan Zhang <siyuanzhang.zsy@alibaba-inc.com>
Date: Thu, 9 May 2024 14:46:53 +0800
Subject: [PATCH] fix(interactive): put pegasus hosts to configmap.yaml (#3777)

---
 charts/graphscope-store/templates/configmap.yaml | 12 ++++++++++++
 1 file changed, 12 insertions(+)

diff --git a/charts/graphscope-store/templates/configmap.yaml b/charts/graphscope-store/templates/configmap.yaml
index b27e19047f64..eb71c90e53a5 100644
--- a/charts/graphscope-store/templates/configmap.yaml
+++ b/charts/graphscope-store/templates/configmap.yaml
@@ -64,6 +64,7 @@ data:
     auth.password={{ .Values.auth.password }}
 
     # Pegasus config
+    pegasus.hosts=PEGASUS_HOSTS  # this is used by ir server to compute the server size
     pegasus.worker.num={{ .Values.pegasus.worker.num }}
     pegasus.timeout={{ .Values.pegasus.timeout }}
     pegasus.batch.size=1024
@@ -103,8 +104,19 @@ data:
     [[ `hostname` =~ -([0-9]+)$ ]] || exit 1
     ordinal=${BASH_REMATCH[1]}
 
+    pegasus_hosts=""
+    i=0
+    while [ $i -ne $STORE_COUNT ]; do
+      pod=`echo $DNS_NAME_PREFIX_STORE | sed -e "s/{}/$i/g"`
+      # 60001 is fixed gaia engine port
+      pegasus_hosts="${pegasus_hosts},${pod}:60001"
+      i=$(($i+1))
+    done
+    pegasus_hosts=${pegasus_hosts:1}
+
     sudo sed  -e "s/ROLE/${ROLE}/g" \
               -e "s/INDEX/${ordinal}/g" \
+              -e "s/PEGASUS_HOSTS/${pegasus_hosts}/g" \
               -e "s@LOG4RS_CONFIG@${GRAPHSCOPE_HOME}/groot/conf/log4rs.yml@g" \
     /etc/groot/groot.config.tpl | sudo tee -a /etc/groot/groot.config