[HUDI-2706] duplicate code

apache · Nov 13, 2021 · 2930c00 · 2930c00
1 parent 7012680
commit 2930c00
Showing 1 changed file with 1 addition and 8 deletions.
diff --git a/hudi-spark-datasource/hudi-spark/src/main/scala/org/apache/hudi/HoodieWriterUtils.scala b/hudi-spark-datasource/hudi-spark/src/main/scala/org/apache/hudi/HoodieWriterUtils.scala
@@ -145,14 +145,7 @@ object HoodieWriterUtils {
         diffConfigs.append(s"PreCombineKey:\t$datasourcePreCombineKey\t$tableConfigPreCombineKey\n")
       }
 
-      val datasourceKeyGen = {
-        val kg = params.getOrElse(KEYGENERATOR_CLASS_NAME.key(), null)
-        if (classOf[SqlKeyGenerator].getCanonicalName == kg) {
-          params.getOrElse(SqlKeyGenerator.ORIGIN_KEYGEN_CLASS_NAME, null)
-        } else {
-          kg
-        }
-      }
+      val datasourceKeyGen = getOriginKeyGenerator(params)
       val tableConfigKeyGen = tableConfig.getString(HoodieTableConfig.KEY_GENERATOR_CLASS_NAME)
       if (null != datasourceKeyGen && null != tableConfigKeyGen
           && datasourceKeyGen != tableConfigKeyGen) {