/
config.go
40 lines (32 loc) · 877 Bytes
/
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
package config
import (
"os"
"github.com/kelseyhightower/envconfig"
"go.uber.org/zap"
"gopkg.in/yaml.v2"
)
type Config struct {
Port int `yaml:"port" envconfig:"PORT"`
DBconfig DBconfig `yaml:"db_config" envconfig:"DB_CONFIG"`
Host string `yaml:"host" envconfig:"HOST"`
}
type DBconfig struct {
DBurl string `yaml:"db_url" envconfig:"DB_URL"`
}
func (c *Config) ReadFromFile(logger *zap.SugaredLogger) error {
configPath := "./configs/app.yaml"
data, err := os.ReadFile(configPath)
if err != nil {
logger.Errorf("incorrect config file: %s", err)
}
if err = yaml.Unmarshal(data, c); err != nil {
logger.Errorf("unnable to unmarshal file: %s", err)
}
return nil
}
func (c *Config) REadFromEnv(logger *zap.SugaredLogger) {
err := envconfig.Process("", &c)
if err != nil {
logger.Fatalf("failed to read config from env: %v", err)
}
}