使用 Java 流将 ArrayList<String> 转换为 Set<ScopeItem>

Question

我想使用 Java 流 将 ArrayList<String> 转换为 Set<ScopeItem>。

ScopeItem is a enum;
items is an ArrayList<String>;

Set<ScopeItem> scopeItems = items.stream()
                    .map(scopeString -> ScopeItem.valueOf(scopeString))
                    .filter(Objects::nonNull)
                    .collect(Collectors.toSet());

对于不在枚举中的字符串，会抛出以下内容：

java.lang.IllegalArgumentException: No enum const...

理想情况下，我想跳过任何不匹配的字符串。

我想可能是使用平面地图？有什么想法吗？

Answer 1

您可以将以下方法添加到您的ScopeItem：

public static ScopeItem valueOfOrNull(String name) {
    try {
        return valueOf(name);
    } catch (IllegalArgumentException e) {
        // no such element
        return null;
    }
}

并使用它来映射您的枚举值：

.map(scopeString -> ScopeItem.valueOfOrNull(scopeString))

非空值（您已经拥有）上的后续.filter() 将过滤掉那些对应于不匹配字符串的空值。

Answer 2

您可以在 map 中放置一个 try-catch 以返回 null 而不是抛出异常：

Set<ScopeItem> scopeItems = items.stream()
    .map(scopeString ->
        {
           try
           {
              return ScopeItem.valueOf(scopeString);
           }
           catch (IllegalArgumentException e)
           {
              return null;
           }
        })
    .filter(Objects::nonNull)
    .collect(Collectors.toSet());

您也可以预先使用filter 来检查值数组是否包含您要查找的字符串：

Set<ScopeItem> scopeItems = items.stream()
    .filter(scopeString -> Arrays.stream(ScopeItem.values())
                               .anyMatch(scopeItem -> scopeItem.name().equals(scopeString)))
    .map(ScopeItem::valueOf)
    .collect(Collectors.toSet());

Answer 3

与其他人不同，我不建议使用异常，因为我认为它们应该用于异常情况，而不是用于可能发生的事情。一个简单的解决方案是拥有一组可接受的静态字符串，然后简单地检查您想要使用valueOf 的字符串是否在所述集合中。

package test;

import java.util.Arrays;
import java.util.HashSet;
import java.util.List;
import java.util.Set;
import java.util.stream.Collectors;

public class Test {

    public static enum ScopeItem {
        ScopeA,
        ScopeB;

        private static Set<String> castableStrings;

        static {
            castableStrings = new HashSet<>();
            for (ScopeItem i : ScopeItem.values()) {
                castableStrings.add(i.name());
            }
        }

        public static boolean acceptable(String s) {
            return castableStrings.contains(s);
        }
    }

    public static void main(String[] args) {

       List<String> items = Arrays.asList("ScopeA", "RandomString", "ScopeB");
       Set<ScopeItem> scopeItems = items.stream()
               .filter(ScopeItem::acceptable)
               .map(ScopeItem::valueOf)
               .collect(Collectors.toSet());

       System.out.println(scopeItems.size());
    }

}

Answer 4

这里有一个更优雅的方法。您不必在枚举中添加任何新字段；您也可以简单地针对 it 运行一个流，并确定您的集合中是否有任何匹配项。

下面的代码假定一个枚举声明：

enum F {
    A, B, C, D, E
}

看起来像这样：

List<String> bad = Arrays.asList("A", "a", "B", "b", "C", "c");

final Set<F> collect = bad.stream()
                                .filter(e -> Arrays.stream(F.values())
                                                     .map(F::name)
                                                     .anyMatch(m -> Objects.equals(e, m)))
                                .map(F::valueOf)
                                .collect(Collectors.toSet());

这里要注意两部分：

• 我们对枚举值进行内部过滤，并通过F::name 将其映射到字符串。
• 我们以 null 安全的方式确定我们的基本集合的元素与我们的枚举元素是否存在任何匹配（Objects.equals 在此处使用空值执行 The Right Thing™）

Answer 5

您可以使用Apache Common's commons-lang3 EnumUtils.getEnum() 代替valueOf()。如果没有匹配的枚举条目，则返回 null（然后您可以完全按照代码中的方式过滤）。

Answer 6

最简单和最清晰的方法是在 items.contains() 上过滤 Enum 的 values() 方法：

Set<ScopeItem> enumVals = Arrays.stream(ScopeItem.values())
        .filter(e -> items.contains(e.name()))
        .collect(Collectors.toSet());

没有添加任何功能只是为了让流做你想做的事，而且一目了然。