Add udt support to TypeCoercion as well

apache · Dec 2, 2019 · a33ed0b · a33ed0b
1 parent 7aaaee2
commit a33ed0b
Showing 1 changed file with 6 additions and 0 deletions.
diff --git a/sql/catalyst/src/main/scala/org/apache/spark/sql/catalyst/analysis/TypeCoercion.scala b/sql/catalyst/src/main/scala/org/apache/spark/sql/catalyst/analysis/TypeCoercion.scala
@@ -103,6 +103,12 @@ object TypeCoercion {
     case (_: TimestampType, _: DateType) | (_: DateType, _: TimestampType) =>
       Some(TimestampType)
 
+    case (t: DataType, udt: UserDefinedType[_]) if t == udt.sqlType =>
+      Some(t)
+
+    case (udt: UserDefinedType[_], t: DataType) if t == udt.sqlType =>
+      Some(t)
+
     case (t1, t2) => findTypeForComplex(t1, t2, findTightestCommonType)
   }