lanyuanxiaoyao/hudi
1
0
Fork 0
Code Activity

handle no such element exception in HoodieSparkSqlWriter

Browse Source
This commit is contained in:
Kent Yao
2019-02-14 16:50:46 +08:00
committed by vinoth chandar
parent a16aa2a78f
commit 8dddecf00f
2 changed files with 14 additions and 4 deletions
Download Patch File Download Diff File Expand all files Collapse all files

7
hoodie-spark/src/main/scala/com/uber/hoodie/HoodieSparkSqlWriter.scala
View File

@@ -52,11 +52,10 @@ private[hoodie] object HoodieSparkSqlWriter {
if (path.isEmpty || tblName.isEmpty) { if (path.isEmpty || tblName.isEmpty) {
throw new HoodieException(s"'${HoodieWriteConfig.TABLE_NAME}', 'path' must be set.") throw new HoodieException(s"'${HoodieWriteConfig.TABLE_NAME}', 'path' must be set.")
} }
val serializer = sparkContext.getConf.get("spark.serializer") sparkContext.getConf.getOption("spark.serializer") match {
if (!serializer.equals("org.apache.spark.serializer.KryoSerializer")) { case Some(ser) if ser.equals("org.apache.spark.serializer.KryoSerializer") =>
throw new HoodieException(s"${serializer} serialization is not supported by hoodie. Please use kryo.") case _ => throw new HoodieException("hoodie only support org.apache.spark.serializer.KryoSerializer as spark.serializer")
} }
val storageType = parameters(STORAGE_TYPE_OPT_KEY) val storageType = parameters(STORAGE_TYPE_OPT_KEY)
val operation = val operation =
// It does not make sense to allow upsert() operation if INSERT_DROP_DUPS_OPT_KEY is true // It does not make sense to allow upsert() operation if INSERT_DROP_DUPS_OPT_KEY is true

11
hoodie-spark/src/test/scala/com/uber/hoodie/HoodieSparkSqlWriterSuite.scala
View File

@@ -19,6 +19,9 @@ package com.uber.hoodie
import org.scalatest.{FunSuite, Matchers} import org.scalatest.{FunSuite, Matchers}
import DataSourceWriteOptions._ import DataSourceWriteOptions._
import com.uber.hoodie.config.HoodieWriteConfig
import com.uber.hoodie.exception.HoodieException
import org.apache.spark.sql.{SaveMode, SparkSession}
class HoodieSparkSqlWriterSuite extends FunSuite with Matchers { class HoodieSparkSqlWriterSuite extends FunSuite with Matchers {
@@ -38,4 +41,12 @@ class HoodieSparkSqlWriterSuite extends FunSuite with Matchers {
} }
} }
test("throw hoodie exception when invalid serializer") {
val session = SparkSession.builder().appName("hoodie_test").master("local").getOrCreate()
val sqlContext = session.sqlContext
val options = Map("path" -> "hoodie/test/path", HoodieWriteConfig.TABLE_NAME -> "hoodie_test_tbl")
val e = intercept[HoodieException](HoodieSparkSqlWriter.write(sqlContext, SaveMode.ErrorIfExists, options, session.emptyDataFrame))
assert(e.getMessage.contains("spark.serializer"))
}
} }