Question

考虑以下示例：

JavaPairRDD<String, Row> R = input.textFile("test").mapToPair(new PairFunction<String, String, Row>() {
        public Tuple2<String, Row> call(String arg0) throws Exception {
            String[] parts = arg0.split(" ");
            Row r = RowFactory.create(parts[0],parts[1]);
            return new Tuple2<String, Row>(r.get(0).toString(), r);
        }}).partitionBy(new HashPartitioner(20));

上面的代码创建了一个名为R的RDD，它通过散列在名为“test”的txt文件的第一列上分成20个部分。

考虑test.txt文件的格式如下：

...
valueA1 valueB1
valueA1 valueB2
valueA1 valueB3
valueA1 valueB4
...

在我的上下文中，我有一个已知值，例如valueA1，我想要检索所有其他值。通过使用具有指定值的现有过滤器操作来完成它是微不足道的。但是，我想避免这种情况，因为基本上过滤操作将在整个RDD上执行。

假设散列（valueA1）= 3，我想仅在分区3上执行给定操作。更一般地说，我有兴趣从RDD中删除/选择特定分区并对它们执行操作。

从SPARK API看来似乎不可能直接有解决方法来实现同样的事情吗？

Answer 1

对于单键，您可以使用lookup方法：

rdd.lookup("a")

// Seq[Int] = ArrayBuffer(1, 4)

为了有效查找，您需要一个分区的RDD，例如使用HashPartitioner，如下所示。

如果您只想过滤包含特定键的分区，可以使用mapPartitionsWithIndex：

完成

import org.apache.spark.HashPartitioner

val rdd = sc.parallelize(
  Seq(("a", 1), ("b", 2), ("c", 3), ("a", 4), ("b", 5)
// A particular number is used only to get a reproducible output
)).partitionBy(new HashPartitioner(8))  

val keys = Set("a", "c")
val parts = keys.map(_.## % rdd.partitions.size)

rdd.mapPartitionsWithIndex((i, iter) =>
  if (parts.contains(i)) iter.filter{ case (k, _) => keys.contains(k) }
  else Iterator()
).collect

// Array[(String, Int)] = Array((a,1), (a,4), (c,3))

如何根据给定的分区过滤RDD？

1 个答案: