Skip to content

Commit 296b3be

Browse files
committed
refactor(udf): null-guard registerUdf arguments and drop dead assertion
1 parent 0d06aa5 commit 296b3be

2 files changed

Lines changed: 5 additions & 2 deletions

File tree

core/src/main/java/org/apache/datafusion/SessionContext.java

Lines changed: 5 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -229,6 +229,11 @@ public void registerUdf(
229229
if (nativeHandle == 0) {
230230
throw new IllegalStateException("SessionContext is closed");
231231
}
232+
java.util.Objects.requireNonNull(name, "name");
233+
java.util.Objects.requireNonNull(udf, "udf");
234+
java.util.Objects.requireNonNull(returnType, "returnType");
235+
java.util.Objects.requireNonNull(argTypes, "argTypes");
236+
java.util.Objects.requireNonNull(volatility, "volatility");
232237
List<Field> fields = new ArrayList<>(argTypes.size() + 1);
233238
fields.add(new Field("return", FieldType.nullable(returnType), null));
234239
for (int i = 0; i < argTypes.size(); i++) {

core/src/test/java/org/apache/datafusion/ScalarUdfTest.java

Lines changed: 0 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -20,7 +20,6 @@
2020
package org.apache.datafusion;
2121

2222
import static org.junit.jupiter.api.Assertions.assertEquals;
23-
import static org.junit.jupiter.api.Assertions.assertNotNull;
2423

2524
import java.util.List;
2625

@@ -79,7 +78,6 @@ void addOne_overConstantTable_returnsIncrementedValues() throws Exception {
7978
assertEquals(2, y.get(0));
8079
assertEquals(3, y.get(1));
8180
assertEquals(4, y.get(2));
82-
assertNotNull(y);
8381
}
8482
}
8583
}

0 commit comments

Comments
 (0)