通过NestJS为所有TypeORM连接设置PostgreSQL角色的方案咨询

方案1：通过 TypeORM 配置透传 pg 连接池 afterCreate 钩子（最推荐） #

该方案利用 PostgreSQL Node.js 驱动（pg）原生的连接池回调能力，每个物理连接创建后仅执行一次指令，性能损耗最低，灵活性最高。
直接在 TypeOrmModule.forRoot() 的配置中添加 extra 字段即可，示例如下：

import { TypeOrmModule } from '@nestjs/typeorm';

@Module({
  imports: [
    TypeOrmModule.forRoot({
      type: 'postgres',
      host: '你的数据库地址',
      port: 5432,
      username: '数据库连接用户名',
      password: '数据库连接密码',
      database: '数据库名',
      entities: [/* 你的实体 */],
      synchronize: false,
      // 核心配置：透传给pg驱动的额外参数
      extra: {
        pool: {
          // 连接池创建新的物理连接后触发的回调
          afterCreate: (connection, done) => {
            connection.query('SET ROLE my-service', (err) => {
              // 执行完成后调用done，把连接返回给连接池
              done(err, connection);
            });
          }
        }
      }
    }),
    // 其他模块
  ],
})
export class AppModule {}

方案优势：

• 符合需求，每个连接仅执行一次SET ROLE指令，无额外冗余执行
• 不需要额外编写生命周期逻辑、事件监听逻辑，配置侵入性极低
• 不限制连接池大小、连接复用规则，所有连接都会生效

方案2：PostgreSQL 数据库层面配置默认角色 #

如果允许修改数据库配置，且所有使用该连接用户的服务都需要使用相同的ROLE，可以直接在数据库层面修改连接用户的默认配置，不需要修改应用代码：

1. 登录PostgreSQL执行以下SQL：

ALTER ROLE "你的数据库连接用户名" SET ROLE "my-service";

2. 配置生效后，所有使用该用户名建立的连接，会自动默认使用指定的ROLE，不需要应用层主动发送指令。
   方案局限性：

• 所有使用该连接用户名的应用都会受影响，不适合多服务共用连接用户的场景
• 角色调整需要修改数据库配置，灵活性低于应用层配置

方案3：监听 TypeORM DataSource 连接事件（备选） #

如果有更复杂的连接初始化逻辑，可以在Nest生命周期钩子中拿到DataSource实例，监听连接创建事件执行指令：

import { Injectable, OnModuleInit } from '@nestjs/common';
import { DataSource } from 'typeorm';

@Injectable()
export class DatabaseInitService implements OnModuleInit {
  constructor(private dataSource: DataSource) {}

  async onModuleInit() {
    this.dataSource.driver.on('connect', async (connection) => {
      await connection.query('SET ROLE my-service');
    });
  }
}

注意该方案需要确认你使用的TypeORM版本驱动支持connect事件，兼容性不如方案1。

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