go-wxhelper/internal/initialize/datatable.go
2024-02-19 14:57:36 +08:00

29 lines
773 B
Go

package initialize
import (
"gitee.ltd/lxh/logger/log"
"wechat-robot/internal/database"
"wechat-robot/model/entity"
)
// databaseTable
// @description: 初始化数据库表
func databaseTable() {
tables := []any{
new(entity.AdminUser), // 用户表
new(entity.Menu), // 菜单表
new(entity.Role), // 角色表
new(entity.RoleMenu), // 角色菜单表
new(entity.AdminUserRole), // 用户角色表
new(entity.SystemConfig), // 系统配置表
new(entity.Robot), // 机器人表
new(entity.AiAssistant), // AI助手表
new(entity.Message), // 微信消息表
}
// 同步表结构
if err := database.Client.AutoMigrate(tables...); err != nil {
log.Panicf("初始化数据库表失败: %v", err)
}
}