mirror of
https://github.com/NVIDIA/nvidia-container-toolkit
synced 2025-06-26 18:18:24 +00:00
Define a basic logger interface
Signed-off-by: Evan Lezar <elezar@nvidia.com>
This commit is contained in:
@@ -26,20 +26,21 @@ import (
|
||||
"strconv"
|
||||
"strings"
|
||||
|
||||
"github.com/NVIDIA/nvidia-container-toolkit/internal/logger"
|
||||
"github.com/sirupsen/logrus"
|
||||
)
|
||||
|
||||
// Logger adds a way to manage output to a log file to a logrus.Logger
|
||||
type Logger struct {
|
||||
*logrus.Logger
|
||||
previousLogger *logrus.Logger
|
||||
logger.Interface
|
||||
previousLogger logger.Interface
|
||||
logFiles []*os.File
|
||||
}
|
||||
|
||||
// NewLogger creates an empty logger
|
||||
func NewLogger() *Logger {
|
||||
return &Logger{
|
||||
Logger: logrus.New(),
|
||||
Interface: logrus.New(),
|
||||
}
|
||||
}
|
||||
|
||||
@@ -116,8 +117,8 @@ func (l *Logger) Update(filename string, logLevel string, argv []string) {
|
||||
}
|
||||
|
||||
*l = Logger{
|
||||
Logger: newLogger,
|
||||
previousLogger: l.Logger,
|
||||
Interface: newLogger,
|
||||
previousLogger: l.Interface,
|
||||
logFiles: logFiles,
|
||||
}
|
||||
}
|
||||
@@ -130,7 +131,7 @@ func (l *Logger) Reset() error {
|
||||
if previous == nil {
|
||||
previous = logrus.New()
|
||||
}
|
||||
l.Logger = previous
|
||||
l.Interface = previous
|
||||
l.previousLogger = nil
|
||||
l.logFiles = nil
|
||||
}()
|
||||
|
||||
Reference in New Issue
Block a user