Add error message for invalid aggregation class

trinodb · May 27, 2022 · 2b9734d · 2b9734d
1 parent 5a76c23
commit 2b9734d
Showing 1 changed file with 6 additions and 1 deletion.
diff --git a/core/trino-main/src/main/java/io/trino/metadata/SqlAggregationFunction.java b/core/trino-main/src/main/java/io/trino/metadata/SqlAggregationFunction.java
@@ -29,7 +29,12 @@ public abstract class SqlAggregationFunction
 
     public static List<SqlAggregationFunction> createFunctionsByAnnotations(Class<?> aggregationDefinition)
     {
-        return ImmutableList.copyOf(AggregationFromAnnotationsParser.parseFunctionDefinitions(aggregationDefinition));
+        try {
+            return ImmutableList.copyOf(AggregationFromAnnotationsParser.parseFunctionDefinitions(aggregationDefinition));
+        }
+        catch (RuntimeException e) {
+            throw new IllegalArgumentException("Invalid aggregation class " + aggregationDefinition.getSimpleName());
+        }
     }
 
     public SqlAggregationFunction(FunctionMetadata functionMetadata, AggregationFunctionMetadata aggregationFunctionMetadata)