shanshanzhong
c582087c0f
Some checks failed
Build docker and publish / build (20.15.1) (push) Failing after 6m27s
feat: 添加版本和构建时间变量 fix: 修正短信队列类型注释错误 style: 清理未使用的代码和测试文件 docs: 更新安装文档中的下载链接 chore: 迁移数据库脚本添加日志和订阅配置
47 lines
981 B
Go
47 lines
981 B
Go
package threading
|
|
|
|
import (
|
|
"bytes"
|
|
"context"
|
|
"runtime"
|
|
"strconv"
|
|
|
|
"github.com/perfect-panel/server/pkg/rescue"
|
|
)
|
|
|
|
// GoSafe runs the given fn using another goroutine, recovers if fn panics.
|
|
func GoSafe(fn func()) {
|
|
go RunSafe(fn)
|
|
}
|
|
|
|
// GoSafeCtx runs the given fn using another goroutine, recovers if fn panics with ctx.
|
|
func GoSafeCtx(ctx context.Context, fn func()) {
|
|
go RunSafeCtx(ctx, fn)
|
|
}
|
|
|
|
// RoutineId is only for debug, never use it in production.
|
|
func RoutineId() uint64 {
|
|
b := make([]byte, 64)
|
|
b = b[:runtime.Stack(b, false)]
|
|
b = bytes.TrimPrefix(b, []byte("goroutine "))
|
|
b = b[:bytes.IndexByte(b, ' ')]
|
|
// if error, just return 0
|
|
n, _ := strconv.ParseUint(string(b), 10, 64)
|
|
|
|
return n
|
|
}
|
|
|
|
// RunSafe runs the given fn, recovers if fn panics.
|
|
func RunSafe(fn func()) {
|
|
defer rescue.Recover()
|
|
|
|
fn()
|
|
}
|
|
|
|
// RunSafeCtx runs the given fn, recovers if fn panics with ctx.
|
|
func RunSafeCtx(ctx context.Context, fn func()) {
|
|
defer rescue.RecoverCtx(ctx)
|
|
|
|
fn()
|
|
}
|