support caller manually

cgxxv · Sep 16, 2020 · 495016b · 495016b
1 parent 7b88485
commit 495016b
Showing 1 changed file with 13 additions and 0 deletions.
diff --git a/entry.go b/entry.go
@@ -94,6 +94,19 @@ func (entry *Entry) String() (string, error) {
 	return str, nil
 }
 
+// Add a caller field to the Entry.
+func (entry *Entry) WithCaller() *Entry {
+	frame := getCaller()
+	if frame != nil {
+		return entry.WithFields(Fields{
+			FieldKeyFile: fmt.Sprintf("%s:%d", frame.File, frame.Line),
+			FieldKeyFunc: frame.Function,
+		})
+	}
+
+	return entry
+}
+
 // Add an error as single field (using the key defined in ErrorKey) to the Entry.
 func (entry *Entry) WithError(err error) *Entry {
 	return entry.WithField(ErrorKey, err)