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() {