Update main.go

mahaonanStart · Apr 9, 2024 · 347a4ed · 347a4ed
1 parent 2539cf6
commit 347a4ed
Showing 1 changed file with 4 additions and 2 deletions.
diff --git a/main.go b/main.go
@@ -28,15 +28,17 @@ func main() {
 
 	_ = godotenv.Load(".env")
 	host := os.Getenv("SERVER_HOST")
-	port := os.Getenv("PORT")
+	port := os.Getenv("SERVER_PORT")
 	tlsCert := os.Getenv("TLS_CERT")
 	tlsKey := os.Getenv("TLS_KEY")
 
 	if host == "" {
 		host = "0.0.0.0"
 	}
 	if port == "" {
-		port = "8080"
+		port = os.Getenv("PORT")
+		if port == "" {
+			port = "8080"
 		}
 	}