diff --git a/relp_handler.go b/relp_handler.go index def50ec..bf0dcd8 100644 --- a/relp_handler.go +++ b/relp_handler.go @@ -9,11 +9,28 @@ import ( "net/url" "os" "path/filepath" + "strconv" "time" "github.com/google/uuid" ) +func getEnvAsInt(name string, defaultVal int) int { + valStr := os.Getenv(name) + if val, err := strconv.Atoi(valStr); err == nil { + return val + } + return defaultVal +} + +func getEnvAsDuration(name string, defaultVal time.Duration) time.Duration { + valStr := os.Getenv(name) + if val, err := time.ParseDuration(valStr); err == nil { + return val + } + return defaultVal +} + var ( cacheDir = os.Getenv("LOGGER_CACHE_DIR") diskBufferLimit = getEnvAsInt("LOGGER_DISK_BUFFER_LIMIT", 100)