Jelajahi Sumber

feat:聚合优化,进度条增加速度显示

ChenYL 2 tahun lalu
induk
melakukan
0090a1bf8f
1 mengubah file dengan 8 tambahan dan 1 penghapusan
  1. 8 1
      src/main/java/top/zhixinghe1/money/AggApplication.java

+ 8 - 1
src/main/java/top/zhixinghe1/money/AggApplication.java

@@ -1,6 +1,7 @@
 package top.zhixinghe1.money;
 
 import me.tongfei.progressbar.ProgressBar;
+import me.tongfei.progressbar.ProgressBarBuilder;
 import org.apache.commons.lang3.StringUtils;
 
 import java.io.BufferedInputStream;
@@ -14,6 +15,7 @@ import java.io.FileReader;
 import java.io.FileWriter;
 import java.io.IOException;
 import java.text.SimpleDateFormat;
+import java.time.temporal.ChronoUnit;
 import java.util.ArrayList;
 import java.util.Arrays;
 import java.util.BitSet;
@@ -181,7 +183,12 @@ public class AggApplication {
 
         SimpleDateFormat sdf = new SimpleDateFormat("yyyyMMddHHmmss");
         String aggFilePath = String.join(File.separator, dataDirPath, String.format("长尾词_合并_聚合_%s.txt", sdf.format(new Date())));
-        try (ProgressBar pb = new ProgressBar("文本聚合计算", totalWord);
+        ProgressBarBuilder progressBarBuilder = new ProgressBarBuilder().setTaskName("文本聚合计算")
+                .setInitialMax(totalWord)
+                .setUnit("个", 1)
+                .setSpeedUnit(ChronoUnit.SECONDS)
+                .showSpeed();
+        try (ProgressBar pb = progressBarBuilder.build();
              FileWriter fileWriter = new FileWriter(aggFilePath);
              BufferedWriter bufferedWriter = new BufferedWriter(fileWriter);) {
             int taskNum = calTasks.size();