File download optimization

lematechvip · Jun 16, 2021 · 56e4986 · 56e4986
1 parent ae33fe4
commit 56e4986
Show file tree

Hide file tree

Showing 2 changed files with 30 additions and 22 deletions.
diff --git a/hrun4j-core/src/main/java/vip/lematech/httprunner4j/helper/OkHttpsHelper.java b/hrun4j-core/src/main/java/vip/lematech/httprunner4j/helper/OkHttpsHelper.java
@@ -29,6 +29,7 @@
 import java.util.Map;
 import java.util.Objects;
 import java.util.Set;
+import java.util.concurrent.CountDownLatch;
 import java.util.concurrent.TimeUnit;
 
 /**
@@ -282,27 +283,34 @@ private static ResponseEntity wrapperResponseEntity(HttpResult httpResult
                 if (workDir.endsWith(Constant.DOT_PATH)) {
                     workDir = LittleHelper.replaceLast(workDir, Constant.DOT_PATH, "");
                 }
+                CountDownLatch latch = new CountDownLatch(1);
                 String finalWorkDir = workDir;
-               body.stepRate(0.1)
-                        .setOnProcess((process -> {
-                            LogHelper.info(I18NFactory.getLocaleMessage("file.download.process"));
-                            long totalBytes = process.getTotalBytes();
-                            LogHelper.info(String.format(I18NFactory.getLocaleMessage("file.total.bytes"),totalBytes));
-                            long doneBytes = process.getDoneBytes();
-                            LogHelper.info(String.format(I18NFactory.getLocaleMessage("file.done.bytes"),doneBytes));
-                            double rate = process.getRate();
-                            LogHelper.info(String.format(I18NFactory.getLocaleMessage("file.done.rate"),rate * 100));
-                            boolean isDone = process.isDone();
-                            LogHelper.info(String.format(I18NFactory.getLocaleMessage("file.is.done"),isDone));
-                        }))
-                        .toFolder(workDir)
-                        .setOnFailure((Download.Failure failure) -> {
-                            LogHelper.info(String.format(I18NFactory.getLocaleMessage("file.failure"),failure.getException().getMessage()));
-                        })
-                        .setOnSuccess((File file) -> {
-                            LogHelper.info(String.format(I18NFactory.getLocaleMessage("file.success"),FileUtil.normalize(finalWorkDir)));
-                        })
-                        .start();
+                body.stepRate(0.1)
+                    .setOnProcess((process -> {
+                        LogHelper.info(I18NFactory.getLocaleMessage("file.download.process"));
+                        long totalBytes = process.getTotalBytes();
+                        LogHelper.info(String.format(I18NFactory.getLocaleMessage("file.total.bytes"),totalBytes));
+                        long doneBytes = process.getDoneBytes();
+                        LogHelper.info(String.format(I18NFactory.getLocaleMessage("file.done.bytes"),doneBytes));
+                        double rate = process.getRate();
+                        LogHelper.info(String.format(I18NFactory.getLocaleMessage("file.done.rate"),rate * 100));
+                        boolean isDone = process.isDone();
+                        LogHelper.info(String.format(I18NFactory.getLocaleMessage("file.is.done"),isDone));
+                    }))
+                    .toFolder(workDir)
+                    .setOnFailure((Download.Failure failure) -> {
+                        LogHelper.info(String.format(I18NFactory.getLocaleMessage("file.failure"),failure.getException().getMessage()));
+                    })
+                    .setOnSuccess((File file) -> {
+                        LogHelper.info(String.format(I18NFactory.getLocaleMessage("file.success"),FileUtil.normalize(finalWorkDir)));
+                        latch.countDown();
+                    })
+                    .start();
+                   try {
+                        latch.await();
+                    } catch (InterruptedException e) {
+                       LogHelper.error("Download interrupt exception");
+                    }
             } else {
                 String responseContent = body.toString();
                 if (JsonHelper.isJson(responseContent)) {

diff --git a/hrun4j-test-demo/src/test/java/vip/lematech/httprunner4j/testcases/demo/DemoTest.java b/hrun4j-test-demo/src/test/java/vip/lematech/httprunner4j/testcases/demo/DemoTest.java
@@ -186,9 +186,9 @@ public void testBeanShellDemo(TestCase testCase) {
      * @param testCase
      *
      */
-    /*@Test(dataProvider = "dataProvider")
+    @Test(dataProvider = "dataProvider")
     public void testFileDownloadDemo(TestCase testCase) {
         TestCaseExecutorEngine.getInstance().execute(testCase);
-    }*/
+    }
 
 }