update: 初始化代码提交

internal/global/g_middleware/middleware.go

@@ -0,0 +1,191 @@
+package g_middleware
+
+import (
+	"context"
+	"github.com/gogf/gf/v2/errors/gerror"
+	"github.com/gogf/gf/v2/frame/g"
+	"github.com/gogf/gf/v2/net/ghttp"
+	"github.com/gogf/gf/v2/os/glog"
+	"github.com/golang-jwt/jwt/v5"
+	"home-network-watcher/manifest"
+	"sync"
+	"time"
+)
+
+type sMiddleware struct{} // 创建结构体
+var (
+	once         = &sync.Once{}    // 创建互锁
+	s            *sMiddleware      // 创建指针
+	SMiddlewares = newMiddleware() // 对外暴露
+)
+
+// defaultHandlerResponse 返回结构体
+type defaultHandlerResponse struct {
+	Status  int         `json:"status"  dc:"Error code"` // 可业务需要可以更改json字段
+	Message string      `json:"msg"     dc:"Error message"`
+	Data    interface{} `json:"data"    dc:"Result data for certain request according API definition"`
+}
+
+// newMiddleware 单例中间件
+func newMiddleware() *sMiddleware {
+	once.Do(func() {
+		s = &sMiddleware{}
+	})
+	return s
+}
+
+func (s *sMiddleware) MiddlewareCORS(r *ghttp.Request) {
+	r.Response.CORSDefault()
+	r.Middleware.Next()
+}
+
+// resWriteJson 返回json输出
+func (s *sMiddleware) resWriteJson(r *ghttp.Request, in defaultHandlerResponse) {
+	r.Response.ClearBuffer()
+	r.Response.WriteJson(defaultHandlerResponse{
+		Status:  in.Status,
+		Message: in.Message,
+		Data:    in.Data,
+	})
+}
+
+// ErrorsStatus 服务器错误码处理
+func (s *sMiddleware) ErrorsStatus(server *ghttp.Server) {
+	server.BindStatusHandlerByMap(map[int]ghttp.HandlerFunc{
+		500: func(r *ghttp.Request) {
+			s.resWriteJson(r, defaultHandlerResponse{
+				Status:  500,
+				Message: "Error 500,Internal Server Error",
+				Data:    "",
+			})
+		},
+		404: func(r *ghttp.Request) {
+			s.resWriteJson(r, defaultHandlerResponse{
+				Status:  404,
+				Message: "Error 404,Not Found",
+				Data:    "",
+			})
+		},
+	})
+}
+
+// JWTAuth 鉴权中间件
+func (s *sMiddleware) JWTAuth(r *ghttp.Request) {
+	JWTString := r.GetHeader("Authorization")
+	if JWTString == "" {
+		s.resWriteJson(r, defaultHandlerResponse{
+			Status:  401,
+			Message: "JWT为空",
+			Data:    nil,
+		})
+		return
+	}
+
+	// 验证JWT
+	token, err := jwt.Parse(JWTString, func(token *jwt.Token) (interface{}, error) {
+		return manifest.JWTKey, nil
+	})
+	if err != nil {
+		s.resWriteJson(r, defaultHandlerResponse{
+			Status:  401,
+			Message: "JWT验证失败",
+			Data:    nil,
+		})
+		return
+	}
+
+	if !token.Valid {
+		s.resWriteJson(r, defaultHandlerResponse{
+			Status:  401,
+			Message: "JWT验证失败",
+			Data:    nil,
+		})
+		return
+	}
+	// 验证是否过期
+	expirationTime, err := token.Claims.GetExpirationTime()
+	if err != nil {
+		s.resWriteJson(r, defaultHandlerResponse{
+			Status:  401,
+			Message: "JWT验证失败",
+			Data:    nil,
+		})
+		return
+	}
+	if expirationTime.Before(time.Now()) {
+		s.resWriteJson(r, defaultHandlerResponse{
+			Status:  401,
+			Message: "JWT已过期",
+			Data:    nil,
+		})
+		return
+	}
+
+	glog.Info(context.TODO(), "JWT验证通过")
+	// 验证通过，设置用户信息
+	audience, err := token.Claims.GetAudience()
+	if err != nil {
+		s.resWriteJson(r, defaultHandlerResponse{
+			Status:  401,
+			Message: "JWT验证失败",
+			Data:    nil,
+		})
+		return
+	}
+
+	r.SetCtxVar("user_id", audience[0])
+	r.Middleware.Next()
+}
+
+// ResponseHandler is the middleware handling handler response object and its error. 中间件处理处理程序响应对象及其错误
+func (s *sMiddleware) ResponseHandler(r *ghttp.Request) {
+	//defer action_log.NewActionLog().LogAdd(r) // 日志钩子
+	r.Middleware.Next()
+
+	// There's custom buffer content, it then exits current handler.
+	if r.Response.BufferLength() > 0 {
+		return
+	}
+
+	// 定义参数和类型.方便调用
+	var (
+		code    int
+		msg     string
+		resData = r.GetHandlerResponse() // 检索并返回处理程序响应对象及其错误
+		err     = r.GetError()           // 返回请求过程中发生的错误。如果没有错误，则返回 nil
+		errCode = gerror.Code(err)       // 将错误信息通过接口解析处理
+	)
+	code = 200      // 默认返回状态码信息码
+	msg = "success" // 默认成功返回响应体文本信息
+	if err != nil {
+		code = 400 // 默认返回状态码信息码
+		switch errCode.Code() {
+		case -1:
+			msg = "Nonstandard error return"
+			resData = "I won't expose it to you. Go back and change it!"
+		case 500:
+			r.Response.Writer.Status = 500
+			return
+		default:
+			msg = errCode.Message()
+			resData = err.Error()
+			r.Response.Writer.Status = 400
+			if msg == "" || msg == "Unknown error reason" {
+				msg = "Nonstandard error return"
+				resData = "I won't expose it to you. Go back and change it!"
+			}
+			if resData == msg {
+				resData = ""
+			}
+		}
+	}
+	// 如果是空res返回，则resData返回空字符串
+	if g.IsNil(resData) {
+		resData = ""
+	}
+	s.resWriteJson(r, defaultHandlerResponse{
+		Status:  code,
+		Message: msg,
+		Data:    resData,
+	})
+}