Updated dataGenerator.sh script

Author: adityauj

tools/dataGenerator.sh

@@ -12,6 +12,7 @@ API_USER="demo"             # User for JWT generation
 # BASE NETWORK CONFIG
 SERVICE_ADDRESS="http://localhost:8080"
 NATS_SERVER="nats://0.0.0.0:4222"
+REST_URL="${SERVICE_ADDRESS}/api/write"
 
 # NATS CREDENTIALS
 NATS_USER="root"
@@ -27,18 +28,22 @@ JWT_STATIC="eyJhbGciOiJFZERTQSIsInR5cCI6IkpXVCJ9.eyJleHAiOjE3NzU3Nzg4NDQsImlhdCI
 ALEX_HOSTS="a0603 a0903 a0832 a0329 a0702 a0122 a1624 a0731 a0224 a0704 a0631 a0225 a0222 a0427 a0603 a0429 a0833 a0705 a0901 a0601 a0227 a0804 a0322 a0226 a0126 a0129 a0605 a0801 a0934 a1622 a0902 a0428 a0537 a1623 a1722 a0228 a0701 a0326 a0327 a0123 a0321 a1621 a0323 a0124 a0534 a0931 a0324 a0933 a0424 a0905 a0128 a0532 a0805 a0521 a0535 a0932 a0127 a0325 a0633 a0831 a0803 a0426 a0425 a0229 a1721 a0602 a0632 a0223 a0422 a0423 a0536 a0328 a0703 anvme7 a0125 a0221 a0604 a0802 a0522 a0531 a0533 a0904"
 FRITZ_HOSTS="f0201 f0202 f0203 f0204 f0205 f0206 f0207 f0208 f0209 f0210 f0211 f0212 f0213 f0214 f0215 f0217 f0218 f0219 f0220 f0221 f0222 f0223 f0224 f0225 f0226 f0227 f0228 f0229 f0230 f0231 f0232 f0233 f0234 f0235 f0236 f0237 f0238 f0239 f0240 f0241 f0242 f0243 f0244 f0245 f0246 f0247 f0248 f0249 f0250 f0251 f0252 f0253 f0254 f0255 f0256 f0257 f0258 f0259 f0260 f0261 f0262 f0263 f0264 f0378"
 
-METRICS_STD="cpu_load cpu_user flops_any cpu_irq cpu_system ipc cpu_idle cpu_iowait core_power clock"
-METRICS_NODE="cpu_irq cpu_load mem_cached net_bytes_in cpu_user cpu_idle nfs4_read mem_used nfs4_write nfs4_total ib_xmit ib_xmit_pkts net_bytes_out cpu_iowait ib_recv cpu_system ib_recv_pkts"
+ALEX_METRICS_HWTHREAD="cpu_user flops_any clock core_power ipc"
+ALEX_METRICS_SOCKET="mem_bw cpu_power"
+ALEX_METRICS_ACC="acc_utilization acc_mem_used acc_power nv_mem_util nv_temp nv_sm_clock"
+ALEX_METRICS_NODE="cpu_load mem_used net_bytes_in net_bytes_out"
+
+FRITZ_METRICS_HWTHREAD="cpu_user flops_any flops_sp flops_dp clock ipc vectorization_ratio"
+FRITZ_METRICS_SOCKET="mem_bw cpu_power mem_power"
+FRITZ_METRICS_NODE="cpu_load mem_used ib_recv ib_xmit ib_recv_pkts ib_xmit_pkts nfs4_read nfs4_total"
+
 ACCEL_IDS="00000000:49:00.0 00000000:0E:00.0 00000000:D1:00.0 00000000:90:00.0 00000000:13:00.0 00000000:96:00.0 00000000:CC:00.0 00000000:4F:00.0"
 
 # ==========================================
 # SETUP ENV (URL & TOKEN)
 # ==========================================
 
-if [ "$CONNECTION_SCOPE" == "INTERNAL" ]; then
-    # 1. Set URL for Internal Mode
-    REST_URL="${SERVICE_ADDRESS}/metricstore/api/write"
-    
+if [ "$CONNECTION_SCOPE" == "INTERNAL" ]; then    
     # 2. Generate JWT dynamically
     echo "Setup: INTERNAL mode selected."
     echo "Generating JWT for user: $API_USER"
@@ -48,10 +53,7 @@ if [ "$CONNECTION_SCOPE" == "INTERNAL" ]; then
         echo "Error: Failed to generate JWT from cc-backend."
         exit 1
     fi
