From 19b49d4c3c03a6640bd82c6888df776dfeef4c69 Mon Sep 17 00:00:00 2001
From: lijikun <ljk190610@163.com>
Date: Fri, 20 Dec 2024 16:57:53 +0800
Subject: [PATCH] =?UTF-8?q?12=E6=9C=8820=E6=97=A5=E4=B8=8B=E5=8D=88,?=
 =?UTF-8?q?=E5=AF=B9=E7=99=BB=E5=85=A5token=E9=AA=8C=E8=AF=81=E5=81=9A?=
 =?UTF-8?q?=E4=BA=86=E8=B0=83=E6=95=B4,=E5=AE=A2=E6=88=B7=E7=AB=AF?=
 =?UTF-8?q?=E5=90=8E=E5=8F=B0=E4=B8=8D=E5=86=B2=E7=AA=81=E4=B8=94=E5=8F=AF?=
 =?UTF-8?q?=E7=94=A8?=
MIME-Version: 1.0
Content-Type: text/plain; charset=UTF-8
Content-Transfer-Encoding: 8bit

---
 link_homework/api/v1/record/Record.go                 |  2 +-
 link_homework/internal/cmd/cmd.go                     |  3 ++-
 .../internal/logic/middleware/interceptor.go          | 19 +++++++++++++++++--
 3 files changed, 20 insertions(+), 4 deletions(-)

diff --git a/link_homework/api/v1/record/Record.go b/link_homework/api/v1/record/Record.go
index ea29366..9c84435 100644
--- a/link_homework/api/v1/record/Record.go
+++ b/link_homework/api/v1/record/Record.go
@@ -44,7 +44,7 @@ type GetShopInfoByDeptIdReq struct {
 }
 
 type GetShopInfoByDeptIdRes struct {
-	ShopId   string `json:"shopId" orm:"db:cms;member_info;column:shopId" dc:"门店id"`
+	ShopId   string `json:"shopId" orm:"db:cms;table:member_info;column:shopId" dc:"门店id"`
 	ShopName string `json:"shopName" orm:"db:cms;member_info;column:shopName" dc:"门店名"`
 }
 
diff --git a/link_homework/internal/cmd/cmd.go b/link_homework/internal/cmd/cmd.go
index 4b4dc72..c67a6ce 100644
--- a/link_homework/internal/cmd/cmd.go
+++ b/link_homework/internal/cmd/cmd.go
@@ -51,7 +51,8 @@ var (
 			//客户端
 			s.Group("/api/homework_client", func(group *ghttp.RouterGroup) {
 				//group.Middleware(middleware.MiddlewareCORS)
-				group.Middleware(middleware.MiddlewareIsLogin)
+				//group.Middleware(middleware.MiddlewareIsLogin)
+				group.Middleware(middleware.MiddlewareIsLoginClient)
 				//gfToken.Middleware(ctx, group)
 				//直接写接口,不用再分组
 				group.POST("/gethomeworklist", clientPage.NewClientPage().GetHomeworkList)
diff --git a/link_homework/internal/logic/middleware/interceptor.go b/link_homework/internal/logic/middleware/interceptor.go
index 464f1ef..a10bf80 100644
--- a/link_homework/internal/logic/middleware/interceptor.go
+++ b/link_homework/internal/logic/middleware/interceptor.go
@@ -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()