mirror of https://github.com/rjbasitali/go-log
90 lines
1.7 KiB
Go
90 lines
1.7 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
|
|
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(level uint8, s ...interface{}) {
|
|
if l.Writer == nil {
|
|
return
|
|
}
|
|
f := fmt.Sprintf("%s%s msg=%q %s", logPrefix(level), l.prefix, fmt.Sprint(s...), l.data)
|
|
fmt.Fprintln(l.Writer, f)
|
|
}
|
|
|
|
func (l myLogger) logf(level uint8, format string, s ...interface{}) {
|
|
if l.Writer == nil {
|
|
return
|
|
}
|
|
format = fmt.Sprintf("%%s %%s msg=\"%s\" %s\n", format, l.data)
|
|
s = append([]interface{}{logPrefix(level), l.prefix}, s...)
|
|
fmt.Fprintf(l.Writer, 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 io.Writer) Logger {
|
|
if w == nil {
|
|
w = os.Stdout
|
|
}
|
|
return myLogger{
|
|
Writer: w,
|
|
begin: time.Now(),
|
|
level: traceFlag,
|
|
}
|
|
}
|