logx

logx/logrus_nested_formatter.go

@@ -0,0 +1,28 @@
+package logx
+
+import (
+	"fmt"
+	"path"
+	"runtime"
+
+	nested "github.com/antonfisher/nested-logrus-formatter"
+)
+
+const defaultTimestampFormat = "2006-01-02 15:04:05.000"
+
+type NestedFormatterOption struct {
+	Color bool
+}
+
+func NewNestedFormatter(option NestedFormatterOption) *nested.Formatter {
+	return &nested.Formatter{
+		TimestampFormat:       defaultTimestampFormat,
+		NoColors:              !option.Color,
+		CustomCallerFormatter: nestedCallerFormatter,
+	}
+}
+
+func nestedCallerFormatter(f *runtime.Frame) string {
+	_, filename := path.Split(f.File)
+	return fmt.Sprintf(" (%s() %s:%d)", f.Function, filename, f.Line)
+}