exception/exception.go

// Package exception ...
//
// Description : exception ...
//
// Author : go_developer@163.com<白茶清欢>
//
// Date *: 2022-06-25 21:04
package exception

import (
	"errors"
	"fmt"
)

// Exception 异常接口的具体实现
//
// Author : go_developer@163.com<白茶清欢>
//
// Date *: 21:09 2022/6/25
type Exception struct {
	code     interface{}
	message  string
	httpCode int
	data     interface{}
}

func (e *Exception) Error() string {
	return e.GetMessage()
}

func (e *Exception) GetCode() interface{} {
	return e.code
}

func (e *Exception) GetMessage() string {
	return e.message
}

func (e *Exception) GetData() interface{} {
	return e.data
}

func (e *Exception) GetHttpCode() int {
	return e.httpCode
}

func (e *Exception) ToError() error {
	if nil == e {
		return nil
	}
	return errors.New(e.Error())
}

// IsCode 判断是否为指定错误码
//
// Author : go_developer@163.com<白茶清欢>
//
// Date : 00:39 2023/9/28
func (e *Exception) IsCode(inputCode interface{}) bool {
	return fmt.Sprintf("%v", inputCode) == fmt.Sprintf("%v", e.GetCode())
}

// NewWithCode 仅使用错误码实例化异常
//
// Author : go_developer@163.com<白茶清欢>
//
// Date : 21:22 2022/6/25
func NewWithCode(code interface{}) IException {
	return New(code, defaultHttpCode, nil, "")
}

// NewWithCodeAndHttpCode 使用 code + http_code 获取实例
//
// Author : go_developer@163.com<白茶清欢>
//
// Date : 21:25 2022/6/25
func NewWithCodeAndHttpCode(code interface{}, httpCode int) IException {
	return New(code, httpCode, nil, "")
}

// NewWithCodeAndData 使用 code + data 获取异常实例
//
// Author : go_developer@163.com<白茶清欢>
//
// Date : 21:28 2022/6/25
func NewWithCodeAndData(code interface{}, data interface{}) IException {
	return New(code, defaultHttpCode, data, "")
}

// New 获取异常实例
//
// Author : go_developer@163.com<白茶清欢>
//
// Date : 21:28 2022/6/25
func New(code interface{}, httpCode int, data interface{}, defaultMessage ...string) IException {
	if nil == data {
		// 保证数据结构的一致性, 同时避免后续使用出现空指针
		data = map[string]interface{}{}
	}
	return &Exception{
		code:     code,
		message:  getMessage(code, defaultMessage...),
		httpCode: httpCode,
		data:     data,
	}
}

// NewFromError 从error转换为Exception
//
// Author : go_developer@163.com<白茶清欢>
//
// Date : 21:36 2022/6/25
func NewFromError(code interface{}, err error) IException {
	if nil == err {
		return nil
	}
	return New(code, defaultHttpCode, map[string]interface{}{
		"error": err.Error(),
	}, err.Error())
}

// NewFromMessage 从 code message 生成exception
//
// Author : go_developer@163.com<白茶清欢>
//
// Date : 22:25 2023/2/11
func NewFromMessage(code interface{}, message string) IException {
	if len(message) == 0 {
		message = getMessage(code, fmt.Sprintf("%v -> 未知异常信息", code))
	}
	return NewFromError(code, errors.New(message))
}

// ToError 转换成内置error
//
// Author : go_developer@163.com<白茶清欢>
//
// Date : 21:37 2022/6/25
func ToError(e *Exception) error {
	if nil == e {
		return nil
	}
	return errors.New(e.GetMessage())
}

// IsSuccess 判断一个异常是否为成功
//
// Author : go_developer@163.com<白茶清欢>
//
// Date : 21:34 2022/6/25
func IsSuccess(e *Exception) bool {
	return nil == e || e.GetCode() == defaultSuccessCode
}

// NewSuccess 代表Success的异常
//
// Author : go_developer@163.com<白茶清欢>
//
// Date : 22:35 2022/6/25
func NewSuccess(data interface{}) IException {
	return New(defaultSuccessCode, defaultHttpCode, data, "")
}