feat: Add connection

recover.go

@@ -0,0 +1,39 @@
+package mq
+
+import (
+	"fmt"
+	"log"
+	"runtime"
+	"runtime/debug"
+
+	"github.com/oarkflow/mq/metrics"
+)
+
+func RecoverPanic(labelGenerator func() string) {
+	if r := recover(); r != nil {
+		pc, file, line, ok := runtime.Caller(2)
+		funcName := "unknown"
+		if ok {
+			fn := runtime.FuncForPC(pc)
+			if fn != nil {
+				funcName = fn.Name()
+			}
+		}
+		log.Printf("[PANIC] - recovered from panic in %s (%s:%d): %v\nStack trace: %s", funcName, file, line, r, debug.Stack())
+		label := labelGenerator()
+		metrics.TasksErrors.WithLabelValues(label).Inc()
+	}
+}
+
+func RecoverTitle() string {
+	pc, _, line, ok := runtime.Caller(1)
+	if !ok {
+		return "Unknown"
+	}
+	fn := runtime.FuncForPC(pc)
+	funcName := "unknown"
+	if fn != nil {
+		funcName = fn.Name()
+	}
+	return fmt.Sprintf("%s:%d", funcName, line)
+}