d669c78c43
There are very few cases, if any, in which a user only wants one of these levels, so combine it into a single level. While we're at it, reduce indirection on the loggers by using an empty function rather than a nil function pointer. It's not like we have retpolines anyway, and we were always calling through a function with a branch prior, so this seems like a net gain. Signed-off-by: Jason A. Donenfeld <Jason@zx2c4.com>
49 lines
1.3 KiB
Go
49 lines
1.3 KiB
Go
/* SPDX-License-Identifier: MIT
|
|
*
|
|
* Copyright (C) 2017-2020 WireGuard LLC. All Rights Reserved.
|
|
*/
|
|
|
|
package device
|
|
|
|
import (
|
|
"log"
|
|
"os"
|
|
)
|
|
|
|
// A Logger provides logging for a Device.
|
|
// The functions are Printf-style functions.
|
|
// They must be safe for concurrent use.
|
|
// They do not require a trailing newline in the format.
|
|
// If nil, that level of logging will be silent.
|
|
type Logger struct {
|
|
Verbosef func(format string, args ...interface{})
|
|
Errorf func(format string, args ...interface{})
|
|
}
|
|
|
|
// Log levels for use with NewLogger.
|
|
const (
|
|
LogLevelSilent = iota
|
|
LogLevelError
|
|
LogLevelVerbose
|
|
)
|
|
|
|
// Function for use in Logger for discarding logged lines.
|
|
func DiscardLogf(format string, args ...interface{}) {}
|
|
|
|
// NewLogger constructs a Logger that writes to stdout.
|
|
// It logs at the specified log level and above.
|
|
// It decorates log lines with the log level, date, time, and prepend.
|
|
func NewLogger(level int, prepend string) *Logger {
|
|
logger := &Logger{DiscardLogf, DiscardLogf}
|
|
logf := func(prefix string) func(string, ...interface{}) {
|
|
return log.New(os.Stdout, prefix+": "+prepend, log.Ldate|log.Ltime).Printf
|
|
}
|
|
if level >= LogLevelVerbose {
|
|
logger.Verbosef = logf("DEBUG")
|
|
}
|
|
if level >= LogLevelError {
|
|
logger.Errorf = logf("ERROR")
|
|
}
|
|
return logger
|
|
}
|