-
Notifications
You must be signed in to change notification settings - Fork 0
/
config.go
56 lines (45 loc) · 1.24 KB
/
config.go
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
// Package config содержит конфигурацию клиента.
package config
import (
"fmt"
"github.com/ilyakaznacheev/cleanenv"
flag "github.com/spf13/pflag"
)
type (
// Config основная конфигурация.
Config struct {
App `yaml:"app"`
Storage `yaml:"storage"`
GRPC `yaml:"grpc"`
}
// App информация о приложении.
App struct {
Name string `yaml:"name" env:"APP_NAME"`
Version string `yaml:"version" env:"APP_VERSION"`
}
// Storage настройки файлового хранилища.
Storage struct {
Path string `env-required:"true" yaml:"path" env:"STORAGE_PATH"`
}
// GRPC настройки gRPC.
GRPC struct {
Address string `env-required:"true" yaml:"address" env:"GRPC_ADDRESS"`
Port string `env-required:"true" yaml:"port" env:"GRPC_PORT"`
}
)
// New создаёт объект Config.
func New() (*Config, error) {
cfg := &Config{}
var cfgFile string
flag.StringVarP(&cfgFile, "config", "c", "", "path to config file (*.yaml)")
flag.Parse()
err := cleanenv.ReadConfig(cfgFile, cfg)
if err != nil {
return nil, fmt.Errorf("config error: %w", err)
}
err = cleanenv.ReadEnv(cfg)
if err != nil {
return nil, err
}
return cfg, nil
}