Как определить DataFrame, используя case и match?

Question

Я хочу создать Spark DataFrame df одним из двух возможных способов:

val dataSourceType = "option1"

dataSourceType.map{
    case "option1" => {
        val df = gu
          .retrieveFromElastic(spark, source_field)
    }
    case "option1" => {
        val df = gu
          .retrieveFromCSV(spark, source_field)
    }
}

// some operations on "df"

Проблема в том, что df не виден вне оператора case.Как правильно с этим бороться?

Answer 1

Просто некоторые незначительные изменения синтаксиса, и вы должны получить это так.

val df = dataSourceType match {
    case "option1" => gu.retrieveFromElastic(spark, source_field)
    case "option2" => gu.retrieveFromCSV(spark, source_field)
}

Удачи: -)