mirror of
https://github.com/siderolabs/talos.git
synced 2025-10-18 02:51:15 +02:00
96f89071c3
This PR enables debug logging of controller-runtime to the server console if the machine configuration .debug field is set to true (log verbosity can be also changed on the fly). For control plane nodes, don't send kubelet logs to the console (as they tend to flood the console with messages), this leaves reasonable amount of logging for early boot troubleshooting. Signed-off-by: Andrey Smirnov <smirnov.andrey@gmail.com>
142 lines
3.2 KiB
Go
142 lines
3.2 KiB
Go
// 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 http://mozilla.org/MPL/2.0/.
|
|
|
|
package logging
|
|
|
|
import (
|
|
"io"
|
|
"log"
|
|
"strings"
|
|
|
|
"go.uber.org/zap"
|
|
"go.uber.org/zap/zapcore"
|
|
)
|
|
|
|
// LogWriter is a wrapper around zap.Logger that implements io.Writer interface.
|
|
type LogWriter struct {
|
|
dest *zap.Logger
|
|
level zapcore.Level
|
|
}
|
|
|
|
// NewWriter creates new log zap log writer.
|
|
func NewWriter(l *zap.Logger, level zapcore.Level) io.Writer {
|
|
return &LogWriter{
|
|
dest: l,
|
|
level: level,
|
|
}
|
|
}
|
|
|
|
// Write implements io.Writer interface.
|
|
func (lw *LogWriter) Write(line []byte) (int, error) {
|
|
checked := lw.dest.Check(lw.level, strings.TrimSpace(string(line)))
|
|
if checked == nil {
|
|
return 0, nil
|
|
}
|
|
|
|
checked.Write()
|
|
|
|
return len(line), nil
|
|
}
|
|
|
|
// LogWrapper wraps around standard logger.
|
|
type LogWrapper struct {
|
|
log *log.Logger
|
|
}
|
|
|
|
// Write implements io.Writer interface.
|
|
func (lw *LogWrapper) Write(line []byte) (int, error) {
|
|
if lw.log == nil {
|
|
log.Print(string(line))
|
|
} else {
|
|
lw.log.Print(string(line))
|
|
}
|
|
|
|
return len(line), nil
|
|
}
|
|
|
|
// StdWriter creates a sync writer that writes all logs to the std logger.
|
|
var StdWriter = &LogWrapper{nil}
|
|
|
|
// LogDestination defines logging destination Config.
|
|
type LogDestination struct {
|
|
// Level log level.
|
|
Level zapcore.LevelEnabler
|
|
writer io.Writer
|
|
config zapcore.EncoderConfig
|
|
}
|
|
|
|
// EncoderOption defines a log destination encoder config setter.
|
|
type EncoderOption func(config *zapcore.EncoderConfig)
|
|
|
|
// WithoutTimestamp disables timestamp.
|
|
func WithoutTimestamp() EncoderOption {
|
|
return func(config *zapcore.EncoderConfig) {
|
|
config.EncodeTime = nil
|
|
}
|
|
}
|
|
|
|
// WithoutLogLevels disable log level.
|
|
func WithoutLogLevels() EncoderOption {
|
|
return func(config *zapcore.EncoderConfig) {
|
|
config.EncodeLevel = nil
|
|
}
|
|
}
|
|
|
|
// WithColoredLevels enables log level colored output.
|
|
func WithColoredLevels() EncoderOption {
|
|
return func(config *zapcore.EncoderConfig) {
|
|
config.EncodeLevel = zapcore.CapitalColorLevelEncoder
|
|
}
|
|
}
|
|
|
|
// NewLogDestination creates new log destination.
|
|
func NewLogDestination(writer io.Writer, logLevel zapcore.LevelEnabler, options ...EncoderOption) *LogDestination {
|
|
config := zap.NewDevelopmentEncoderConfig()
|
|
config.ConsoleSeparator = " "
|
|
config.StacktraceKey = "error"
|
|
|
|
for _, option := range options {
|
|
option(&config)
|
|
}
|
|
|
|
return &LogDestination{
|
|
Level: logLevel,
|
|
config: config,
|
|
writer: writer,
|
|
}
|
|
}
|
|
|
|
// Wrap is a simple helper to wrap io.Writer with default arguments.
|
|
func Wrap(writer io.Writer) *zap.Logger {
|
|
return ZapLogger(
|
|
NewLogDestination(writer, zapcore.DebugLevel),
|
|
)
|
|
}
|
|
|
|
// ZapLogger creates new default Zap Logger.
|
|
func ZapLogger(dests ...*LogDestination) *zap.Logger {
|
|
if len(dests) == 0 {
|
|
panic("at least one writer must be defined")
|
|
}
|
|
|
|
cores := []zapcore.Core{}
|
|
|
|
for _, dest := range dests {
|
|
consoleEncoder := zapcore.NewConsoleEncoder(dest.config)
|
|
|
|
cores = append(cores, zapcore.NewCore(consoleEncoder, zapcore.AddSync(dest.writer), dest.Level))
|
|
}
|
|
|
|
core := zapcore.NewTee(cores...)
|
|
|
|
logger := zap.New(core)
|
|
|
|
return logger
|
|
}
|
|
|
|
// Component helper for creating zap.Field.
|
|
func Component(name string) zapcore.Field {
|
|
return zap.String("component", name)
|
|
}
|