caller.go (1489B)
1 // GoToSocial 2 // Copyright (C) GoToSocial Authors admin@gotosocial.org 3 // SPDX-License-Identifier: AGPL-3.0-or-later 4 // 5 // This program is free software: you can redistribute it and/or modify 6 // it under the terms of the GNU Affero General Public License as published by 7 // the Free Software Foundation, either version 3 of the License, or 8 // (at your option) any later version. 9 // 10 // This program is distributed in the hope that it will be useful, 11 // but WITHOUT ANY WARRANTY; without even the implied warranty of 12 // MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the 13 // GNU Affero General Public License for more details. 14 // 15 // You should have received a copy of the GNU Affero General Public License 16 // along with this program. If not, see <http://www.gnu.org/licenses/>. 17 18 package log 19 20 import ( 21 "runtime" 22 "strings" 23 ) 24 25 // Caller fetches the calling function name, skipping 'depth'. Results are cached per PC. 26 func Caller(depth int) string { 27 var pcs [1]uintptr 28 29 // Fetch calling function using calldepth 30 _ = runtime.Callers(depth, pcs[:]) 31 fn := runtime.FuncForPC(pcs[0]) 32 33 if fn == nil { 34 return "" 35 } 36 37 // Get func name. 38 name := fn.Name() 39 40 // Drop all but the package name and function name, no mod path 41 if idx := strings.LastIndex(name, "/"); idx >= 0 { 42 name = name[idx+1:] 43 } 44 45 const params = `[...]` 46 47 // Drop any generic type parameter markers 48 if idx := strings.Index(name, params); idx >= 0 { 49 name = name[:idx] + name[idx+len(params):] 50 } 51 52 return name 53 }