NestJS创建记录前动态选择集合名称的实现问询

1. 定义输入DTO #

先创建DTO规范输入格式，保证类型安全：

// customer.dto.ts
import { IsString, IsObject } from 'class-validator';

export class CustomerDataDto {
  @IsString()
  customer_uuid: string; // 注意你输入示例里的拼写是cusomer_uuid，建议修正为customer_uuid

  @IsObject()
  value: {
    action: string;
    value: string;
  };
}

2. 实现动态集合插入的服务逻辑 #

利用Mongoose的连接实例动态获取目标集合，执行插入操作：

// customer.service.ts
import { Injectable } from '@nestjs/common';
import { InjectConnection } from '@nestjs/mongoose';
import { Connection } from 'mongoose';
import { CustomerDataDto } from './customer.dto';

@Injectable()
export class CustomerService {
  constructor(@InjectConnection() private readonly connection: Connection) {}

  async insertToDynamicCollection(dto: CustomerDataDto) {
    // 生成集合名称，统一小写适配MongoDB命名习惯
    const collectionName = `customer_${dto.customer_uuid.toLowerCase()}`;
    // 获取动态集合
    const collection = this.connection.collection(collectionName);
    // 插入数据，可按需添加额外字段（比如创建时间）
    await collection.insertOne({
      ...dto.value,
      createdAt: new Date(),
    });
    return { success: true, collectionName };
  }
}

3. 编写控制器接口 #

暴露HTTP接口接收请求，调用服务层逻辑：

// customer.controller.ts
import { Controller, Post, Body, ValidationPipe } from '@nestjs/common';
import { CustomerService } from './customer.service';
import { CustomerDataDto } from './customer.dto';

@Controller('customer')
export class CustomerController {
  constructor(private readonly customerService: CustomerService) {}

  @Post('data')
  async addCustomerData(@Body(ValidationPipe) dto: CustomerDataDto) {
    return this.customerService.insertToDynamicCollection(dto);
  }
}

4. 关键注意事项 #

• 确保项目已完成Mongoose的基础配置，能正常连接MongoDB。
• 输入示例里的cusomer_uuid存在拼写错误，建议修正为customer_uuid避免逻辑异常。
• 如果需要对集合做Schema校验或索引，可动态创建Schema并关联集合；若只是简单插入文档，直接使用connection.collection()更高效。
• 可在服务层添加权限校验，比如限制用户仅能操作自身对应的集合。

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