v1.0.0
This commit is contained in:
3
.gitignore
vendored
3
.gitignore
vendored
@@ -1,2 +1,3 @@
|
|||||||
.idea/
|
.idea/
|
||||||
test/
|
test/
|
||||||
|
*.log
|
||||||
21
examples/main.go
Normal file
21
examples/main.go
Normal file
@@ -0,0 +1,21 @@
|
|||||||
|
package main
|
||||||
|
|
||||||
|
import "git.nix13.pw/scuroneko/slog"
|
||||||
|
|
||||||
|
func main() {
|
||||||
|
logger := slog.CreateLogger().JsonPretty(false)
|
||||||
|
defer logger.Close()
|
||||||
|
logger.AddWriter(logger.CreateTextStdoutWriter())
|
||||||
|
f, err := logger.CreateJsonFileWriter("main.log")
|
||||||
|
if err != nil {
|
||||||
|
panic(err)
|
||||||
|
}
|
||||||
|
logger.AddWriter(f)
|
||||||
|
logger.Infoln("Test")
|
||||||
|
logger.Warnln("Test")
|
||||||
|
logger.Errorln("Test")
|
||||||
|
logger.Debugln("Test") // No output
|
||||||
|
logger.Level(slog.DEBUG)
|
||||||
|
logger.Debugln("Test") // Now we have output
|
||||||
|
logger.Fatalln("Test") // Exit code 1
|
||||||
|
}
|
||||||
8
go.sum
8
go.sum
@@ -1,13 +1,9 @@
|
|||||||
github.com/fatih/color v1.18.0 h1:S8gINlzdQ840/4pfAwic/ZE0djQEH3wM94VfqLTZcOM=
|
github.com/fatih/color v1.18.0 h1:S8gINlzdQ840/4pfAwic/ZE0djQEH3wM94VfqLTZcOM=
|
||||||
github.com/fatih/color v1.18.0/go.mod h1:4FelSpRwEGDpQ12mAdzqdOukCy4u8WUtOY6lkT/6HfU=
|
github.com/fatih/color v1.18.0/go.mod h1:4FelSpRwEGDpQ12mAdzqdOukCy4u8WUtOY6lkT/6HfU=
|
||||||
github.com/mattn/go-colorable v0.1.13 h1:fFA4WZxdEF4tXPZVKMLwD8oUnCTTo08duU7wxecdEvA=
|
github.com/mattn/go-colorable v0.1.14 h1:9A9LHSqF/7dyVVX6g0U9cwm9pG3kP9gSzcuIPHPsaIE=
|
||||||
github.com/mattn/go-colorable v0.1.13/go.mod h1:7S9/ev0klgBDR4GtXTXX8a3vIGJpMovkB8vQcUbaXHg=
|
|
||||||
github.com/mattn/go-colorable v0.1.14/go.mod h1:6LmQG8QLFO4G5z1gPvYEzlUgJ2wF+stgPZH1UqBm1s8=
|
github.com/mattn/go-colorable v0.1.14/go.mod h1:6LmQG8QLFO4G5z1gPvYEzlUgJ2wF+stgPZH1UqBm1s8=
|
||||||
github.com/mattn/go-isatty v0.0.16/go.mod h1:kYGgaQfpe5nmfYZH+SKPsOc2e4SrIfOl2e/yFXSvRLM=
|
|
||||||
github.com/mattn/go-isatty v0.0.20 h1:xfD0iDuEKnDkl03q4limB+vH+GxLEtL/jb4xVJSWWEY=
|
github.com/mattn/go-isatty v0.0.20 h1:xfD0iDuEKnDkl03q4limB+vH+GxLEtL/jb4xVJSWWEY=
|
||||||
github.com/mattn/go-isatty v0.0.20/go.mod h1:W+V8PltTTMOvKvAeJH7IuucS94S2C6jfK/D7dTCTo3Y=
|
github.com/mattn/go-isatty v0.0.20/go.mod h1:W+V8PltTTMOvKvAeJH7IuucS94S2C6jfK/D7dTCTo3Y=
|
||||||
golang.org/x/sys v0.0.0-20220811171246-fbc7d0a398ab/go.mod h1:oPkhp1MJrh7nUepCBck5+mAzfO9JrbApNNgaTdGDITg=
|
|
||||||
golang.org/x/sys v0.6.0/go.mod h1:oPkhp1MJrh7nUepCBck5+mAzfO9JrbApNNgaTdGDITg=
|
golang.org/x/sys v0.6.0/go.mod h1:oPkhp1MJrh7nUepCBck5+mAzfO9JrbApNNgaTdGDITg=
|
||||||
golang.org/x/sys v0.25.0 h1:r+8e+loiHxRqhXVl6ML1nO3l1+oFoWbnlu2Ehimmi34=
|
golang.org/x/sys v0.40.0 h1:DBZZqJ2Rkml6QMQsZywtnjnnGvHza6BTfYFWY9kjEWQ=
|
||||||
golang.org/x/sys v0.25.0/go.mod h1:/VUhepiaJMQUp4+oa/7Zr1D23ma6VTLIYjOOTFZPUcA=
|
|
||||||
golang.org/x/sys v0.40.0/go.mod h1:OgkHotnGiDImocRcuBABYBEXf8A9a87e/uXjp9XT3ks=
|
golang.org/x/sys v0.40.0/go.mod h1:OgkHotnGiDImocRcuBABYBEXf8A9a87e/uXjp9XT3ks=
|
||||||
|
|||||||
91
io.go
Normal file
91
io.go
Normal file
@@ -0,0 +1,91 @@
|
|||||||
|
package slog
|
||||||
|
|
||||||
|
import (
|
||||||
|
"fmt"
|
||||||
|
"os"
|
||||||
|
)
|
||||||
|
|
||||||
|
func (l *Logger) Infof(format string, args ...any) {
|
||||||
|
l.print(INFO, fmt.Sprintf(format, args...))
|
||||||
|
}
|
||||||
|
func (l *Logger) Info(m ...any) {
|
||||||
|
l.print(INFO, m...)
|
||||||
|
}
|
||||||
|
func (l *Logger) Infoln(m ...any) {
|
||||||
|
l.println(INFO, m...)
|
||||||
|
}
|
||||||
|
|
||||||
|
func (l *Logger) Warnf(format string, args ...any) {
|
||||||
|
l.print(WARN, fmt.Sprintf(format, args...))
|
||||||
|
}
|
||||||
|
func (l *Logger) Warn(m ...any) {
|
||||||
|
l.print(WARN, m...)
|
||||||
|
}
|
||||||
|
func (l *Logger) Warnln(m ...any) {
|
||||||
|
l.println(WARN, m...)
|
||||||
|
}
|
||||||
|
|
||||||
|
func (l *Logger) Errorf(format string, args ...any) {
|
||||||
|
l.print(ERROR, fmt.Sprintf(format, args...))
|
||||||
|
}
|
||||||
|
func (l *Logger) Error(m ...any) {
|
||||||
|
l.print(ERROR, m...)
|
||||||
|
}
|
||||||
|
func (l *Logger) Errorln(m ...any) {
|
||||||
|
l.println(ERROR, m...)
|
||||||
|
}
|
||||||
|
|
||||||
|
func (l *Logger) Fatalf(format string, args ...any) {
|
||||||
|
l.print(FATAL, fmt.Sprintf(format, args...))
|
||||||
|
os.Exit(1)
|
||||||
|
}
|
||||||
|
func (l *Logger) Fatal(m ...any) {
|
||||||
|
l.print(FATAL, m...)
|
||||||
|
os.Exit(1)
|
||||||
|
}
|
||||||
|
func (l *Logger) Fatalln(m ...any) {
|
||||||
|
l.println(FATAL, m...)
|
||||||
|
os.Exit(1)
|
||||||
|
}
|
||||||
|
|
||||||
|
func (l *Logger) Debugf(format string, args ...any) {
|
||||||
|
l.print(DEBUG, fmt.Sprintf(format, args...))
|
||||||
|
}
|
||||||
|
func (l *Logger) Debug(m ...any) {
|
||||||
|
l.print(DEBUG, m...)
|
||||||
|
}
|
||||||
|
func (l *Logger) Debugln(m ...any) {
|
||||||
|
l.println(DEBUG, m...)
|
||||||
|
}
|
||||||
|
|
||||||
|
// Write message without trailing "\n"
|
||||||
|
// Good for database
|
||||||
|
func (l *Logger) print(level LogLevel, m ...any) {
|
||||||
|
if l.level.n < level.n {
|
||||||
|
return
|
||||||
|
}
|
||||||
|
|
||||||
|
tb := getFullTraceback(0)
|
||||||
|
for _, writer := range l.writers {
|
||||||
|
err := writer.Print(level, l.prefix, tb, m...)
|
||||||
|
if err != nil {
|
||||||
|
l.Error(err)
|
||||||
|
}
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
// Docker requires "\n" at end to write to log.
|
||||||
|
// print not work for docker, otherwise it will work and write into stdout
|
||||||
|
func (l *Logger) println(level LogLevel, m ...any) {
|
||||||
|
if l.level.n < level.n {
|
||||||
|
return
|
||||||
|
}
|
||||||
|
|
||||||
|
tb := getFullTraceback(0)
|
||||||
|
for _, writer := range l.writers {
|
||||||
|
err := writer.Print(level, l.prefix, tb, append(m, "\n")...)
|
||||||
|
if err != nil {
|
||||||
|
l.Error(err)
|
||||||
|
}
|
||||||
|
}
|
||||||
|
}
|
||||||
300
logger.go
300
logger.go
@@ -1,105 +1,22 @@
|
|||||||
package laniakea
|
package slog
|
||||||
|
|
||||||
import (
|
import (
|
||||||
"bufio"
|
|
||||||
"encoding/json"
|
|
||||||
"fmt"
|
"fmt"
|
||||||
"io"
|
"io"
|
||||||
"os"
|
|
||||||
"path/filepath"
|
|
||||||
"runtime"
|
|
||||||
"sort"
|
|
||||||
"strings"
|
"strings"
|
||||||
"time"
|
"time"
|
||||||
|
|
||||||
"github.com/fatih/color"
|
"github.com/fatih/color"
|
||||||
)
|
)
|
||||||
|
|
||||||
type LoggerWriter interface {
|
|
||||||
Close() error
|
|
||||||
Write(p []byte) (n int, err error)
|
|
||||||
Print(level LogLevel, prefix string, traceback []*MethodTraceback, messages ...any) error
|
|
||||||
}
|
|
||||||
type LoggerTextWriter struct {
|
|
||||||
LoggerWriter
|
|
||||||
writer io.Writer
|
|
||||||
printTraceback bool
|
|
||||||
printTime bool
|
|
||||||
}
|
|
||||||
|
|
||||||
func (w *LoggerTextWriter) Write(p []byte) (n int, err error) {
|
|
||||||
n, err = w.writer.Write(p)
|
|
||||||
if err != nil {
|
|
||||||
return n, err
|
|
||||||
}
|
|
||||||
err = bufio.NewWriter(w.writer).Flush()
|
|
||||||
return n, err
|
|
||||||
}
|
|
||||||
func (w *LoggerTextWriter) Print(level LogLevel, prefix string, tb []*MethodTraceback, messages ...any) error {
|
|
||||||
s := buildString(level, prefix, true, w.printTraceback, w.printTime, messages...)
|
|
||||||
_, err := w.Write([]byte(s))
|
|
||||||
return err
|
|
||||||
}
|
|
||||||
func (w *LoggerTextWriter) Close() error {
|
|
||||||
return w.writer.(io.Closer).Close()
|
|
||||||
}
|
|
||||||
|
|
||||||
type LoggerJsonWriter struct {
|
|
||||||
LoggerWriter
|
|
||||||
writer io.Writer
|
|
||||||
pretty bool
|
|
||||||
}
|
|
||||||
type LoggerJsonMessage struct {
|
|
||||||
Time time.Time `json:"time"`
|
|
||||||
Level string `json:"level"`
|
|
||||||
Prefix string `json:"prefix"`
|
|
||||||
Message string `json:"message"`
|
|
||||||
Traceback []*MethodTraceback `json:"traceback"`
|
|
||||||
}
|
|
||||||
|
|
||||||
func (w *LoggerJsonWriter) Write(data []byte) (int, error) {
|
|
||||||
n, err := w.writer.Write(data)
|
|
||||||
if err != nil {
|
|
||||||
return n, err
|
|
||||||
}
|
|
||||||
err = bufio.NewWriter(w.writer).Flush()
|
|
||||||
return n, err
|
|
||||||
}
|
|
||||||
func (w *LoggerJsonWriter) Print(level LogLevel, prefix string, traceback []*MethodTraceback, messages ...any) error {
|
|
||||||
msg := Map(messages, func(el any) string {
|
|
||||||
return fmt.Sprintf("%v", el)
|
|
||||||
})
|
|
||||||
m := LoggerJsonMessage{
|
|
||||||
Time: time.Now(),
|
|
||||||
Level: level.GetName(),
|
|
||||||
Prefix: prefix,
|
|
||||||
Message: strings.Join(msg, " "),
|
|
||||||
Traceback: traceback,
|
|
||||||
}
|
|
||||||
var data []byte
|
|
||||||
var err error
|
|
||||||
if w.pretty {
|
|
||||||
data, err = json.MarshalIndent(m, "", " ")
|
|
||||||
} else {
|
|
||||||
data, err = json.Marshal(m)
|
|
||||||
}
|
|
||||||
if err != nil {
|
|
||||||
return err
|
|
||||||
}
|
|
||||||
_, err = w.Write(append(data, []byte("\n")...))
|
|
||||||
return err
|
|
||||||
}
|
|
||||||
func (w *LoggerJsonWriter) Close() error {
|
|
||||||
return w.writer.(io.Closer).Close()
|
|
||||||
}
|
|
||||||
|
|
||||||
type Logger struct {
|
type Logger struct {
|
||||||
prefix string
|
prefix string
|
||||||
level LogLevel
|
level LogLevel
|
||||||
|
writers []LoggerWriter
|
||||||
|
|
||||||
printTraceback bool
|
printTraceback bool
|
||||||
printTime bool
|
printTime bool
|
||||||
jsonPretty bool
|
jsonPretty bool
|
||||||
writers []LoggerWriter
|
|
||||||
}
|
}
|
||||||
|
|
||||||
type LogLevel struct {
|
type LogLevel struct {
|
||||||
@@ -113,12 +30,11 @@ func (l *LogLevel) GetName() string {
|
|||||||
}
|
}
|
||||||
|
|
||||||
type MethodTraceback struct {
|
type MethodTraceback struct {
|
||||||
Package string `json:"package"`
|
|
||||||
Method string `json:"method"`
|
Method string `json:"method"`
|
||||||
fullPath string
|
|
||||||
signature string
|
|
||||||
Filename string `json:"filename"`
|
Filename string `json:"filename"`
|
||||||
Line int `json:"line"`
|
Line int `json:"line"`
|
||||||
|
Signature string `json:"signature"`
|
||||||
|
FullPath string `json:"fullPath"`
|
||||||
}
|
}
|
||||||
|
|
||||||
var (
|
var (
|
||||||
@@ -138,60 +54,6 @@ func CreateLogger() *Logger {
|
|||||||
}
|
}
|
||||||
}
|
}
|
||||||
|
|
||||||
func (l *Logger) CreateTextFileWriter(path string) (*LoggerTextWriter, error) {
|
|
||||||
err := os.MkdirAll(filepath.Dir(path), os.ModePerm)
|
|
||||||
if err != nil {
|
|
||||||
return nil, err
|
|
||||||
}
|
|
||||||
file, err := os.OpenFile(path, os.O_APPEND|os.O_CREATE|os.O_WRONLY, 0644)
|
|
||||||
if err != nil {
|
|
||||||
return nil, err
|
|
||||||
}
|
|
||||||
|
|
||||||
writer := &LoggerTextWriter{
|
|
||||||
writer: file, printTraceback: l.printTraceback, printTime: l.printTime,
|
|
||||||
}
|
|
||||||
return writer, nil
|
|
||||||
}
|
|
||||||
func (l *Logger) CreateTextStdoutWriter() *LoggerTextWriter {
|
|
||||||
writer := &LoggerTextWriter{
|
|
||||||
writer: os.Stdout, printTraceback: l.printTraceback, printTime: l.printTime,
|
|
||||||
}
|
|
||||||
return writer
|
|
||||||
}
|
|
||||||
func (l *Logger) CreateJsonStdoutWriter() *LoggerJsonWriter {
|
|
||||||
writer := &LoggerJsonWriter{
|
|
||||||
writer: os.Stdout, pretty: l.jsonPretty,
|
|
||||||
}
|
|
||||||
return writer
|
|
||||||
}
|
|
||||||
func (l *Logger) CreateJsonFileWriter(path string) (*LoggerJsonWriter, error) {
|
|
||||||
err := os.MkdirAll(filepath.Dir(path), os.ModePerm)
|
|
||||||
if err != nil {
|
|
||||||
return nil, err
|
|
||||||
}
|
|
||||||
file, err := os.OpenFile(path, os.O_APPEND|os.O_CREATE|os.O_WRONLY, 0644)
|
|
||||||
if err != nil {
|
|
||||||
return nil, err
|
|
||||||
}
|
|
||||||
writer := &LoggerJsonWriter{
|
|
||||||
writer: file, pretty: l.jsonPretty,
|
|
||||||
}
|
|
||||||
return writer, nil
|
|
||||||
}
|
|
||||||
func (l *Logger) CreateTextWriter(w io.Writer) *LoggerTextWriter {
|
|
||||||
writer := &LoggerTextWriter{
|
|
||||||
writer: w, printTraceback: l.printTraceback, printTime: l.printTime,
|
|
||||||
}
|
|
||||||
return writer
|
|
||||||
}
|
|
||||||
func (l *Logger) CreateJsonWriter(w io.Writer) *LoggerJsonWriter {
|
|
||||||
writer := &LoggerJsonWriter{
|
|
||||||
writer: w, pretty: l.jsonPretty,
|
|
||||||
}
|
|
||||||
return writer
|
|
||||||
}
|
|
||||||
|
|
||||||
func (l *Logger) Prefix(prefix string) *Logger {
|
func (l *Logger) Prefix(prefix string) *Logger {
|
||||||
l.prefix = prefix
|
l.prefix = prefix
|
||||||
return l
|
return l
|
||||||
@@ -220,153 +82,65 @@ func (l *Logger) AddWriter(writer LoggerWriter) *Logger {
|
|||||||
l.writers = append(l.writers, writer)
|
l.writers = append(l.writers, writer)
|
||||||
return l
|
return l
|
||||||
}
|
}
|
||||||
|
func (l *Logger) Close() error {
|
||||||
func (l *Logger) Infof(format string, args ...any) {
|
for _, writer := range l.writers {
|
||||||
l.print(INFO, fmt.Sprintf(format, args...))
|
err := writer.Close()
|
||||||
}
|
if err != nil {
|
||||||
func (l *Logger) Info(m ...any) {
|
l.Error(err)
|
||||||
l.println(INFO, m...)
|
}
|
||||||
|
}
|
||||||
|
return nil
|
||||||
}
|
}
|
||||||
|
|
||||||
func (l *Logger) Warnf(format string, args ...any) {
|
func (l *Logger) CreateTextWriter(w io.Writer) *LoggerTextWriter {
|
||||||
l.print(WARN, fmt.Sprintf(format, args...))
|
return CreateTextWriter(w, l.printTraceback, l.printTime)
|
||||||
}
|
}
|
||||||
func (l *Logger) Warn(m ...any) {
|
func (l *Logger) CreateTextStdoutWriter() *LoggerTextWriter {
|
||||||
l.println(WARN, m...)
|
return CreateTextStdoutWriter(l.printTraceback, l.printTime)
|
||||||
|
}
|
||||||
|
func (l *Logger) CreateTextFileWriter(filename string) (*LoggerTextWriter, error) {
|
||||||
|
return CreateTextFileWriter(filename, l.printTraceback, l.printTime)
|
||||||
|
}
|
||||||
|
func (l *Logger) CreateJsonWriter(w io.Writer) *LoggerJsonWriter {
|
||||||
|
return CreateJsonWriter(w, l.jsonPretty)
|
||||||
|
}
|
||||||
|
func (l *Logger) CreateJsonStdoutWriter() *LoggerJsonWriter {
|
||||||
|
return CreateJsonStdoutWriter(l.jsonPretty)
|
||||||
|
}
|
||||||
|
func (l *Logger) CreateJsonFileWriter(filename string) (*LoggerJsonWriter, error) {
|
||||||
|
return CreateJsonFileWriter(filename, l.jsonPretty)
|
||||||
}
|
}
|
||||||
|
|
||||||
func (l *Logger) Error(m ...any) {
|
func FormatTime(t time.Time) string {
|
||||||
l.println(ERROR, m...)
|
|
||||||
}
|
|
||||||
|
|
||||||
func (l *Logger) Fatal(m ...any) {
|
|
||||||
l.println(FATAL, m...)
|
|
||||||
os.Exit(1)
|
|
||||||
}
|
|
||||||
|
|
||||||
func (l *Logger) Debug(m ...any) {
|
|
||||||
l.println(DEBUG, m...)
|
|
||||||
}
|
|
||||||
|
|
||||||
func formatTime(t time.Time) string {
|
|
||||||
return fmt.Sprintf("%02d.%02d.%02d %02d:%02d:%02d", t.Day(), t.Month(), t.Year(), t.Hour(), t.Minute(), t.Second())
|
return fmt.Sprintf("%02d.%02d.%02d %02d:%02d:%02d", t.Day(), t.Month(), t.Year(), t.Hour(), t.Minute(), t.Second())
|
||||||
}
|
}
|
||||||
func formatTraceback(mt *MethodTraceback) string {
|
func FormatTraceback(mt *MethodTraceback) string {
|
||||||
return fmt.Sprintf("%s:%s:%d", mt.Filename, mt.Method, mt.Line)
|
return fmt.Sprintf("%s:%s:%d", mt.Filename, mt.Method, mt.Line)
|
||||||
}
|
}
|
||||||
func FormatFullTraceback(tracebacks []*MethodTraceback) string {
|
func FormatFullTraceback(tracebacks []*MethodTraceback) string {
|
||||||
formatted := make([]string, 0)
|
formatted := make([]string, 0)
|
||||||
for _, tb := range tracebacks {
|
for _, tb := range tracebacks {
|
||||||
formatted = append(formatted, formatTraceback(tb))
|
formatted = append(formatted, FormatTraceback(tb))
|
||||||
}
|
}
|
||||||
return strings.Join(formatted, "->")
|
return strings.Join(formatted, "->")
|
||||||
}
|
}
|
||||||
|
func BuildString(level LogLevel, prefix string, printTime, printTraceback bool, m ...any) string {
|
||||||
func getTraceback() *MethodTraceback {
|
|
||||||
caller, _, _, _ := runtime.Caller(4)
|
|
||||||
details := runtime.FuncForPC(caller)
|
|
||||||
signature := details.Name()
|
|
||||||
path, line := details.FileLine(caller)
|
|
||||||
splitPath := strings.Split(path, "/")
|
|
||||||
|
|
||||||
splitSignature := strings.Split(signature, ".")
|
|
||||||
pkg, method := splitSignature[0], splitSignature[len(splitSignature)-1]
|
|
||||||
|
|
||||||
tb := &MethodTraceback{
|
|
||||||
Filename: splitPath[len(splitPath)-1],
|
|
||||||
fullPath: path,
|
|
||||||
Line: line,
|
|
||||||
signature: signature,
|
|
||||||
Package: pkg,
|
|
||||||
Method: method,
|
|
||||||
}
|
|
||||||
|
|
||||||
return tb
|
|
||||||
}
|
|
||||||
func getFullTraceback(skip int) []*MethodTraceback {
|
|
||||||
pc := make([]uintptr, 15)
|
|
||||||
runtime.Callers(skip, pc)
|
|
||||||
list := make([]*MethodTraceback, 0)
|
|
||||||
frames := runtime.CallersFrames(pc)
|
|
||||||
for {
|
|
||||||
frame, more := frames.Next()
|
|
||||||
if !more {
|
|
||||||
break
|
|
||||||
}
|
|
||||||
details := runtime.FuncForPC(frame.PC)
|
|
||||||
signature := details.Name()
|
|
||||||
path, line := details.FileLine(frame.PC)
|
|
||||||
splitPath := strings.Split(path, "/")
|
|
||||||
|
|
||||||
splitSignature := strings.Split(signature, ".")
|
|
||||||
pkg, method := splitSignature[0], splitSignature[len(splitSignature)-1]
|
|
||||||
|
|
||||||
tb := &MethodTraceback{
|
|
||||||
Filename: splitPath[len(splitPath)-1],
|
|
||||||
fullPath: path,
|
|
||||||
Line: line,
|
|
||||||
signature: signature,
|
|
||||||
Package: pkg,
|
|
||||||
Method: method,
|
|
||||||
}
|
|
||||||
list = append(list, tb)
|
|
||||||
}
|
|
||||||
sort.Slice(list, func(i, j int) bool {
|
|
||||||
return j < i
|
|
||||||
})
|
|
||||||
return list
|
|
||||||
}
|
|
||||||
|
|
||||||
func buildString(level LogLevel, prefix string, newline, printTime, printTraceback bool, m ...any) string {
|
|
||||||
args := []string{
|
args := []string{
|
||||||
fmt.Sprintf("[%s]", prefix),
|
fmt.Sprintf("[%s]", prefix),
|
||||||
fmt.Sprintf("[%s]", strings.ToUpper(level.t)),
|
fmt.Sprintf("[%s]", strings.ToUpper(level.t)),
|
||||||
}
|
}
|
||||||
|
|
||||||
if printTraceback {
|
if printTraceback {
|
||||||
args = append(args, fmt.Sprintf("[%s]", formatTraceback(getTraceback())))
|
args = append(args, fmt.Sprintf("[%s]", FormatTraceback(getTraceback())))
|
||||||
}
|
}
|
||||||
|
|
||||||
if printTime {
|
if printTime {
|
||||||
args = append(args, fmt.Sprintf("[%s]", formatTime(time.Now())))
|
args = append(args, fmt.Sprintf("[%s]", FormatTime(time.Now())))
|
||||||
}
|
}
|
||||||
|
|
||||||
msg := Map(m, func(el any) string {
|
msg := Map(m, func(el any) string {
|
||||||
return fmt.Sprintf("%v", el)
|
return fmt.Sprintf("%v", el)
|
||||||
})
|
})
|
||||||
s := fmt.Sprintf("%s %s", strings.Join(args, " "), strings.Join(msg, " "))
|
s := fmt.Sprintf("%s %s", strings.Join(args, " "), strings.Join(msg, " "))
|
||||||
if newline {
|
|
||||||
s += "\n"
|
|
||||||
}
|
|
||||||
return s
|
return s
|
||||||
}
|
}
|
||||||
|
|
||||||
func (l *Logger) print(level LogLevel, m ...any) {
|
|
||||||
if l.level.n < level.n {
|
|
||||||
return
|
|
||||||
}
|
|
||||||
|
|
||||||
tb := getFullTraceback(0)
|
|
||||||
for _, writer := range l.writers {
|
|
||||||
err := writer.Print(level, l.prefix, tb, m...)
|
|
||||||
if err != nil {
|
|
||||||
l.Error(err)
|
|
||||||
}
|
|
||||||
}
|
|
||||||
}
|
|
||||||
|
|
||||||
// Docker requires "\n" at end to write to log.
|
|
||||||
// print not work for docker, otherwise it will work and write into stdout
|
|
||||||
func (l *Logger) println(level LogLevel, m ...any) {
|
|
||||||
if l.level.n < level.n {
|
|
||||||
return
|
|
||||||
}
|
|
||||||
|
|
||||||
tb := getFullTraceback(0)
|
|
||||||
for _, writer := range l.writers {
|
|
||||||
err := writer.Print(level, l.prefix, tb, m...)
|
|
||||||
if err != nil {
|
|
||||||
l.Error(err)
|
|
||||||
}
|
|
||||||
}
|
|
||||||
}
|
|
||||||
|
|||||||
60
traceback.go
Normal file
60
traceback.go
Normal file
@@ -0,0 +1,60 @@
|
|||||||
|
package slog
|
||||||
|
|
||||||
|
import (
|
||||||
|
"runtime"
|
||||||
|
"sort"
|
||||||
|
"strings"
|
||||||
|
)
|
||||||
|
|
||||||
|
func getTraceback() *MethodTraceback {
|
||||||
|
caller, _, _, _ := runtime.Caller(5)
|
||||||
|
details := runtime.FuncForPC(caller)
|
||||||
|
signature := details.Name()
|
||||||
|
path, line := details.FileLine(caller)
|
||||||
|
splitPath := strings.Split(path, "/")
|
||||||
|
|
||||||
|
splitSignature := strings.Split(signature, ".")
|
||||||
|
method := splitSignature[len(splitSignature)-1]
|
||||||
|
|
||||||
|
tb := &MethodTraceback{
|
||||||
|
Filename: splitPath[len(splitPath)-1],
|
||||||
|
FullPath: path,
|
||||||
|
Line: line,
|
||||||
|
Signature: signature,
|
||||||
|
Method: method,
|
||||||
|
}
|
||||||
|
|
||||||
|
return tb
|
||||||
|
}
|
||||||
|
func getFullTraceback(skip int) []*MethodTraceback {
|
||||||
|
pc := make([]uintptr, 15)
|
||||||
|
runtime.Callers(skip, pc)
|
||||||
|
list := make([]*MethodTraceback, 0)
|
||||||
|
frames := runtime.CallersFrames(pc)
|
||||||
|
for {
|
||||||
|
frame, more := frames.Next()
|
||||||
|
if !more {
|
||||||
|
break
|
||||||
|
}
|
||||||
|
details := runtime.FuncForPC(frame.PC)
|
||||||
|
signature := details.Name()
|
||||||
|
path, line := details.FileLine(frame.PC)
|
||||||
|
splitPath := strings.Split(path, "/")
|
||||||
|
|
||||||
|
splitSignature := strings.Split(signature, ".")
|
||||||
|
method := splitSignature[len(splitSignature)-1]
|
||||||
|
|
||||||
|
tb := &MethodTraceback{
|
||||||
|
Filename: splitPath[len(splitPath)-1],
|
||||||
|
FullPath: path,
|
||||||
|
Line: line,
|
||||||
|
Signature: signature,
|
||||||
|
Method: method,
|
||||||
|
}
|
||||||
|
list = append(list, tb)
|
||||||
|
}
|
||||||
|
sort.Slice(list, func(i, j int) bool {
|
||||||
|
return j < i
|
||||||
|
})
|
||||||
|
return list
|
||||||
|
}
|
||||||
9
utils.go
Normal file
9
utils.go
Normal file
@@ -0,0 +1,9 @@
|
|||||||
|
package slog
|
||||||
|
|
||||||
|
func Map[T, R any](s []T, f func(T) R) []R {
|
||||||
|
out := make([]R, len(s))
|
||||||
|
for i, el := range s {
|
||||||
|
out[i] = f(el)
|
||||||
|
}
|
||||||
|
return out
|
||||||
|
}
|
||||||
134
writers.go
Normal file
134
writers.go
Normal file
@@ -0,0 +1,134 @@
|
|||||||
|
package slog
|
||||||
|
|
||||||
|
import (
|
||||||
|
"bufio"
|
||||||
|
"encoding/json"
|
||||||
|
"fmt"
|
||||||
|
"io"
|
||||||
|
"os"
|
||||||
|
"path/filepath"
|
||||||
|
"strings"
|
||||||
|
"time"
|
||||||
|
)
|
||||||
|
|
||||||
|
type LoggerWriter interface {
|
||||||
|
Close() error
|
||||||
|
Write(p []byte) (n int, err error)
|
||||||
|
Print(level LogLevel, prefix string, traceback []*MethodTraceback, messages ...any) error
|
||||||
|
}
|
||||||
|
|
||||||
|
type LoggerTextWriter struct {
|
||||||
|
LoggerWriter
|
||||||
|
writer io.Writer
|
||||||
|
printTraceback bool
|
||||||
|
printTime bool
|
||||||
|
}
|
||||||
|
|
||||||
|
func (w *LoggerTextWriter) Write(p []byte) (n int, err error) {
|
||||||
|
n, err = w.writer.Write(p)
|
||||||
|
if err != nil {
|
||||||
|
return n, err
|
||||||
|
}
|
||||||
|
err = bufio.NewWriter(w.writer).Flush()
|
||||||
|
return n, err
|
||||||
|
}
|
||||||
|
func (w *LoggerTextWriter) Print(level LogLevel, prefix string, _ []*MethodTraceback, messages ...any) error {
|
||||||
|
s := BuildString(level, prefix, w.printTime, w.printTraceback, messages...)
|
||||||
|
_, err := w.Write([]byte(s))
|
||||||
|
return err
|
||||||
|
}
|
||||||
|
func (w *LoggerTextWriter) Close() error {
|
||||||
|
return w.writer.(io.Closer).Close()
|
||||||
|
}
|
||||||
|
|
||||||
|
// LoggerJsonWriter write into writer JSON
|
||||||
|
type LoggerJsonWriter struct {
|
||||||
|
LoggerWriter
|
||||||
|
writer io.Writer
|
||||||
|
pretty bool
|
||||||
|
}
|
||||||
|
type LoggerJsonMessage struct {
|
||||||
|
Time time.Time `json:"time"`
|
||||||
|
Level string `json:"level"`
|
||||||
|
Prefix string `json:"prefix"`
|
||||||
|
Message string `json:"message"`
|
||||||
|
Traceback []*MethodTraceback `json:"traceback"`
|
||||||
|
}
|
||||||
|
|
||||||
|
func (w *LoggerJsonWriter) Write(data []byte) (int, error) {
|
||||||
|
n, err := w.writer.Write(data)
|
||||||
|
if err != nil {
|
||||||
|
return n, err
|
||||||
|
}
|
||||||
|
err = bufio.NewWriter(w.writer).Flush()
|
||||||
|
return n, err
|
||||||
|
}
|
||||||
|
func (w *LoggerJsonWriter) Print(level LogLevel, prefix string, traceback []*MethodTraceback, messages ...any) error {
|
||||||
|
msg := Map(messages, func(el any) string {
|
||||||
|
return fmt.Sprintf("%v", el)
|
||||||
|
})
|
||||||
|
m := LoggerJsonMessage{
|
||||||
|
Time: time.Now(),
|
||||||
|
Level: level.GetName(),
|
||||||
|
Prefix: prefix,
|
||||||
|
Message: strings.Join(msg, " "),
|
||||||
|
Traceback: traceback,
|
||||||
|
}
|
||||||
|
var data []byte
|
||||||
|
var err error
|
||||||
|
if w.pretty {
|
||||||
|
data, err = json.MarshalIndent(m, "", " ")
|
||||||
|
} else {
|
||||||
|
data, err = json.Marshal(m)
|
||||||
|
}
|
||||||
|
if err != nil {
|
||||||
|
return err
|
||||||
|
}
|
||||||
|
_, err = w.Write(append(data, []byte("\n")...))
|
||||||
|
return err
|
||||||
|
}
|
||||||
|
func (w *LoggerJsonWriter) Close() error {
|
||||||
|
return w.writer.(io.Closer).Close()
|
||||||
|
}
|
||||||
|
|
||||||
|
func CreateTextWriter(w io.Writer, printTraceback, printTime bool) *LoggerTextWriter {
|
||||||
|
writer := &LoggerTextWriter{
|
||||||
|
writer: w, printTraceback: printTraceback, printTime: printTime,
|
||||||
|
}
|
||||||
|
return writer
|
||||||
|
}
|
||||||
|
func CreateTextFileWriter(path string, printTraceback, printTime bool) (*LoggerTextWriter, error) {
|
||||||
|
err := os.MkdirAll(filepath.Dir(path), os.ModePerm)
|
||||||
|
if err != nil {
|
||||||
|
return nil, err
|
||||||
|
}
|
||||||
|
file, err := os.OpenFile(path, os.O_APPEND|os.O_CREATE|os.O_WRONLY, 0644)
|
||||||
|
if err != nil {
|
||||||
|
return nil, err
|
||||||
|
}
|
||||||
|
return CreateTextWriter(file, printTraceback, printTime), nil
|
||||||
|
}
|
||||||
|
func CreateTextStdoutWriter(printTraceback, printTime bool) *LoggerTextWriter {
|
||||||
|
return CreateTextWriter(os.Stdout, printTraceback, printTime)
|
||||||
|
}
|
||||||
|
|
||||||
|
func CreateJsonWriter(w io.Writer, pretty bool) *LoggerJsonWriter {
|
||||||
|
writer := &LoggerJsonWriter{
|
||||||
|
writer: w, pretty: pretty,
|
||||||
|
}
|
||||||
|
return writer
|
||||||
|
}
|
||||||
|
func CreateJsonStdoutWriter(pretty bool) *LoggerJsonWriter {
|
||||||
|
return CreateJsonWriter(os.Stdout, pretty)
|
||||||
|
}
|
||||||
|
func CreateJsonFileWriter(path string, pretty bool) (*LoggerJsonWriter, error) {
|
||||||
|
err := os.MkdirAll(filepath.Dir(path), os.ModePerm)
|
||||||
|
if err != nil {
|
||||||
|
return nil, err
|
||||||
|
}
|
||||||
|
file, err := os.OpenFile(path, os.O_APPEND|os.O_CREATE|os.O_WRONLY, 0644)
|
||||||
|
if err != nil {
|
||||||
|
return nil, err
|
||||||
|
}
|
||||||
|
return CreateJsonWriter(file, pretty), nil
|
||||||
|
}
|
||||||
Reference in New Issue
Block a user