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" );
     }
 
 }