TypeORM DataSource初始化后动态添加实体：可行性与最佳实践

import { DataSource } from "typeorm";
import * as path from "path";
// 假设 Product 和 Cart 实体定义在 ../entity/ 目录下

// Product 实体定义示例 (确保包含所有列)
// import { Entity, PrimaryGeneratedColumn, Column } from "typeorm";
// @Entity()
// export class Product {
//     @PrimaryGeneratedColumn()
//     id: number;
//     @Column()
//     name: string;
//     @Column({ nullable: true })
//     description: string;
// }

// Cart 实体定义示例 (确保包含所有列)
// @Entity()
// export class Cart {
//     @PrimaryGeneratedColumn()
//     id: number;
//     @Column()
//     userId: number;
//     @Column()
//     quantity: number;
// }


export const AppDataSource = new DataSource({
    type: "postgres",
    host: "localhost",
    port: 5432,
    username: "engineerhead",
    password: "",
    database: "test",
    synchronize: true, // 生产环境请谨慎使用 synchronize: true
    logging: false,
    entities: [
        // 使用通配符加载所有位于 '../entity/' 目录下的 .ts 或 .js 实体文件
        // 生产环境部署时，通常编译为 .js 文件，所以可能需要同时包含 .ts 和 .js
        path.join(__dirname, "../entity/*.ts"),
        path.join(__dirname, "../entity/*.js"),
        // 如果需要明确列出，也可以这样：
        // Product,
        // Cart,
    ],
    migrations: [],
    subscribers: [],
});

export default async () => {
    try {
        await AppDataSource.initialize();
        console.log("DataSource initialized successfully with all entities.");
    } catch (error) {
        console.error("Error during DataSource initialization:", error);
        process.exit(1); // 初始化失败，退出应用
    }
};

import { AppDataSource } from "./data-source";
import { AnotherNewEntity } from "../entity/AnotherNewEntity"; // 假设这是需要动态添加的实体

async function addEntityAndReinitialize() {
    // 1. 关闭现有 DataSource
    if (AppDataSource.isInitialized) {
        await AppDataSource.destroy();
        console.log("Existing DataSource destroyed.");
    }

    // 2. 获取原有实体列表，并添加新实体
    // 注意：这里需要重新构建一个全新的 DataSource 配置对象
    const currentEntities = AppDataSource.options.entities; // 这只能获取到初始化时的配置
    const newEntitiesConfig = Array.isArray(currentEntities) ? [...currentEntities, AnotherNewEntity] : [AnotherNewEntity];

    // 3. 创建并初始化新的 DataSource 实例
    const newDataSource = new DataSource({
        ...AppDataSource.options, // 复制原有配置
        entities: newEntitiesConfig, // 使用新的实体列表
    });

    try {
        await newDataSource.initialize();
        console.log("DataSource re-initialized with new entity.");
        // 现在你的应用程序应该使用 newDataSource 实例
        // 替换掉旧的 AppDataSource 引用
        // 例如：global.AppDataSource = newDataSource;
    } catch (error) {
        console.error("Error during DataSource re-initialization:", error);
        // 错误处理
    }
}

import { DataSource } from "typeorm";
import { Product } from "../entity/Product";
import { Cart } from "../entity/Cart";
import { User } from "../entity/User"; // 假设是另一个独立的实体

// 第一个 DataSource，用于管理核心业务实体
export const CoreDataSource = new DataSource({
    type: "postgres",
    host: "localhost",
    port: 5432,
    username: "core_user",
    password: "",
    database: "core_db",
    entities: [Product, Cart],
    synchronize: true,
});

// 第二个 DataSource，可能用于管理用户认证或特定模块的实体
export const AuthDataSource = new DataSource({
    type: "postgres",
    host: "localhost",
    port: 5432,
    username: "auth_user",
    password: "",
    database: "auth_db",
    entities: [User],
    synchronize: true,
});

export async function initAllDataSources() {
    await CoreDataSource.initialize();
    await AuthDataSource.initialize();
    console.log("All DataSources initialized.");
}

import { Entity, PrimaryGeneratedColumn, Column } from "typeorm";

@Entity()
export class Product {
    @PrimaryGeneratedColumn()
    id: number;

    @Column()
    name: string; // 必须使用 @Column() 标记

    @Column({ nullable: true }) // 可以定义列选项，如是否可空
    description: string;

    @Column({ type: "decimal", precision: 10, scale: 2 })
    price: number;
}

@Entity()
export class Cart {
    @PrimaryGeneratedColumn()
    id: number;

    @Column()
    userId: number; // 必须使用 @Column() 标记

    @Column()
    productId: number;

    @Column()
    quantity: number;
}