fix

dataroaring · Jun 20, 2024 · 0222f3f · 0222f3f
1 parent c91d1d2
commit 0222f3f
Showing 1 changed file with 10 additions and 2 deletions.
diff --git a/fe/fe-core/src/main/java/org/apache/doris/load/loadv2/LoadJob.java b/fe/fe-core/src/main/java/org/apache/doris/load/loadv2/LoadJob.java
@@ -1177,11 +1177,19 @@ protected String getTimeZone() {
     }
 
     public int getLoadParallelism() {
-        return (int) jobProperties.get(LoadStmt.LOAD_PARALLELISM);
+        if (jobProperties.get(LoadStmt.LOAD_PARALLELISM).getClass() == Integer.class) {
+            return (int) jobProperties.get(LoadStmt.LOAD_PARALLELISM);
+        } else {
+            return ((Long) jobProperties.get(LoadStmt.LOAD_PARALLELISM)).intValue();
+        }
     }
 
     public int getSendBatchParallelism() {
-        return (int) jobProperties.get(LoadStmt.SEND_BATCH_PARALLELISM);
+        if (jobProperties.get(LoadStmt.SEND_BATCH_PARALLELISM).getClass() == Integer.class) {
+            return (int) jobProperties.get(LoadStmt.SEND_BATCH_PARALLELISM);
+        } else {
+            return ((Long) jobProperties.get(LoadStmt.SEND_BATCH_PARALLELISM)).intValue();
+        }
     }
 
     public LoadTask.Priority getPriority() {