ExpressionFunctionParameter.setType

mP1 · Dec 6, 2023 · c13781b · c13781b
1 parent 6ebc46b
commit c13781b
Show file tree

Hide file tree

Showing 2 changed files with 64 additions and 1 deletion.
diff --git a/src/main/java/walkingkooka/tree/expression/function/ExpressionFunctionParameter.java b/src/main/java/walkingkooka/tree/expression/function/ExpressionFunctionParameter.java
@@ -88,7 +88,7 @@ public static <T> ExpressionFunctionParameter<T> with(final ExpressionFunctionPa
                                                           final ExpressionFunctionParameterCardinality cardinality,
                                                           final Set<ExpressionFunctionParameterKind> kinds) {
         Objects.requireNonNull(name, "name");
-        Objects.requireNonNull(type, "type");
+        checkType(type);
         Objects.requireNonNull(typeParameters, "typeParameters");
         Objects.requireNonNull(cardinality, "cardinality");
         Objects.requireNonNull(kinds, "kinds");
@@ -102,6 +102,10 @@ public static <T> ExpressionFunctionParameter<T> with(final ExpressionFunctionPa
         );
     }
 
+    private static <T> Class<T> checkType(final Class<T> type) {
+        return Objects.requireNonNull(type, "type");
+    }
+
     private ExpressionFunctionParameter(final ExpressionFunctionParameterName name,
                                         final Class<T> type,
                                         final ExpressionFunctionParameterCardinality cardinality,
@@ -142,6 +146,25 @@ public Class<T> type() {
         return this.type;
     }
 
+    /**
+     * Would be setter that returns a {@link ExpressionFunctionParameter} with the new type but copying all other properties.
+     * This is particularly useful when updating a {@link ExpressionFunction} parameters such as COUNT which initially has its parameters declared but Object but within a
+     * spreadsheet it needs all parameter values converted to a number.
+     */
+    public <TT> ExpressionFunctionParameter<TT> setType(final Class<TT> type) {
+        checkType(type);
+
+        return this.type.equals(type) ?
+                Cast.to(this) :
+                new ExpressionFunctionParameter<>(
+                        this.name,
+                        type,
+                        this.cardinality,
+                        this.typeParameters,
+                        this.kinds
+                );
+    }
+
     private final Class<T> type;
 
     public List<Class<?>> typeParameters() {

diff --git a/src/test/java/walkingkooka/tree/expression/function/ExpressionFunctionParameterTest.java b/src/test/java/walkingkooka/tree/expression/function/ExpressionFunctionParameterTest.java
@@ -167,6 +167,46 @@ public void testSetNameWithDifferent() {
         this.checkKinds(parameter);
     }
 
+    // setType.........................................................................................................
+
+    @Test
+    public void testSetTypeWithNullFails() {
+        assertThrows(
+                NullPointerException.class,
+                () -> this.createObject().setType(null)
+        );
+    }
+
+    @Test
+    public void testSetTypeWithSame() {
+        final ExpressionFunctionParameter<?> parameter = this.createObject();
+        assertSame(
+                parameter,
+                parameter.setType(TYPE)
+        );
+    }
+
+    @Test
+    public void testSetTypeWithDifferent() {
+        final ExpressionFunctionParameter<?> parameter = this.createObject();
+        final Class<ExpressionFunctionParameterTest> type = ExpressionFunctionParameterTest.class;
+
+        final ExpressionFunctionParameter<?> different = parameter.setType(type);
+        assertNotSame(parameter, different);
+
+        this.checkName(different);
+        this.checkType(different, type);
+        this.checkTypeParameters(different);
+        this.checkCardinality(different);
+        this.checkKinds(different);
+
+        this.checkName(parameter);
+        this.checkType(parameter);
+        this.checkTypeParameters(parameter);
+        this.checkCardinality(parameter);
+        this.checkKinds(parameter);
+    }
+
     // setCardinality...................................................................................................
 
     @Test