JavaPairPartitionMapper directly implements PartitionMapper, so we can remove a helper object