-else
-    # 1. Set URL for External Mode
-    REST_URL="${SERVICE_ADDRESS}/api/write"
-    
+else    
     # 2. Use Static JWT
     echo "Setup: EXTERNAL mode selected."
     echo "Using static JWT."
@@ -96,7 +98,7 @@ while [ true ]; do
     # 1. ALEX: HWTHREAD
     echo "Generating Alex: hwthread"
     {
-        for metric in $METRICS_STD; do
+        for metric in $ALEX_METRICS_HWTHREAD; do
             for hostname in $ALEX_HOSTS; do
                 for id in {0..127}; do
                     echo "$metric,cluster=alex,hostname=$hostname,type=hwthread,type-id=$id value=$((1 + RANDOM % 100)).0 $timestamp"
@@ -109,7 +111,7 @@ while [ true ]; do
     # 2. FRITZ: HWTHREAD
     echo "Generating Fritz: hwthread"
     {
-        for metric in $METRICS_STD; do
+        for metric in $FRITZ_METRICS_HWTHREAD; do
             for hostname in $FRITZ_HOSTS; do
                 for id in {0..71}; do
                     echo "$metric,cluster=fritz,hostname=$hostname,type=hwthread,type-id=$id value=$((1 + RANDOM % 100)).0 $timestamp"
@@ -122,7 +124,7 @@ while [ true ]; do
     # 3. ALEX: ACCELERATOR
     echo "Generating Alex: accelerator"
     {
-        for metric in $METRICS_STD; do
+        for metric in $ALEX_METRICS_ACC; do
             for hostname in $ALEX_HOSTS; do
                 for id in $ACCEL_IDS; do
                     echo "$metric,cluster=alex,hostname=$hostname,type=accelerator,type-id=$id value=$((1 + RANDOM % 100)).0 $timestamp"
@@ -132,23 +134,10 @@ while [ true ]; do
     } > sample_alex.txt
     send_payload "sample_alex.txt" "alex"
 
-    # 4. ALEX: MEMORY DOMAIN
-    echo "Generating Alex: memoryDomain"
-    {
-        for metric in $METRICS_STD; do
-            for hostname in $ALEX_HOSTS; do
-                for id in {0..7}; do
-                    echo "$metric,cluster=alex,hostname=$hostname,type=memoryDomain,type-id=$id value=$((1 + RANDOM % 100)).0 $timestamp"
-                done
-            done
-        done
-    } > sample_alex.txt
-    send_payload "sample_alex.txt" "alex"
-
     # 5. ALEX: SOCKET
     echo "Generating Alex: socket"
     {
-        for metric in $METRICS_STD; do
+        for metric in $ALEX_METRICS_SOCKET; do
             for hostname in $ALEX_HOSTS; do
                 for id in {0..1}; do
                     echo "$metric,cluster=alex,hostname=$hostname,type=socket,type-id=$id value=$((1 + RANDOM % 100)).0 $timestamp"
@@ -161,7 +150,7 @@ while [ true ]; do
     # 6. FRITZ: SOCKET
     echo "Generating Fritz: socket"
     {
-        for metric in $METRICS_STD; do
+        for metric in $FRITZ_METRICS_SOCKET; do
             for hostname in $FRITZ_HOSTS; do
                 for id in {0..1}; do
                     echo "$metric,cluster=fritz,hostname=$hostname,type=socket,type-id=$id value=$((1 + RANDOM % 100)).0 $timestamp"
@@ -174,7 +163,7 @@ while [ true ]; do
     # 7. ALEX: NODE
     echo "Generating Alex: node"
     {
-        for metric in $METRICS_NODE; do
+        for metric in $ALEX_METRICS_NODE; do
             for hostname in $ALEX_HOSTS; do
                 echo "$metric,cluster=alex,hostname=$hostname,type=node value=$((1 + RANDOM % 100)).0 $timestamp"
             done
@@ -185,7 +174,7 @@ while [ true ]; do
     # 8. FRITZ: NODE
     echo "Generating Fritz: node"
     {
-        for metric in $METRICS_NODE; do
+        for metric in $FRITZ_METRICS_NODE; do
             for hostname in $FRITZ_HOSTS; do
                 echo "$metric,cluster=fritz,hostname=$hostname,type=node value=$((1 + RANDOM % 100)).0 $timestamp"
             done