added support to read port from environment variable

flags/flags.go

@@ -1,10 +1,13 @@
 package flags
 
 import (
-	actuatorCommons "github.com/sinhashubham95/go-actuator/commons"
+	"github.com/sinhashubham95/moxy/commons"
 	flag "github.com/spf13/pflag"
+	"os"
+	"strconv"
+	"strings"
 
-	"github.com/sinhashubham95/moxy/commons"
+	actuatorCommons "github.com/sinhashubham95/go-actuator/commons"
 )
 
 var (
@@ -41,6 +44,12 @@ func PersistencePath() string {
 
 // Port is the port number where the application is running
 func Port() int {
+	// see if the port is available from the environment variable
+	p, err := strconv.Atoi(os.Getenv(strings.ToUpper(commons.Port)))
+	if err == nil {
+		return p
+	}
+	// otherwise use the flag
 	return *port
 }
 

flags/flags_test.go

@@ -4,6 +4,8 @@ import (
 	"github.com/sinhashubham95/moxy/commons"
 	"github.com/sinhashubham95/moxy/flags"
 	"github.com/stretchr/testify/assert"
+	"os"
+	"strings"
 	"testing"
 )
 
@@ -23,6 +25,11 @@ func TestPort(t *testing.T) {
 	assert.Equal(t, commons.PortDefaultValue, flags.Port())
 }
 
+func TestPortFromEnv(t *testing.T) {
+	assert.NoError(t, os.Setenv(strings.ToUpper(commons.Port), "5050"))
+	assert.Equal(t, 5050, flags.Port())
+}
+
 func TestTLSEnabled(t *testing.T) {
 	assert.Equal(t, commons.TLSEnabledDefaultValue, flags.TLSEnabled())
 }