NestJS中使用TypeORM时指定从PostgreSQL主库执行查询

针对你的问题，有几种可靠的方法让特定SELECT查询绕过从库、强制走PostgreSQL主库，以下是适配NestJS+TypeORM场景的具体实现：

1. 使用QueryRunner绑定主库连接 #

TypeORM的读写分离配置中，主库对应write连接，从库对应read连接。你可以通过QueryRunner显式获取主库连接执行查询：

import { DataSource, Repository } from 'typeorm';
import { InjectDataSource, InjectRepository } from '@nestjs/typeorm';
// 导入你的实体类
import { Entity } from './path-to-entity';

export class SomeClassName {
    constructor(
        @InjectRepository(Entity) private readonly entityRepository: Repository<Entity>,
        @InjectDataSource() private readonly dataSource: DataSource
    ) {}

    async findSomething(id: string): Promise<Entity> {
        // 创建主库连接的QueryRunner（连接名根据你的配置调整，默认主库是'write'）
        const queryRunner = this.dataSource.createQueryRunner('write');
        try {
            // 用主库连接执行查询
            return await queryRunner.manager.findOne(Entity, { where: { job_id: id } });
        } finally {
            // 必须释放QueryRunner，避免连接泄漏
            await queryRunner.release();
        }
    }
}

2. 用QueryBuilder指定主库连接 #

如果习惯使用QueryBuilder构建查询，可直接从主库连接创建QueryBuilder：

async findSomething(id: string): Promise<Entity> {
    const queryRunner = this.dataSource.createQueryRunner('write');
    try {
        return await this.dataSource
            .getRepository(Entity)
            .createQueryBuilder('entity')
            .setQueryRunner(queryRunner)
            .where('entity.job_id = :id', { id })
            .getOne();
    } finally {
        await queryRunner.release();
    }
}

3. 直接获取主库连接的Repository #

如果需要多次执行主库查询，可以直接获取主库连接对应的Repository实例：

async findSomething(id: string): Promise<Entity> {
    // 获取主库连接的Repository（连接名对应你的配置）
    const masterRepo = this.dataSource.getRepository(Entity, 'write');
    return masterRepo.findOne({ where: { job_id: id } });
}

4. 原生SQL兜底方案 #

如果以上Repository/QueryBuilder方式不适用，可直接用主库连接执行原生SQL：

async findSomething(id: string): Promise<Entity> {
    const result = await this.dataSource.query(
        'SELECT * FROM your_table_name WHERE job_id = $1',
        [id],
        'write' // 指定主库连接
    );
    return result[0] as Entity;
}

注意事项 #

• 确保你的TypeORM配置已正确开启读写分离：在DataSource配置中，主库信息放在顶层配置，从库信息放在replica数组中。
• 连接名（如write）需与配置一致，默认主库连接名为write，从库为read；若你自定义了主库连接名，需替换为对应名称。

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