/*
* 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.axis2.AxisFault;
import org.apache.axis2.Constants;
import org.apache.axis2.addressing.EndpointReference;
import org.apache.axis2.client.Call;
import org.apache.axis2.client.Options;
import org.apache.axis2.om.OMAbstractFactory;
import org.apache.axis2.om.OMElement;
import org.apache.axis2.om.OMNamespace;
import org.apache.axis2.soap.SOAPEnvelope;
import org.apache.axis2.soap.SOAPFactory;
import org.apache.axis2.soap.SOAPHeaderBlock;
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("===============================================================================");
Call call = new Call();
Options options = new Options();
options.setListenerTransportProtocol(Constants.TRANSPORT_HTTP);
options.setUseSeparateListener(false);
call.setClientOptions(options);
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...");
SOAPEnvelope result = call.invokeBlocking(opStr, getRequestEnvelope(opStr, param1, param2,
serviceGroupContextId));
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...");
SOAPEnvelope result = call.invokeBlocking(opStr, getPreviousRequestEnvelope(opStr, param,
serviceGroupContextId));
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) {
String serviceGroupContextId = resultEnvelope.getHeader()
.getFirstChildWithName(new QName("ReplyTo"))
.getFirstChildWithName(new QName("ReferenceParameters"))
.getFirstChildWithName(new QName("ServiceGroupId")).getText();
return serviceGroupContextId;
}
}