Skip to content

Commit db04ce1

Browse files
committed
Add port validation
1 parent 5142c39 commit db04ce1

File tree

1 file changed

+8
-1
lines changed

1 file changed

+8
-1
lines changed

cmd/podinfo/main.go

Lines changed: 8 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -12,6 +12,7 @@ import (
1212
"io/ioutil"
1313
"os"
1414
"path/filepath"
15+
"strconv"
1516
"strings"
1617
"time"
1718
)
@@ -82,6 +83,12 @@ func main() {
8283
// start stress tests if any
8384
beginStressTest(viper.GetInt("stress-cpu"), viper.GetInt("stress-memory"), logger)
8485

86+
// validate port
87+
if _, err := strconv.Atoi(viper.GetString("port")); err != nil {
88+
port, _ := fs.GetInt("port")
89+
viper.Set("port", strconv.Itoa(port))
90+
}
91+
8592
// load HTTP server config
8693
var srvCfg api.Config
8794
if err := viper.Unmarshal(&srvCfg); err != nil {
@@ -92,7 +99,7 @@ func main() {
9299
logger.Info("Starting podinfo",
93100
zap.String("version", viper.GetString("version")),
94101
zap.String("revision", viper.GetString("revision")),
95-
zap.String("port", viper.GetString("port")),
102+
zap.String("port", srvCfg.Port),
96103
)
97104

98105
// start HTTP server

0 commit comments

Comments
 (0)