添加Task触发外键约束异常，如何验证关联User已存在？

尝试添加关联数据库中不存在的用户的任务时，出现如下错误：

org.h2.jdbc.JdbcSQLIntegrityConstraintViolationException: Referential integrity 
constraint violation: "FK2HQOTYIYWTEUJXLQ0PHEMTLMF: PUBLIC.TBL_TASK FOREIGN 
KEY(USER_ID) REFERENCES PUBLIC.TBL_USER(USER_ID) (CAST(13 AS BIGINT))"; SQL

实体类定义 #

Task 类

@Table(name = "tbl_task")
public class Task {
    @Id
    @GeneratedValue(strategy = GenerationType.AUTO)
    private Long taskId;
    private String title;
    private String description;
    private Status status;

    @OneToOne
    @JoinColumn(name = "user_id", referencedColumnName = "userId")
    private User assignee;
}

User 类

@Table(name = "tbl_user")
public class User {
    @Id
    @GeneratedValue(strategy = GenerationType.AUTO)
    private Long userId;
    private String name;
    @Column(unique = true)
    private String email;
    private Active active;
    private String password;
}

控制器代码 #

@ResponseStatus(HttpStatus.CREATED)
public TaskResponse addTask(@Valid @RequestBody Task task){
    return taskService.addTask(task);
}

问题场景：若在用户数据入库前调用addTask()接口，会触发上述外键约束错误，需要确保调用接口前关联用户已存在于数据库中。

1. 业务层提前校验用户存在性 #

在taskService.addTask方法中，先校验关联用户是否存在，不存在则抛出业务异常，避免触发数据库层面的约束错误：

@Service
public class TaskServiceImpl implements TaskService {
    private final TaskRepository taskRepository;
    private final UserRepository userRepository;

    // 构造注入
    public TaskServiceImpl(TaskRepository taskRepository, UserRepository userRepository) {
        this.taskRepository = taskRepository;
        this.userRepository = userRepository;
    }

    @Override
    public TaskResponse addTask(Task task) {
        Long userId = task.getAssignee().getUserId();
        // 校验用户是否存在
        if (!userRepository.existsById(userId)) {
            throw new IllegalArgumentException("关联的用户不存在，用户ID：" + userId);
            // 也可以自定义业务异常，比如UserNotFoundException
        }
        // 用户存在则执行保存逻辑
        Task savedTask = taskRepository.save(task);
        return convertToResponse(savedTask);
    }
}

2. 全局异常处理返回友好提示 #

配合自定义异常，添加全局异常处理器，给前端返回标准化的错误响应：

@RestControllerAdvice
public class GlobalExceptionHandler {
    @ExceptionHandler(IllegalArgumentException.class)
    @ResponseStatus(HttpStatus.BAD_REQUEST)
    public Map<String, String> handleIllegalArgument(IllegalArgumentException ex) {
        Map<String, String> errorMap = new HashMap<>();
        errorMap.put("error", ex.getMessage());
        return errorMap;
    }

    // 若使用自定义异常
    @ExceptionHandler(UserNotFoundException.class)
    @ResponseStatus(HttpStatus.NOT_FOUND)
    public Map<String, String> handleUserNotFound(UserNotFoundException ex) {
        Map<String, String> errorMap = new HashMap<>();
        errorMap.put("error", ex.getMessage());
        return errorMap;
    }
}

3. 使用DTO接收请求（更规范的做法） #

避免直接用实体类接收请求，定义TaskCreateDTO仅接收必要字段，先校验userId对应的用户存在，再转换为Task实体：

// DTO类
public class TaskCreateDTO {
    @NotBlank(message = "标题不能为空")
    private String title;
    private String description;
    private Status status;
    @NotNull(message = "用户ID不能为空")
    private Long userId;

    // getter、setter
}

// 控制器修改
@ResponseStatus(HttpStatus.CREATED)
public TaskResponse addTask(@Valid @RequestBody TaskCreateDTO taskCreateDTO){
    return taskService.addTask(taskCreateDTO);
}

// Service层逻辑
@Override
public TaskResponse addTask(TaskCreateDTO dto) {
    // 校验用户存在，不存在则抛出异常
    User user = userRepository.findById(dto.getUserId())
            .orElseThrow(() -> new UserNotFoundException("用户不存在，ID：" + dto.getUserId()));
    // 转换为Task实体
    Task task = new Task();
    task.setTitle(dto.getTitle());
    task.setDescription(dto.getDescription());
    task.setStatus(dto.getStatus());
    task.setAssignee(user);
    // 保存任务
    Task savedTask = taskRepository.save(task);
    return convertToResponse(savedTask);
}

4. 禁用级联保存（避免误操作） #

确保@OneToOne注解没有添加cascade = CascadeType.PERSIST或CascadeType.ALL，防止JPA自动保存未入库的用户，保证只有已存在的用户才能关联任务：

@OneToOne
@JoinColumn(name = "user_id", referencedColumnName = "userId")
// 不要添加cascade属性，或明确设置cascade = {}
private User assignee;

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