使用Java Stream API的分层过滤

Question

private  Map<Integer,Thing> thingMap = new HashMap<Integer,Thing>();
// populate thingMap

List<Thing> things = new ArrayList<Thing>();

for (Thing thing : thingMap.values()) {
    if (thing.getCategory().equals(category)) {
        if (location == null) {
            things.add(thing);
        } else if (thing.getLocation().equals(location)) {
            things.add(thing);
        }
    }
}

List<Thing> things = thingMap.entrySet()
                      .stream()
                      .filter(t -> t.getValue().getCategory().equals(category))
                      .filter(t -> 
                          location == null || 
                          t.getValue().getLocation().equals(location)
                       )
                      .map(Map.Entry::getValue)
                      .collect(Collectors.toList());
    

Answer 1

在筛选器之后链接的操作将只对谓词接受的元素执行。所以没有必要担心这个。

您还可以将这些条件连接到一个filter步骤中，就像您可以将嵌套的if语句连接到一个if语句一样，方法是使用&组合这些条件。结果是一样的。

但请注意，该循环使用条件location==null，引用的是在发布的代码段之外声明的变量，而不是thing.getLocation()==null。

List<Thing> things = thingMap.values().stream()
    .filter(thing -> thing.getCategory().equals(category))
    .filter(thing -> location == null || thing.getLocation().equals(location))
    .collect(Collectors.toList());