package org.apache.maven.wagon.tck.http.fixture;

 /*

  * Licensed to the Apache Software Foundation (ASF) under one

  * or more contributor license agreements.  See the NOTICE file

  * distributed with this work for additional information

  * regarding copyright ownership.  The ASF licenses this file

  * to you under the Apache License, Version 2.0 (the

  * "License"); you may not use this file except in compliance

  * with the License.  You may obtain a copy of the License at

  *

  *   http://www.apache.org/licenses/LICENSE-2.0

  *

  * Unless required by applicable law or agreed to in writing,

  * software distributed under the License is distributed on an

  * "AS IS" BASIS, WITHOUT WARRANTIES OR CONDITIONS OF ANY

  * KIND, either express or implied.  See the License for the

  * specific language governing permissions and limitations

  * under the License.

  */

 import java.io.File;

 import java.io.FileInputStream;

 import java.io.IOException;

 import java.io.OutputStream;

 import javax.servlet.ServletException;

 import javax.servlet.http.HttpServlet;

 import javax.servlet.http.HttpServletRequest;

 import javax.servlet.http.HttpServletResponse;

 import org.apache.log4j.Logger;

 import org.codehaus.plexus.util.IOUtil;

 public class LatencyServlet

     extends HttpServlet

 {

     private static Logger logger = Logger.getLogger( LatencyServlet.class );

     private static final long serialVersionUID = 1L;

     private static final int BUFFER_SIZE = 32;

     private final int latencyMs;

     public LatencyServlet( final int latencyMs )

     {

         this.latencyMs = latencyMs;

     }

     @Override

     protected void doGet( final HttpServletRequest req, final HttpServletResponse resp )

         throws ServletException, IOException

     {

         if ( latencyMs < 0 )

         {

             logger.info( "Starting infinite wait." );

             synchronized ( this )

             {

                 try

                 {

                     wait();

                 }

                 catch ( InterruptedException e )

                 {

                 }

             }

             return;

         }

         String path = req.getPathInfo();

         // ignore the servlet's path here, since the servlet path is really only to provide a

         // binding for the servlet.

         String realPath = getServletContext().getRealPath( path );

         File f = new File( realPath );

         FileInputStream in = null;

         long total = 0;

         long start = System.currentTimeMillis();

         try

         {

             in = new FileInputStream( f );

             OutputStream out = resp.getOutputStream();

             logger.info( "Starting high-latency transfer. This should take about "

                 + ( ( f.length() / BUFFER_SIZE * latencyMs / 1000 ) + ( latencyMs / 1000 ) ) + " seconds." );

             int read = -1;

             byte[] buf = new byte[BUFFER_SIZE];

             while ( ( read = in.read( buf ) ) > -1 )

             {

                 try

                 {

                     Thread.sleep( latencyMs );

                 }

                 catch ( InterruptedException e )

                 {

                     e.printStackTrace();

                 }

                 logger.info( "Writing bytes " + total + "-" + ( total + read - 1 ) + " of " + f.length()

                     + ". Elapsed time so far: " + ( ( System.currentTimeMillis() - start ) / 1000 ) + " seconds" );

                 out.write( buf, 0, read );

                 total += read;

             }

         }

         finally

         {

             IOUtil.close( in );

         }

         logger.info( "High-latency transfer done in " + ( System.currentTimeMillis() - start ) + "ms" );

     }

 }