mirror of
				https://github.com/k3s-io/kubernetes.git
				synced 2025-10-31 13:50:01 +00:00 
			
		
		
		
	
		
			
				
	
	
		
			324 lines
		
	
	
		
			7.9 KiB
		
	
	
	
		
			Go
		
	
	
	
	
	
			
		
		
	
	
			324 lines
		
	
	
		
			7.9 KiB
		
	
	
	
		
			Go
		
	
	
	
	
	
| package logrus
 | |
| 
 | |
| import (
 | |
| 	"io"
 | |
| 	"os"
 | |
| 	"sync"
 | |
| 	"sync/atomic"
 | |
| )
 | |
| 
 | |
| type Logger struct {
 | |
| 	// The logs are `io.Copy`'d to this in a mutex. It's common to set this to a
 | |
| 	// file, or leave it default which is `os.Stderr`. You can also set this to
 | |
| 	// something more adventorous, such as logging to Kafka.
 | |
| 	Out io.Writer
 | |
| 	// Hooks for the logger instance. These allow firing events based on logging
 | |
| 	// levels and log entries. For example, to send errors to an error tracking
 | |
| 	// service, log to StatsD or dump the core on fatal errors.
 | |
| 	Hooks LevelHooks
 | |
| 	// All log entries pass through the formatter before logged to Out. The
 | |
| 	// included formatters are `TextFormatter` and `JSONFormatter` for which
 | |
| 	// TextFormatter is the default. In development (when a TTY is attached) it
 | |
| 	// logs with colors, but to a file it wouldn't. You can easily implement your
 | |
| 	// own that implements the `Formatter` interface, see the `README` or included
 | |
| 	// formatters for examples.
 | |
| 	Formatter Formatter
 | |
| 	// The logging level the logger should log at. This is typically (and defaults
 | |
| 	// to) `logrus.Info`, which allows Info(), Warn(), Error() and Fatal() to be
 | |
| 	// logged.
 | |
| 	Level Level
 | |
| 	// Used to sync writing to the log. Locking is enabled by Default
 | |
| 	mu MutexWrap
 | |
| 	// Reusable empty entry
 | |
| 	entryPool sync.Pool
 | |
| }
 | |
| 
 | |
| type MutexWrap struct {
 | |
| 	lock     sync.Mutex
 | |
| 	disabled bool
 | |
| }
 | |
| 
 | |
| func (mw *MutexWrap) Lock() {
 | |
| 	if !mw.disabled {
 | |
| 		mw.lock.Lock()
 | |
| 	}
 | |
| }
 | |
| 
 | |
| func (mw *MutexWrap) Unlock() {
 | |
| 	if !mw.disabled {
 | |
| 		mw.lock.Unlock()
 | |
| 	}
 | |
| }
 | |
| 
 | |
| func (mw *MutexWrap) Disable() {
 | |
| 	mw.disabled = true
 | |
| }
 | |
| 
 | |
| // Creates a new logger. Configuration should be set by changing `Formatter`,
 | |
| // `Out` and `Hooks` directly on the default logger instance. You can also just
 | |
| // instantiate your own:
 | |
| //
 | |
| //    var log = &Logger{
 | |
| //      Out: os.Stderr,
 | |
| //      Formatter: new(JSONFormatter),
 | |
| //      Hooks: make(LevelHooks),
 | |
| //      Level: logrus.DebugLevel,
 | |
| //    }
 | |
| //
 | |
| // It's recommended to make this a global instance called `log`.
 | |
| func New() *Logger {
 | |
| 	return &Logger{
 | |
| 		Out:       os.Stderr,
 | |
| 		Formatter: new(TextFormatter),
 | |
| 		Hooks:     make(LevelHooks),
 | |
| 		Level:     InfoLevel,
 | |
| 	}
 | |
| }
 | |
| 
 | |
| func (logger *Logger) newEntry() *Entry {
 | |
| 	entry, ok := logger.entryPool.Get().(*Entry)
 | |
| 	if ok {
 | |
| 		return entry
 | |
| 	}
 | |
| 	return NewEntry(logger)
 | |
| }
 | |
| 
 | |
| func (logger *Logger) releaseEntry(entry *Entry) {
 | |
| 	logger.entryPool.Put(entry)
 | |
| }
 | |
| 
 | |
| // Adds a field to the log entry, note that it doesn't log until you call
 | |
| // Debug, Print, Info, Warn, Fatal or Panic. It only creates a log entry.
 | |
| // If you want multiple fields, use `WithFields`.
 | |
| func (logger *Logger) WithField(key string, value interface{}) *Entry {
 | |
| 	entry := logger.newEntry()
 | |
| 	defer logger.releaseEntry(entry)
 | |
| 	return entry.WithField(key, value)
 | |
| }
 | |
| 
 | |
