浏览代码

✨ feat(server): 增强数据库初始化功能

- 导入initializeDataSource函数用于更灵活的数据源初始化
- 添加UserEntity、Role和File实体到数据库初始化过程
- 通过initializeDataSource显式指定需要加载的实体类,优化数据库连接设置
yourname 4 周之前
父节点
当前提交
69dfcbe501
共有 1 个文件被更改,包括 4 次插入1 次删除
  1. 4 1
      packages/server/src/index.ts

+ 4 - 1
packages/server/src/index.ts

@@ -1,6 +1,6 @@
 import { OpenAPIHono } from '@hono/zod-openapi'
 import { swaggerUI } from '@hono/swagger-ui'
-import { errorHandler } from '@d8d/shared-utils'
+import { errorHandler, initializeDataSource } from '@d8d/shared-utils'
 import { userRoutes as userModuleRoutes } from '@d8d/user-module'
 import { authRoutes as authModuleRoutes } from '@d8d/auth-module'
 import { fileRoutes as fileModuleRoutes } from '@d8d/file-module'
@@ -8,8 +8,11 @@ import { AuthContext } from '@d8d/shared-types'
 import { AppDataSource } from '@d8d/shared-utils'
 import { Hono } from 'hono'
 import { databaseBackup } from './utils/backup'
+import { UserEntity, Role } from '@d8d/user-module'
+import { File } from '@d8d/file-module'
 
 if(!AppDataSource.isInitialized) {
+  initializeDataSource([UserEntity, Role, File])
   await AppDataSource.initialize();
   console.log('数据库初始化完成')