在NextJS+Mongo用户注册API中，单userModel.ts管理多角色必填字段

你的代码存在几个问题，无法直接正确运行，下面逐一说明并给出最优方案：

原代码的问题 #

1. Schema语法错误：Mongoose Schema不支持用TypeScript的?:语法标记可选字段，正确的做法是通过required: false（或不指定required，默认就是可选）来定义可选字段。
2. 缺少角色与专属字段的关联校验：当前代码没有限制角色和对应专属字段的匹配关系，比如候选人用户可能同时存在企业专属字段，导致数据混乱。
3. 专属字段未定义具体结构：空的candidateSpecific和businessSpecific对象无法发挥Mongoose的数据校验、类型约束作用。

最优方案选择 #

根据角色字段的差异程度，推荐两种方案：

方案一：嵌入式专属字段（适合字段差异小的场景） #

将不同角色的专属字段嵌套在Schema中，通过自定义校验确保角色与字段匹配：

import mongoose from 'mongoose';

const userSchema = new mongoose.Schema({
  email: {
    type: String,
    required: true,
    unique: true,
  },
  password: {
    type: String,
    required: true,
  },
  role: {
    type: String,
    enum: ['candidate', 'business'],
    required: true,
  },
  // 候选人专属字段，按需定义必填/可选
  candidateSpecific: {
    fullName: { type: String, required: false },
    resumeUrl: { type: String, required: false },
    skills: [{ type: String }]
    // 补充其他候选人字段
  },
  // 企业专属字段，按需定义必填/可选
  businessSpecific: {
    companyName: { type: String, required: false },
    industry: { type: String, required: false },
    companySize: { type: String, enum: ['1-10', '11-50', '51-200', '200+'] }
    // 补充其他企业字段
  }
});

// 前置校验：确保角色与专属字段匹配
userSchema.pre('save', function(next) {
  const user = this;
  if (user.role === 'candidate') {
    // 候选人不能有企业专属字段
    if (user.businessSpecific && Object.keys(user.businessSpecific).length > 0) {
      return next(new Error('候选人用户不允许包含企业专属字段'));
    }
    // 可选：校验候选人必填的专属字段
    // if (!user.candidateSpecific?.fullName) {
    //   return next(new Error('候选人姓名不能为空'));
    // }
  } else if (user.role === 'business') {
    // 企业不能有候选人专属字段
    if (user.candidateSpecific && Object.keys(user.candidateSpecific).length > 0) {
      return next(new Error('企业用户不允许包含候选人专属字段'));
    }
    // 可选：校验企业必填的专属字段
    // if (!user.businessSpecific?.companyName) {
    //   return next(new Error('企业名称不能为空'));
    // }
  }
  next();
});

export default mongoose.models.User || mongoose.model('User', userSchema);

方案二：Mongoose鉴别器（适合字段差异大、逻辑复杂的场景） #

鉴别器是Mongoose的继承实现方案，共享基础字段，各自定义专属字段，数据存储在同一集合，同时保留独立的模型逻辑：

import mongoose from 'mongoose';

// 基础用户Schema：仅包含公共字段
const baseUserSchema = new mongoose.Schema({
  email: {
    type: String,
    required: true,
    unique: true,
  },
  password: {
    type: String,
    required: true,
  },
  role: {
    type: String,
    enum: ['candidate', 'business'],
    required: true,
  }
}, { discriminatorKey: 'role' }); // 指定role为鉴别器字段

// 创建基础用户模型
const BaseUser = mongoose.models.BaseUser || mongoose.model('BaseUser', baseUserSchema);

// 候选人鉴别器模型：定义专属字段及校验
const Candidate = BaseUser.discriminator('candidate', new mongoose.Schema({
  fullName: { type: String, required: true },
  resumeUrl: { type: String, required: true },
  skills: [{ type: String }]
}));

// 企业鉴别器模型：定义专属字段及校验
const Business = BaseUser.discriminator('business', new mongoose.Schema({
  companyName: { type: String, required: true },
  industry: { type: String, required: true },
  companySize: { type: String, enum: ['1-10', '11-50', '51-200', '200+'], required: true }
}));

// 导出所有模型，按需使用
export { BaseUser, Candidate, Business };

方案对比 #

• 嵌入式字段：实现简单，查询便捷，适合角色间字段差异小、业务逻辑统一的场景。
• 鉴别器：分离不同角色的模型逻辑，校验更精准，适合角色间字段差异大、各自有独立业务规则的场景。

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