Require not null for CI

JiajunBernoulli · Jan 28, 2024 · 59c7415 · 59c7415
1 parent 2602f08
commit 59c7415
Showing 1 changed file with 3 additions and 2 deletions.
diff --git a/core/src/main/java/org/apache/calcite/tools/RelBuilder.java b/core/src/main/java/org/apache/calcite/tools/RelBuilder.java
@@ -2535,13 +2535,14 @@ private AggregateCall removeRedundantDistinct(
       ImmutableBitSet groupSet,
       RelNode r) {
     if (aggregateCall.isDistinct() && !config.redundantDistinct()) {
-      final RelMetadataQuery mq = r.getCluster().getMetadataQuery();
+      final RelNode relNode = requireNonNull(r, "RelNode");
+      final RelMetadataQuery mq = relNode.getCluster().getMetadataQuery();
       final List<Integer> argList = aggregateCall.getArgList();
       final ImmutableBitSet distinctArg = ImmutableBitSet.builder()
           .addAll(argList)
           .build();
       final ImmutableBitSet columns = groupSet.union(distinctArg);
-      final boolean alreadyUnique = mq.areColumnsUnique(r, columns);
+      final boolean alreadyUnique = mq.areColumnsUnique(relNode, columns);
       if (alreadyUnique) {
         return aggregateCall.withDistinct(false);
       }