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

}
