如何在Spring Boot中通过MongoTemplate实现MongoDB多字段唯一验证

我的MongoDB中有以下员工文档：

{
    "_id" : ObjectId("64cc954d8e68f438ca3c44b0"),
    "empId" : "EMP-30066",
    "empName" : "Alex_Mortin",
    "dept" : {
        "name" : "CE"
    },
    "country" : {
        "name" : "US"
    },
    "qualifications" : [ ],
    "_class" : "ama.me.model.backend.Employee"
},
{
    "_id" : ObjectId("64cc954d8e68f438ca3c44a8"),
    "empId" : "EMP-30077",
    "empName" : "Luisa_Deo",
    "dept" : {
        "name" : "DG"
    },
    "country" : {
        "name" : "US"
    },
    "qualifications" : [ ],
    "_class" : "ama.me.model.backend.Employee"
}

我需要在Spring Boot应用的POST和PUT请求前添加验证，确保每个对象的empId、empName、dept、country组合唯一，忽略qualifications字段。

我已经用MongoDB原生查询实现了该验证：

db.empLists.countDocuments({
    "empId" : "EMP-30066",
    "empName" : "Alex_Mortin",
    "dept" : {
            "name" : "CE"
        },
    "country" : {
            "name" : "US"
        }
    },{limit: 1})

现在需要把这个逻辑转换成Spring Boot中使用MongoTemplate的实现方式。

1. 核心查询条件构建 #

针对嵌套的dept.name和country.name字段，用Criteria类构建精确匹配的组合条件，同时设置limit(1)减少查询开销：

// 从请求中获取待验证的员工对象
Employee employee = ...;

Criteria criteria = Criteria.where("empId").is(employee.getEmpId())
    .and("empName").is(employee.getEmpName())
    .and("dept.name").is(employee.getDept().getName())
    .and("country.name").is(employee.getCountry().getName());

Query query = new Query(criteria);
query.limit(1);

2. POST请求的验证逻辑 #

POST是新增员工，直接统计符合条件的文档数量，若数量大于0则说明组合重复：

@Autowired
private MongoTemplate mongoTemplate;

// 新增时验证组合唯一性
public boolean isEmployeeCombinationUnique(Employee employee) {
    Criteria criteria = Criteria.where("empId").is(employee.getEmpId())
        .and("empName").is(employee.getEmpName())
        .and("dept.name").is(employee.getDept().getName())
        .and("country.name").is(employee.getCountry().getName());
    
    long count = mongoTemplate.count(new Query(criteria), Employee.class);
    return count == 0;
}

在POST接口中调用该方法，返回false时抛出重复数据异常即可。

3. PUT请求的验证逻辑 #

PUT是更新员工，需要排除当前要更新的文档本身，需额外添加_id不等于当前文档ID的条件：

// 更新时验证组合唯一性（排除自身）
public boolean isEmployeeCombinationUniqueForUpdate(Employee employee) {
    Criteria criteria = Criteria.where("empId").is(employee.getEmpId())
        .and("empName").is(employee.getEmpName())
        .and("dept.name").is(employee.getDept().getName())
        .and("country.name").is(employee.getCountry().getName())
        .and("_id").ne(employee.getId()); // 排除当前待更新的文档
    
    long count = mongoTemplate.count(new Query(criteria), Employee.class);
    return count == 0;
}

4. 性能优化建议 #

为了提升验证查询的性能，同时从数据库层面强制约束唯一性，建议给目标字段创建复合唯一索引：

db.empLists.createIndex(
    { empId: 1, empName: 1, "dept.name": 1, "country.name": 1 },
    { unique: true }
)

该索引可以有效避免并发场景下出现的重复数据问题。

内容的提问来源于stack exchange，提问作者Abhijit Mondal Abhi