Address Paddy's comments

longvu-db · Feb 10, 2025 · 37254ec · 37254ec
1 parent 123a5ce
commit 37254ec
Showing 1 changed file with 3 additions and 6 deletions.
diff --git a/spark/src/main/scala/org/apache/spark/sql/delta/DeltaColumnMapping.scala b/spark/src/main/scala/org/apache/spark/sql/delta/DeltaColumnMapping.scala
@@ -872,12 +872,9 @@ case object NameMapping extends DeltaColumnMappingMode {
 
 object DeltaColumnMappingMode {
   def apply(columnMappingModeString: String): DeltaColumnMappingMode = {
-    val columnMappingModeLowerCaseString = try {
-      columnMappingModeString.toLowerCase(Locale.ROOT)
-    } catch {
-      case ex: NullPointerException =>
-        throw DeltaErrors.unsupportedColumnMappingModeException(columnMappingModeString)
-    }
+    val columnMappingModeLowerCaseString =
+      Option(columnMappingModeString).map(_.toLowerCase(Locale.ROOT)).
+        getOrElse(throw DeltaErrors.unsupportedColumnMappingModeException(columnMappingModeString))
     columnMappingModeLowerCaseString match {
       case NoMapping.name => NoMapping
       case IdMapping.name => IdMapping