使用lambda的内联过滤避免意外的null

Question

我有我想要这样做的每个元素的列表(使用Java 8):

disabledUsersOnLDAP.stream().forEach(user ->usersRepository.findEnabledByUsername(user.getUserName()).setEnabled(false));

怎么usersRepository.findEnabledByUsername可能会返回null.当然,我可以这样做:

disabledUsersOnLDAP.stream().forEach(user -> {
            UserEntity userEntity = usersRepository.findEnabledByUsername(user.getUserName());
            {
                if (userEntity != null) {
                    userEntity.setEnabled(false);
                }
            }
        });

但我想知道我是否可以进行内联空检查(在第一个选项中)？

Answer 1

你可以在运行forEach部件之前进行映射+过滤,这也使它更具可读性:

disabledUsersOnLDAP.stream()
            .map(user -> usersRepository.findEnabledByUsername(user.getUserName()))
            .filter(userEntity -> userEntity != null)
            .forEach(userEntity -> userEntity.setEnabled(false));

或作为替代方案:

disabledUsersOnLDAP.stream()
            .map(User::getUsername)
            .map(usersRepository::findEnabledByUsername)
            .filter(Objects::nonNull)
            .forEach(userEntity -> userEntity.setEnabled(false));

Answer 2

@assylias 答案的一些替代方案.

使用对象的方法引用== nonNull进行null检查:

disabledUsersOnLDAP.stream()
    .map(User::getUsername)
    .map(usersRepository::findEnabledByUsername)
    .filter(Objects::nonNull)
    .forEach(userEntity -> userEntity.setEnabled(false));

如果你可以UserEntity用disable方法 更新

class UserEntity {
    public void disable() {
        setEnabled(false);
    }
}

你可以再次使用方法reference(UserEntity::disable):

disabledUsersOnLDAP.stream()
    .map(User::getUsername)
    .map(usersRepository::findEnabledByUsername)
    .filter(Objects::nonNull)
    .forEach(UserEntity::disable);

一些资源:

• 关于java教程的方法参考
• JSR 335:JavaTM编程语言的Lambda表达式