129 lines
2.7 KiB
Go
129 lines
2.7 KiB
Go
package log
|
|
|
|
import (
|
|
"bytes"
|
|
"encoding/json"
|
|
"fmt"
|
|
"log"
|
|
"runtime"
|
|
"strings"
|
|
"sync"
|
|
"time"
|
|
|
|
"github.com/sirupsen/logrus"
|
|
)
|
|
|
|
type levelSkip struct {
|
|
SkipLogrus int
|
|
SkipBasic int
|
|
}
|
|
|
|
// JSONFormatter formats logs into parsable json
|
|
type JSONFormatter struct {
|
|
skip []*levelSkip
|
|
once sync.Once
|
|
}
|
|
|
|
// Format renders a single log entry
|
|
func (h *JSONFormatter) Format(e *logrus.Entry) ([]byte, error) {
|
|
|
|
skipOnce := h.skip[int(e.Level)]
|
|
|
|
if skipOnce.SkipLogrus == 0 {
|
|
var skipLogrus OpenClose
|
|
|
|
for i := 4; i < 50; i++ {
|
|
// log.Println(i)
|
|
if pc, file, line, ok := runtime.Caller(i); ok {
|
|
funcStruct := runtime.FuncForPC(pc)
|
|
log.Println(funcStruct.Name(), file, line)
|
|
|
|
skipLogrus.OpenFunc(func() bool {
|
|
return strings.Contains(funcStruct.Name(), "github.com/sirupsen/logrus.")
|
|
}, func() {
|
|
|
|
// skip = i
|
|
skipOnce.SkipLogrus = i
|
|
i = 10000
|
|
})
|
|
|
|
} else {
|
|
break
|
|
}
|
|
|
|
}
|
|
}
|
|
|
|
if skipOnce.SkipBasic == 0 {
|
|
var skipBasic OpenClose
|
|
for i := 4; i < 50; i++ {
|
|
// log.Println(i)
|
|
if pc, file, line, ok := runtime.Caller(i); ok {
|
|
funcStruct := runtime.FuncForPC(pc)
|
|
log.Println(funcStruct.Name(), file, line)
|
|
|
|
skipBasic.OpenFunc(func() bool {
|
|
return strings.Contains(funcStruct.Name(), "basic.(*Response).")
|
|
}, func() {
|
|
skipOnce.SkipBasic = i + 1
|
|
})
|
|
} else {
|
|
break
|
|
}
|
|
}
|
|
}
|
|
|
|
var fileinfo string
|
|
if pc, _, _, ok := runtime.Caller(skipOnce.SkipBasic - 1); ok {
|
|
funcStruct := runtime.FuncForPC(pc)
|
|
log.Println(funcStruct.Name())
|
|
if strings.Contains(funcStruct.Name(), "basic.(*Response).") {
|
|
if _, file, line, ok := runtime.Caller(skipOnce.SkipBasic); ok {
|
|
if e.Level == logrus.InfoLevel {
|
|
fileinfo = fmt.Sprintf("%s:%d", file, line)
|
|
} else {
|
|
ps := strings.Split(file, "/")
|
|
// ps = ps[len(ps)-4:]
|
|
fileinfo = fmt.Sprintf("%s:%d", strings.Join(ps, "/"), line)
|
|
}
|
|
}
|
|
}
|
|
}
|
|
|
|
if fileinfo != "" {
|
|
if _, file, line, ok := runtime.Caller(skipOnce.SkipLogrus); ok {
|
|
if e.Level == logrus.InfoLevel {
|
|
fileinfo = fmt.Sprintf("%s:%d", file, line)
|
|
} else {
|
|
ps := strings.Split(file, "/")
|
|
// ps = ps[len(ps)-4:]
|
|
fileinfo = fmt.Sprintf("%s:%d", strings.Join(ps, "/"), line)
|
|
}
|
|
|
|
}
|
|
}
|
|
|
|
var Data map[string]any = make(map[string]any, 4)
|
|
|
|
Data[logrus.FieldKeyTime] = e.Time.Format(time.RFC3339)
|
|
Data[logrus.FieldKeyMsg] = e.Message
|
|
Data[logrus.FieldKeyLevel] = e.Level
|
|
Data[logrus.FieldKeyFile] = fileinfo
|
|
|
|
var b *bytes.Buffer
|
|
if e.Buffer != nil {
|
|
b = e.Buffer
|
|
} else {
|
|
b = &bytes.Buffer{}
|
|
}
|
|
|
|
encoder := json.NewEncoder(b)
|
|
encoder.SetEscapeHTML(false)
|
|
|
|
if err := encoder.Encode(Data); err != nil {
|
|
return nil, fmt.Errorf("failed to marshal fields to JSON, %w", err)
|
|
}
|
|
|
|
return b.Bytes(), nil
|
|
}
|