SPARK-2895: Add mapPartitionsWithContext related support on Spark Java API.

can this call mapPartitionsWithContext and just rap it around?