to #57132470 fix export rows CountStat

polardb · Jun 18, 2024 · 8513d6b · 8513d6b
1 parent 1d02b48
commit 8513d6b
Showing 1 changed file with 2 additions and 0 deletions.
diff --git a/batch-tool/src/main/java/worker/export/BaseExportWorker.java b/batch-tool/src/main/java/worker/export/BaseExportWorker.java
@@ -29,6 +29,7 @@
 import org.apache.commons.lang3.StringUtils;
 import org.slf4j.Logger;
 import org.slf4j.LoggerFactory;
+import util.CountStat;
 import util.DataSourceUtil;
 import util.FileUtil;
 import util.IOUtil;
@@ -154,6 +155,7 @@ protected void produceData() {
             logger.error("{} 导出发生错误: {}", topology, e.getMessage());
         } finally {
             IOUtil.close(os);
+            CountStat.addDbRowCount(rowCount);
             logger.info("{} 导出行数：{}", topology, rowCount);
         }
     }