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 设置生产环境模式
func SetProd() {
prod = true
func SetProd(isProd bool) {
if prod == isProd {
return
}
prod = isProd
if logger != nil {
_ = logger.Sync()
}
@ -156,8 +159,11 @@ func SetLogDir(dir string) {
}
// SetWriteFileWithDev 设置开发环境下写入文件
func SetWriteFileWithDev() {
logDevWrite = true
func SetWriteFileWithDev(isWrite bool) {
if isWrite == logDevWrite {
return
}
logDevWrite = isWrite
if logger != nil {
_ = logger.Sync()
}