在NestJS与Mongoose中实现创建账号时自动关联创建用户

已附上数据库设计图：

我需要实现创建账号后自动创建关联的用户并保存到数据库，当前使用Mongoose + NestJS，已编写如下Schema：

Account Schema #

// Account Schema
export type AccountDocument = HydratedDocument<Account>

@Schema()
export class Account {
  // @Prop()
  // id: string;

  @Prop({ required: true, unique: true, immutable: true })
  username: string;

  @Prop({ required: true })
  password: string;

  @Prop({ required: true, default: false, immutable: true })
  isAdmin: boolean;

  @Prop({
    type: mongoose.Schema.Types.ObjectId,
    ref: 'User',
    unique: true
  })
  user: User;
}

export const AccountSchema = SchemaFactory.createForClass(Account)

User Schema #

//User schema
export type UserDocument = HydratedDocument<User>

@Schema()
export class User {
  // @Prop()
  // id: string;

  @Prop({ required: false })
  displayName: string;

  @Prop({ required: false })
  address: string;

  @Prop({ required: false })
  phoneNumber: string;

  @Prop({ required: false, enum: genderEnum })
  gender: genderEnum;

  @Prop({ required: false })
  avatar: string;

  @Prop({
    type: mongoose.Schema.Types.ObjectId,
    ref: 'Account',
    unique: true
  })
  account: Account
}

export const UserSchema = SchemaFactory.createForClass(User)

求实现创建账号时自动创建关联用户的方法，感谢帮助！

有两种实用方式可以实现这个需求，根据你的场景选择：

方式一：使用Mongoose pre('save') 中间件（推荐） #

在Account Schema上添加保存前钩子，自动创建User并完成双向关联：

// 在AccountSchema定义后添加中间件
AccountSchema.pre('save', async function(next) {
  // 仅在新建Account时执行，避免更新操作重复创建User
  if (this.isNew) {
    try {
      const userModel = mongoose.model('User');
      // 创建默认User实例，可根据需求设置默认值
      const newUser = new userModel({
        displayName: this.username // 比如用账号名作为默认昵称
      });
      await newUser.save();
      
      // 关联User到Account
      this.user = newUser._id;
      // 反向关联Account到User
      await userModel.findByIdAndUpdate(newUser._id, { account: this._id });
    } catch (err) {
      next(err);
      return;
    }
  }
  next();
});

注意：确保User模型已在NestJS模块中注册，否则mongoose.model('User')会报错。

方式二：在Service层手动处理逻辑 #

在Account的创建服务中，先创建User再关联Account，全程手动控制流程：

// account.service.ts
import { Injectable } from '@nestjs/common';
import { InjectModel } from '@nestjs/mongoose';
import { Model } from 'mongoose';
import { Account, AccountDocument } from './schemas/account.schema';
import { User, UserDocument } from '../user/schemas/user.schema';

@Injectable()
export class AccountService {
  constructor(
    @InjectModel(Account.name) private accountModel: Model<AccountDocument>,
    @InjectModel(User.name) private userModel: Model<UserDocument>,
  ) {}

  async create(createAccountDto: any): Promise<Account> {
    // 1. 创建基础User实例
    const newUser = new this.userModel({
      displayName: createAccountDto.username
    });
    await newUser.save();

    // 2. 创建Account并关联User
    const newAccount = new this.accountModel({
      ...createAccountDto,
      user: newUser._id
    });
    const savedAccount = await newAccount.save();

    // 3. 完成反向关联
    await this.userModel.findByIdAndUpdate(newUser._id, {
      account: savedAccount._id
    });

    return savedAccount;
  }
}

注意事项 #

1. 因为你的Schema是双向关联，必须同时维护两边的关联字段，避免数据不一致；
2. 密码建议用bcrypt等库哈希后再保存，不要明文存储；
3. 中间件方式要做好异常捕获，避免创建Account失败但User已被创建的情况，必要时可以加事务处理。

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