Store each partition to file and load it on same partition in Scala Spark

Question

I have situation where have to store each partition\'s data to a file and load stored data at same partition later. Here is my code

Base class

case cla