如何在Go中使用Zap Logger临时禁用特定函数的日志输出？

方法一：临时替换标准log包的输出目标 #

如果你的代码基于标准库log包，可以直接模仿你给出的fmt方案思路，临时将日志输出重定向到io.Discard（一个安全的空写入器，比设为nil更稳妥，避免panic风险）：

package main

import (
	"io"
	"log"
	"os"
)

func functionToMask() {
	log.Println("这条日志会被屏蔽")
}

func main() {
	// 保存原日志输出目标
	originalWriter := log.Writer()
	// 将日志输出重定向到丢弃设备
	log.SetOutput(io.Discard)
	
	// 调用需要屏蔽日志的函数
	functionToMask()
	
	// 恢复原输出目标
	log.SetOutput(originalWriter)
	
	log.Println("这条日志会正常输出")
}

方法二：自定义带开关的日志工具 #

如果需要更细粒度的控制，比如针对不同函数独立开关日志，可以封装一个简单的日志工具：

全局开关版本 #

package main

import (
	"log"
)

var logEnabled = true

func customLog(msg string) {
	if logEnabled {
		log.Println(msg)
	}
}

func functionToMask() {
	// 临时关闭日志
	logEnabled = false
	customLog("这条日志被屏蔽")
	// 恢复日志状态
	logEnabled = true
}

func main() {
	customLog("正常输出的日志")
	functionToMask()
	customLog("恢复后的正常日志")
}

多函数独立开关版本 #

package main

import (
	"log"
	"sync"
)

var (
	funcLogSwitches = make(map[string]bool)
	mu              sync.RWMutex
)

func setFuncLogEnabled(funcName string, enabled bool) {
	mu.Lock()
	defer mu.Unlock()
	funcLogSwitches[funcName] = enabled
}

func logForFunc(funcName, msg string) {
	mu.RLock()
	enabled := funcLogSwitches[funcName]
	mu.RUnlock()
	if enabled {
		log.Printf("[%s] %s\n", funcName, msg)
	}
}

func functionA() {
	logForFunc("functionA", "functionA的日志")
}

func functionB() {
	logForFunc("functionB", "functionB的日志")
}

func main() {
	// 启用所有函数日志
	setFuncLogEnabled("functionA", true)
	setFuncLogEnabled("functionB", true)
	functionA() // 正常输出
	functionB() // 正常输出

	// 屏蔽functionB的日志
	setFuncLogEnabled("functionB", false)
	functionB() // 无输出
	functionA() // 正常输出
}

方法三：利用第三方日志库的原生控制 #

如果使用logrus或zap这类功能丰富的日志库，可以直接借助它们的原生能力实现：

logrus级别过滤示例 #

package main

import (
	"github.com/sirupsen/logrus"
)

func functionToMask() {
	// 用Debug级别输出日志
	logrus.Debug("这条日志会被屏蔽")
}

func main() {
	// 设置全局日志级别为Info，低于该级别的日志会被自动过滤
	logrus.SetLevel(logrus.InfoLevel)
	
	functionToMask()
	
	logrus.Info("这条日志正常输出")
}

logrus字段过滤示例 #

package main

import (
	"github.com/sirupsen/logrus"
)

func main() {
	// 添加钩子，过滤带有mask: true标记的日志
	logrus.AddHook(&LogMaskHook{})
	
	logrus.WithField("mask", true).Info("这条日志被屏蔽")
	logrus.Info("这条日志正常输出")
}

type LogMaskHook struct{}

func (h *LogMaskHook) Levels() []logrus.Level {
	return logrus.AllLevels
}

func (h *LogMaskHook) Fire(entry *logrus.Entry) error {
	if mask, ok := entry.Data["mask"].(bool); ok && mask {
		// 丢弃该日志，返回nil终止后续写入
		return nil
	}
	return nil
}

内容的提问来源于stack exchange，提问作者Gopi Kumar