Test Largefile download

Author: sswguo

bindings/jaxrs/src/main/java/org/commonjava/indy/core/bind/jaxrs/util/TransferStreamingOutput.java

@@ -61,7 +61,7 @@ public void write( final OutputStream out )
         throws IOException, WebApplicationException
     {
         long start = System.nanoTime();
-        try
+        /*try
         {
             CountingOutputStream cout = new CountingOutputStream( out );
             IOUtils.copy( stream, cout );
@@ -81,6 +81,36 @@ public void write( final OutputStream out )
         finally
         {
             IOUtils.closeQuietly( stream );
+        }*/
+
+        byte[] buffer = new byte[8192]; // 8KB buffer
+        int bytesRead;
+
+        try
+        {
+            CountingOutputStream cout = new CountingOutputStream( out );
+
+            while ((bytesRead = stream.read(buffer)) != -1)
+            {
+                cout.write(buffer, 0, bytesRead);
+                cout.flush();
+            }
+
+            Logger logger = LoggerFactory.getLogger( getClass() );
+            logger.trace( "Wrote: {} bytes", cout.getByteCount() );
+
+            String name = getName( metricsConfig.getNodePrefix(), TRANSFER_METRIC_NAME,
+                    getDefaultName( TransferStreamingOutput.class, "write" ), METER );
+
+            long end = System.nanoTime();
+            double elapsed = (end-start)/NANOS_PER_SEC;
+
+            Meter meter = metricsManager.getMeter( name );
+            meter.mark( Math.round( cout.getByteCount() / elapsed ) );
+        }
+        finally
+        {
+            IOUtils.closeQuietly(stream);  // Ensure the InputStream is closed properly
         }
     }