u

code/go-project/base-farmework/pkg/utils/error.go

@@ -0,0 +1,38 @@
+package utils
+
+import "runtime/debug"
+
+// StackError 保存原始错误和堆栈
+type StackError struct {
+	err   error
+	stack string
+}
+
+// 实现 error 接口
+func (s *StackError) Error() string {
+	if s.err == nil {
+		return ""
+	}
+	return s.err.Error()
+}
+
+// StackTrace 返回堆栈信息
+func (s *StackError) StackTrace() string {
+	return s.stack
+}
+
+// WrapWithStack 给错误附加堆栈，如果已经是 StackError 则直接返回
+func WrapWithStack(err error) *StackError {
+	if err == nil {
+		return nil
+	}
+
+	if se, ok := err.(*StackError); ok {
+		return se
+	}
+
+	return &StackError{
+		err:   err,
+		stack: string(debug.Stack()),
+	}
+}