extern.go

package runtime

import (
	"fmt"
	"path"
	"runtime"
	"strings"
)

// GetCaller returns the caller of the function that calls it.
//The argument skip is the number of stack frames to skip before recording in pc, with 0 identifying the frame for Callers itself and 1 identifying the caller of Callers
func GetCallerWithSkip(skip int) string {
	var pc [1]uintptr
	runtime.Callers(skip+1, pc[:])
	f := runtime.FuncForPC(pc[0])
	if f == nil {
		return fmt.Sprintf("Unable to find caller")
	}
	return f.Name()
}

func GetCaller() string {
	//4 skip, 1 GetCaller, 1 GetCallerWithSkip, 1 runtime.Callers, 1 caller of  GetCaller
	return GetCallerWithSkip(3)
}

func GetShortCaller() string {
	fn := GetCallerWithSkip(3)
	return strings.TrimPrefix(path.Ext(fn), ".")
}

func GetCallStackTrace() string {
	const size = 64 << 10
	stacktrace := make([]byte, size)
	stacktrace = stacktrace[:runtime.Stack(stacktrace, false)]

	return string(stacktrace)
}