|
|
@ -3,6 +3,7 @@ package middleware |
|
|
|
import ( |
|
|
|
"github.com/gogf/gf/v2/net/ghttp" |
|
|
|
"link_homework/internal/model/dto" |
|
|
|
"link_homework/internal/service" |
|
|
|
) |
|
|
|
|
|
|
|
func MiddlewareCORS(r *ghttp.Request) { |
|
|
@ -12,10 +13,24 @@ func MiddlewareCORS(r *ghttp.Request) { |
|
|
|
|
|
|
|
func MiddlewareIsLogin(r *ghttp.Request) { |
|
|
|
token := r.GetHeader("token") |
|
|
|
if token == "" { |
|
|
|
valid, err := service.LoginLogic().ValidateToken(r.Context(), token) |
|
|
|
if err != nil || !valid { |
|
|
|
r.Response.WriteJsonExit(dto.Result{ |
|
|
|
Code: 401, |
|
|
|
Message: "未登录", |
|
|
|
Message: "Token 无效123", |
|
|
|
}) |
|
|
|
} else { |
|
|
|
r.Middleware.Next() |
|
|
|
} |
|
|
|
} |
|
|
|
|
|
|
|
func MiddlewareIsLoginClient(r *ghttp.Request) { |
|
|
|
token := r.GetHeader("token") |
|
|
|
const LocalToken = "9ior41AF0xTIbIG2pRnnbZi0+fEeMx8pywnIlrmTwo5FbqJ9lWrSWOxp9MkpKiNtedtUafqvzIwpFKrwuMs" |
|
|
|
if token != LocalToken { |
|
|
|
r.Response.WriteJsonExit(dto.Result{ |
|
|
|
Code: 401, |
|
|
|
Message: "Token 无效123", |
|
|
|
}) |
|
|
|
} else { |
|
|
|
r.Middleware.Next() |
|
|
|