Fixed bugs

prestodb · Feb 5, 2025 · dc3786f · dc3786f
1 parent 4e20c1c
commit dc3786f
Showing 1 changed file with 10 additions and 4 deletions.
diff --git a/presto-main/src/main/java/com/facebook/presto/sql/analyzer/ExpressionAnalyzer.java b/presto-main/src/main/java/com/facebook/presto/sql/analyzer/ExpressionAnalyzer.java
@@ -890,8 +890,11 @@ protected Type visitGenericLiteral(GenericLiteral node, StackableAstVisitorConte
             try {
                 type = functionAndTypeResolver.getType(parseTypeSignature(node.getType()));
             }
-            catch (IllegalArgumentException e) {
-                throw new SemanticException(TYPE_MISMATCH, node, "Unknown type: " + node.getType());
+            catch (PrestoException e) {
+                if (e.getErrorCode() == UNKNOWN_TYPE.toErrorCode()) {
+                    throw new SemanticException(TYPE_MISMATCH, node, "Unknown type: " + node.getType());
+                }
+                throw e;
             }
 
             if (!JSON.equals(type)) {
@@ -913,8 +916,11 @@ protected Type visitEnumLiteral(EnumLiteral node, StackableAstVisitorContext<Con
             try {
                 type = functionAndTypeResolver.getType(parseTypeSignature(node.getType()));
             }
-            catch (IllegalArgumentException e) {
-                throw new SemanticException(TYPE_MISMATCH, node, "Unknown type: " + node.getType());
+            catch (PrestoException e) {
+                if (e.getErrorCode() == UNKNOWN_TYPE.toErrorCode()) {
+                    throw new SemanticException(TYPE_MISMATCH, node, "Unknown type: " + node.getType());
+                }
+                throw e;
             }
 
             return setExpressionType(node, type);