[SPARK-2534] Avoid pulling in the entire RDD in groupByKey.