lxh/logger
1
0
Fork 0
Code Issues Pull Requests Projects Releases Wiki Activity

🎨 配置优化

Browse Source
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
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

22
config.go
View File

@ -2,22 +2,22 @@ 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地址
LokiPort int `env:"LOG_LOKI_PORT"` // Loki端口 LokiPort int `env:"LOG_LOKI_PORT"` // Loki端口
LokiSource string `env:"LOG_LOKI_SOURCE_NAME"` // Loki的source名称 LokiSource string `env:"LOG_LOKI_SOURCE_NAME"` // Loki的source名称
LokiJob string `env:"LOG_LOKI_JOB_NAME"` // Loki的job名称 LokiJob string `env:"LOG_LOKI_JOB_NAME"` // Loki的job名称
} }
func (c LogConfig) getLokiPushURL() string { func (c LogConfig) getLokiPushURL() string {

4
logger.go
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)
} }
} }