java - 如何使用javapairrdd中的containsAll和contains来使用过滤器

Question

我有 2 个集合，一个是“list”，另一个是“pairRdd2”，其中包含如下所述的数据。

我正在尝试使用 containsAll 应用过滤器，其中如果 mypairRdd2 包含列表中提到的所有值。预期结果是 joe,{US,UK}

List<String> list = Arrays.asList("US","UK");

JavaRDD pairRdd = ctx.parallelize(Arrays.asList(new Tuple2("john","US"),new Tuple2("john","UAE"),new Tuple2("joe","US"),new Tuple2("joe","UK")));

JavaPairRDD<String, String> pairRdd2 = JavaPairRDD.fromJavaRDD(pairRdd);

pairRdd2.groupByKey().filter(x-> Arrays.asList(x._2).containsAll(list)).foreach(new VoidFunction<Tuple2<String,Iterable<String>>>() {

    @Override
    public void call(Tuple2<String, Iterable<String>> t) throws Exception {
        System.out.println(t._1());             
    }
});

有人可以强调一下我做错了什么......

Answer 1

问题出在 Arrays.asList() 上。这将创建一个 Iterable 列表，这不是执行过滤器所需的列表。您应该使用 groupBy 本身给出的列表:

    pairRdd2.groupByKey().filter(f -> {
        Set<String> set = new HashSet<>();
        for(String s: f._2())
            set.add(s);

        return list.containsAll(set);
    });

您还可以找到一种将迭代器/迭代器转换为集合并完全避免循环的快速方法。