mirror of https://github.com/rjbasitali/go-log
115 lines
2.1 KiB
Go
115 lines
2.1 KiB
Go
package log
|
|
|
|
import (
|
|
"fmt"
|
|
"io"
|
|
"os"
|
|
"runtime"
|
|
"strings"
|
|
"time"
|
|
)
|
|
|
|
var (
|
|
DefaultTimeFormat = time.RFC3339
|
|
)
|
|
|
|
type Fields map[string]interface{}
|
|
|
|
type myLogger struct {
|
|
Writer io.Writer
|
|
ErrWriter io.Writer
|
|
prefix string
|
|
begin time.Time
|
|
level uint8
|
|
data Fields
|
|
}
|
|
|
|
func logPrefix(level uint8) string {
|
|
return fmt.Sprintf("time=%q level=%s function=%s", time.Now().Format(DefaultTimeFormat), unmarshalLevel(level), funcName())
|
|
}
|
|
|
|
func (fields Fields) String() string {
|
|
if len(fields) == 0 {
|
|
return ""
|
|
}
|
|
var buffer strings.Builder
|
|
for k, v := range fields {
|
|
switch v := v.(type) {
|
|
case string:
|
|
buffer.WriteString(fmt.Sprintf(" %s=%q", k, v))
|
|
default:
|
|
buffer.WriteString(fmt.Sprintf(" %s=%v", k, v))
|
|
}
|
|
}
|
|
return buffer.String()
|
|
}
|
|
|
|
func (l myLogger) log(flag uint8, s ...interface{}) {
|
|
var w io.Writer
|
|
{
|
|
if flag != errorFlag && l.Writer == nil {
|
|
return
|
|
} else {
|
|
w = l.Writer
|
|
}
|
|
if flag == errorFlag && l.ErrWriter == nil {
|
|
return
|
|
} else {
|
|
w = l.ErrWriter
|
|
}
|
|
}
|
|
f := fmt.Sprintf("%s%s msg=%q%s", logPrefix(flag), l.prefix, fmt.Sprint(s...), l.data)
|
|
fmt.Fprintln(w, f)
|
|
}
|
|
|
|
func (l myLogger) logf(flag uint8, format string, s ...interface{}) {
|
|
var w io.Writer
|
|
{
|
|
if flag != errorFlag && l.Writer == nil {
|
|
return
|
|
} else {
|
|
w = l.Writer
|
|
}
|
|
if flag == errorFlag && l.ErrWriter == nil {
|
|
return
|
|
} else {
|
|
w = l.ErrWriter
|
|
}
|
|
}
|
|
format = fmt.Sprintf("%%s%%s msg=\"%s\"%s\n", format, l.data)
|
|
s = append([]interface{}{logPrefix(flag), l.prefix}, s...)
|
|
fmt.Fprintf(w, format, s...)
|
|
}
|
|
|
|
func funcName() string {
|
|
pc := make([]uintptr, 10)
|
|
runtime.Callers(2, pc)
|
|
for _, p := range pc {
|
|
fn := runtime.FuncForPC(p)
|
|
if !strings.HasPrefix(fn.Name(), "github.com/rjbasitali/go-log") {
|
|
fname := fn.Name()
|
|
i := strings.LastIndexByte(fname, byte('/'))
|
|
if i == -1 {
|
|
return fname
|
|
}
|
|
return fname[i+1:]
|
|
}
|
|
}
|
|
return ""
|
|
}
|
|
|
|
func NewLogger(w, errw io.Writer) Logger {
|
|
if w == nil {
|
|
w = os.Stdout
|
|
}
|
|
if errw == nil {
|
|
errw = os.Stderr
|
|
}
|
|
return myLogger{
|
|
Writer: w,
|
|
ErrWriter: errw,
|
|
begin: time.Now(),
|
|
level: traceFlag,
|
|
}
|
|
}
|