refactor: 日志设置生产模式和开发模式写入文件支持开关

This commit is contained in:
kercylan98 2023-07-03 13:35:31 +08:00
parent cc0853a700
commit c6073a97a8
1 changed files with 10 additions and 4 deletions

View File

@ -138,8 +138,11 @@ func ErrorWithStack(msg, stack string, fields ...zap.Field) {
} }
// SetProd 设置生产环境模式 // SetProd 设置生产环境模式
func SetProd() { func SetProd(isProd bool) {
prod = true if prod == isProd {
return
}
prod = isProd
if logger != nil { if logger != nil {
_ = logger.Sync() _ = logger.Sync()
} }
@ -156,8 +159,11 @@ func SetLogDir(dir string) {
} }
// SetWriteFileWithDev 设置开发环境下写入文件 // SetWriteFileWithDev 设置开发环境下写入文件
func SetWriteFileWithDev() { func SetWriteFileWithDev(isWrite bool) {
logDevWrite = true if isWrite == logDevWrite {
return
}
logDevWrite = isWrite
if logger != nil { if logger != nil {
_ = logger.Sync() _ = logger.Sync()
} }