com.puppetlabs.puppetdb.javaclient.impl.DefaultModule.java Source code

Introduction

Here is the source code for com.puppetlabs.puppetdb.javaclient.impl.DefaultModule.java

Source

/**
 * Copyright (c) 2013 Puppet Labs, Inc. and other contributors, as listed below.
 * All rights reserved. This program and the accompanying materials
 * are made available under the terms of the Apache License, Version 2.0
 * which accompanies this distribution, and is available at
 * http://www.apache.org/licenses/LICENSE-2.0
 * 
 * Contributors:
 *   Puppet Labs
 */
package com.puppetlabs.puppetdb.javaclient.impl;

import org.apache.http.client.HttpClient;
import org.apache.http.conn.scheme.Scheme;
import org.apache.http.conn.ssl.SSLSocketFactory;
import org.apache.http.impl.client.DefaultHttpClient;
import org.apache.http.params.BasicHttpParams;
import org.apache.http.params.CoreConnectionPNames;
import org.apache.http.params.HttpConnectionParams;
import org.apache.http.params.HttpParams;

import com.google.gson.Gson;
import com.google.inject.AbstractModule;
import com.google.inject.Provides;
import com.google.inject.Singleton;
import com.google.inject.name.Names;
import com.puppetlabs.puppetdb.javaclient.APIPreferences;
import com.puppetlabs.puppetdb.javaclient.HttpConnector;
import com.puppetlabs.puppetdb.javaclient.PuppetDBClient;

/**
 * Default Guice Injection module
 */
public class DefaultModule extends AbstractModule {
    private final APIPreferences preferences;

    /**
     * Create a module with preferences.
     * 
     * @param preferences
     *            The preferences
     */
    public DefaultModule(APIPreferences preferences) {
        this.preferences = preferences;
    }

    @Override
    protected void configure() {
        bind(APIPreferences.class).toInstance(preferences);
        bind(Integer.class).annotatedWith(Names.named(CoreConnectionPNames.CONNECTION_TIMEOUT))
                .toInstance(preferences.getConnectTimeout());
        bind(Integer.class).annotatedWith(Names.named(CoreConnectionPNames.SO_TIMEOUT))
                .toInstance(preferences.getSoTimeout());
        bind(Gson.class).toProvider(GsonProvider.class);
        if (preferences.getCertPEM() != null)
            bind(SSLSocketFactory.class).toProvider(PEM_SSLSocketFactoryProvider.class).in(Singleton.class);
        else
            bind(SSLSocketFactory.class).toInstance(SSLSocketFactory.getSocketFactory());
        bind(HttpConnector.class).to(HttpComponentsConnector.class);
        bind(PuppetDBClient.class).to(PuppetDBClientImpl.class);
    }

    /**
     * Provides a HttpClient that is configured with the preferences of this module and the
     * injected <code>sslSocketFactory</code>.
     * 
     * @param sslSocketFactory
     *            The injected SSL socket factory
     * @return The new HttpClient instance
     */
    @Provides
    public HttpClient provideHttpClient(SSLSocketFactory sslSocketFactory) {
        HttpParams params = new BasicHttpParams();
        HttpConnectionParams.setConnectionTimeout(params, preferences.getConnectTimeout());
        HttpConnectionParams.setSoTimeout(params, preferences.getSoTimeout());

        DefaultHttpClient httpClient = new DefaultHttpClient(params);
        if (preferences.getCertPEM() != null)
            httpClient.getConnectionManager().getSchemeRegistry()
                    .register(new Scheme("https", 443, sslSocketFactory));
        return httpClient;
    }
}