异常增加namespace

Author: yumaojun03

exception/base.go

@@ -8,6 +8,7 @@ type APIException interface {
 	ErrorCode() int
 	Meta() interface{}
 	Is(code int) bool
+	Namespace() string
 }
 
 // WithAPIException 携带元信息的异常
@@ -16,20 +17,31 @@ type WithAPIException interface {
 	WithMeta(m interface{}) APIException
 }
 
+func newException(namespace Namespace, code int, reason, format string, a ...interface{}) *exception {
+	return &exception{
+		namespace: namespace,
+		code:      code,
+		reason:    reason,
+		message:   fmt.Sprintf(format, a...),
+	}
+}
+
 // APIException is impliment for api exception
 type exception struct {
-	code int
-	msg  string
-	meta interface{}
+	namespace Namespace
+	code      int
+	reason    string
+	message   string
+	meta      interface{}
 }
 
 func (e *exception) Error() string {
-	return e.msg
+	return e.message
 }
 
 // Code exception's code, 如果code不存在返回-1
 func (e *exception) ErrorCode() int {
-	return e.code
+	return int(e.code)
 }
 
 // WithMeta 携带一些额外信息
@@ -46,16 +58,17 @@ func (e *exception) Is(code int) bool {
 	return e.ErrorCode() == code
 }
 
+func (e *exception) Namespace() string {
+	return e.namespace.String()
+}
+
 // NewAPIException 创建一个API异常
 // 用于其他模块自定义异常
-func NewAPIException(code int, format string, a ...interface{}) WithAPIException {
+func NewAPIException(code int, reason, format string, a ...interface{}) WithAPIException {
 	// 0表示正常状态, 但是要排除变量的零值
 	if code == 0 {
 		code = -1
 	}
 
-	return &exception{
-		msg:  fmt.Sprintf(format, a...),
-		code: code,
-	}
+	return newException(usedNamespace, code, reason, format, a...)
 }

exception/code.go

@@ -3,6 +3,7 @@ package exception
 import "net/http"
 
 const (
+	// TokenExpired token过期
 	TokenExpired = 1000
 
 	// 1xx - 5xx copy from http status code
@@ -14,3 +15,24 @@ const (
 
 	UnKnownException = 9999
 )
+
+var (
+	reasonMap = map[int]string{
+		Unauthorized:        "认证失败",
+		NotFound:            "资源为找到",
+		BadRequest:          "请求不合法",
+		InternalServerError: "系统内部错误",
+		Forbidden:           "访问未授权",
+		UnKnownException:    "未知异常",
+		TokenExpired:        "访问过期, 请重新访问",
+	}
+)
+
+func codeReason(code int) string {
+	v, ok := reasonMap[code]
+	if !ok {
+		v = reasonMap[UnKnownException]
+	}
+
+	return v
+}

exception/exception.go

@@ -1,53 +1,33 @@
 package exception
 
-import "fmt"
-
 // NewUnauthorized 未认证
 func NewUnauthorized(format string, a ...interface{}) APIException {
-	return &exception{
-		msg:  fmt.Sprintf(format, a...),
-		code: Unauthorized,
-	}
+	return newException(usedNamespace, Unauthorized, codeReason(Unauthorized), format, a...)
 }
 
 // NewPermissionDeny 没有权限访问
 func NewPermissionDeny(format string, a ...interface{}) APIException {
-	return &exception{
-		msg:  fmt.Sprintf(format, a...),
-		code: Forbidden,
-	}
+	return newException(usedNamespace, Forbidden, codeReason(Forbidden), format, a...)
 }
 
 // NewTokenExpired token过期
 func NewTokenExpired(format string, a ...interface{}) APIException {
-	return &exception{
-		msg:  fmt.Sprintf(format, a...),
-		code: TokenExpired,
-	}
+	return newException(usedNamespace, TokenExpired, codeReason(TokenExpired), format, a...)
 }
 
 // NewBadRequest todo
 func NewBadRequest(format string, a ...interface{}) APIException {
-	return &exception{
-		msg:  fmt.Sprintf(format, a...),
-		code: BadRequest,
-	}
+	return newException(usedNamespace, BadRequest, codeReason(BadRequest), format, a...)
 }
 
 // NewNotFound todo
 func NewNotFound(format string, a ...interface{}) APIException {
-	return &exception{
-		msg:  fmt.Sprintf(format, a...),
-		code: NotFound,
-	}
+	return newException(usedNamespace, NotFound, codeReason(BadRequest), format, a...)
 }
 
 // NewInternalServerError 500
 func NewInternalServerError(format string, a ...interface{}) APIException {
-	return &exception{
-		msg:  fmt.Sprintf(format, a...),
-		code: InternalServerError,
-	}
+	return newException(usedNamespace, InternalServerError, codeReason(InternalServerError), format, a...)
 }
 
 // IsNotFoundError 判断是否是NotFoundError

exception/namespace.go

@@ -0,0 +1,24 @@
+package exception
+
+var (
+	usedNamespace = GlobalNamespace
+)
+
+// Namespace 异常所属空间
+type Namespace string
+
+func (ns Namespace) String() string {
+	return string(ns)
+}
+
+const (
+	// GlobalNamespace 所有服务公用的一些异常
+	GlobalNamespace = Namespace("global")
+	// Keyauth 权限系统, 每个子系统有自己的异常空间
+	Keyauth = Namespace("keyauth")
+)
+
+// ChangeNameSapce 切换异常空间
+func ChangeNameSapce(ns Namespace) {
+	usedNamespace = ns
+}

go.mod

@@ -23,21 +23,21 @@ require (
 	github.com/julienschmidt/httprouter v1.3.0
 	github.com/modern-go/concurrent v0.0.0-20180306012644-bacd9c7ef1dd // indirect
 	github.com/modern-go/reflect2 v1.0.1 // indirect
-	github.com/onsi/ginkgo v1.10.3 // indirect
-	github.com/onsi/gomega v1.7.1 // indirect
+	github.com/onsi/ginkgo v1.11.0 // indirect
+	github.com/onsi/gomega v1.8.1 // indirect
 	github.com/pkg/errors v0.8.1
-	github.com/prometheus/client_golang v1.2.1 // indirect
+	github.com/prometheus/client_golang v1.3.0 // indirect
 	github.com/soheilhy/cmux v0.1.4 // indirect
 	github.com/stretchr/testify v1.4.0
 	github.com/tmc/grpc-websocket-proxy v0.0.0-20190109142713-0ad062ec5ee5 // indirect
 	github.com/xiang90/probing v0.0.0-20190116061207-43a291ad63a2 // indirect
 	go.etcd.io/bbolt v1.3.3 // indirect
-	go.etcd.io/etcd v3.3.17+incompatible
-	go.mongodb.org/mongo-driver v1.1.3
-	go.uber.org/zap v1.12.0
-	golang.org/x/crypto v0.0.0-20191108234033-bd318be0434a
-	golang.org/x/sys v0.0.0-20191105231009-c1f44814a5cd
+	go.etcd.io/etcd v3.3.18+incompatible
+	go.mongodb.org/mongo-driver v1.2.0
+	go.uber.org/zap v1.13.0
+	golang.org/x/crypto v0.0.0-20191219195013-becbf705a915
+	golang.org/x/sys v0.0.0-20191220220014-0732a990476f
 	golang.org/x/time v0.0.0-20191024005414-555d28b269f0 // indirect
-	google.golang.org/grpc v1.25.1 // indirect
+	google.golang.org/grpc v1.26.0 // indirect
 	sigs.k8s.io/yaml v1.1.0 // indirect
 )