Add logr.Logger.WithCallDepth() support

go-logr · Dec 15, 2020 · 5c54c20 · 5c54c20
1 parent d6464ae
commit 5c54c20
Show file tree

Hide file tree

Showing 2 changed files with 16 additions and 0 deletions.
diff --git a/example/main.go b/example/main.go
@@ -20,6 +20,7 @@ import (
 	"flag"
 
 	"github.com/go-logr/glogr"
+	"github.com/go-logr/logr"
 	"github.com/golang/glog"
 )
 
@@ -31,6 +32,14 @@ func (e E) Error() string {
 	return e.str
 }
 
+func Helper(log logr.Logger, msg string) {
+	helper2(log, msg)
+}
+
+func helper2(log logr.Logger, msg string) {
+	log.WithCallDepth(2).Info(msg)
+}
+
 func main() {
 	flag.Set("v", "1")
 	flag.Set("logtostderr", "true")
@@ -41,5 +50,6 @@ func main() {
 	log.V(1).V(1).Info("you should NOT see this")
 	log.Error(nil, "uh oh", "trouble", true, "reasons", []float64{0.1, 0.11, 3.14})
 	log.Error(E{"an error occurred"}, "goodbye", "code", -1)
+	Helper(log, "thru a helper")
 	glog.Flush()
 }
diff --git a/glogr.go b/glogr.go
@@ -176,4 +176,10 @@ func (l glogger) WithValues(kvList ...interface{}) logr.Logger {
 	return new
 }
 
+func (l glogger) WithCallDepth(depth int) logr.Logger {
+	new := l.clone()
+	new.depth += depth
+	return new
+}
+
 var _ logr.Logger = glogger{}