From 1cdec2f9fa63801be654a8aa2c3bc119cb234732 Mon Sep 17 00:00:00 2001
From: ravinsp <33562092+ravinsp@users.noreply.github.com>
Date: Wed, 17 Nov 2021 16:52:13 +0530
Subject: [PATCH] Customer port specifier for signalr streamer.

---
 test/vm-cluster/stream.js | 12 +++++++++---
 1 file changed, 9 insertions(+), 3 deletions(-)

diff --git a/test/vm-cluster/stream.js b/test/vm-cluster/stream.js
index 40021904..d16a2366 100644
--- a/test/vm-cluster/stream.js
+++ b/test/vm-cluster/stream.js
@@ -172,9 +172,15 @@ function metricsTracker() {
 async function streamCluster(cluster) {
     console.log(`Starting to stream cluster '${cluster.name}'...`);
 
-    // TODO: Resolve Vultr vm groups.
-
-    cluster.hosts.forEach((h, idx) => streamNode(cluster.name, (idx + 1), h, cluster.userPort));
+    cluster.hosts.forEach((host, idx) => {
+        let port = cluster.userPort;
+        if (host.includes(":")) {
+            const parts = host.split(":");
+            host = parts[0];
+            port = parseInt(parts[1]);
+        }
+        streamNode(cluster.name, (idx + 1), host, cluster.userPort);
+    });
 }
 
 function streamNode(clusterName, nodeIdx, host, port) {