从spark的Angular 来看，这并没有直接的支持：最接近的是mappartitions，preservepartitions=true。然而，这将不会直接帮助你的情况下，因为钥匙可能不会改变。

/**
   * Return a new RDD by applying a function to each partition of this RDD.
   *
   * `preservesPartitioning` indicates whether the input function preserves the partitioner, which
   * should be `false` unless this is a pair RDD and the input function doesn't modify the keys.
   */
  def mapPartitions[U: ClassTag](
      f: Iterator[T] => Iterator[U], preservesPartitioning: Boolean = false): RDD[U] = {
    val func = (context: TaskContext, index: Int, iter: Iterator[T]) => f(iter)
    new MapPartitionsRDD(this, sc.clean(func), preservesPartitioning)
  }

如果您能够确切地知道没有一个键会移动到它们原来的分区之外，那么上面的方法就可以工作了。但边界上的价值观可能不会合作。
与迁移密钥相比，数据的规模有多大？您可以考虑添加一个后处理步骤。首先为所有迁移密钥构造一个分区。Map器将为需要迁移的键输出一个特殊的键值。然后对结果进行后处理，以便对标准分区进行某种附加。这是额外的麻烦，所以您需要在额外的步骤和管道复杂性中评估权衡。