[SPARK-2024] Allow users to specify batch size on reading