| // Adds a struct of fields to the log entry. All it does is call `WithField` for
 | |
| // each `Field`.
 | |
| func (logger *Logger) WithFields(fields Fields) *Entry {
 | |
| 	entry := logger.newEntry()
 | |
| 	defer logger.releaseEntry(entry)
 | |
| 	return entry.WithFields(fields)
 | |
| }
 | |
| 
 | |
| // Add an error as single field to the log entry.  All it does is call
 | |
| // `WithError` for the given `error`.
 | |
| func (logger *Logger) WithError(err error) *Entry {
 | |
| 	entry := logger.newEntry()
 | |
| 	defer logger.releaseEntry(entry)
 | |
| 	return entry.WithError(err)
 | |
| }
 | |
| 
 | |
| func (logger *Logger) Debugf(format string, args ...interface{}) {
 | |
| 	if logger.level() >= DebugLevel {
 | |
| 		entry := logger.newEntry()
 | |
| 		entry.Debugf(format, args...)
 | |
| 		logger.releaseEntry(entry)
 | |
| 	}
 | |
| }
 | |
| 
 | |
| func (logger *Logger) Infof(format string, args ...interface{}) {
 | |
| 	if logger.level() >= InfoLevel {
 | |
| 		entry := logger.newEntry()
 | |
| 		entry.Infof(format, args...)
 | |
| 		logger.releaseEntry(entry)
 | |
| 	}
 | |
| }
 | |
| 
 | |
| func (logger *Logger) Printf(format string, args ...interface{}) {
 | |
| 	entry := logger.newEntry()
 | |
| 	entry.Printf(format, args...)
 | |
| 	logger.releaseEntry(entry)
 | |
| }
 | |
| 
 | |
| func (logger *Logger) Warnf(format string, args ...interface{}) {
 | |
| 	if logger.level() >= WarnLevel {
 | |
| 		entry := logger.newEntry()
 | |
| 		entry.Warnf(format, args...)
 | |
| 		logger.releaseEntry(entry)
 | |
| 	}
 | |
| }
 | |
| 
 | |
| func (logger *Logger) Warningf(format string, args ...interface{}) {
 | |
| 	if logger.level() >= WarnLevel {
 | |
| 		entry := logger.newEntry()
 | |
| 		entry.Warnf(format, args...)
 | |
| 		logger.releaseEntry(entry)
 | |
| 	}
 | |
| }
 | |
| 
 | |
| func (logger *Logger) Errorf(format string, args ...interface{}) {
 | |
| 	if logger.level() >= ErrorLevel {
 | |
| 		entry := logger.newEntry()
 | |
| 		entry.Errorf(format, args...)
 | |
| 		logger.releaseEntry(entry)
 | |
| 	}
 | |
| }
 | |
| 
 | |
| func (logger *Logger) Fatalf(format string, args ...interface{}) {
 | |
| 	if logger.level() >= FatalLevel {
 | |
| 		entry := logger.newEntry()
 | |
| 		entry.Fatalf(format, args...)
 | |
| 		logger.releaseEntry(entry)
 | |
| 	}
 | |
| 	Exit(1)
 | |
| }
 | |
| 
 | |
| func (logger *Logger) Panicf(format string, args ...interface{}) {
 | |
| 	if logger.level() >= PanicLevel {
 | |
| 		entry := logger.newEntry()
 | |
| 		entry.Panicf(format, args...)
 | |
| 		logger.releaseEntry(entry)
 | |
| 	}
 | |
| }
 | |
| 
 | |
| func (logger *Logger) Debug(args ...interface{}) {
 | |
| 	if logger.level() >= DebugLevel {
 | |
| 		entry := logger.newEntry()
 | |
| 		entry.Debug(args...)
 | |
| 		logger.releaseEntry(entry)
 | |
| 	}
 | |
| }
 | |
| 
 | |
| func (logger *Logger) Info(args ...interface{}) {
 | |
| 	if logger.level() >= InfoLevel {
 | |
| 		entry := logger.newEntry()
 | |
| 		entry.Info(args...)
 | |
| 		logger.releaseEntry(entry)
 | |
| 	}
 | |
| }
 | |
| 
 | |
| func (logger *Logger) Print(args ...interface{}) {
 | |
| 	entry := logger.newEntry()
 | |
| 	entry.Info(args...)
 | |
| 	logger.releaseEntry(entry)
 | |
| }
 | |
| 
 | |
| func (logger *Logger) Warn(args ...interface{}) {
 | |
| 	if logger.level() >= WarnLevel {
 | |
| 		entry := logger.newEntry()
 | |
| 		entry.Warn(args...)
 | |
| 		logger.releaseEntry(entry)
 | |
| 	}
 | |
| }
 | |
| 
 | |
