github.com/kaydxh/golang@v0.0.131/go/runtime/extern.go (about)

     1  /*
     2   *Copyright (c) 2022, kaydxh
     3   *
     4   *Permission is hereby granted, free of charge, to any person obtaining a copy
     5   *of this software and associated documentation files (the "Software"), to deal
     6   *in the Software without restriction, including without limitation the rights
     7   *to use, copy, modify, merge, publish, distribute, sublicense, and/or sell
     8   *copies of the Software, and to permit persons to whom the Software is
     9   *furnished to do so, subject to the following conditions:
    10   *
    11   *The above copyright notice and this permission notice shall be included in all
    12   *copies or substantial portions of the Software.
    13   *
    14   *THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR
    15   *IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,
    16   *FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE
    17   *AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER
    18   *LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,
    19   *OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN THE
    20   *SOFTWARE.
    21   */
    22  package runtime
    23  
    24  import (
    25  	"fmt"
    26  	"path"
    27  	"runtime"
    28  	"strings"
    29  )
    30  
    31  // GetCaller returns the caller of the function that calls it.
    32  //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
    33  func GetCallerWithSkip(skip int) string {
    34  	var pc [1]uintptr
    35  	runtime.Callers(skip+1, pc[:])
    36  	f := runtime.FuncForPC(pc[0])
    37  	if f == nil {
    38  		return fmt.Sprintf("Unable to find caller")
    39  	}
    40  	return f.Name()
    41  }
    42  
    43  func GetCaller() string {
    44  	//4 skip, 1 GetCaller, 1 GetCallerWithSkip, 1 runtime.Callers, 1 caller of  GetCaller
    45  	return GetCallerWithSkip(3)
    46  }
    47  
    48  func GetShortCaller() string {
    49  	fn := GetCallerWithSkip(3)
    50  	return strings.TrimPrefix(path.Ext(fn), ".")
    51  }
    52  
    53  func GetCallStackTrace() string {
    54  	const size = 64 << 10
    55  	stacktrace := make([]byte, size)
    56  	stacktrace = stacktrace[:runtime.Stack(stacktrace, false)]
    57  
    58  	return string(stacktrace)
    59  }