fix(interactive): put pegasus hosts to configmap.yaml (alibaba#3777)

shirly121 · May 9, 2024 · 5307437 · 5307437
1 parent 039fce0
commit 5307437
Showing 1 changed file with 12 additions and 0 deletions.
diff --git 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