init: 1.0.0

pkg/threading/routines.go

@@ -0,0 +1,46 @@
+package threading
+
+import (
+	"bytes"
+	"context"
+	"runtime"
+	"strconv"
+
+	"github.com/perfect-panel/ppanel-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()
+}