添加功能表创建及功能路由

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 {

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()
 	//定时任务