Question

我目前正在使用弹簧靴来实现电子公告板API。我将向您展示的源代码场景是关于公告板或注释修改的，

在进行修改之前，在输入评论或公告栏时都需要提交密码。

因此，我只需使用'isBoard'作为输入：隐藏值即可在'boardRepository'中执行'findById'，然后将其与passwordEncoder的匹配功能进行比较，然后更新布尔值。

但是，下面的代码显示重复并使用了相同的逻辑。有没有办法使它更简单？

 @Transactional
public boolean tryToUpdateArticleOrReply(Long id, String password, boolean isBoard) throws ApiException {
    AtomicBoolean result = new AtomicBoolean(false);

    if(isBoard) {
         boardRepository.findById(id).ifPresent(
                board -> {
                    result.compareAndSet(passwordEncoder.matches(board.getUserPass(), password), true);
                }
         );
    } else {
        replyRepository.findById(id).ifPresent(
                reply -> {
                    result.compareAndSet(passwordEncoder.matches(reply.getUserPass(), password), true);
                }
        );
    }
    if(result.get())
      return result.get();

    throw new ApiException("INVALID_USER_PASS", "you submitted invaild password.", new ApiExceptionData().add("user_pass", password));
}

Answer 1

如果您的findById和getUserPass方法来自接口（并且findById方法返回getUserPass来自的接口），那么您可以简单地做到： >

private boolean check(FindByIdInterface repository,
                      Long id,
                      String password) {
    return repository.findById(id)
            .map(result -> 
                passwordEncoder.matches(
                    result.getUserPass(), password))
            .orElse(false); 
}

然后在根据您的repository变量选择要传递的isBoard的地方调用它。 isBoard ? boardRepository : replyRepository

如果它们不来自共享接口，则可以使用Java中的某些功能接口有效地“假装”它们。

/* R = repository type, T = return type of 'findById' */
private <R, T> check(Function<R, Optional<T>> getByIdMapper,
                     Function<T, String> getUserPassMapper,
                     R repository,
                     Long id,
                     String password) {
    return getByIdMapper.apply(repository)
            .map(result ->
                passwordEncoder.matches(
                    getUserPassMapper.apply(result),
                    password))
            .orElse(false);
}

然后用类似这样的名称来调用它：

if (isBoard) {
    return check(BoardRepository::getById,
                 Board::getUserPass,
                 boardRepository, id, password);
} else {
    // ...
}

注意，在任何情况下都不需要AtomicBoolean。

Answer 2

您可以这样做：

@Transactional
public boolean tryToUpdateArticleOrReply(Long id, String password, boolean isBoard) throws ApiException {
    Optional<String> optUserPass;
    if (isBoard) {
        optUserPass = boardRepository.findById(id).map(Board::getUserPass);
    else {
        optUserPass = replyRepository.findById(id).map(Reply::getUserPass);
    }
    if (optUserPass.isPresent() && passwordEncoder.matches(password, optUserPass.get()))
        return true;
    throw new ApiException("INVALID_USER_PASS", "you submitted invaild password.", new ApiExceptionData().add("user_pass", password));
}

注1：方法名称似乎有点不正确。

注释2：为什么boolean只能返回true时返回类型？

有没有一种方法可以简化该源代码？

2 个答案: