org.apache.axis2.rpc.receivers.RPCUtil Maven / Gradle / Ivy
Go to download
Show more of this group Show more artifacts with this name
Show all versions of axis2-adb Show documentation
Show all versions of axis2-adb Show documentation
Axis2 Data Binding module
/*
* Licensed to the Apache Software Foundation (ASF) under one
* or more contributor license agreements. See the NOTICE file
* distributed with this work for additional information
* regarding copyright ownership. The ASF licenses this file
* to you 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 org.apache.axis2.rpc.receivers;
import org.apache.axiom.om.OMAbstractFactory;
import org.apache.axiom.om.OMElement;
import org.apache.axiom.om.OMNamespace;
import org.apache.axiom.om.OMText;
import org.apache.axiom.om.impl.builder.StAXOMBuilder;
import org.apache.axiom.om.impl.llom.factory.OMXMLBuilderFactory;
import org.apache.axiom.om.util.Base64;
import org.apache.axiom.soap.SOAPEnvelope;
import org.apache.axiom.soap.SOAPFactory;
import org.apache.axis2.AxisFault;
import org.apache.axis2.context.MessageContext;
import org.apache.axis2.databinding.typemapping.SimpleTypeMapper;
import org.apache.axis2.databinding.utils.BeanUtil;
import org.apache.axis2.databinding.utils.reader.NullXMLStreamReader;
import org.apache.axis2.description.AxisMessage;
import org.apache.axis2.description.AxisService;
import org.apache.axis2.description.java2wsdl.TypeTable;
import org.apache.axis2.engine.ObjectSupplier;
import org.apache.axis2.util.StreamWrapper;
import javax.xml.namespace.QName;
import javax.xml.stream.XMLStreamReader;
import java.lang.reflect.Array;
import java.lang.reflect.InvocationTargetException;
import java.lang.reflect.Method;
import java.util.Collection;
import java.util.Iterator;
public class RPCUtil {
private static String RETURN_WRAPPER = "return";
public static void processResponse(SOAPFactory fac, Object resObject,
OMElement bodyContent,
OMNamespace ns,
SOAPEnvelope envelope,
Method method,
boolean qualified,
TypeTable typeTable) {
if (resObject != null) {
//simple type
if (resObject instanceof OMElement) {
OMElement result = (OMElement) resObject;
bodyContent = fac.createOMElement(
method.getName() + "Response", ns);
OMElement resWrapper;
if (qualified) {
resWrapper = fac.createOMElement(RETURN_WRAPPER, ns.getNamespaceURI(),
ns.getPrefix());
} else {
resWrapper = fac.createOMElement(RETURN_WRAPPER, null);
}
resWrapper.addChild(result);
bodyContent.addChild(resWrapper);
} else if (SimpleTypeMapper.isSimpleType(resObject)) {
bodyContent = fac.createOMElement(
method.getName() + "Response", ns);
OMElement child;
if (qualified) {
child = fac.createOMElement(RETURN_WRAPPER, ns);
} else {
child = fac.createOMElement(RETURN_WRAPPER, null);
}
child.addChild(fac.createOMText(child, SimpleTypeMapper.getStringValue(resObject)));
bodyContent.addChild(child);
} else {
bodyContent = fac.createOMElement(
method.getName() + "Response", ns);
// Java Beans
QName returnWrapper;
if (qualified) {
returnWrapper = new QName(ns.getNamespaceURI(), RETURN_WRAPPER, ns.getPrefix());
} else {
returnWrapper = new QName(RETURN_WRAPPER);
}
XMLStreamReader xr = BeanUtil.getPullParser(resObject,
returnWrapper, typeTable, qualified, false);
StAXOMBuilder stAXOMBuilder =
OMXMLBuilderFactory.createStAXOMBuilder(
OMAbstractFactory.getOMFactory(), new StreamWrapper(xr));
OMElement documentElement = stAXOMBuilder.getDocumentElement();
if (documentElement != null) {
bodyContent.addChild(documentElement);
}
}
}
if (bodyContent != null) {
envelope.getBody().addChild(bodyContent);
}
}
public static void processObjectAsDocLitBare(SOAPFactory fac,
Object resObject,
OMElement bodyContent,
OMNamespace ns,
SOAPEnvelope envelope,
boolean qualified,
TypeTable typeTable,
String partName) {
if (resObject instanceof OMElement) {
OMElement result = (OMElement) resObject;
bodyContent = fac.createOMElement(
partName, ns);
bodyContent.addChild(result);
} else if (SimpleTypeMapper.isSimpleType(resObject)) {
bodyContent = fac.createOMElement(
partName, ns);
bodyContent.addChild(fac.createOMText(bodyContent,
SimpleTypeMapper.getStringValue(resObject)));
} else {
QName returnWrapper = new QName(ns.getNamespaceURI(), partName, ns.getPrefix());
XMLStreamReader xr = BeanUtil.getPullParser(resObject,
returnWrapper, typeTable, qualified, true);
StAXOMBuilder stAXOMBuilder =
OMXMLBuilderFactory.createStAXOMBuilder(
OMAbstractFactory.getOMFactory(), new StreamWrapper(xr));
OMElement documentElement = stAXOMBuilder.getDocumentElement();
if (documentElement != null) {
envelope.getBody().addChild(documentElement);
}
}
if (bodyContent != null) {
envelope.getBody().addChild(bodyContent);
}
}
public static Object[] processRequest(OMElement methodElement,
Method method, ObjectSupplier objectSupplier)
throws AxisFault {
Class[] parameters = method.getParameterTypes();
return BeanUtil.deserialize(methodElement, parameters, objectSupplier);
}
public static Object invokeServiceClass(AxisMessage inAxisMessage,
Method method,
Object implClass,
String messageNameSpace,
OMElement methodElement,
MessageContext inMessage) throws AxisFault,
IllegalAccessException, InvocationTargetException {
if (inAxisMessage.getElementQName() == null) {
// method accept empty SOAPbody
return method.invoke(implClass, new Object[0]);
} else {
QName elementQName = inAxisMessage.getElementQName();
messageNameSpace = elementQName.getNamespaceURI();
OMNamespace namespace = methodElement.getNamespace();
if (messageNameSpace != null) {
if (namespace == null) {
throw new AxisFault("namespace mismatch require " +
messageNameSpace +
" found none");
}
if (!messageNameSpace.equals(namespace.getNamespaceURI())) {
throw new AxisFault("namespace mismatch require " +
messageNameSpace +
" found " + methodElement.getNamespace().getNamespaceURI());
}
} else if (namespace != null) {
throw new AxisFault(
"namespace mismatch. Axis Oepration expects non-namespace " +
"qualified element. But received a namespace qualified element");
}
Object[] objectArray;
if (inAxisMessage.isWrapped()) {
objectArray = RPCUtil.processRequest(methodElement,
method, inMessage.getAxisService().getObjectSupplier());
} else {
objectArray = RPCUtil.processRequest((OMElement) methodElement.getParent(),
method, inMessage.getAxisService().getObjectSupplier());
}
return method.invoke(implClass, objectArray);
}
}
public static OMElement getResponseElement(QName resname,
Object[] objs,
boolean qualified,
TypeTable typeTable) {
if (qualified) {
return BeanUtil.getOMElement(resname, objs,
new QName(resname.getNamespaceURI(),
RETURN_WRAPPER,
resname.getPrefix()),
qualified,
typeTable);
} else {
return BeanUtil.getOMElement(resname, objs,
new QName(RETURN_WRAPPER), qualified,
typeTable);
}
}
public static void processResonseAsDocLitBare(Object resObject,
AxisService service,
SOAPEnvelope envelope,
SOAPFactory fac,
OMNamespace ns,
OMElement bodyContent,
MessageContext outMessage
) throws Exception {
QName elementQName = outMessage.getAxisMessage().getElementQName();
String partName = outMessage.getAxisMessage().getPartName();
if (resObject == null) {
processNullReturns(service, envelope, partName);
} else {
if (resObject instanceof Object[]) {
QName resName = new QName(elementQName.getNamespaceURI(),
partName,
elementQName.getPrefix());
OMElement bodyChild = RPCUtil.getResponseElement(resName,
(Object[]) resObject,
service.isElementFormDefault(),
service.getTypeTable());
envelope.getBody().addChild(bodyChild);
} else {
if (resObject.getClass().isArray()) {
int length = Array.getLength(resObject);
Object objArray[];
if (resObject instanceof byte[]) {
objArray = new Object[1];
objArray[0] = Base64.encode((byte[]) resObject);
} else {
objArray = new Object[length];
for (int i = 0; i < length; i++) {
objArray[i] = Array.get(resObject, i);
}
}
QName resName = new QName(elementQName.getNamespaceURI(),
partName,
elementQName.getPrefix());
OMElement bodyChild = RPCUtil.getResponseElement(resName,
objArray,
service.isElementFormDefault(),
service.getTypeTable());
envelope.getBody().addChild(bodyChild);
} else {
if (SimpleTypeMapper.isCollection(resObject.getClass())) {
Collection collection = (Collection) resObject;
int size = collection.size();
Object values[] = new Object[size];
int count = 0;
for (Iterator iterator = collection.iterator(); iterator.hasNext();) {
values[count] = iterator.next();
count++;
}
QName resName = new QName(elementQName.getNamespaceURI(),
partName,
elementQName.getPrefix());
OMElement bodyChild = RPCUtil.getResponseElement(resName,
values,
service.isElementFormDefault(),
service.getTypeTable());
envelope.getBody().addChild(bodyChild);
} else if (SimpleTypeMapper.isDataHandler(resObject.getClass())) {
OMElement resElemt;
if (service.isElementFormDefault()) {
resElemt = fac.createOMElement(partName, ns);
} else {
resElemt = fac.createOMElement(partName, null);
}
OMText text = fac.createOMText(resObject, true);
resElemt.addChild(text);
envelope.getBody().addChild(resElemt);
} else {
if (service.isElementFormDefault()) {
RPCUtil.processObjectAsDocLitBare(fac,
resObject,
bodyContent,
ns,
envelope,
service.isElementFormDefault(),
service.getTypeTable(),
partName);
} else {
RPCUtil.processObjectAsDocLitBare(fac,
resObject,
bodyContent,
ns,
envelope,
service.isElementFormDefault(),
null,
partName);
}
}
}
}
}
outMessage.setEnvelope(envelope);
}
/**
* This method is use to to crete the reposne when , the return value is null
*
* @param service Current AxisService
* @param envelope response enevelop
*/
private static void processNullReturns(AxisService service,
SOAPEnvelope envelope, String partName) {
QName resName;
if (service.isElementFormDefault()) {
resName = new QName(service.getSchemaTargetNamespace(),
partName,
service.getSchemaTargetNamespacePrefix());
} else {
resName = new QName(partName);
}
XMLStreamReader xr = new NullXMLStreamReader(resName);
StreamWrapper parser = new StreamWrapper(xr);
StAXOMBuilder stAXOMBuilder =
OMXMLBuilderFactory.createStAXOMBuilder(
OMAbstractFactory.getSOAP11Factory(), parser);
envelope.getBody().addChild(stAXOMBuilder.getDocumentElement());
}
public static void processResponseAsDocLitWrapped(Object resObject,
AxisService service,
Method method,
SOAPEnvelope envelope,
SOAPFactory fac,
OMNamespace ns,
OMElement bodyContent,
MessageContext outMessage
) throws Exception {
QName elementQName = outMessage.getAxisMessage().getElementQName();
if (resObject == null) {
QName resName;
if (service.isElementFormDefault()) {
resName = new QName(service.getSchemaTargetNamespace(),
RETURN_WRAPPER,
service.getSchemaTargetNamespacePrefix());
} else {
resName = new QName(RETURN_WRAPPER);
}
XMLStreamReader xr = new NullXMLStreamReader(resName);
StreamWrapper parser = new StreamWrapper(xr);
StAXOMBuilder stAXOMBuilder =
OMXMLBuilderFactory.createStAXOMBuilder(
OMAbstractFactory.getSOAP11Factory(), parser);
ns = fac.createOMNamespace(service.getSchemaTargetNamespace(),
service.getSchemaTargetNamespacePrefix());
OMElement bodyChild = fac.createOMElement(method.getName() + "Response", ns);
bodyChild.addChild(stAXOMBuilder.getDocumentElement());
envelope.getBody().addChild(bodyChild);
} else {
if (resObject instanceof Object[]) {
QName resName = new QName(elementQName.getNamespaceURI(),
method.getName() + "Response",
elementQName.getPrefix());
OMElement bodyChild = RPCUtil.getResponseElement(resName,
(Object[]) resObject,
service.isElementFormDefault(),
service.getTypeTable());
envelope.getBody().addChild(bodyChild);
} else {
if (resObject.getClass().isArray()) {
int length = Array.getLength(resObject);
Object objArray[];
if (resObject instanceof byte[]) {
objArray = new Object[1];
objArray[0] = Base64.encode((byte[]) resObject);
} else {
objArray = new Object[length];
for (int i = 0; i < length; i++) {
objArray[i] = Array.get(resObject, i);
}
}
QName resName = new QName(elementQName.getNamespaceURI(),
method.getName() + "Response",
elementQName.getPrefix());
OMElement bodyChild = RPCUtil.getResponseElement(resName,
objArray,
service.isElementFormDefault(),
service.getTypeTable());
envelope.getBody().addChild(bodyChild);
} else {
if (SimpleTypeMapper.isCollection(resObject.getClass())) {
Collection collection = (Collection) resObject;
int size = collection.size();
Object values[] = new Object[size];
int count = 0;
for (Iterator iterator = collection.iterator(); iterator.hasNext();) {
values[count] = iterator.next();
count++;
}
QName resName = new QName(elementQName.getNamespaceURI(),
method.getName() + "Response",
elementQName.getPrefix());
OMElement bodyChild = RPCUtil.getResponseElement(resName,
values,
service.isElementFormDefault(),
service.getTypeTable());
envelope.getBody().addChild(bodyChild);
} else if (SimpleTypeMapper.isDataHandler(resObject.getClass())) {
OMElement resElemt = fac.createOMElement(method.getName() + "Response", ns);
OMText text = fac.createOMText(resObject, true);
OMElement returnElement;
if (service.isElementFormDefault()) {
returnElement = fac.createOMElement(RETURN_WRAPPER, ns);
} else {
returnElement = fac.createOMElement(RETURN_WRAPPER, null);
}
returnElement.addChild(text);
resElemt.addChild(returnElement);
envelope.getBody().addChild(resElemt);
} else {
if (service.isElementFormDefault()) {
RPCUtil.processResponse(fac, resObject, bodyContent, ns,
envelope, method,
service.isElementFormDefault(),
service.getTypeTable());
} else {
RPCUtil.processResponse(fac, resObject, bodyContent, ns,
envelope, method,
service.isElementFormDefault(),
service.getTypeTable());
}
}
}
}
}
outMessage.setEnvelope(envelope);
}
}