如何在NestJS中通过TypeORM实现Mongo与Postgres的无缝切换？

一、简便实现Postgres与MongoDB切换的方式 #

• 环境变量驱动连接配置：将数据库连接参数（类型、地址、用户名、密码等）抽离到环境变量或独立配置文件中，TypeORM会根据配置的type字段自动适配对应数据库驱动。
  示例配置（config/database.ts）：

  import { DataSource } from "typeorm";
  
  export default new DataSource({
    type: process.env.DB_TYPE as any, // 取值为 'postgres' 或 'mongodb'
    host: process.env.DB_HOST,
    port: parseInt(process.env.DB_PORT),
    username: process.env.DB_USER,
    password: process.env.DB_PASSWORD,
    database: process.env.DB_NAME,
    entities: [__dirname + '/../entities/**/*.ts'],
    synchronize: true,
  });
  

  启动服务时通过指定环境变量切换，比如：

  DB_TYPE=postgres npm run start
  # 或
  DB_TYPE=mongodb npm run start
  

• 封装通用数据访问层：把CRUD、查询等操作封装成通用工具类或Service，业务层仅调用这些通用方法，不直接依赖TypeORM的Repository或EntityManager，彻底屏蔽底层数据库差异。

二、TypeORM实体通用定义方案（无需手动修改字段） #

核心思路是通过自定义装饰器工厂，根据当前数据库类型动态应用对应的主键或字段装饰器，避免切换数据库时手动注释代码。

1. 自定义主键装饰器 #

创建一个适配不同数据库的主键装饰器：

// src/decorators/PrimaryColumn.ts
import { ObjectIdColumn, PrimaryGeneratedColumn, ColumnOptions } from "typeorm";

export function PrimaryColumn(options?: ColumnOptions) {
  return function (target: object, propertyKey: string) {
    if (process.env.DB_TYPE === 'mongodb') {
      ObjectIdColumn(options)(target, propertyKey);
    } else {
      PrimaryGeneratedColumn(options)(target, propertyKey);
    }
  };
}

2. 通用实体示例 #

在实体中使用自定义装饰器，实现主键自动适配：

// src/entities/User.ts
import { Entity, Column } from "typeorm";
import { PrimaryColumn } from "../decorators/PrimaryColumn";

@Entity()
export class User {
  @PrimaryColumn()
  id: string | number; // 根据数据库类型自动对应ObjectId或自增ID

  @Column()
  username: string;

  // 适配数组/JSON字段的示例
  @Column({
    type: process.env.DB_TYPE === 'mongodb' ? 'array' : 'json',
    nullable: true
  })
  tags: string[];
}

3. 注意事项 #

• 确保环境变量DB_TYPE在实体加载前已被正确加载（可在项目启动时优先加载.env文件）；
• 对于MongoDB特有的字段类型（如嵌套文档），可通过条件判断使用@Column的type: 'object'，在Postgres中对应jsonb类型；
• 迁移时需注意数据类型的兼容性，比如MongoDB的ObjectID可转为Postgres的UUID或字符串类型。

内容的提问来源于stack exchange，提问作者Dyram