From 1a05537b1f4e8000329902047021d7c296850bf7 Mon Sep 17 00:00:00 2001
From: junleea <354425203@qq.com>
Date: Fri, 21 Mar 2025 16:32:55 +0800
Subject: [PATCH] =?UTF-8?q?=E5=AE=8C=E5=96=84=E4=BC=9A=E8=AF=9D=E7=AE=A1?=
 =?UTF-8?q?=E7=90=86=E3=80=81=E6=A8=A1=E5=9E=8B=E7=AE=A1=E7=90=86?=
MIME-Version: 1.0
Content-Type: text/plain; charset=UTF-8
Content-Transfer-Encoding: 8bit

---
 main.go | 10 ++++++----
 1 file changed, 6 insertions(+), 4 deletions(-)

diff --git a/main.go b/main.go
index 3bb60b3..4a4dd33 100644
--- a/main.go
+++ b/main.go
@@ -31,10 +31,12 @@ func main() {
 		panic("failed to connect redis:" + err.Error())
 	}
 	r.Use(handler.CrosHandler())
-	r.Use(JWTAuthMiddleware()) // 使用 JWT 认证中间件
-	handler.SetUpUserGroup(r)  // User
-	handler.SetUpToolGroup(r)  // Tool
-	handler.SetUpFileGroup(r)  // File
+	r.Use(JWTAuthMiddleware())   // 使用 JWT 认证中间件
+	handler.SetUpUserGroup(r)    // User
+	handler.SetUpToolGroup(r)    // Tool
+	handler.SetUpFileGroup(r)    // File
+	handler.SetUpSessionGroup(r) // Session
+	handler.SetUpModelGroup(r)   // Model
 	defer dao.Close()
 	defer worker.CloseRedis()
 	//定时任务