com.tasktop.c2c.server.auth.service.proxy.ProxyPreAuthHttpRequestFactory.java Source code

Introduction

Here is the source code for com.tasktop.c2c.server.auth.service.proxy.ProxyPreAuthHttpRequestFactory.java

Source

/*******************************************************************************
 * Copyright (c) 2010, 2012 Tasktop Technologies
 * Copyright (c) 2010, 2011 SpringSource, a division of VMware
 * 
 * All rights reserved. This program and the accompanying materials
 * are made available under the terms of the Eclipse Public License v1.0
 * which accompanies this distribution, and is available at
 * http://www.eclipse.org/legal/epl-v10.html
 * 
 * Contributors:
 *     Tasktop Technologies - initial API and implementation
 ******************************************************************************/
package com.tasktop.c2c.server.auth.service.proxy;

import java.util.Map.Entry;

import org.apache.commons.httpclient.DefaultHttpMethodRetryHandler;
import org.apache.commons.httpclient.HttpMethodBase;
import org.apache.commons.httpclient.params.HttpMethodParams;
import org.springframework.beans.factory.InitializingBean;
import org.springframework.beans.factory.annotation.Autowired;
import org.springframework.http.client.CommonsClientHttpRequestFactory;

import com.tasktop.c2c.server.auth.service.AuthenticationToken;
import com.tasktop.c2c.server.common.internal.tenancy.InternalTenancyContextHttpHeaderProvider;

public class ProxyPreAuthHttpRequestFactory extends CommonsClientHttpRequestFactory implements InitializingBean {

    private AuthenticationTokenSerializer tokenSerializer = new AuthenticationTokenSerializer();

    @Autowired
    private InternalTenancyContextHttpHeaderProvider tenancySerializer;

    @Override
    public void postProcessCommonsHttpMethod(final HttpMethodBase httpMethod) {
        final AuthenticationToken authenticationToken = ProxyClient.getAuthenticationToken();
        if (authenticationToken != null) {
            tokenSerializer.serialize(new RequestHeaders() {
                @Override
                public void addHeader(String name, String value) {
                    httpMethod.addRequestHeader(name, value);
                }
            }, authenticationToken);
        }

        // Also setup the tenancy context headers
        for (Entry<String, String> header : tenancySerializer.computeHeaders().entrySet()) {
            httpMethod.addRequestHeader(header.getKey(), header.getValue());
        }
    }

    @Override
    public void afterPropertiesSet() throws Exception {
        getHttpClient().getParams().setParameter(HttpMethodParams.RETRY_HANDLER,
                new DefaultHttpMethodRetryHandler(0, false));
    }
}