时间戳

首页 > 解决方案 > 如果条件不满足，如何跳过映射元素

java - 如果条件不满足，如何跳过映射元素

问题描述

假设我有以下

public class UserEntity{
    String id;
    List<String> relatedEntity;
}

public class EmployeeEntity{
    String id;
    String name;
    Boolean isActive;
    List<String> relatedEntityDetails;
}

现在有了一个 UserEntity 列表，我必须映射到一个 EmployeeEntity 列表：

private List<EmployeeEntity> getEmployees(List<UserEntity> users)
     return users.stream()
         .filter(x-> !x.getRelatedEntity().isEmpty())
         .map(this::mapToEmployee)
         .collect(Collectors.toList());
}

private EmployeeEntity mapToEmployee(UserEntity userEntity){
    // retrieve EmployeeEntity from DB and perform a validations like
    // employeeEntity.isActive = true
    return employeeEntity;
}

现在，一切正常，但是当数据库中不存在 EmployeeEntity 或 isActive = false 时，我需要处理这种情况，在这些情况下，应该跳过 map()，所以如果有 3 个元素 UserEntity 和 for其中一个用户，一名员工不活跃，那么返回的列表应该只有 2 个元素。

关于如何添加该行为的任何建议？

标签： javajava-stream

解决方案

使mapToEmployee返回一个Optional<EmployeeEntity>，并过滤一个不为空的：

private List<EmployeeEntity> getEmployees(List<UserEntity> users)
     return users.stream()
         .filter(x-> !x.getRelatedEntity().isEmpty())
         .map(this::mapToEmployee)
         .filter(Optional::isPresent)
         .map(Optional::get)
         .collect(Collectors.toList());
}

private Optional<EmployeeEntity> mapToEmployee(UserEntity userEntity){
    ... optional depending on whether it is present in DB and/or is active
}

推荐阅读