首页 > 解决方案 > 根据优先级过滤列表中的值

问题描述

我有一个类型的有效值列表:

Set<String> validTypes = ImmutableSet.of("TypeA", "TypeB", "TypeC");

从给定的列表中,我想提取具有有效类型的第一个值。在这种情况下,我会写这样的东西:

public class A{
    private String type;
    private String member;
}

List<A> classAList;
classAList.stream()
    .filter(a -> validTypes.contains(a.getType()))
    .findFirst();

但是,我想优先考虑TypeA,即如果classAListTypeA并且TypeB,我想要拥有的对象typeA。要做到这一点,我的一种方法是:

Set<String> preferredValidTypes = ImmutableSet.of("TypeA");
classAList.stream()
    .filter(a -> preferredValidTypes.contains(a.getType()))
    .findFirst()
    .orElseGet(() -> {
        return classAList.stream()
        .filter(a -> validTypes.contains(a.getType()))
        .findFirst();
    }

有更好的方法吗?

标签: javajava-8

解决方案


按类型过滤列表,按类型排序,收集到列表,然后只获取第一个元素

List<A> collect = classAList.stream()
                  .filter(a -> validTypes.contains(a.getType()))
                  .sorted(Comparator.comparing(A::getType))
                  .collect(Collectors.toList());
System.out.println(collect.get(0));

推荐阅读