modules/samples/deprecated/sgccalculator/src/sample/sgccalculator/CalcClient.java - axis-axis2-java-rampart - Git at Google

 /*
  * Copyright 2004,2005 The Apache Software Foundation.
  *
  * Licensed 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 sample.sgccalculator;

 import org.apache.axiom.om.OMAbstractFactory;
 import org.apache.axiom.om.OMElement;
 import org.apache.axiom.om.OMNamespace;
 import org.apache.axiom.soap.SOAPEnvelope;
 import org.apache.axiom.soap.SOAPFactory;
 import org.apache.axiom.soap.SOAPHeaderBlock;
 import org.apache.axis2.AxisFault;
 import org.apache.axis2.Constants;
 import org.apache.axis2.wsdl.WSDLConstants;
 import org.apache.axis2.addressing.EndpointReference;
 import org.apache.axis2.client.OperationClient;
 import org.apache.axis2.client.Options;
 import org.apache.axis2.client.ServiceClient;
 import org.apache.axis2.context.MessageContext;

 import javax.xml.namespace.QName;
 import javax.xml.stream.FactoryConfigurationError;
 import javax.xml.stream.XMLOutputFactory;
 import javax.xml.stream.XMLStreamException;
 import javax.xml.stream.XMLStreamWriter;
 import java.io.BufferedReader;
 import java.io.IOException;
 import java.io.InputStreamReader;

 public class CalcClient {

     final static String addService = "http://localhost:8080/axis2/services/AddService";

     final static String substractService = "http://localhost:8080/axis2/services/SubstractService";

     final static String multiplyService = "http://localhost:8080/axis2/services/MultiplyService";

     public static void main(String[] args) throws AxisFault, IOException {

         System.out.println("\nTHIS IS A SAMPLE APPLICATION TO DEMONSTRATE THE FUNCTIONALITY OF SERVICE GROUPS");
         System.out.println("===============================================================================");
         Options options = new Options();
         options.setTransportInProtocol(Constants.TRANSPORT_HTTP);

         boolean exit = false;
         String serviceGroupContextId = null;
         while (!exit) {
             BufferedReader reader = new BufferedReader(new InputStreamReader(System.in));

             System.out.print("\n\nNew round (n) /continue round (c).../exit (e)");
             String option = reader.readLine();
             if ("e".equalsIgnoreCase(option)) {
                 System.out.println("Exiting calculator...");
                 return;
             }

             if (!"n".equalsIgnoreCase(option) && !"c".equalsIgnoreCase(option)) {
                 System.out.println("Error: Invalid option");
                 continue;
             }

             System.out.print("Please Select the service ( '+' / '-' / '*' )....");
             String operation = reader.readLine();
             if (!"+".equalsIgnoreCase(operation) && !"-".equalsIgnoreCase(operation) && !"*".equalsIgnoreCase(operation)) {
                 System.out.println("Error: Invalid option");
                 continue;
             }

             if ("+".equals(operation))
                 options.setTo(new EndpointReference(addService));
             else if ("-".equals(operation))
                 options.setTo(new EndpointReference(substractService));
             else if ("*".equals(operation))
                 options.setTo(new EndpointReference(multiplyService));

             if ("n".equalsIgnoreCase(option)) {
                 System.out.print("Enter parameter 1...");
                 String param1Str = reader.readLine();
                 System.out.print("Enter parameter 2...");
                 String param2Str = reader.readLine();
                 int param1 = Integer.parseInt(param1Str);
                 int param2 = Integer.parseInt(param2Str);

                 String opStr = null;
                 if ("+".equals(operation)) {
                     opStr = "add";
                 } else if ("-".equals(operation)) {
                     opStr = "substract";
                 } else if ("*".equals(operation)) {
                     opStr = "multiply";
                 }
                 System.out.println("Invoking...");

                 ServiceClient serviceClient = new ServiceClient();
                 serviceClient.engageModule(new QName("addressing"));
                 serviceClient.setOptions(options);
                 options.setAction("urn:" + opStr);
                 MessageContext requetMessageContext = new MessageContext();
                 requetMessageContext.setEnvelope(getRequestEnvelope(opStr, param1, param2,
                         serviceGroupContextId));

                 OperationClient opClient = serviceClient.createClient(ServiceClient.ANON_OUT_IN_OP);
                 opClient.addMessageContext(requetMessageContext);
                 opClient.setOptions(options);

                 opClient.execute(true);

                 SOAPEnvelope result = opClient.getMessageContext(
                         WSDLConstants.MESSAGE_LABEL_IN_VALUE).getEnvelope();

                 printResult(result);

                 if (serviceGroupContextId == null)
                     serviceGroupContextId = getServiceGroupContextId(result);


             } else if ("c".equalsIgnoreCase(option)) {
                 if (serviceGroupContextId == null) {
                     System.out.println("Error: First operation must be a New one. Please select 'n'");
                     continue;
                 }

                 System.out.print("Enter parameter...");
                 String paramStr = reader.readLine();
                 int param = Integer.parseInt(paramStr);
                 String opStr = null;
                 if ("+".equals(operation)) {
                     opStr = "addPrevious";
                 } else if ("-".equals(operation)) {
                     opStr = "substractPrevious";
                 } else if ("*".equals(operation)) {
                     opStr = "multiplyPrevious";
                 }

                 System.out.println("Invoking...");


                 ServiceClient serviceClient = new ServiceClient();
                 serviceClient.setOptions(options);
                 options.setAction("urn:" + opStr);
                 MessageContext requetMessageContext = new MessageContext();
                 requetMessageContext.setEnvelope(getPreviousRequestEnvelope(opStr, param,
                         serviceGroupContextId));

                 OperationClient opClient = serviceClient.createClient(ServiceClient.ANON_OUT_IN_OP);
                 opClient.addMessageContext(requetMessageContext);
                 opClient.setOptions(options);

                 opClient.execute(true);

                 SOAPEnvelope result = opClient.getMessageContext(
                         WSDLConstants.MESSAGE_LABEL_IN_VALUE).getEnvelope();

                 printResult(result);

             }

         }
     }

     public static void printResult(SOAPEnvelope result) {
         try {
             XMLStreamWriter writer = XMLOutputFactory.newInstance()
                     .createXMLStreamWriter(System.out);
             if (result != null) {
                 OMElement resultOM = result.getBody().getFirstChildWithName(new QName("result"));
                 System.out.println("Result is:" + resultOM.getText());
             } else
                 System.out.println("Result is null");
         } catch (XMLStreamException e) {
             e.printStackTrace();
         } catch (FactoryConfigurationError e) {
             e.printStackTrace();
         }
     }

     public static SOAPEnvelope getRequestEnvelope(String operationName,
                                                   int param1, int param2, String groupContextId) {
         SOAPFactory fac = OMAbstractFactory.getSOAP11Factory();
         SOAPEnvelope envelope = fac.getDefaultEnvelope();
         OMNamespace namespace = fac.createOMNamespace(
                 "http://axis2/test/namespace1", "ns1");

         OMElement params = fac.createOMElement(operationName, namespace);
         OMElement param1OM = fac.createOMElement("param1", namespace);
         OMElement param2OM = fac.createOMElement("param2", namespace);
         param1OM.setText(Integer.toString(param1));
         param2OM.setText(Integer.toString(param2));
         params.addChild(param1OM);
         params.addChild(param2OM);
         envelope.getBody().setFirstChild(params);

         if (groupContextId != null) {
             OMNamespace axis2Namespace = fac.createOMNamespace(
                     Constants.AXIS2_NAMESPACE_URI,
                     Constants.AXIS2_NAMESPACE_PREFIX);
             SOAPHeaderBlock soapHeaderBlock = envelope.getHeader()
                     .addHeaderBlock(Constants.SERVICE_GROUP_ID, axis2Namespace);
             soapHeaderBlock.setText(groupContextId);
         }

         return envelope;
     }

     public static SOAPEnvelope getPreviousRequestEnvelope(String operationName,
                                                           int param, String groupContextId) {
         SOAPFactory fac = OMAbstractFactory.getSOAP11Factory();
         SOAPEnvelope envelope = fac.getDefaultEnvelope();
         OMNamespace namespace = fac.createOMNamespace(
                 "http://axis2/test/namespace1", "ns1");

         OMElement params = fac.createOMElement(operationName, namespace);
         OMElement paramOM = fac.createOMElement("param", namespace);
         paramOM.setText(Integer.toString(param));
         params.addChild(paramOM);
         envelope.getBody().setFirstChild(params);

         if (groupContextId != null) {
             OMNamespace axis2Namespace = fac.createOMNamespace(
                     Constants.AXIS2_NAMESPACE_URI,
                     Constants.AXIS2_NAMESPACE_PREFIX);
             SOAPHeaderBlock soapHeaderBlock = envelope.getHeader()
                     .addHeaderBlock(Constants.SERVICE_GROUP_ID, axis2Namespace);
             soapHeaderBlock.setText(groupContextId);
         }

         return envelope;
     }

     public static String getServiceGroupContextId(SOAPEnvelope resultEnvelope) {
         return resultEnvelope.getHeader()
                 .getFirstChildWithName(new QName("ReplyTo"))
                 .getFirstChildWithName(new QName("ReferenceParameters"))
                 .getFirstChildWithName(new QName("ServiceGroupId")).getText();
     }
 }
	/*
	* Copyright 2004,2005 The Apache Software Foundation.
	*
	* Licensed 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 sample.sgccalculator;

	import org.apache.axiom.om.OMAbstractFactory;
	import org.apache.axiom.om.OMElement;
	import org.apache.axiom.om.OMNamespace;
	import org.apache.axiom.soap.SOAPEnvelope;
	import org.apache.axiom.soap.SOAPFactory;
	import org.apache.axiom.soap.SOAPHeaderBlock;
	import org.apache.axis2.AxisFault;
	import org.apache.axis2.Constants;
	import org.apache.axis2.wsdl.WSDLConstants;
	import org.apache.axis2.addressing.EndpointReference;
	import org.apache.axis2.client.OperationClient;
	import org.apache.axis2.client.Options;
	import org.apache.axis2.client.ServiceClient;
	import org.apache.axis2.context.MessageContext;

	import javax.xml.namespace.QName;
	import javax.xml.stream.FactoryConfigurationError;
	import javax.xml.stream.XMLOutputFactory;
	import javax.xml.stream.XMLStreamException;
	import javax.xml.stream.XMLStreamWriter;
	import java.io.BufferedReader;
	import java.io.IOException;
	import java.io.InputStreamReader;

	public class CalcClient {

	final static String addService = "http://localhost:8080/axis2/services/AddService";

	final static String substractService = "http://localhost:8080/axis2/services/SubstractService";

	final static String multiplyService = "http://localhost:8080/axis2/services/MultiplyService";

	public static void main(String[] args) throws AxisFault, IOException {

	System.out.println("\nTHIS IS A SAMPLE APPLICATION TO DEMONSTRATE THE FUNCTIONALITY OF SERVICE GROUPS");
	System.out.println("===============================================================================");
	Options options = new Options();
	options.setTransportInProtocol(Constants.TRANSPORT_HTTP);

	boolean exit = false;
	String serviceGroupContextId = null;
	while (!exit) {
	BufferedReader reader = new BufferedReader(new InputStreamReader(System.in));

	System.out.print("\n\nNew round (n) /continue round (c).../exit (e)");
	String option = reader.readLine();
	if ("e".equalsIgnoreCase(option)) {
	System.out.println("Exiting calculator...");
	return;
	}

	if (!"n".equalsIgnoreCase(option) && !"c".equalsIgnoreCase(option)) {
	System.out.println("Error: Invalid option");
	continue;
	}

	System.out.print("Please Select the service ( '+' / '-' / '*' )....");
	String operation = reader.readLine();
	if (!"+".equalsIgnoreCase(operation) && !"-".equalsIgnoreCase(operation) && !"*".equalsIgnoreCase(operation)) {
	System.out.println("Error: Invalid option");
	continue;
	}

	if ("+".equals(operation))
	options.setTo(new EndpointReference(addService));
	else if ("-".equals(operation))
	options.setTo(new EndpointReference(substractService));
	else if ("*".equals(operation))
	options.setTo(new EndpointReference(multiplyService));

	if ("n".equalsIgnoreCase(option)) {
	System.out.print("Enter parameter 1...");
	String param1Str = reader.readLine();
	System.out.print("Enter parameter 2...");
	String param2Str = reader.readLine();
	int param1 = Integer.parseInt(param1Str);
	int param2 = Integer.parseInt(param2Str);

	String opStr = null;
	if ("+".equals(operation)) {
	opStr = "add";
	} else if ("-".equals(operation)) {
	opStr = "substract";
	} else if ("*".equals(operation)) {
	opStr = "multiply";
	}
	System.out.println("Invoking...");

	ServiceClient serviceClient = new ServiceClient();
	serviceClient.engageModule(new QName("addressing"));
	serviceClient.setOptions(options);
	options.setAction("urn:" + opStr);
	MessageContext requetMessageContext = new MessageContext();
	requetMessageContext.setEnvelope(getRequestEnvelope(opStr, param1, param2,
	serviceGroupContextId));

	OperationClient opClient = serviceClient.createClient(ServiceClient.ANON_OUT_IN_OP);
	opClient.addMessageContext(requetMessageContext);
	opClient.setOptions(options);

	opClient.execute(true);

	SOAPEnvelope result = opClient.getMessageContext(
	WSDLConstants.MESSAGE_LABEL_IN_VALUE).getEnvelope();

	printResult(result);

	if (serviceGroupContextId == null)
	serviceGroupContextId = getServiceGroupContextId(result);


	} else if ("c".equalsIgnoreCase(option)) {
	if (serviceGroupContextId == null) {
	System.out.println("Error: First operation must be a New one. Please select 'n'");
	continue;
	}

	System.out.print("Enter parameter...");
	String paramStr = reader.readLine();
	int param = Integer.parseInt(paramStr);
	String opStr = null;
	if ("+".equals(operation)) {
	opStr = "addPrevious";
	} else if ("-".equals(operation)) {
	opStr = "substractPrevious";
	} else if ("*".equals(operation)) {
	opStr = "multiplyPrevious";
	}

	System.out.println("Invoking...");


	ServiceClient serviceClient = new ServiceClient();
	serviceClient.setOptions(options);
	options.setAction("urn:" + opStr);
	MessageContext requetMessageContext = new MessageContext();
	requetMessageContext.setEnvelope(getPreviousRequestEnvelope(opStr, param,
	serviceGroupContextId));

	OperationClient opClient = serviceClient.createClient(ServiceClient.ANON_OUT_IN_OP);
	opClient.addMessageContext(requetMessageContext);
	opClient.setOptions(options);

	opClient.execute(true);

	SOAPEnvelope result = opClient.getMessageContext(
	WSDLConstants.MESSAGE_LABEL_IN_VALUE).getEnvelope();

	printResult(result);

	}

	}
	}

	public static void printResult(SOAPEnvelope result) {
	try {
	XMLStreamWriter writer = XMLOutputFactory.newInstance()
	.createXMLStreamWriter(System.out);
	if (result != null) {
	OMElement resultOM = result.getBody().getFirstChildWithName(new QName("result"));
	System.out.println("Result is:" + resultOM.getText());
	} else
	System.out.println("Result is null");
	} catch (XMLStreamException e) {
	e.printStackTrace();
	} catch (FactoryConfigurationError e) {
	e.printStackTrace();
	}
	}

	public static SOAPEnvelope getRequestEnvelope(String operationName,
	int param1, int param2, String groupContextId) {
	SOAPFactory fac = OMAbstractFactory.getSOAP11Factory();
	SOAPEnvelope envelope = fac.getDefaultEnvelope();
	OMNamespace namespace = fac.createOMNamespace(
	"http://axis2/test/namespace1", "ns1");

	OMElement params = fac.createOMElement(operationName, namespace);
	OMElement param1OM = fac.createOMElement("param1", namespace);
	OMElement param2OM = fac.createOMElement("param2", namespace);
	param1OM.setText(Integer.toString(param1));
	param2OM.setText(Integer.toString(param2));
	params.addChild(param1OM);
	params.addChild(param2OM);
	envelope.getBody().setFirstChild(params);

	if (groupContextId != null) {
	OMNamespace axis2Namespace = fac.createOMNamespace(
	Constants.AXIS2_NAMESPACE_URI,
	Constants.AXIS2_NAMESPACE_PREFIX);
	SOAPHeaderBlock soapHeaderBlock = envelope.getHeader()
	.addHeaderBlock(Constants.SERVICE_GROUP_ID, axis2Namespace);
	soapHeaderBlock.setText(groupContextId);
	}

	return envelope;
	}

	public static SOAPEnvelope getPreviousRequestEnvelope(String operationName,
	int param, String groupContextId) {
	SOAPFactory fac = OMAbstractFactory.getSOAP11Factory();
	SOAPEnvelope envelope = fac.getDefaultEnvelope();
	OMNamespace namespace = fac.createOMNamespace(
	"http://axis2/test/namespace1", "ns1");

	OMElement params = fac.createOMElement(operationName, namespace);
	OMElement paramOM = fac.createOMElement("param", namespace);
	paramOM.setText(Integer.toString(param));
	params.addChild(paramOM);
	envelope.getBody().setFirstChild(params);

	if (groupContextId != null) {
	OMNamespace axis2Namespace = fac.createOMNamespace(
	Constants.AXIS2_NAMESPACE_URI,
	Constants.AXIS2_NAMESPACE_PREFIX);
	SOAPHeaderBlock soapHeaderBlock = envelope.getHeader()
	.addHeaderBlock(Constants.SERVICE_GROUP_ID, axis2Namespace);
	soapHeaderBlock.setText(groupContextId);
	}

	return envelope;
	}

	public static String getServiceGroupContextId(SOAPEnvelope resultEnvelope) {
	return resultEnvelope.getHeader()
	.getFirstChildWithName(new QName("ReplyTo"))
	.getFirstChildWithName(new QName("ReferenceParameters"))
	.getFirstChildWithName(new QName("ServiceGroupId")).getText();
	}
	}