如何确保同一Company仅存在一个main=true的Office？JPA方案问询

当然可以通过JPA结合Bean Validation（校验框架）实现这个唯一性约束，但要注意应用层验证的局限性，建议搭配数据库层面的约束来彻底规避风险：

一、应用层实现：自定义JPA校验逻辑 #

你可以给Company实体的Office集合添加自定义校验注解，确保集合中标记为main=true的Office数量不超过1：

1. 定义自定义校验注解

@Target({ElementType.FIELD})
@Retention(RetentionPolicy.RUNTIME)
@Constraint(validatedBy = SingleMainOfficeValidator.class)
public @interface SingleMainOffice {
    String message() default "一个公司只能设置一个主办公地点";
    Class<?>[] groups() default {};
    Class<? extends Payload>[] payload() default {};
}

2. 实现校验器逻辑

public class SingleMainOfficeValidator implements ConstraintValidator<SingleMainOffice, List<Office>> {
    @Override
    public boolean isValid(List<Office> offices, ConstraintValidatorContext context) {
        if (offices == null || offices.isEmpty()) {
            return true;
        }
        // 统计main为true的Office数量
        long mainOfficeCount = offices.stream()
                .filter(Office::isMain)
                .count();
        return mainOfficeCount <= 1;
    }
}

3. 在Company实体中使用注解

@Entity
public class Company {
    // 其他字段和映射关系
    @OneToMany(mappedBy = "company", cascade = CascadeType.ALL, orphanRemoval = true)
    @SingleMainOffice
    private List<Office> offices = new ArrayList<>();

    // getter、setter及业务方法
}

4. 触发校验
   在保存或更新Company的时候，需要主动触发校验：

• 如果你用Spring，可以在Service层方法的参数上加上@Valid注解，让Spring自动执行校验；
• 也可以手动调用校验器：validator.validate(company)，校验不通过会抛出ConstraintViolationException。

二、关键局限性：并发风险 #

这种应用层验证有个致命问题：无法处理并发场景。比如两个请求同时修改同一个Company的Office，都将不同的Office设置为main=true，应用层的校验可能因为事务隔离的问题，各自都检测不到对方的修改，最终导致数据库中出现多个主Office。

三、数据库层面兜底约束（必加） #

为了彻底解决这个问题，一定要配合数据库的部分唯一索引（PostgreSQL、MySQL 8.0+等主流数据库都支持）：

CREATE UNIQUE INDEX idx_company_main_office ON office (company_id) WHERE main = true;

这个索引会强制要求：同一个company_id下，只能有一条main=true的记录。即使应用层校验失效，数据库也会直接抛出错误，阻止违规操作。

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