Golang自定义error实现业务校验方法

type BusinessError struct {
    Code    int
    Message string
}

func (e *BusinessError) Error() string {
    return e.Message
}

// 预定义一些常见的业务错误
var (
    ErrInvalidEmail = &BusinessError{Code: 1001, Message: "邮箱格式不正确"}
    ErrUserExists   = &BusinessError{Code: 1002, Message: "用户已存在"}
    ErrWeakPassword = &BusinessError{Code: 1003, Message: "密码强度不足"}
)

func ValidateUserRegistration(email, password string) error {
    if !isValidEmail(email) {
        return ErrInvalidEmail
    }
    if len(password) < 6 {
        return ErrWeakPassword
    }
    // 假设检查数据库发现用户已存在
    if userExists(email) {
        return ErrUserExists
    }
    return nil
}

// 使用示例
func RegisterUser(email, password string) {
    if err := ValidateUserRegistration(email, password); err != nil {
        switch e := err.(*BusinessError); e.Code {
        case 1001:
            log.Println("输入错误：", e.Message)
        case 1002:
            log.Println("注册失败：", e.Message)
        case 1003:
            log.Println("安全提示：", e.Message)
        default:
            log.Println("未知错误：", e.Message)
        }
        return
    }
    // 继续注册流程...
}

type BusinessErrorInterface interface {
    Error() string
    Code() int
}

func (e *BusinessError) Code() int {
    return e.Code
}

if be, ok := err.(interface{ Code() int }); ok {
    // 是业务错误，按错误码处理
    switch be.Code() {
    case 1001:
        // 处理邮箱错误
    }
}

if userExists(email) {
    return fmt.Errorf("注册失败: %w", ErrUserExists)
}

if errors.Is(err, ErrUserExists) {
    // 处理用户已存在的场景
}

var be *BusinessError
if errors.As(err, &be) {
    log.Printf("业务错误码: %d, 消息: %s", be.Code, be.Message)
}