diff --git a/entry.go b/entry.go index c6da56291..27b14bfb1 100644 --- a/entry.go +++ b/entry.go @@ -180,15 +180,20 @@ func getPackageName(f string) string { // getCaller retrieves the name of the first non-logrus calling function func getCaller() *runtime.Frame { - // cache this package's fully-qualified name callerInitOnce.Do(func() { - pcs := make([]uintptr, 2) + pcs := make([]uintptr, maximumCallerDepth) _ = runtime.Callers(0, pcs) - logrusPackage = getPackageName(funcName(pcs)) - // now that we have the cache, we can skip a minimum count of known-logrus functions - // XXX this is dubious, the number of frames may vary + // dynamic get the package name and the minimum caller depth + for i := 0; i < maximumCallerDepth; i++ { + funcName := runtime.FuncForPC(pcs[i]).Name() + if strings.Contains(funcName, "getCaller") { + logrusPackage = getPackageName(funcName) + break + } + } + minimumCallerDepth = knownLogrusFrames }) diff --git a/versions_go1_14.go b/versions_go1_14.go deleted file mode 100644 index 8db33606a..000000000 --- a/versions_go1_14.go +++ /dev/null @@ -1,10 +0,0 @@ -// +build go1.14 - -package logrus - -import "runtime" - -// funcName returns the function name that logrus calls -func funcName(pcs []uintptr) string { - return runtime.FuncForPC(pcs[0]).Name() -} diff --git a/versions_others.go b/versions_others.go deleted file mode 100644 index 8b3a68204..000000000 --- a/versions_others.go +++ /dev/null @@ -1,10 +0,0 @@ -// +build !go1.14 - -package logrus - -import "runtime" - -// funcName returns the function name that logrus calls -func funcName(pcs []uintptr) string { - return runtime.FuncForPC(pcs[1]).Name() -}