se.riv.itinfra.tp.ping.PingConsumer Maven / Gradle / Ivy

Go to download
Show more of this group Show more artifacts with this name
Show all versions of itinfra-tp-ping-test-consumer Show documentation
There is a newer version: 3.2.0-RC5
Show newest version
/**
 * Copyright (c) 2013 Center for eHalsa i samverkan (CeHis).
 * 							
 *
 * This file is part of SKLTP.
 *
 * This library is free software; you can redistribute it and/or
 * modify it under the terms of the GNU Lesser General Public
 * License as published by the Free Software Foundation; either
 * version 2.1 of the License, or (at your option) any later version.
 *
 * This library is distributed in the hope that it will be useful,
 * but WITHOUT ANY WARRANTY; without even the implied warranty of
 * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the GNU
 * Lesser General Public License for more details.
 *
 * You should have received a copy of the GNU Lesser General Public
 * License along with this library; if not, write to the Free Software
 * Foundation, Inc., 51 Franklin Street, Fifth Floor, Boston, MA  02110-1301  USA
 */
package se.riv.itinfra.tp.ping;

import java.io.FileInputStream;
import java.net.URL;
import java.security.KeyStore;
import java.util.Random;

import javax.net.ssl.KeyManagerFactory;
import javax.net.ssl.TrustManagerFactory;
import javax.xml.ws.BindingProvider;

import org.apache.cxf.configuration.jsse.TLSClientParameters;
import org.apache.cxf.endpoint.Client;
import org.apache.cxf.frontend.ClientProxy;
import org.apache.cxf.transport.http.HTTPConduit;
import org.apache.cxf.transports.http.configuration.HTTPClientPolicy;
import org.w3.wsaddressing10.AttributedURIType;

import se.riv.itinfra.tp.ping.v1.rivtabp20.PingResponderInterface;
import se.riv.itinfra.tp.ping.v1.rivtabp20.PingResponderService;
import se.riv.itinfra.tp.pingresponder.v1.PingRequestType;
import se.riv.itinfra.tp.pingresponder.v1.PingResponseType;

public final class PingConsumer {

	private static String logicalAddress = "Ping";
	private static long sleepTime = 5000L;
	private static String endpointAddress = "https://localhost:20000/vp/Ping/1/rivtabp20";

	private static PingResponderInterface service = null;

	public static void main(String[] args) throws Exception {

		if (args.length > 0) {
			sleepTime = Long.parseLong(args[0]);
			endpointAddress = args[1];
			logicalAddress = args[2];
		}

		long randomness = sleepTime / 3L;
		Random random = new Random();
		System.out.println("Consumer connecting to " + endpointAddress);
		service = setupService(endpointAddress, logicalAddress);
		while (true) {
			long start = System.nanoTime();
			try {
				String p = callService(logicalAddress);
				System.out.println("Returned: " + p);
			} catch (Exception e) {
				System.out.println("Error: " + e.toString());
			} finally {
				System.out.println("Total time: " + (System.nanoTime() - start)
						/ 1000000L + " ms");
			}
			Thread.sleep((long) (sleepTime - randomness + 2L * randomness
					* random.nextDouble()));
		}

	}

	public static String callService(String logicalAddresss) {

		AttributedURIType logicalAddressHeader = new AttributedURIType();
		logicalAddressHeader.setValue(logicalAddresss);

		PingRequestType request = new PingRequestType();

		// The below request is created to mimic the sample xml-message
		request.setPingIn("Pinging!");

		try {
			PingResponseType result = service.ping(logicalAddressHeader,
					request);

			return ("Ping response=" + result.getPingUt());

		} catch (Exception ex) {
			System.out.println("Exception=" + ex.getMessage());
			return null;
		}
	}

	public static PingResponderInterface setupService(String serviceAddress,
			String logicalAddresss) {
		try {
			ClassLoader loader = Thread.currentThread().getContextClassLoader();
			URL wsdlUrl = loader
					.getResource("PingInteraction_1.0_rivtabp20.wsdl");

			PingResponderService service = new PingResponderService(wsdlUrl);
			PingResponderInterface serviceInterface = service
					.getPingResponderPort();

			BindingProvider provider = (BindingProvider) serviceInterface;
			provider.getRequestContext().put(
					"javax.xml.ws.service.endpoint.address", serviceAddress);

			Client client = ClientProxy.getClient(serviceInterface);
			HTTPConduit http = (HTTPConduit) client.getConduit();

			HTTPClientPolicy httpClientPolicy = new HTTPClientPolicy();
			httpClientPolicy.setConnectionTimeout(36000L);
			httpClientPolicy.setAllowChunking(false);
			httpClientPolicy.setReceiveTimeout(32000L);
			http.setClient(httpClientPolicy);

			if (serviceAddress.startsWith("https")) {
				TLSClientParameters tlsCP = setUpTlsClientParams();
				http.setTlsClientParameters(tlsCP);
			}

			return serviceInterface;
		} catch (Throwable ex) {
			System.out.println("Exception={}" + ex.getMessage());
			throw new RuntimeException(ex);
		}
	}

	private static TLSClientParameters setUpTlsClientParams() throws Exception {

		KeyStore trustStore = KeyStore.getInstance("JKS");
		String trustStoreLoc = "src/main/resources/test-certs/truststore.jks";
		String trustPassword = "password";
		trustStore.load(new FileInputStream(trustStoreLoc),
				trustPassword.toCharArray());

		String keyPassword = "password";
		KeyStore keyStore = KeyStore.getInstance("jks");
		String keyStoreLoc = "src/main/resources/test-certs/client.jks";
		keyStore.load(new FileInputStream(keyStoreLoc),
				keyPassword.toCharArray());

		TrustManagerFactory tmf = TrustManagerFactory
				.getInstance(TrustManagerFactory.getDefaultAlgorithm());
		tmf.init(trustStore);

		KeyManagerFactory kmf = KeyManagerFactory.getInstance(KeyManagerFactory
				.getDefaultAlgorithm());
		kmf.init(keyStore, keyPassword.toCharArray());

		TLSClientParameters tlsCP = new TLSClientParameters();
		tlsCP.setTrustManagers(tmf.getTrustManagers());
		tlsCP.setKeyManagers(kmf.getKeyManagers());

		// The following is not recommended and would not be done in a
		// prodcution environment,
		// this is just for illustrative purpose
		tlsCP.setDisableCNCheck(true);

		return tlsCP;

	}
}