commit 7c16245d40b36591567660ca9b8757dd2f6f39d4
author Evgenii Kliuchnikov <eustas@google.com> Wed Jul 26 06:45:56 2023 -0700
committer Copybara-Service <copybara-worker@google.com> Thu Jul 27 01:36:06 2023 -0700
tree 33e089ab4b763f11fa6ceea9480508f73eba9e26
parent dbfebd13dcf51e1360f4f943306282b396a3e8cd

add "repeat" to Java toy decoder

PiperOrigin-RevId: 551190598

java/org/brotli/dec/Decoder.java

diff --git a/java/org/brotli/dec/Decoder.java b/java/org/brotli/dec/Decoder.java
index 599588e..e33f5a9 100644
--- a/java/org/brotli/dec/Decoder.java
+++ b/java/org/brotli/dec/Decoder.java
@@ -24,21 +24,15 @@
     return totalOut;
   }
 
-  public static void main(String... args) throws IOException {
-    if (args.length != 2) {
-      System.out.println("Usage: decoder <compressed_in> <decompressed_out>");
-      return;
-    }
-
-    byte[] buffer = new byte[1024 * 1024];
+  private static void decompress(String fromPath, String toPath, byte[] buffer) throws IOException {
     long start;
     long bytesDecoded;
     long end;
     InputStream in = null;
     OutputStream out = null;
     try {
-      in = new FileInputStream(args[0]);
-      out = new FileOutputStream(args[1]);
+      in = new FileInputStream(fromPath);
+      out = new FileOutputStream(toPath);
       start = System.nanoTime();
       bytesDecoded = decodeBytes(in, out, buffer);
       end = System.nanoTime();
@@ -58,4 +52,21 @@
     double mbDecoded = bytesDecoded / (1024.0 * 1024.0);
     System.out.println(mbDecoded / timeDelta + " MiB/s");
   }
+
+  public static void main(String... args) throws IOException {
+    if (args.length != 2 && args.length != 3) {
+      System.out.println("Usage: decoder <compressed_in> <decompressed_out> [repeat]");
+      return;
+    }
+
+    int repeat = 1;
+    if (args.length == 3) {
+      repeat = Integer.parseInt(args[2]);
+    }
+
+    byte[] buffer = new byte[1024 * 1024];
+    for (int i = 0; i < repeat; ++i) {
+      decompress(args[0], args[1], buffer);
+    }
+  }
 }