Как назначить значение по умолчанию, если переменная окружения не установлена в Go?
В Python я мог сделать mongo_password = os.getenv('MONGO_PASS', 'pass'), где pass является значением по умолчанию, если MONGO_PASS env var не установлен.
Я пробовал оператор if, основанный на том, что os.Getenv пуст, но, похоже, это не работает из-за объема присваивания переменных внутри оператора if. И я проверяю наличие нескольких env var, поэтому я не могу использовать эту информацию в операторе if.
Нет встроенного, чтобы вернуться к значению по умолчанию, так что вам нужно сделать добрый старомодный if-else.
Но вы всегда можете создать вспомогательную функцию, чтобы сделать это проще:
func getenv(key, fallback string) string {
value := os.Getenv(key)
if len(value) == 0 {
return fallback
}
return value
}
Обратите внимание, что в @michael-hausenblas в комментарии, имейте в виду, что если значение переменной окружения действительно пустое, вы получите вместо этого запасное значение.
Еще лучше, чем @ŁukaszWojciechowski, используя os.LookupEnv:
func getEnv(key, fallback string) string {
if value, ok := os.LookupEnv(key); ok {
return value
}
return fallback
}
Что вам нужно, так это os.LookupEnv сочетании с оператором if.
Вот ответ от janos, обновленный для использования LookupEnv:
func getEnv(key, fallback string) string {
value, exists := os.LookupEnv(key)
if !exists {
value = fallback
}
return value
}
Go не имеет такой же функциональности, как Python; самый идиоматический способ сделать это, хотя, я могу думать, есть:
mongo_password := "pass"
if mp := os.Getenv("MONGO_PASS"); mp != "" {
mongo_password = mp
}
Чтобы иметь чистый код, я делаю это:
myVar := getEnv("MONGO_PASS", "default-pass")
Я определил функцию, которая используется во всем приложении
// getEnv get key environment variable if exist otherwise return defalutValue
func getEnv(key, defaultValue string) string {
value := os.Getenv(key)
if len(value) == 0 {
return defaultValue
}
return value
}
У него был тот же вопрос, что и у ОП, и он нашел, что кто-то заключил ответы из этой темы в изящную библиотеку, довольно простую в использовании, надеюсь, это поможет другим!
Если у вас все в порядке с добавлением небольшой зависимости, вы можете использовать что-то вроде https://github.com/urfave/cli
package main
import (
"os"
"github.com/urfave/cli"
)
func main() {
app := cli.NewApp()
app.Flags = []cli.Flag {
cli.StringFlag{
Name: "lang, l",
Value: "english",
Usage: "language for the greeting",
EnvVar: "APP_LANG",
},
}
app.Run(os.Args)
}