From d1a2ab79c932fccf7160a4a2ed501f0ebb9bd297 Mon Sep 17 00:00:00 2001
From: junleea <354425203@qq.com>
Date: Mon, 24 Mar 2025 13:48:58 +0800
Subject: [PATCH] =?UTF-8?q?=E6=B7=BB=E5=8A=A0=E5=8A=9F=E8=83=BD=E8=A1=A8?=
 =?UTF-8?q?=E5=88=9B=E5=BB=BA=E5=8F=8A=E5=8A=9F=E8=83=BD=E8=B7=AF=E7=94=B1?=
MIME-Version: 1.0
Content-Type: text/plain; charset=UTF-8
Content-Transfer-Encoding: 8bit

---
 dao/db.go |  6 ++++++
 main.go   | 15 ++++++++-------
 2 files changed, 14 insertions(+), 7 deletions(-)

diff --git a/dao/db.go b/dao/db.go
index 6050f57..742dfa2 100644
--- a/dao/db.go
+++ b/dao/db.go
@@ -6,6 +6,7 @@ import (
 	"gorm.io/driver/mysql"
 	"gorm.io/driver/postgres"
 	"gorm.io/gorm"
+	"log"
 )
 
 var DB *gorm.DB
@@ -59,6 +60,11 @@ func Init() error {
 		fmt.Println("model table:", err)
 		return err
 	}
+	err = db.AutoMigrate(&FunctionModel{})
+	if err != nil {
+		log.Println("function model table:", err)
+		return err
+	}
 
 	err = db.AutoMigrate(&Session{})
 	if err != nil {
diff --git a/main.go b/main.go
index 3263608..10896c5 100644
--- a/main.go
+++ b/main.go
@@ -31,13 +31,14 @@ 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
-	handler.SetUpSessionGroup(r) // Session
-	handler.SetUpModelGroup(r)   // Model
-	handler.SetUpIMGroup(r)      // IM
+	r.Use(JWTAuthMiddleware())     // 使用 JWT 认证中间件
+	handler.SetUpUserGroup(r)      // User
+	handler.SetUpToolGroup(r)      // Tool
+	handler.SetUpFileGroup(r)      // File
+	handler.SetUpSessionGroup(r)   // Session
+	handler.SetUpModelGroup(r)     // Model
+	handler.SetUpFuncModelGroup(r) // FuncModel
+	handler.SetUpIMGroup(r)        // IM
 	defer dao.Close()
 	defer worker.CloseRedis()
 	//定时任务