45 lines
1.0 KiB
Go
45 lines
1.0 KiB
Go
package utility
|
||
|
||
import (
|
||
"fmt"
|
||
"runtime"
|
||
"runtime/debug"
|
||
"strings"
|
||
|
||
"github.com/go-admin-team/go-admin-core/logger"
|
||
)
|
||
|
||
// SafeGo 安全地启动一个 goroutine，捕获 panic
|
||
func SafeGo(fn func()) {
|
||
go func() {
|
||
defer func() {
|
||
if r := recover(); r != nil {
|
||
// 记录 Goroutine ID、panic 信息和堆栈
|
||
logger.Error(fmt.Sprintf("Recovered from panic in Goroutine %s: %v\nStack Trace:\n%s", getGoroutineID(), r, string(debug.Stack())))
|
||
}
|
||
}()
|
||
fn()
|
||
}()
|
||
}
|
||
|
||
// 获取 Goroutine ID
|
||
func getGoroutineID() string {
|
||
buf := make([]byte, 64)
|
||
n := runtime.Stack(buf, false)
|
||
stack := string(buf[:n])
|
||
// 提取 Goroutine ID
|
||
id := strings.Split(stack, " ")[1]
|
||
return id
|
||
}
|
||
|
||
func SafeGoParam[T any](fn func(T), param T) {
|
||
go func() {
|
||
defer func() {
|
||
if r := recover(); r != nil {
|
||
logger.Error(fmt.Sprintf(" SafeGoParam Recovered from panic in Goroutine %s: %v\nStack Trace:\n%s", getGoroutineID(), r, string(debug.Stack())))
|
||
}
|
||
}()
|
||
fn(param) // 执行传入的函数
|
||
}()
|
||
}
|