95 lines
2.2 KiB
Go
95 lines
2.2 KiB
Go
package params
|
|
|
|
import (
|
|
"os"
|
|
"path/filepath"
|
|
"sync"
|
|
|
|
"github.com/ethereum/go-ethereum/log"
|
|
)
|
|
|
|
// Logger is wrapper for custom logging
|
|
type Logger struct {
|
|
origHandler log.Handler
|
|
handler log.Handler
|
|
config *NodeConfig
|
|
}
|
|
|
|
var (
|
|
onceInitNodeLogger sync.Once
|
|
nodeLoggerInstance *Logger
|
|
)
|
|
|
|
// SetupLogger configs logger using parameters in config
|
|
func SetupLogger(config *NodeConfig) (*Logger, error) {
|
|
if !config.LogEnabled {
|
|
return nil, nil
|
|
}
|
|
|
|
onceInitNodeLogger.Do(func() {
|
|
nodeLoggerInstance = &Logger{
|
|
config: config,
|
|
origHandler: log.Root().GetHandler(),
|
|
}
|
|
nodeLoggerInstance.handler = nodeLoggerInstance.makeLogHandler(parseLogLevel(config.LogLevel))
|
|
})
|
|
|
|
if err := nodeLoggerInstance.Start(); err != nil {
|
|
return nil, err
|
|
}
|
|
|
|
return nodeLoggerInstance, nil
|
|
}
|
|
|
|
// SetV allows to dynamically change log level of messages being written
|
|
func (l *Logger) SetV(logLevel string) {
|
|
log.Root().SetHandler(l.makeLogHandler(parseLogLevel(logLevel)))
|
|
}
|
|
|
|
// Start installs logger handler
|
|
func (l *Logger) Start() error {
|
|
log.Root().SetHandler(l.handler)
|
|
return nil
|
|
}
|
|
|
|
// Stop replaces our handler back to the original log handler
|
|
func (l *Logger) Stop() error {
|
|
log.Root().SetHandler(l.origHandler)
|
|
return nil
|
|
}
|
|
|
|
// makeLogHandler creates a log handler for a given level and node configuration
|
|
func (l *Logger) makeLogHandler(lvl log.Lvl) log.Handler {
|
|
var handler log.Handler
|
|
logFilePath := filepath.Join(l.config.DataDir, l.config.LogFile)
|
|
fileHandler := log.Must.FileHandler(logFilePath, log.LogfmtFormat())
|
|
stderrHandler := log.StreamHandler(os.Stderr, log.TerminalFormat(true))
|
|
if l.config.LogToStderr {
|
|
handler = log.MultiHandler(
|
|
log.LvlFilterHandler(lvl, log.CallerFileHandler(log.CallerFuncHandler(stderrHandler))),
|
|
log.LvlFilterHandler(lvl, fileHandler))
|
|
} else {
|
|
handler = log.LvlFilterHandler(lvl, fileHandler)
|
|
}
|
|
|
|
return handler
|
|
}
|
|
|
|
// parseLogLevel parses string and returns logger.* constant
|
|
func parseLogLevel(logLevel string) log.Lvl {
|
|
switch logLevel {
|
|
case "ERROR":
|
|
return log.LvlError
|
|
case "WARNING":
|
|
return log.LvlWarn
|
|
case "INFO":
|
|
return log.LvlInfo
|
|
case "DEBUG":
|
|
return log.LvlDebug
|
|
case "TRACE":
|
|
return log.LvlTrace
|
|
}
|
|
|
|
return log.LvlInfo
|
|
}
|