On notification server, get server host and port from environement if…

… set (#728)
haiwen · Dec 16, 2024 · da31024 · da31024
1 parent c11e8cc
commit da31024
Showing 1 changed file with 12 additions and 1 deletion.
diff --git a/notification-server/server.go b/notification-server/server.go
@@ -11,6 +11,7 @@ import (
 	"os/signal"
 	"path/filepath"
 	"strings"
+	"strconv"
 	"syscall"
 	"time"
 
@@ -45,8 +46,18 @@ func init() {
 }
 
 func loadNotifConfig() {
-	host = "0.0.0.0"
+	host := os.Getenv("NOTIFICATION_SERVER_HOST")
+	if host == "" {
+		host = "0.0.0.0"
+	}
+
 	port = 8083
+	if os.Getenv("NOTIFICATION_SERVER_PORT") != "" {
+		i, err := strconv.Atoi(os.Getenv("NOTIFICATION_SERVER_PORT"))
+		if err == nil {
+			port = i
+		}
+	}
 
 	logLevel := os.Getenv("NOTIFICATION_SERVER_LOG_LEVEL")
 	if logLevel == "" {