fix: handle LISTEN_ADDRESS env variable

leonsteinhaeuser · Nov 21, 2024 · c4d7edf · c4d7edf
1 parent 0a6d958
commit c4d7edf
Show file tree

Hide file tree

Showing 3 changed files with 24 additions and 4 deletions.
diff --git a/services/number/main.go b/services/number/main.go
@@ -8,12 +8,21 @@ import (
 	"os"
 )
 
+var (
+	envListenAddress = os.Getenv("LISTEN_ADDRESS")
+)
+
 func init() {
 	logger := slog.New(slog.NewJSONHandler(os.Stdout, &slog.HandlerOptions{
 		AddSource: true,
 		Level:     slog.LevelDebug,
 	}))
 	slog.SetDefault(logger)
+
+	if envListenAddress == "" {
+		slog.Warn("LISTEN_ADDRESS is not set, using default value", "default", ":8081")
+		envListenAddress = ":8081"
+	}
 }
 
 func main() {
@@ -22,8 +31,8 @@ func main() {
 		w.WriteHeader(http.StatusOK)
 	})
 	http.HandleFunc("GET /number", logMiddleware(getNumber))
-	slog.Info("starting server on port 8081")
-	http.ListenAndServe(":8081", nil)
+	slog.Info("starting server on: " + envListenAddress)
+	http.ListenAndServe(envListenAddress, nil)
 }
 
 func logMiddleware(next http.HandlerFunc) http.HandlerFunc {

diff --git a/services/status/main.go b/services/status/main.go
@@ -43,6 +43,11 @@ const (
 )
 
 func init() {
+	if envListenAddress == "" {
+		slog.Warn("LISTEN_ADDRESS is not set, using default value", "default", ":8082")
+		envListenAddress = ":8082"
+	}
+
 	// Parse the envExternalServicesToWatch and populate the externalServicesToWatch map.
 	// If the envExternalServicesToWatch is empty, then we should watch the default services.
 	if envExternalServicesToWatch == "" {

diff --git a/services/view/main.go b/services/view/main.go
@@ -13,6 +13,7 @@ import (
 var (
 	envNumberServiceURL = os.Getenv("NUMBER_SERVICE_URL")
 	envStatusServiceURL = os.Getenv("STATUS_SERVICE_URL")
+	envListenAddress    = os.Getenv("LISTEN_ADDRESS")
 
 	//go:embed view.html
 	websiteHTML string
@@ -24,6 +25,11 @@ func init() {
 		Level:     slog.LevelDebug,
 	}))
 	slog.SetDefault(logger)
+
+	if envListenAddress == "" {
+		slog.Warn("LISTEN_ADDRESS is not set, using default value", "default", ":8080")
+		envListenAddress = ":8080"
+	}
 }
 
 func main() {
@@ -33,8 +39,8 @@ func main() {
 	})
 	http.HandleFunc("GET /api/v1/status", proxyEndpoint)
 	http.HandleFunc("GET /", logMiddleware(getMain))
-	slog.Info("starting server on port 8080")
-	http.ListenAndServe(":8080", nil)
+	slog.Info("starting server on port: " + envListenAddress)
+	http.ListenAndServe(envListenAddress, nil)
 }
 
 func proxyEndpoint(w http.ResponseWriter, r *http.Request) {