summaryrefslogtreecommitdiff
diff options
context:
space:
mode:
authorwangkuan <[email protected]>2024-08-02 10:16:58 +0800
committerwangkuan <[email protected]>2024-08-02 10:16:58 +0800
commit778ee36be04ce08c8230d0b3247b4166f90a05ae (patch)
treef73875fc1ed910f36cc3f14093ab9e0994c13cf0
parentc0a2e18f14fd495d80e047b90e5afb0abcb9602c (diff)
[improve][core]优化部分函数校验逻辑
-rw-r--r--groot-core/src/main/java/com/geedgenetworks/core/udf/Flatten.java3
-rw-r--r--groot-core/src/main/java/com/geedgenetworks/core/udf/PathCombine.java1
-rw-r--r--groot-core/src/main/java/com/geedgenetworks/core/udf/Rename.java4
3 files changed, 6 insertions, 2 deletions
diff --git a/groot-core/src/main/java/com/geedgenetworks/core/udf/Flatten.java b/groot-core/src/main/java/com/geedgenetworks/core/udf/Flatten.java
index 3153ef7..84c2c2a 100644
--- a/groot-core/src/main/java/com/geedgenetworks/core/udf/Flatten.java
+++ b/groot-core/src/main/java/com/geedgenetworks/core/udf/Flatten.java
@@ -24,6 +24,9 @@ public class Flatten implements ScalarFunction {
@Override
public void open(RuntimeContext runtimeContext, UDFContext udfContext) {
+ if(udfContext.getParameters()==null){
+ throw new GrootStreamRuntimeException(CommonErrorCode.ILLEGAL_ARGUMENT, "Missing required parameters");
+ }
prefix = udfContext.getParameters().getOrDefault("prefix", "").toString();
delimiter = udfContext.getParameters().getOrDefault("delimiter", ".").toString();
flattenKeys = new HashSet<>();
diff --git a/groot-core/src/main/java/com/geedgenetworks/core/udf/PathCombine.java b/groot-core/src/main/java/com/geedgenetworks/core/udf/PathCombine.java
index e48a503..874735d 100644
--- a/groot-core/src/main/java/com/geedgenetworks/core/udf/PathCombine.java
+++ b/groot-core/src/main/java/com/geedgenetworks/core/udf/PathCombine.java
@@ -31,7 +31,6 @@ public class PathCombine implements ScalarFunction {
if (udfContext.getParameters() != null
&& !udfContext.getParameters().isEmpty()) {
String paths = udfContext.getParameters().getOrDefault("path","").toString();
- // 使用逗号分隔项并转换为数组
if (!paths.isEmpty()) {
List<String> pathList;
try {
diff --git a/groot-core/src/main/java/com/geedgenetworks/core/udf/Rename.java b/groot-core/src/main/java/com/geedgenetworks/core/udf/Rename.java
index ba9b4d2..6a77c3a 100644
--- a/groot-core/src/main/java/com/geedgenetworks/core/udf/Rename.java
+++ b/groot-core/src/main/java/com/geedgenetworks/core/udf/Rename.java
@@ -26,7 +26,9 @@ public class Rename implements ScalarFunction {
@Override
public void open(RuntimeContext runtimeContext, UDFContext udfContext) {
-
+ if(udfContext.getParameters()==null ){
+ throw new GrootStreamRuntimeException(CommonErrorCode.ILLEGAL_ARGUMENT, "Missing required parameters");
+ }
String parentFields = udfContext.getParameters().getOrDefault("parent_fields", "").toString();
this.parentFields = new HashSet<>();
if (!parentFields.isEmpty()) {