diff options
Diffstat (limited to 'cmd/grus/env.go')
-rw-r--r-- | cmd/grus/env.go | 17 |
1 files changed, 17 insertions, 0 deletions
diff --git a/cmd/grus/env.go b/cmd/grus/env.go new file mode 100644 index 0000000..4d72030 --- /dev/null +++ b/cmd/grus/env.go @@ -0,0 +1,17 @@ +package main + +import "os" + +// getEnv will check if the the key exists, if it does then it'll +// return the value otherwise it will return fallback string. +func getEnv(key, fallback string) string { + // We use os.LookupEnv instead of using os.GetEnv and checking + // if the length equals 0 because environment variable can be + // set and be of length 0. User could've set key="" which + // means the variable was set but the length is 0. + value, exists := os.LookupEnv(key) + if !exists { + value = fallback + } + return value +} |