piecework.client.LoadTester.java Source code

Introduction

Here is the source code for piecework.client.LoadTester.java

Source

/*
 * Copyright 2013 University of Washington
 *
 * Licensed under the Educational Community License, Version 1.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.opensource.org/licenses/ecl1.php
 *
 * 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 piecework.client;

import org.apache.http.HttpResponse;
import org.apache.http.client.HttpClient;
import org.apache.http.client.methods.HttpGet;
import org.apache.http.conn.ClientConnectionManager;
import org.apache.http.conn.scheme.Scheme;
import org.apache.http.conn.scheme.SchemeRegistry;
import org.apache.http.conn.ssl.SSLSocketFactory;
import org.apache.http.conn.ssl.X509HostnameVerifier;
import org.apache.http.impl.client.DefaultHttpClient;
import org.apache.http.impl.conn.BasicClientConnectionManager;
import org.apache.http.impl.conn.PoolingClientConnectionManager;
import org.springframework.context.annotation.AnnotationConfigApplicationContext;
import org.springframework.core.env.StandardEnvironment;
import piecework.security.KeyManagerCabinet;
import piecework.settings.SecuritySettings;

import java.security.KeyStore;

/**
 * @author James Renfro
 */
public class LoadTester {

    private final HttpClient client;

    public LoadTester(KeyStore keystore, SecuritySettings securitySettings) {
        ClientConnectionManager cm;
        try {
            SSLSocketFactory sslSocketFactory = new SSLSocketFactory(keystore,
                    new String(securitySettings.getKeystorePassword()));
            X509HostnameVerifier hostnameVerifier = SSLSocketFactory.ALLOW_ALL_HOSTNAME_VERIFIER;
            sslSocketFactory.setHostnameVerifier(hostnameVerifier);

            SchemeRegistry schemeRegistry = new SchemeRegistry();
            schemeRegistry.register(new Scheme("https", 8443, sslSocketFactory));

            cm = new PoolingClientConnectionManager(schemeRegistry);
        } catch (Exception e) {
            cm = new BasicClientConnectionManager();
        }
        this.client = new DefaultHttpClient(cm);
    }

    public void retrieveAllTasks() throws Exception {
        HttpGet get = new HttpGet("https://localhost:8443/piecework/api/v1/task.json");
        HttpResponse response = client.execute(get);
        System.out.println(response.getStatusLine());
    }

    public static final void main(String[] args) throws Exception {
        String profile = args.length > 0 ? args[0] : "workstation";

        StandardEnvironment environment = new StandardEnvironment();
        environment.setActiveProfiles(profile);
        AnnotationConfigApplicationContext ctx = new AnnotationConfigApplicationContext();
        ctx.setEnvironment(environment);
        ctx.register(ClientConfiguration.class);
        ctx.refresh();

        KeyManagerCabinet cabinet = ctx.getBean(KeyManagerCabinet.class);
        SecuritySettings securitySettings = ctx.getBean(SecuritySettings.class);

        LoadTester loadTester = new LoadTester(cabinet.getKeystore(), securitySettings);
        loadTester.retrieveAllTasks();
    }

}