Merge pull request #952 from ashitsalesforce/master

minor improvement in showing results progress in UI
forcedotcom · Jan 20, 2024 · 817596e · 817596e
2 parents fee014c + 9e1139a
commit 817596e
Show file tree

Hide file tree

Showing 2 changed files with 2 additions and 0 deletions.
diff --git a/src/main/java/com/salesforce/dataloader/action/visitor/AbstractQueryVisitor.java b/src/main/java/com/salesforce/dataloader/action/visitor/AbstractQueryVisitor.java
@@ -184,6 +184,7 @@ protected void startWriteExtraction(int size) {
         getRateCalculator().start(size);
         // start the Progress Monitor
         getProgressMonitor().beginTask(Messages.getMessage(getClass(), "extracting"), size); //$NON-NLS-1$
+        getProgressMonitor().setSubTask(getRateCalculator().calculateSubTask(getNumberOfRows(), getNumberErrors()));
     }
 
     @Override

diff --git a/src/main/java/com/salesforce/dataloader/action/visitor/DAOLoadVisitor.java b/src/main/java/com/salesforce/dataloader/action/visitor/DAOLoadVisitor.java
@@ -228,6 +228,7 @@ private void initLoadRateCalculator() {
         try {
             DataReader dao = (DataReader)getController().getDao();
             getRateCalculator().start(dao.getTotalRows());
+            getProgressMonitor().setSubTask(getRateCalculator().calculateSubTask(getNumberOfRows(), getNumberErrors()));
         } catch (Exception e) {
             logger.error("Unable to get total rows to upload from CSV or database");
             getRateCalculator().start(0);