scalaapache-sparkapache-spark-datasetcase-classclasstag

Passing case class into function arguments


sorry for asking a simple question. I want to pass a case class to a function argument and I want to use it further inside the function. Till now I have tried this with TypeTag and ClassTag but for some reason, I am unable to properly use it or may be I am not looking at the correct place.

Use cases is something similar to this:

case class infoData(colA:Int,colB:String)
case class someOtherData(col1:String,col2:String,col3:Int)

def readCsv[T:???](path:String,passedCaseClass:???): Dataset[???] = {
  sqlContext
    .read
    .option("header", "true")
    .csv(path)
    .as[passedCaseClass]
}

It will be called something like this:

val infoDf = readCsv("/src/main/info.csv",infoData)
val otherDf = readCsv("/src/main/someOtherData.csv",someOtherData)

Solution

  • First change your function definition to:

    object t0 {
        def readCsv[T] (path: String)(implicit spark: SparkSession, encoder: Encoder[T]): Dataset[T] = {
          spark
            .read
            .option("header", "true")
            .csv(path)
            .as[T]
        }
    }
    

    You don´t need to perform any kind of reflection to create a generic readCsv function. The key here is that Spark needs the encoder at compile time. So you can pass it as implicit parameter and the compiler will add it.

    Because Spark SQL can deserialize product types(your case classes) including the default encoders, it is easy to call your function like:

    case class infoData(colA: Int, colB: String)
    case class someOtherData(col1: String, col2: String, col3: Int)
    
    object test {
      import t0._
    
      implicit val spark = SparkSession.builder().getOrCreate()
    
      import spark.implicits._
      readCsv[infoData]("/tmp")
    
    }
    

    Hope it helps