🎨 配置优化

This commit is contained in:
李寻欢 2022-05-18 11:17:22 +08:00
parent 52e9be31c3
commit 3ea919e06c
2 changed files with 15 additions and 11 deletions

View File

@ -2,16 +2,16 @@ package logger
import "fmt" import "fmt"
type mode int type mode string
var ( const (
Dev mode = 0 Dev mode = "development"
Prod mode = 1 Prod mode = "production"
) )
// LogConfig 日志配置 // LogConfig 日志配置
type LogConfig struct { type LogConfig struct {
Mode mode `env:"LOG_MODE"` // dev, prod Mode mode `env:"LOG_MODE" envDefault:"production"` // dev, prod
LokiEnable bool `env:"LOG_LOKI_ENABLE"` // 是否启用Loki LokiEnable bool `env:"LOG_LOKI_ENABLE"` // 是否启用Loki
FileEnable bool `env:"LOG_FILE_ENABLE"` // 是否输出到文件 FileEnable bool `env:"LOG_FILE_ENABLE"` // 是否输出到文件
LokiHost string `env:"LOG_LOKI_HOST"` // Loki地址 LokiHost string `env:"LOG_LOKI_HOST"` // Loki地址

View File

@ -19,6 +19,10 @@ func init() {
fmt.Println("日志配置解析错误: " + err.Error()) fmt.Println("日志配置解析错误: " + err.Error())
c = LogConfig{Mode: Dev, LokiEnable: false, FileEnable: false} c = LogConfig{Mode: Dev, LokiEnable: false, FileEnable: false}
} }
// 如果值错了直接默认为Prod
if c.Mode != Dev && c.Mode != Prod {
c.Mode = Prod
}
InitLogger(c) InitLogger(c)
} }
} }