auther 使用http request 作为参数

Author: yumaojun03

http/mock/auth.go

@@ -20,8 +20,8 @@ func NewMockAuther() router.Auther {
 
 type mockAuther struct{}
 
-func (m *mockAuther) Auth(h http.Header) (authInfo interface{}, err error) {
-	authHeader := h.Get("Authorization")
+func (m *mockAuther) Auth(r *http.Request) (authInfo interface{}, err error) {
+	authHeader := r.Header.Get("Authorization")
 	if authHeader == "" {
 		return nil, exception.NewUnauthorized("Authorization missed in header")
 	}

http/router/auther.go

@@ -6,7 +6,7 @@ import "net/http"
 // Header 用于鉴定身份
 // Entry 用于鉴定权限
 type Auther interface {
-	Auth(http.Header) (authInfo interface{}, err error)
+	Auth(*http.Request) (authInfo interface{}, err error)
 	Permission(authInfo interface{}, entry Entry) (err error)
 }
 

http/router/httprouter/httprouter.go

@@ -147,7 +147,7 @@ func (r *httpRouter) addHandler(method, path string, h http.Handler) {
 			}
 
 			if entry.AuthEnable {
-				ai, err := r.auther.Auth(req.Header)
+				ai, err := r.auther.Auth(req)
 				if err != nil {
 					response.Failed(w, exception.NewUnauthorized(err.Error()))
 					return