fix: load host or IP from .env

pilinux · Aug 5, 2023 · 5e8d37f · 5e8d37f
1 parent 8038852
commit 5e8d37f
Show file tree

Hide file tree

Showing 5 changed files with 5 additions and 2 deletions.
diff --git a/config/config.go b/config/config.go
@@ -551,6 +551,7 @@ func security() (securityConfig SecurityConfig, err error) {
 
 // server - port and env
 func server() (serverConfig ServerConfig, err error) {
+	serverConfig.ServerHost = strings.TrimSpace(os.Getenv("APP_HOST"))
 	serverConfig.ServerPort = strings.TrimSpace(os.Getenv("APP_PORT"))
 	serverConfig.ServerEnv = strings.TrimSpace(os.Getenv("APP_ENV"))
 

diff --git a/config/config_test.go b/config/config_test.go
@@ -137,6 +137,7 @@ func TestGetConfig(t *testing.T) {
 	}
 	expected.Logger.SentryDsn = "https://xyz.ingest.sentry.io/123456"
 
+	expected.Server.ServerHost = "localhost"
 	expected.Server.ServerPort = "3000"
 	expected.Server.ServerEnv = "development"
 

diff --git a/config/server.go b/config/server.go
@@ -2,6 +2,7 @@ package config
 
 // ServerConfig ...
 type ServerConfig struct {
+	ServerHost string
 	ServerPort string // public port of server
 	ServerEnv  string
 }
diff --git a/example/main.go b/example/main.go
@@ -72,7 +72,7 @@ func main() {
 		return
 	}
 	// Attaches the router to a http.Server and starts listening and serving HTTP requests
-	err = r.Run(":" + configure.Server.ServerPort)
+	err = r.Run(configure.Server.ServerHost + ":" + configure.Server.ServerPort)
 	if err != nil {
 		fmt.Println(err)
 		return

diff --git a/example/router/setupRouter.go b/example/router/setupRouter.go
@@ -15,7 +15,7 @@ import (
 // SetupRouter sets up all the routes
 func SetupRouter(configure *gconfig.Configuration) (*gin.Engine, error) {
 	if configure.Server.ServerEnv == "production" {
-		gin.SetMode(gin.ReleaseMode) // Omit this line to enable debug mode
+		gin.SetMode(gin.ReleaseMode)
 	}
 
 	// Write log file