| func (logger *Logger) Warning(args ...interface{}) {
 | |
| 	if logger.level() >= WarnLevel {
 | |
| 		entry := logger.newEntry()
 | |
| 		entry.Warn(args...)
 | |
| 		logger.releaseEntry(entry)
 | |
| 	}
 | |
| }
 | |
| 
 | |
| func (logger *Logger) Error(args ...interface{}) {
 | |
| 	if logger.level() >= ErrorLevel {
 | |
| 		entry := logger.newEntry()
 | |
| 		entry.Error(args...)
 | |
| 		logger.releaseEntry(entry)
 | |
| 	}
 | |
| }
 | |
| 
 | |
| func (logger *Logger) Fatal(args ...interface{}) {
 | |
| 	if logger.level() >= FatalLevel {
 | |
| 		entry := logger.newEntry()
 | |
| 		entry.Fatal(args...)
 | |
| 		logger.releaseEntry(entry)
 | |
| 	}
 | |
| 	Exit(1)
 | |
| }
 | |
| 
 | |
| func (logger *Logger) Panic(args ...interface{}) {
 | |
| 	if logger.level() >= PanicLevel {
 | |
| 		entry := logger.newEntry()
 | |
| 		entry.Panic(args...)
 | |
| 		logger.releaseEntry(entry)
 | |
| 	}
 | |
| }
 | |
| 
 | |
| func (logger *Logger) Debugln(args ...interface{}) {
 | |
| 	if logger.level() >= DebugLevel {
 | |
| 		entry := logger.newEntry()
 | |
| 		entry.Debugln(args...)
 | |
| 		logger.releaseEntry(entry)
 | |
| 	}
 | |
| }
 | |
| 
 | |
| func (logger *Logger) Infoln(args ...interface{}) {
 | |
| 	if logger.level() >= InfoLevel {
 | |
| 		entry := logger.newEntry()
 | |
| 		entry.Infoln(args...)
 | |
| 		logger.releaseEntry(entry)
 | |
| 	}
 | |
| }
 | |
| 
 | |
| func (logger *Logger) Println(args ...interface{}) {
 | |
| 	entry := logger.newEntry()
 | |
| 	entry.Println(args...)
 | |
| 	logger.releaseEntry(entry)
 | |
| }
 | |
| 
 | |
| func (logger *Logger) Warnln(args ...interface{}) {
 | |
| 	if logger.level() >= WarnLevel {
 | |
| 		entry := logger.newEntry()
 | |
| 		entry.Warnln(args...)
 | |
| 		logger.releaseEntry(entry)
 | |
| 	}
 | |
| }
 | |
| 
 | |
| func (logger *Logger) Warningln(args ...interface{}) {
 | |
| 	if logger.level() >= WarnLevel {
 | |
| 		entry := logger.newEntry()
 | |
| 		entry.Warnln(args...)
 | |
| 		logger.releaseEntry(entry)
 | |
| 	}
 | |
| }
 | |
| 
 | |
| func (logger *Logger) Errorln(args ...interface{}) {
 | |
| 	if logger.level() >= ErrorLevel {
 | |
| 		entry := logger.newEntry()
 | |
| 		entry.Errorln(args...)
 | |
| 		logger.releaseEntry(entry)
 | |
| 	}
 | |
| }
 | |
| 
 | |
| func (logger *Logger) Fatalln(args ...interface{}) {
 | |
| 	if logger.level() >= FatalLevel {
 | |
| 		entry := logger.newEntry()
 | |
| 		entry.Fatalln(args...)
 | |
| 		logger.releaseEntry(entry)
 | |
| 	}
 | |
| 	Exit(1)
 | |
| }
 | |
| 
 | |
| func (logger *Logger) Panicln(args ...interface{}) {
 | |
| 	if logger.level() >= PanicLevel {
 | |
| 		entry := logger.newEntry()
 | |
| 		entry.Panicln(args...)
 | |
| 		logger.releaseEntry(entry)
 | |
| 	}
 | |
| }
 | |
| 
 | |
| //When file is opened with appending mode, it's safe to
 | |
| //write concurrently to a file (within 4k message on Linux).
 | |
| //In these cases user can choose to disable the lock.
 | |
| func (logger *Logger) SetNoLock() {
 | |
| 	logger.mu.Disable()
 | |
| }
 | |
| 
 | |
| func (logger *Logger) level() Level {
 | |
| 	return Level(atomic.LoadUint32((*uint32)(&logger.Level)))
 | |
| }
 | |
| 
 | |
| func (logger *Logger) SetLevel(level Level) {
 | |
| 	atomic.StoreUint32((*uint32)(&logger.Level), uint32(level))
 | |
| }
 | |
| 
 | |
| func (logger *Logger) AddHook(hook Hook) {
 | |
| 	logger.mu.Lock()
 | |
| 	defer logger.mu.Unlock()
 | |
| 	logger.Hooks.Add(hook)
 | |
| }
 |