[hotfix][table-common] ConstraintArgumentTypeStrategy uses Predicate …

…rather than function
Treydone · Apr 13, 2022 · 8aebfb1 · 8aebfb1
1 parent ffc3475
commit 8aebfb1
Show file tree

Hide file tree

Showing 2 changed files with 6 additions and 6 deletions.
diff --git a/...able-common/src/main/java/org/apache/flink/table/types/inference/InputTypeStrategies.java b/...able-common/src/main/java/org/apache/flink/table/types/inference/InputTypeStrategies.java
@@ -48,7 +48,7 @@
 
 import java.util.Arrays;
 import java.util.List;
-import java.util.function.Function;
+import java.util.function.Predicate;
 import java.util.stream.Collectors;
 
 /**
@@ -268,7 +268,7 @@ public static FamilyArgumentTypeStrategy logical(
 
     /** Strategy for an argument that must fulfill a given constraint. */
     public static ConstraintArgumentTypeStrategy constraint(
-            String constraintMessage, Function<List<DataType>, Boolean> evaluator) {
+            String constraintMessage, Predicate<List<DataType>> evaluator) {
         return new ConstraintArgumentTypeStrategy(constraintMessage, evaluator);
     }
 

diff --git a/...ava/org/apache/flink/table/types/inference/strategies/ConstraintArgumentTypeStrategy.java b/...ava/org/apache/flink/table/types/inference/strategies/ConstraintArgumentTypeStrategy.java
@@ -28,18 +28,18 @@
 import java.util.List;
 import java.util.Objects;
 import java.util.Optional;
-import java.util.function.Function;
+import java.util.function.Predicate;
 
 /** Strategy for an argument that must fulfill a given constraint. */
 @Internal
 public final class ConstraintArgumentTypeStrategy implements ArgumentTypeStrategy {
 
     private final String constraintMessage;
 
-    private final Function<List<DataType>, Boolean> evaluator;
+    private final Predicate<List<DataType>> evaluator;
 
     public ConstraintArgumentTypeStrategy(
-            String constraintMessage, Function<List<DataType>, Boolean> evaluator) {
+            String constraintMessage, Predicate<List<DataType>> evaluator) {
         this.constraintMessage = constraintMessage;
         this.evaluator = evaluator;
     }
@@ -50,7 +50,7 @@ public Optional<DataType> inferArgumentType(
         final List<DataType> actualDataTypes = callContext.getArgumentDataTypes();
 
         // type fulfills constraint
-        if (evaluator.apply(actualDataTypes)) {
+        if (evaluator.test(actualDataTypes)) {
             return Optional.of(actualDataTypes.get(argumentPos));
         }