🎨 配置优化

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"
type mode int
type mode string
var (
Dev mode = 0
Prod mode = 1
const (
Dev mode = "development"
Prod mode = "production"
)
// LogConfig 日志配置
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
FileEnable bool `env:"LOG_FILE_ENABLE"` // 是否输出到文件
LokiHost string `env:"LOG_LOKI_HOST"` // Loki地址

View File

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