处理事务中特定的 DataIntegrityViolationException

Question

我有一个非常基本的创建用户控制器。

    @PostMapping(consumes = MediaType.APPLICATION_JSON_VALUE, produces = MediaType.APPLICATION_JSON_VALUE)
    public ResponseEntity<String> createUser(@RequestBody UserInput userInput) {
        userControllerService.createUser(userInput);
        return ResponseEntity.ok("success");
    }

UserControllerService#createUser 方法是一个包含多个 SimpleJpaRepository#save 调用的事务。例如，

    @Transactional
    @Override
    public void createUser(UserInput userInput) {
        userRepository.save(userInput);
        profileRepository.save(userInput.getProfile());
    }

我希望能够让数据库处理唯一约束违规，并能够通知客户端特定的违规。

例如，如果我想当且仅当我遇到特定的约束违规时才通知客户。

    @PostMapping(consumes = MediaType.APPLICATION_JSON_VALUE, produces = MediaType.APPLICATION_JSON_VALUE)
    public ResponseEntity<String> createUser(@RequestBody UserInput userInput) {
        try {
            userControllerService.createUser(userInput);
        } catch (DuplicateUserNameException e) {
            return new ResponseEntity<>("", HttpStatus.BAD_REQUEST);
        } catch (DuplicateEmailException e) {
            return new ResponseEntity<>("", HttpStatus.BAD_REQUEST);
        } catch (Exception e) {
            return new ResponseEntity<>(HttpStatus.INTERNAL_SERVER_ERROR);
        }
        return ResponseEntity.ok("success");
    }

但是，任何约束违规都会在 UserControllerService#createUser 末尾引发 DataIntegrityViolationException。而且 DataIntegrityViolationException 太脆弱，无法依赖。它只有原因 SQLState: 23505（违反唯一约束）和非结构化消息，例如：

错误：重复的键值违反了唯一约束“unique_user” 详细信息：密钥（电子邮件）=([电子邮件受保护]) 已存在。

即使我添加自定义异常处理，它也永远不会运行，因为直到第一次实际调用数据库时，直到方法结束时才会遇到 DataIntegrityViolationException。例如，这没有效果。

    @Transactional
    @Override
    public void createUser(UserInput userInput) {
        try
          userRepository.save(userInput);
        } catch (Exception e) {
          throw new DuplicateUserNameException();
        }
        try {
          profileRepository.save(userInput.getProfile());
        } catch (Exception e) {
          throw new DuplicateEmailException();
        }
    }

我是否以错误的方式处理这个问题？看起来这是非常基本的功能，应该以某种方式实现。

我能想到的最好方法是添加一些代码来解析来自 DataIntegrityViolationException 的消息，但这有其局限性，例如，对于同一个表的两次插入对于应用程序具有不同的含义。第一个插入可能直接来自用户，第二个插入可能是应用程序生成的内容。仅通过解析详细消息可能无法从事务结束时区分两者。

我应该考虑其他实现吗？

Answer 1

如果我正确理解的话，听起来您想拥有一种可靠的方法来确定何时投掷dataintegrityViolationException是什么是导致它的确切原因，例如是由于重复的电子邮件还是重复的用户名。对于特定的用例或其他任何东西。

最简单的方法是不依靠抛出的例外来确定，而是在将数据保存到DB之前，积极发出某些SQL来对其进行验证，例如：

@Transactional
@Override
public void createUser(UserInput userInput) {

    if(userRepository.existUsername(userInput.getUsername()){
       throw new DuplicateUserNameException();
    }

    if(userRepository.existEmail(userInput.getEmail())){
      throw new DuplicateEmailException();
    }

    userRepository.save(userInput);
    profileRepository.save(userInput.getProfile());
}

Answer 2

你提到的问题

即使我添加自定义异常处理，它也永远不会运行，因为直到第一次实际调用数据库时，直到方法结束时才遇到 DataIntegrityViolationException。例如，这没有效果。

应该可以通过告诉 Hibernate 立即执行事务来解决，通过调用

userRepository.flush();

This 至少应该导致在该行上抛出异常。