partitionByKey

RichPairDataset