package fserr import ( "errors" "fmt" "strconv" "strings" ) // New 创建新的错误,支持格式化占位符 func New(format string, args ...any) error { msg := format if len(args) > 0 { msg = fmt.Sprintf(format, args...) } return &fundamental{ msg: msg, stack: callers(), } } // Wrap 包装已有错误,支持格式化占位符 func Wrap(err error, format string, args ...any) error { if err == nil { return nil } msg := format if len(args) > 0 { msg = fmt.Sprintf(format, args...) } return &withMessage{ cause: wrapStack(err), msg: msg, } } // WithCode 创建带有错误码的error,支持格式化占位符 // 使用option可以替换其中信息 func WithCode[T codeType](err error, businessCode T, options ...Option) error { code := getCode(serviceCode + int(businessCode)) ret := &withCode{ cause: wrapStack(err), Msg: code.Message, HttpCode: code.HttpCode, BusinessCode: code.BusinessCode, } for _, option := range options { option(ret) } if err == nil { return &withStack{ error: ret, stack: callers(), } } return ret } // UnWrap 获取包装过的error // 若error1使用 Wrap 包装后产出错误error2 // 当使用 UmWrap 后,返回的是error1 func UnWrap(err error) error { type causer interface { Cause() error } for err != nil { cause, ok := err.(causer) if !ok { break } err = cause.Cause() } return err } // Is 判断err是否是target类型 // 相比==判断错误,该方式会进行不断地类似于 UmWrap 的操作, // 将 UmWrap 后的错误进行比较 func Is(err, target error) bool { return errors.Is(err, target) } // As 匹配最外层的与target类型相同的error,将其赋值给target func As(err error, target any) bool { return errors.As(err, target) } // ParseCode 将错误解析为错误码错误 // 若err不是错误码错误,则包裹传递错误,其他信息为默认错误码信息 // 若err为错误码错误,将其转换,将最外层错误信息作为最终错误信息返回 // 若想得到原始的错误码错误,可以使用As方法 func ParseCode(err error) *withCode { var target *withCode if !As(err, &target) { return &withCode{ cause: nil, Msg: err.Error(), HttpCode: defaultErrCode.HttpCode, BusinessCode: serviceCode * 10000, } } return &withCode{ cause: target.cause, Msg: outerMsg(err), HttpCode: target.HttpCode, BusinessCode: target.BusinessCode, } } // IsCode 判断某个错误是否为某个错误码 func IsCode[T codeType](err error, code T) bool { var target *withCode if !As(err, &target) { return false } return target.BusinessCode == serviceCode+int(code) } // SetAppCode 设置服务错误码 // 例如设置服务码为300000,用户不存在错误码为2001, // 那么最终业务错误码为302001 func SetAppCode[T codeType](code T) { strCode := strconv.Itoa(int(code)) if len(strCode) == 1 { strCode = "0" + strCode } c, err := strconv.ParseInt(strCode+strings.Repeat("0", 6-len(strCode)), 10, 64) if err != nil { panic(err) } serviceCode = int(c) } // outerMsg 获取最外层的错误信息 func outerMsg(err error) string { if err == nil { return "" } if e, ok := err.(*withCode); ok { return e.Msg } if e, ok := err.(*withMessage); ok { return e.msg } if e, ok := err.(*withStack); ok { return outerMsg(e.error) } return err.Error() } func wrapStack(err error) error { if _, ok := err.(*fundamental); ok { return err } if _, ok := err.(*withCode); ok { return err } if _, ok := err.(*withMessage); ok { return err } if _, ok := err.(*withStack); ok { return err } return &withStack{ error: err, stack: callers(), } }