Fix ut

Signed-off-by: wyb <[email protected]>
StarRocks · Nov 8, 2024 · 3ddafdb · 3ddafdb
1 parent 14a14f2
commit 3ddafdb
Showing 1 changed file with 7 additions and 5 deletions.
diff --git a/fe/fe-core/src/main/java/com/starrocks/load/pipe/Pipe.java b/fe/fe-core/src/main/java/com/starrocks/load/pipe/Pipe.java
@@ -79,6 +79,8 @@ public class Pipe implements GsonPostProcessable {
     public static final long DEFAULT_BATCH_SIZE = 1 << 30; // 1 GB
     public static final long DEFAULT_BATCH_FILES = 256;
     public static final int FAILED_TASK_THRESHOLD = 5;
+
+    private static final String TASK_PROPERTY_PREFIX = "task.";
 
     @SerializedName(value = "name")
     private final String name;
@@ -138,9 +140,9 @@ public static Pipe fromStatement(long id, CreatePipeStmt stmt) {
 
     public void processProperties(Map<String, String> properties) {
         for (Map.Entry<String, String> entry : properties.entrySet()) {
-            String key = entry.getKey();
+            String key = entry.getKey().toLowerCase();
             String value = entry.getValue();
-            switch (key.toLowerCase()) {
+            switch (key) {
                 case PipeAnalyzer.PROPERTY_POLL_INTERVAL: {
                     this.pollIntervalSecond = Integer.parseInt(value);
                     break;
@@ -164,11 +166,11 @@ public void processProperties(Map<String, String> properties) {
                 }
                 default: {
                     // task execution variables
-                    if (key.toUpperCase().startsWith("TASK.")) {
-                        String taskVariable = StringUtils.removeStart(key.toUpperCase(), "TASK.");
+                    if (key.startsWith(TASK_PROPERTY_PREFIX)) {
+                        String taskVariable = StringUtils.removeStart(key, TASK_PROPERTY_PREFIX);
                         this.taskExecutionVariables.put(taskVariable, value);
                     } else {
-                        throw new IllegalArgumentException("unsupported property: " + key);
+                        throw new IllegalArgumentException("unsupported property: " + entry.getKey());
                     }
                 }
             }