如何在Nest.js+TypeORM中实现启动时执行PostgreSQL多表关联查询

下面是针对需求的具体实现步骤，涵盖多对多关系配置和服务启动时自动执行目标查询的逻辑：

一、正确配置多对多实体关系 #

TypeORM 支持直接通过实体注解定义多对多关系，无需手动操作中间表（若中间表无额外字段）。

1. Category 实体类 #

import { Entity, PrimaryGeneratedColumn, Column, ManyToMany, JoinTable } from 'typeorm';
import { Subcategory } from './subcategory.entity';

@Entity('category')
export class Category {
  @PrimaryGeneratedColumn()
  category_id: number;

  @Column()
  name: string;

  @ManyToMany(() => Subcategory)
  @JoinTable({
    name: 'category_subcategory', // 指定中间表名称
    joinColumn: {
      name: 'category_id',
      referencedColumnName: 'category_id',
    },
    inverseJoinColumn: {
      name: 'subcategory_id',
      referencedColumnName: 'subcategory_id',
    },
  })
  subcategories: Subcategory[];
}

2. Subcategory 实体类 #

import { Entity, PrimaryGeneratedColumn, Column, ManyToMany } from 'typeorm';
import { Category } from './category.entity';

@Entity('subcategory')
export class Subcategory {
  @PrimaryGeneratedColumn()
  subcategory_id: number;

  @Column()
  name: string;

  @ManyToMany(() => Category, category => category.subcategories)
  categories: Category[];
}

二、实现服务启动时自动执行查询 #

利用 Nest.js 的 OnModuleInit 生命周期钩子，在模块初始化阶段执行目标查询。

1. 编写业务服务类 #

import { Injectable, OnModuleInit } from '@nestjs/common';
import { InjectRepository } from '@nestjs/typeorm';
import { Repository } from 'typeorm';
import { Category } from './entities/category.entity';

@Injectable()
export class CategoryService implements OnModuleInit {
  constructor(
    @InjectRepository(Category)
    private readonly categoryRepo: Repository<Category>,
  ) {}

  async onModuleInit() {
    // 复刻 pgAdmin 中的查询逻辑
    const queryResult = await this.categoryRepo
      .createQueryBuilder('category')
      .select(['category.name', 'subcategory.name'])
      .innerJoin('category.subcategories', 'subcategory')
      .orderBy('category.name', 'ASC')
      .addOrderBy('subcategory.name', 'ASC')
      .getRawMany(); // 获取与 pgAdmin 格式一致的原始结果集

    // 可根据需求处理结果，比如打印日志、存入缓存等
    console.log('启动时执行查询结果：', queryResult);
  }
}

2. 模块注册配置 #

在对应业务模块中注册实体与服务：

import { Module } from '@nestjs/common';
import { TypeOrmModule } from '@nestjs/typeorm';
import { Category } from './entities/category.entity';
import { Subcategory } from './entities/subcategory.entity';
import { CategoryService } from './category.service';

@Module({
  imports: [TypeOrmModule.forFeature([Category, Subcategory])],
  providers: [CategoryService],
})
export class CategoryModule {}

三、常见问题排查 #

• 若手动创建过 category_subcategory 表，需确保表结构（字段名、外键关联）与 TypeORM 生成的一致
• 检查 TypeORM 配置，确认数据库连接正常、实体路径被正确扫描
• 若使用 getMany() 替代 getRawMany()，返回的是实体对象而非扁平结果集，可能与预期格式不符

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