mirror of https://github.com/status-im/go-waku.git
60 lines
1.2 KiB
Go
60 lines
1.2 KiB
Go
package utils
|
|
|
|
import (
|
|
"strings"
|
|
|
|
logging "github.com/ipfs/go-log/v2"
|
|
|
|
"go.uber.org/zap"
|
|
)
|
|
|
|
var log *zap.Logger = nil
|
|
|
|
// Logger creates a zap.Logger with some reasonable defaults
|
|
func Logger() *zap.Logger {
|
|
if log == nil {
|
|
InitLogger("console", "stdout")
|
|
}
|
|
return log
|
|
}
|
|
|
|
// InitLogger initializes a global logger using an specific encoding
|
|
func InitLogger(encoding string, output string) {
|
|
cfg := logging.GetConfig()
|
|
|
|
if encoding == "json" {
|
|
cfg.Format = logging.JSONOutput
|
|
} else if encoding == "nocolor" {
|
|
cfg.Format = logging.PlaintextOutput
|
|
} else {
|
|
cfg.Format = logging.ColorizedOutput
|
|
}
|
|
|
|
if output == "stdout" || output == "" {
|
|
cfg.Stdout = true
|
|
cfg.Stderr = false
|
|
} else {
|
|
if encoding == "console" {
|
|
cfg.Format = logging.PlaintextOutput
|
|
}
|
|
cfg.Stdout = false
|
|
cfg.Stderr = false
|
|
|
|
outputParts := strings.Split(output, ":")
|
|
if len(outputParts) == 2 {
|
|
cfg.File = outputParts[1]
|
|
} else {
|
|
if len(outputParts) > 2 || outputParts[0] != "file" {
|
|
panic("invalid output format")
|
|
}
|
|
cfg.File = "./waku.log"
|
|
}
|
|
}
|
|
|
|
logging.SetupLogging(cfg)
|
|
|
|
log = logging.Logger("gowaku").Desugar()
|
|
|
|
logging.SetAllLoggers(logging.LevelInfo)
|
|
}
|