Ross Smith II
3058aa6315
### Purpose Re-enables LOGGER_DISCARD. See #10262. ### Documentation No changes needed, as the docs already mention this variable.
59 lines
1.4 KiB
Go
59 lines
1.4 KiB
Go
// Copyright (C) 2025 The Syncthing Authors.
|
|
//
|
|
// This Source Code Form is subject to the terms of the Mozilla Public
|
|
// License, v. 2.0. If a copy of the MPL was not distributed with this file,
|
|
// You can obtain one at https://mozilla.org/MPL/2.0/.
|
|
|
|
package slogutil
|
|
|
|
import (
|
|
"io"
|
|
"log/slog"
|
|
"os"
|
|
"strings"
|
|
)
|
|
|
|
var (
|
|
GlobalRecorder = &lineRecorder{level: -1000}
|
|
ErrorRecorder = &lineRecorder{level: slog.LevelError}
|
|
globalLevels = &levelTracker{
|
|
levels: make(map[string]slog.Level),
|
|
descrs: make(map[string]string),
|
|
}
|
|
slogDef = slog.New(&formattingHandler{
|
|
recs: []*lineRecorder{GlobalRecorder, ErrorRecorder},
|
|
out: logWriter(),
|
|
})
|
|
)
|
|
|
|
func logWriter() io.Writer {
|
|
if os.Getenv("LOGGER_DISCARD") != "" {
|
|
// Hack to completely disable logging, for example when running
|
|
// benchmarks.
|
|
return io.Discard
|
|
}
|
|
|
|
return os.Stdout
|
|
}
|
|
|
|
func init() {
|
|
slog.SetDefault(slogDef)
|
|
|
|
// Handle legacy STTRACE var
|
|
pkgs := strings.Split(os.Getenv("STTRACE"), ",")
|
|
for _, pkg := range pkgs {
|
|
pkg = strings.TrimSpace(pkg)
|
|
if pkg == "" {
|
|
continue
|
|
}
|
|
level := slog.LevelDebug
|
|
if cutPkg, levelStr, ok := strings.Cut(pkg, ":"); ok {
|
|
pkg = cutPkg
|
|
if err := level.UnmarshalText([]byte(levelStr)); err != nil {
|
|
slog.Warn("Bad log level requested in STTRACE", slog.String("pkg", pkg), slog.String("level", levelStr), Error(err))
|
|
}
|
|
}
|
|
globalLevels.Set(pkg, level)
|
|
}
|
|
}
|