|
1 | 1 | package logger |
2 | 2 |
|
3 | 3 | import ( |
| 4 | + "fmt" |
4 | 5 | "os" |
5 | | - "strconv" |
6 | 6 | "strings" |
7 | 7 |
|
| 8 | + "github.com/joho/godotenv" |
| 9 | + "github.com/kelseyhightower/envconfig" |
| 10 | + "github.com/pkg/errors" |
8 | 11 | "go.uber.org/zap" |
9 | 12 | "go.uber.org/zap/zapcore" |
10 | 13 | ) |
11 | 14 |
|
| 15 | +type Env string |
| 16 | + |
| 17 | +type cfg struct { |
| 18 | + Environment Env `envconfig:"ENVIRONMENT" default:"production"` |
| 19 | + LogLevel zapcore.Level `envconfig:"LOG_LEVEL" default:"info"` |
| 20 | +} |
| 21 | + |
| 22 | +const ( |
| 23 | + EnvironmentProd Env = "production" |
| 24 | + EnvironmentDev Env = "development" |
| 25 | +) |
| 26 | + |
| 27 | +func (e *Env) UnmarshalText(text []byte) error { |
| 28 | + switch strings.ToLower(string(text)) { |
| 29 | + case string(EnvironmentProd): |
| 30 | + *e = EnvironmentProd |
| 31 | + case string(EnvironmentDev): |
| 32 | + *e = EnvironmentDev |
| 33 | + default: |
| 34 | + return errors.Errorf("unknown environment type: '%s'", string(text)) |
| 35 | + } |
| 36 | + return nil |
| 37 | +} |
| 38 | + |
12 | 39 | func init() { |
13 | | - // constructs a logger and replaces the default global logger |
| 40 | + godotenv.Load("../.env", ".env") //nolint:errcheck |
| 41 | + |
| 42 | + var c cfg |
| 43 | + envconfig.MustProcess("", &c) |
| 44 | + |
14 | 45 | var config zap.Config |
15 | | - if d, e := strconv.ParseBool(os.Getenv("DEVELOPMENT")); d && e == nil || isInTests() { |
| 46 | + if c.Environment == EnvironmentDev || isInTests() { |
16 | 47 | config = zap.NewDevelopmentConfig() |
17 | 48 | } else { |
18 | 49 | config = zap.NewProductionConfig() |
19 | 50 | } |
| 51 | + |
| 52 | + config.Level.SetLevel(c.LogLevel) |
20 | 53 | config.EncoderConfig.EncodeTime = zapcore.ISO8601TimeEncoder |
21 | | - config.DisableStacktrace = true |
22 | | - if d, e := strconv.ParseBool(os.Getenv("DEBUG")); d && e == nil || isInTests() { |
23 | | - config.Level = zap.NewAtomicLevelAt(zap.DebugLevel) |
24 | | - } |
| 54 | + |
25 | 55 | logger, err := config.Build() |
26 | 56 | if err != nil { |
27 | | - panic(err) |
| 57 | + fmt.Println("Error during logging config:", err) |
| 58 | + os.Exit(1) |
28 | 59 | } |
29 | 60 | zap.ReplaceGlobals(logger) |
| 61 | + |
| 62 | + zap.L().Info("logger configured", |
| 63 | + zap.String("level", c.LogLevel.String()), |
| 64 | + zap.String("env", string(c.Environment))) |
| 65 | + |
30 | 66 | } |
31 | 67 |
|
32 | 68 | func isInTests() bool { |
|
0 commit comments