NestJS项目中实现多数据源配置的最佳实践

在NestJS项目中，当我们需要同时连接多个数据库时，TypeORM提供了良好的支持。本文将详细介绍如何在NestJS框架中正确配置和使用多个数据源。

多数据源配置的基本原理

TypeORM支持在一个应用中创建多个连接实例，每个连接可以指向不同的数据库。在NestJS中，我们可以通过@nestjs/typeorm模块来实现这一功能。

配置步骤详解

1. 安装必要依赖

首先确保项目中已经安装了@nestjs/typeorm和typeorm包。

2. 创建数据源模块

为每个数据源创建独立的模块是推荐的做法。例如：

// database-a.module.ts
@Module({
  imports: [
    TypeOrmModule.forRootAsync({
      name: 'DsSchemaA',
      imports: [ConfigModule],
      inject: [ConfigService],
      useFactory: (config: ConfigService) => ({
        type: 'mysql',
        host: config.get('DB_A_HOST'),
        port: config.get('DB_A_PORT'),
        username: config.get('DB_A_USERNAME'),
        password: config.get('DB_A_PASSWORD'),
        database: config.get('DB_A_DATABASE'),
        entities: [UserEntity],
        synchronize: true,
      }),
    }),
  ],
  exports: [TypeOrmModule],
})
export class DatabaseAModule {}

3. 实体注册

对于每个数据源，需要明确指定其对应的实体：

// user.module.ts
@Module({
  imports: [
    TypeOrmModule.forFeature([UserEntity], 'DsSchemaA'),
  ],
  providers: [UserService],
  controllers: [UserController],
})
export class UserModule {}

4. 服务层注入

在服务层使用时，需要指定数据源名称：

@Injectable()
export class UserService {
  constructor(
    @InjectRepository(UserEntity, 'DsSchemaA')
    private readonly userRepository: Repository<UserEntity>,
  ) {}
}

高级用法：跨数据库查询

TypeORM支持跨数据库查询，但需要注意以下几点：

1. 所有数据库必须位于同一MySQL实例中
2. 查询时需要明确指定表所属的数据库

const results = await this.userRepository
  .createQueryBuilder('user', 'DsSchemaA')
  .innerJoin('DsSchemaB.company', 'company')
  .where('company.id = :companyId', { companyId })
  .getMany();

常见问题解决方案

1. 连接失败：检查各数据源的连接配置是否正确
2. 实体未注册：确保每个模块都正确注册了对应的实体
3. 注入错误：检查@InjectRepository装饰器中是否指定了正确的数据源名称

最佳实践建议

1. 为每个数据源创建独立的模块
2. 使用环境变量管理数据库连接配置
3. 在开发环境启用synchronize，生产环境禁用
4. 考虑使用迁移来管理数据库结构变更

通过以上配置，开发者可以在NestJS项目中灵活地使用多个数据源，满足复杂的业务需求。