forked from jshiffer/matterbridge
86 lines
1.9 KiB
Go
86 lines
1.9 KiB
Go
|
package logger
|
||
|
|
|
||
|
|
import (
|
||
|
|
"fmt"
|
||
|
|
"strings"
|
||
|
|
"time"
|
||
|
|
)
|
||
|
|
|
||
|
|
// DefaultLogger is used if no logger has been set up.
|
||
|
|
type defaultLogger struct {
|
||
|
|
namespaces []string
|
||
|
|
}
|
||
|
|
|
||
|
|
// log simply logs the given message to stdout if the message
|
||
|
|
// caller is allowed to log.
|
||
|
|
func (d *defaultLogger) log(level, caller, msg string) {
|
||
|
|
if !d.shouldLog(caller) {
|
||
|
|
// return
|
||
|
|
}
|
||
|
|
t := time.Now()
|
||
|
|
fmt.Println(
|
||
|
|
"["+level+"]",
|
||
|
|
t.Format(time.RFC3339),
|
||
|
|
caller,
|
||
|
|
"▶ ",
|
||
|
|
msg,
|
||
|
|
)
|
||
|
|
}
|
||
|
|
|
||
|
|
// shouldLog determines whether or not the given caller should
|
||
|
|
// be allowed to log messages.
|
||
|
|
func (d *defaultLogger) shouldLog(caller string) bool {
|
||
|
|
shouldLog := false
|
||
|
|
d.ensureNamespaces()
|
||
|
|
for _, namespace := range d.namespaces {
|
||
|
|
if namespace == "all" {
|
||
|
|
shouldLog = true
|
||
|
|
}
|
||
|
|
if strings.Contains(caller, namespace) {
|
||
|
|
shouldLog = true
|
||
|
|
}
|
||
|
|
}
|
||
|
|
|
||
|
|
return shouldLog
|
||
|
|
}
|
||
|
|
|
||
|
|
// ensureNamespaces checks to see if our list of loggable namespaces
|
||
|
|
// has been initialized or not. If not, it defaults to log all.
|
||
|
|
func (d *defaultLogger) ensureNamespaces() {
|
||
|
|
if d.namespaces == nil {
|
||
|
|
d.namespaces = []string{"all"}
|
||
|
|
}
|
||
|
|
}
|
||
|
|
|
||
|
|
// Debug is used to log debug messages.
|
||
|
|
func (d *defaultLogger) Debug(caller, msg string) {
|
||
|
|
//d.log("DEBUG", caller, msg)
|
||
|
|
}
|
||
|
|
|
||
|
|
// Info is used to log info messages.
|
||
|
|
func (d *defaultLogger) Info(caller, msg string) {
|
||
|
|
d.log("INFO", caller, msg)
|
||
|
|
}
|
||
|
|
|
||
|
|
// Warning is used to log warning messages.
|
||
|
|
func (d *defaultLogger) Warning(caller, msg string) {
|
||
|
|
d.log("WARNING", caller, msg)
|
||
|
|
}
|
||
|
|
|
||
|
|
// Error is used to log error messages.
|
||
|
|
func (d *defaultLogger) Error(caller, msg string) {
|
||
|
|
d.log("ERROR", caller, msg)
|
||
|
|
}
|
||
|
|
|
||
|
|
// Configure takes a configuration string separated by commas
|
||
|
|
// that contains all the callers that should be logged. This
|
||
|
|
// allows granular logging of different go files.
|
||
|
|
//
|
||
|
|
// Example:
|
||
|
|
// logger.Configure("RootKey.go,Curve.go")
|
||
|
|
// logger.Configure("all")
|
||
|
|
//
|
||
|
|
func (d *defaultLogger) Configure(settings string) {
|
||
|
|
d.namespaces = strings.Split(settings, ",")
|
||
|
|
}
|