SparkWriter.scala

trait SparkWriter {
  def write(df: DataFrame, mode: WriteMode = Overwrite): Unit
}

class SparkJsonWriter(path: String, partitions: Int = 1) extends SparkWriter {
  def write(df: DataFrame, mode: WriteMode = Overwrite): Unit = {
    df
      .coalesce(partitions)
      .write
      .mode(mode)
      .json(path)
  }
}