chore(logging): 更新日志重定向与捕获功能
- 在 .gitignore 中添加调试日志文件的忽略规则,避免不必要的调试信息被提交 - 修改 application.go 中的 stdout 重定向逻辑,使用 log.CaptureStream 以支持更灵活的日志捕获 - 更新 README 文档,增加对日志重定向功能的说明
This commit is contained in:
+86
-4
@@ -18,6 +18,27 @@ import (
|
||||
"github.com/rs/zerolog"
|
||||
)
|
||||
|
||||
const maxCaptureLine = 10 << 20 // 10MB,支持 GetReqMap 等大 body 单行日志
|
||||
|
||||
var (
|
||||
origStderr io.Writer
|
||||
origStderrOnce sync.Once
|
||||
)
|
||||
|
||||
func saveOrigStderr() {
|
||||
origStderrOnce.Do(func() {
|
||||
origStderr = os.Stderr
|
||||
})
|
||||
}
|
||||
|
||||
func seqWriteStderr(format string, args ...interface{}) {
|
||||
if origStderr != nil {
|
||||
fmt.Fprintf(origStderr, format, args...)
|
||||
} else {
|
||||
fmt.Fprintf(os.Stderr, format, args...)
|
||||
}
|
||||
}
|
||||
|
||||
// ErrorRecord 错误历史记录条目
|
||||
type ErrorRecord struct {
|
||||
Err error
|
||||
@@ -43,6 +64,7 @@ type Logger struct {
|
||||
// logFile: 文件路径模板(如 "logs/20060102.txt"),空则不写文件
|
||||
// maxErrors: 错误历史最大条数,0 则不记录
|
||||
func NewLogger(logLevel int, logFile string, maxErrors int) *Logger {
|
||||
saveOrigStderr()
|
||||
zerolog.CallerMarshalFunc = callerMarshalFunc
|
||||
zerolog.TimeFieldFormat = "2006-01-02 15:04:05"
|
||||
|
||||
@@ -97,6 +119,7 @@ func NewLogger(logLevel int, logFile string, maxErrors int) *Logger {
|
||||
|
||||
// NewLoggerNoCaller 创建不带自动 Caller 字段的日志实例(用于访问日志等 caller 无意义的场景)
|
||||
func NewLoggerNoCaller(logLevel int, logFile string, maxErrors int) *Logger {
|
||||
saveOrigStderr()
|
||||
zerolog.CallerMarshalFunc = callerMarshalFunc
|
||||
zerolog.TimeFieldFormat = "2006-01-02 15:04:05"
|
||||
|
||||
@@ -181,6 +204,62 @@ func (l *Logger) GetLevel() int {
|
||||
return l.logLevel
|
||||
}
|
||||
|
||||
// EmitCaptured 将捕获的 stdout/stderr 等输出直写 multiWriter,绕过主 Logger 的 level 过滤。
|
||||
// 保证 logLevel=0 时 fmt.Println / log.Println 仍能进 Seq。
|
||||
func (l *Logger) EmitCaptured(level zerolog.Level, source, msg string) {
|
||||
if l == nil || l.mw == nil || msg == "" {
|
||||
return
|
||||
}
|
||||
zl := zerolog.New(l.mw)
|
||||
zl.WithLevel(level).Timestamp().Str("source", source).Msg(msg)
|
||||
}
|
||||
|
||||
// EmitRecoveredPanic 记录 recover 到的 panic:错误原因 + 调用栈,source=panic。
|
||||
func (l *Logger) EmitRecoveredPanic(err interface{}) {
|
||||
if l == nil || l.mw == nil || err == nil {
|
||||
return
|
||||
}
|
||||
zl := zerolog.New(l.mw)
|
||||
zl.WithLevel(zerolog.ErrorLevel).Timestamp().
|
||||
Str("source", "panic").
|
||||
Str("stack", string(debugStack())).
|
||||
Msg(fmt.Sprint(err))
|
||||
}
|
||||
|
||||
func debugStack() []byte {
|
||||
buf := make([]byte, 64*1024)
|
||||
n := runtime.Stack(buf, false)
|
||||
return buf[:n]
|
||||
}
|
||||
|
||||
// CaptureStream 从 reader 持续读取行并写入 Logger(用于 stdout/stderr 管道)。
|
||||
// 单行超过 maxCaptureLine 时截断并标注,读取出错后重启读取,避免永久失效。
|
||||
func CaptureStream(l *Logger, level zerolog.Level, source string, r io.Reader) {
|
||||
rd := bufio.NewReader(r)
|
||||
for {
|
||||
line, err := rd.ReadString('\n')
|
||||
if len(line) > 0 {
|
||||
msg := strings.TrimRight(line, "\r\n")
|
||||
if msg != "" {
|
||||
if len(msg) > maxCaptureLine {
|
||||
msg = msg[:maxCaptureLine] + "...(truncated)"
|
||||
}
|
||||
l.EmitCaptured(level, source, msg)
|
||||
}
|
||||
}
|
||||
if err != nil {
|
||||
if err == io.EOF {
|
||||
if len(line) == 0 {
|
||||
return
|
||||
}
|
||||
continue
|
||||
}
|
||||
l.EmitCaptured(zerolog.ErrorLevel, source+"-capture", err.Error())
|
||||
return
|
||||
}
|
||||
}
|
||||
}
|
||||
|
||||
// --- 链式调用 API ---
|
||||
// 以下方法兼容两种调用风格:
|
||||
// 无参数:返回 *zerolog.Event 用于链式调用,如 l.Error().Str("k","v").Msg("...")
|
||||
@@ -501,7 +580,10 @@ func (w *SeqWriter) Write(p []byte) (int, error) {
|
||||
select {
|
||||
case w.queue <- clef:
|
||||
default:
|
||||
atomic.AddInt64(&w.dropped, 1)
|
||||
n := atomic.AddInt64(&w.dropped, 1)
|
||||
if n == 1 || n%100 == 0 {
|
||||
seqWriteStderr("[seq] queue full, dropped=%d\n", n)
|
||||
}
|
||||
}
|
||||
return len(p), nil
|
||||
}
|
||||
@@ -600,7 +682,7 @@ func (w *SeqWriter) flush(batch [][]byte) {
|
||||
}
|
||||
req, err := http.NewRequest("POST", w.seqUrl+"/api/events/raw?clef", &buf)
|
||||
if err != nil {
|
||||
fmt.Fprintf(os.Stderr, "[seq] build request error: %v\n", err)
|
||||
seqWriteStderr("[seq] build request error: %v\n", err)
|
||||
return
|
||||
}
|
||||
req.Header.Set("Content-Type", "application/vnd.serilog.clef")
|
||||
@@ -609,13 +691,13 @@ func (w *SeqWriter) flush(batch [][]byte) {
|
||||
}
|
||||
resp, err := w.client.Do(req)
|
||||
if err != nil {
|
||||
fmt.Fprintf(os.Stderr, "[seq] send error: %v (dropped=%d)\n", err, atomic.LoadInt64(&w.dropped))
|
||||
seqWriteStderr("[seq] send error: %v (dropped=%d)\n", err, atomic.LoadInt64(&w.dropped))
|
||||
return
|
||||
}
|
||||
defer resp.Body.Close()
|
||||
if resp.StatusCode >= 400 {
|
||||
body, _ := io.ReadAll(resp.Body)
|
||||
fmt.Fprintf(os.Stderr, "[seq] server returned %d: %s\n", resp.StatusCode, string(body))
|
||||
seqWriteStderr("[seq] server returned %d: %s\n", resp.StatusCode, string(body))
|
||||
}
|
||||
}
|
||||
|
||||
|
||||
Reference in New Issue
Block a user