special error

pkg/common/mw/specialerror/error.go

@@ -0,0 +1,33 @@
+package specialerror
+
+import "OpenIM/pkg/errs"
+
+var handlers []func(err error) errs.CodeError
+
+func AddErrHandler(h func(err error) errs.CodeError) {
+	if h == nil {
+		panic("nil handler")
+	}
+	handlers = append(handlers, h)
+}
+
+func AddReplace(target error, codeErr errs.CodeError) {
+	AddErrHandler(func(err error) errs.CodeError {
+		if err == target {
+			return codeErr
+		}
+		return nil
+	})
+}
+
+func ErrCode(err error) errs.CodeError {
+	if codeErr, ok := err.(errs.CodeError); ok {
+		return codeErr
+	}
+	for i := 0; i < len(handlers); i++ {
+		if codeErr := handlers[i](err); codeErr != nil {
+			return codeErr
+		}
+	}
+	return nil
+}