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

This commit is contained in:
junleea 2025-03-24 13:48:58 +08:00
parent c8c55b4cf4
commit d1a2ab79c9
2 changed files with 14 additions and 7 deletions

View File

@ -6,6 +6,7 @@ import (
"gorm.io/driver/mysql" "gorm.io/driver/mysql"
"gorm.io/driver/postgres" "gorm.io/driver/postgres"
"gorm.io/gorm" "gorm.io/gorm"
"log"
) )
var DB *gorm.DB var DB *gorm.DB
@ -59,6 +60,11 @@ func Init() error {
fmt.Println("model table:", err) fmt.Println("model table:", err)
return err return err
} }
err = db.AutoMigrate(&FunctionModel{})
if err != nil {
log.Println("function model table:", err)
return err
}
err = db.AutoMigrate(&Session{}) err = db.AutoMigrate(&Session{})
if err != nil { if err != nil {

15
main.go
View File

@ -31,13 +31,14 @@ func main() {
panic("failed to connect redis:" + err.Error()) panic("failed to connect redis:" + err.Error())
} }
r.Use(handler.CrosHandler()) r.Use(handler.CrosHandler())
r.Use(JWTAuthMiddleware()) // 使用 JWT 认证中间件 r.Use(JWTAuthMiddleware()) // 使用 JWT 认证中间件
handler.SetUpUserGroup(r) // User handler.SetUpUserGroup(r) // User
handler.SetUpToolGroup(r) // Tool handler.SetUpToolGroup(r) // Tool
handler.SetUpFileGroup(r) // File handler.SetUpFileGroup(r) // File
handler.SetUpSessionGroup(r) // Session handler.SetUpSessionGroup(r) // Session
handler.SetUpModelGroup(r) // Model handler.SetUpModelGroup(r) // Model
handler.SetUpIMGroup(r) // IM handler.SetUpFuncModelGroup(r) // FuncModel
handler.SetUpIMGroup(r) // IM
defer dao.Close() defer dao.Close()
defer worker.CloseRedis() defer worker.CloseRedis()
//定时任务 //定时任务