modify: use node port from arg url

anniemon · Aug 19, 2024 · 249f2a4 · 249f2a4
1 parent 990f9e4
commit 249f2a4
Show file tree

Hide file tree

Showing 2 changed files with 2 additions and 1 deletion.
diff --git a/backend_server.go b/backend_server.go
@@ -6,6 +6,7 @@ import (
 	"math/rand"
 	"net/http"
 	"net/url"
+	"strings"
 )
 
 func startBackendServer(id int, targetURL string) {
@@ -27,7 +28,7 @@ func startBackendServer(id int, targetURL string) {
 		w.WriteHeader(http.StatusOK)
 	})
 	server := &http.Server{
-		Addr:    fmt.Sprintf(":%d", 8080+id),
+		Addr:    fmt.Sprintf(":%s", strings.Split(targetURL, ":")[2]),
 		Handler: mux,
 	}
 	log.Printf("Starting node %d on port %d", id, 8080+id)

diff --git a/goLoadBalancer b/goLoadBalancer