diff --git a/config/helper.go b/config/helper.go index d3849db..e17dc02 100644 --- a/config/helper.go +++ b/config/helper.go @@ -26,3 +26,16 @@ func getEnvSeconds(key string, fallback int) time.Duration { } return time.Duration(fallback) * time.Second } + +func getEnvUint(key string, fallback uint) uint { + var def uint = 100 + if value, ok := os.LookupEnv(key); ok { + num, err := strconv.ParseUint(value, 10, 64) + if err != nil { + log.Printf("Error converting %v to uint, using default value - %v", key, def) + return def + } + return uint(num) + } + return fallback +}