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

can we remove this one? I don't think it is needed for Hive, is it?