initial commit
This commit is contained in:
255
log.go
Normal file
255
log.go
Normal file
@@ -0,0 +1,255 @@
|
||||
package lol
|
||||
|
||||
import (
|
||||
"fmt"
|
||||
"io"
|
||||
"os"
|
||||
"runtime"
|
||||
"strings"
|
||||
"time"
|
||||
|
||||
"github.com/davecgh/go-spew/spew"
|
||||
"github.com/gookit/color"
|
||||
"go.uber.org/atomic"
|
||||
)
|
||||
|
||||
var l = GetStd()
|
||||
|
||||
func GetStd() (ll *Log) {
|
||||
ll, _ = New(os.Stdout)
|
||||
return
|
||||
}
|
||||
|
||||
func init() {
|
||||
switch strings.ToUpper(os.Getenv("GODEBUG")) {
|
||||
case "1", "TRUE", "ON":
|
||||
SetLogLevel(Debug)
|
||||
l.D.Ln("printing logs at this level and lower")
|
||||
case "INFO":
|
||||
SetLogLevel(Info)
|
||||
case "DEBUG":
|
||||
SetLogLevel(Debug)
|
||||
l.D.Ln("printing logs at this level and lower")
|
||||
case "TRACE":
|
||||
SetLogLevel(Trace)
|
||||
l.T.Ln("printing logs at this level and lower")
|
||||
case "WARN":
|
||||
SetLogLevel(Warn)
|
||||
case "ERROR":
|
||||
SetLogLevel(Error)
|
||||
case "FATAL":
|
||||
SetLogLevel(Fatal)
|
||||
case "0", "OFF", "FALSE":
|
||||
SetLogLevel(Off)
|
||||
default:
|
||||
SetLogLevel(Info)
|
||||
}
|
||||
|
||||
}
|
||||
|
||||
const (
|
||||
Off = iota
|
||||
Fatal
|
||||
Error
|
||||
Warn
|
||||
Info
|
||||
Debug
|
||||
Trace
|
||||
)
|
||||
|
||||
type (
|
||||
// LevelPrinter defines a set of terminal printing primitives that output with
|
||||
// extra data, time, log logLevelList, and code location
|
||||
|
||||
// Ln prints lists of interfaces with spaces in between
|
||||
Ln func(a ...interface{})
|
||||
// F prints like fmt.Println surrounded by log details
|
||||
F func(format string, a ...interface{})
|
||||
// S prints a spew.Sdump for an interface slice
|
||||
S func(a ...interface{})
|
||||
// C accepts a function so that the extra computation can be avoided if it is
|
||||
// not being viewed
|
||||
C func(closure func() string)
|
||||
// Chk is a shortcut for printing if there is an error, or returning true
|
||||
Chk func(e error) bool
|
||||
// Err is a pass-through function that uses fmt.Errorf to construct an error
|
||||
// and returns the error after printing it to the log
|
||||
Err func(format string, a ...interface{}) error
|
||||
LevelPrinter struct {
|
||||
Ln
|
||||
F
|
||||
S
|
||||
C
|
||||
Chk
|
||||
Err
|
||||
}
|
||||
LevelSpec struct {
|
||||
ID int
|
||||
Name string
|
||||
Colorizer func(a ...interface{}) string
|
||||
}
|
||||
|
||||
// Entry is a log entry to be printed as json to the log file
|
||||
Entry struct {
|
||||
Time time.Time
|
||||
Level string
|
||||
Package string
|
||||
CodeLocation string
|
||||
Text string
|
||||
}
|
||||
)
|
||||
|
||||
var (
|
||||
// sep is just a convenient shortcut for this very longwinded expression
|
||||
sep = string(os.PathSeparator)
|
||||
currentLevel = atomic.NewInt32(Info)
|
||||
// writer can be swapped out for any io.*writer* that you want to use instead of
|
||||
// stdout.
|
||||
writer io.Writer = os.Stderr
|
||||
// LevelSpecs specifies the id, string name and color-printing function
|
||||
LevelSpecs = []LevelSpec{
|
||||
{Off, " ", color.Bit24(0, 0, 0, false).Sprint},
|
||||
{Fatal, "FTL", color.Bit24(128, 0, 0, false).Sprint},
|
||||
{Error, "ERR", color.Bit24(255, 0, 0, false).Sprint},
|
||||
{Warn, "WRN", color.Bit24(0, 255, 0, false).Sprint},
|
||||
{Info, "INF", color.Bit24(255, 255, 0, false).Sprint},
|
||||
{Debug, "DBG", color.Bit24(0, 125, 255, false).Sprint},
|
||||
{Trace, "TRC", color.Bit24(125, 0, 255, false).Sprint},
|
||||
}
|
||||
)
|
||||
|
||||
// Log is a set of log printers for the various Level items.
|
||||
type Log struct {
|
||||
F, E, W, I, D, T LevelPrinter
|
||||
}
|
||||
|
||||
type Check struct {
|
||||
F, E, W, I, D, T Chk
|
||||
}
|
||||
|
||||
func JoinStrings(a ...any) (s string) {
|
||||
for i := range a {
|
||||
s += fmt.Sprint(a[i])
|
||||
if i < len(a)-1 {
|
||||
s += " "
|
||||
}
|
||||
}
|
||||
return
|
||||
}
|
||||
|
||||
func GetPrinter(l int32, writer io.Writer) LevelPrinter {
|
||||
return LevelPrinter{
|
||||
Ln: func(a ...interface{}) {
|
||||
fmt.Fprintf(writer,
|
||||
"%s %s %s %s\n",
|
||||
UnixNanoAsFloat(),
|
||||
LevelSpecs[l].Colorizer(LevelSpecs[l].Name),
|
||||
JoinStrings(a...),
|
||||
GetLoc(2),
|
||||
)
|
||||
},
|
||||
F: func(format string, a ...interface{}) {
|
||||
fmt.Fprintf(writer,
|
||||
"%s %s %s %s\n",
|
||||
UnixNanoAsFloat(),
|
||||
LevelSpecs[l].Colorizer(LevelSpecs[l].Name),
|
||||
fmt.Sprintf(format, a...),
|
||||
GetLoc(2),
|
||||
)
|
||||
},
|
||||
S: func(a ...interface{}) {
|
||||
fmt.Fprintf(writer,
|
||||
"%s %s %s %s\n",
|
||||
UnixNanoAsFloat(),
|
||||
LevelSpecs[l].Colorizer(LevelSpecs[l].Name),
|
||||
spew.Sdump(a...),
|
||||
GetLoc(2),
|
||||
)
|
||||
},
|
||||
C: func(closure func() string) {
|
||||
fmt.Fprintf(writer,
|
||||
"%s %s %s %s\n",
|
||||
UnixNanoAsFloat(),
|
||||
LevelSpecs[l].Colorizer(LevelSpecs[l].Name),
|
||||
closure(),
|
||||
GetLoc(2),
|
||||
)
|
||||
},
|
||||
Chk: func(e error) bool {
|
||||
if e != nil {
|
||||
fmt.Fprintf(writer,
|
||||
"%s %s %s %s\n",
|
||||
UnixNanoAsFloat(),
|
||||
LevelSpecs[l].Colorizer(LevelSpecs[l].Name),
|
||||
e.Error(),
|
||||
GetLoc(2),
|
||||
)
|
||||
return true
|
||||
}
|
||||
return false
|
||||
},
|
||||
Err: func(format string, a ...interface{}) error {
|
||||
fmt.Fprintf(writer,
|
||||
"%s %s %s %s\n",
|
||||
UnixNanoAsFloat(),
|
||||
LevelSpecs[l].Colorizer(LevelSpecs[l].Name, " "),
|
||||
fmt.Sprintf(format, a...),
|
||||
GetLoc(2),
|
||||
)
|
||||
return fmt.Errorf(format, a...)
|
||||
},
|
||||
}
|
||||
}
|
||||
|
||||
func New(writer io.Writer) (l *Log, c *Check) {
|
||||
l = &Log{
|
||||
F: GetPrinter(Fatal, writer),
|
||||
E: GetPrinter(Error, writer),
|
||||
W: GetPrinter(Warn, writer),
|
||||
I: GetPrinter(Info, writer),
|
||||
D: GetPrinter(Debug, writer),
|
||||
T: GetPrinter(Trace, writer),
|
||||
}
|
||||
c = &Check{
|
||||
F: l.F.Chk,
|
||||
E: l.E.Chk,
|
||||
W: l.W.Chk,
|
||||
I: l.I.Chk,
|
||||
D: l.D.Chk,
|
||||
T: l.T.Chk,
|
||||
}
|
||||
return
|
||||
}
|
||||
|
||||
// SetLogLevel sets the log level via a string, which can be truncated down to
|
||||
// one character, similar to nmcli's argument processor, as the first letter is
|
||||
// unique. This could be used with a linter to make larger command sets.
|
||||
func SetLogLevel(l int) {
|
||||
currentLevel.Store(int32(l))
|
||||
}
|
||||
|
||||
func GetLogLevel() (l int) {
|
||||
return int(currentLevel.Load())
|
||||
}
|
||||
|
||||
// UnixNanoAsFloat e
|
||||
func UnixNanoAsFloat() (s string) {
|
||||
timeText := fmt.Sprint(time.Now().UnixNano())
|
||||
lt := len(timeText)
|
||||
lb := lt + 1
|
||||
var timeBytes = make([]byte, lb)
|
||||
copy(timeBytes[lb-9:lb], timeText[lt-9:lt])
|
||||
timeBytes[lb-10] = '.'
|
||||
lb -= 10
|
||||
lt -= 9
|
||||
copy(timeBytes[:lb], timeText[:lt])
|
||||
return color.Bit24(0, 128, 255, false).Sprint(string(timeBytes))
|
||||
}
|
||||
|
||||
func GetLoc(skip int) (output string) {
|
||||
_, file, line, _ := runtime.Caller(skip)
|
||||
output = color.Bit24(0, 128, 255, false).Sprint(
|
||||
file, ":", line,
|
||||
)
|
||||
return
|
||||
}
|
||||
Reference in New Issue
Block a user