mercury-1.0.0-alpha-2/mercury-transport/mercury-transport-http/src/main/java/org/apache/maven/mercury/spi/http/client/HandshakeExchange.java

/**
 * 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.
 */

package org.apache.maven.mercury.spi.http.client;

import java.util.Map;

import javax.servlet.http.HttpServletResponse;

import org.mortbay.io.Buffer;
import org.mortbay.jetty.client.HttpClient;
import org.mortbay.jetty.client.HttpExchange;

/**
 * HandshakeExchange
 * <p/>
 * Asychronous http message sending. Used to generate a commit or discard
 * message to the remote server after the mercury upload is finished.
 */
public abstract class HandshakeExchange extends HttpExchange
{
    private HttpClient _httpClient;
    private String _method;
    private String _url;
    private Map<String, String> _headers;
    private int _status;

    public abstract void onHandshakeComplete( String url );

    public abstract void onHandshakeError( String url, Exception e );

    /**
     * @param httpClient
     * @param method
     * @param url
     * @param headers
     */
    public HandshakeExchange( HttpClient httpClient, String method, String url, Map<String, String> headers )
    {
        _httpClient = httpClient;
        _method = method;
        _url = url;
        _headers = headers;
    }

    public void send()
    {
        try
        {
            setMethod( _method );
            setURL( _url );
            if ( _headers != null )
            {
                for ( Map.Entry<String, String> e : _headers.entrySet() )
                {
                    setRequestHeader( e.getKey(), e.getValue() );
                }
            }
            _httpClient.send( this );
        }
        catch ( Exception e )
        {
            onHandshakeError( _url, new Exception( null, e ) );
        }
    }


    protected void onResponseStatus( Buffer version, int status, Buffer reason )
    {
        _status = status;
    }

    protected void onException( Throwable ex )
    {
        onHandshakeError( _url, new Exception( ex ) );
    }

    protected void onExpire()
    {
        onHandshakeError( _url, new Exception( "Timeout occurred" ) );
    }

    protected void onResponseComplete()
    {
        if ( _status != HttpServletResponse.SC_OK )
        {
            onHandshakeError( _url, new Exception( "Http Error Code:" + _status ) );
        }
        else
        {
            onHandshakeComplete( _url );
        }
    }

}