/* | |
* Copyright 2004,2005 The Apache Software Foundation. | |
* Copyright 2006 International Business Machines Corp. | |
* | |
* 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 org.apache.axis2.jaxws.param; | |
import java.io.ByteArrayInputStream; | |
import javax.xml.stream.XMLInputFactory; | |
import javax.xml.stream.XMLStreamReader; | |
import javax.xml.ws.WebServiceException; | |
import javax.xml.ws.Service.Mode; | |
import org.apache.axiom.om.OMAbstractFactory; | |
import org.apache.axiom.om.OMElement; | |
import org.apache.axiom.om.impl.builder.StAXOMBuilder; | |
import org.apache.axiom.soap.SOAPBody; | |
import org.apache.axiom.soap.SOAPEnvelope; | |
import org.apache.axiom.soap.SOAPFactory; | |
import org.apache.axiom.soap.impl.builder.StAXSOAPModelBuilder; | |
import org.apache.axis2.jaxws.util.SoapUtils; | |
/* | |
* The StringParameter is a simple implementation of the Parameter | |
* interface for binding XML strings. | |
*/ | |
public class StringParameter implements Parameter { | |
private String value; | |
public StringParameter() { | |
//do nothing | |
} | |
public StringParameter(String s) { | |
this.value = s; | |
} | |
/** | |
* @see Parameter#getValue() | |
*/ | |
public Object getValue() { | |
return value; | |
} | |
/** | |
* @see Parameter#setValue(Object) | |
*/ | |
public void setValue(Object v) { | |
this.value = (String) v; | |
} | |
/** | |
* @see Parameter#getValueAsStreamReader() | |
*/ | |
public XMLStreamReader getValueAsStreamReader() { | |
if (value != null) { | |
try { | |
XMLStreamReader reader = XMLInputFactory.newInstance().createXMLStreamReader( | |
new ByteArrayInputStream(value.getBytes())); | |
return reader; | |
} catch (Exception e) { | |
e.printStackTrace(); | |
} | |
} | |
else { | |
throw new WebServiceException("Cannot create XMLStreamReader from null value"); | |
} | |
return null; | |
} | |
/** | |
* @see Parameter#toOMElement() | |
*/ | |
public OMElement toOMElement() { | |
try { | |
StAXOMBuilder builder = | |
new StAXOMBuilder(OMAbstractFactory.getOMFactory(), getValueAsStreamReader()); | |
OMElement documentElement = builder.getDocumentElement(); | |
return documentElement; | |
} catch (Exception e) { | |
throw new WebServiceException(e.getMessage()); | |
} | |
} | |
/** | |
* @see Parameter#fromOM(OMElement) | |
*/ | |
public void fromOM(OMElement documentElement){ | |
this.value = documentElement.toString(); | |
} | |
public SOAPEnvelope toEnvelope(Mode mode, String soapVersionURI) { | |
// TODO Auto-generated method stub | |
SOAPFactory soapfactory = SoapUtils.getSoapFactory(soapVersionURI); | |
SOAPEnvelope env = null; | |
try{ | |
if(mode !=null && mode.equals(Mode.MESSAGE)){ | |
/* I am assuming that if the mode is message then param has a soap xml with | |
* Envelope as root node. If not then I will throw a webservice exception. | |
*/ | |
XMLStreamReader reader = XMLInputFactory.newInstance().createXMLStreamReader(new ByteArrayInputStream(value.getBytes())); | |
StAXSOAPModelBuilder builder = new StAXSOAPModelBuilder(reader, soapfactory, soapVersionURI); | |
return builder.getSOAPEnvelope(); | |
} | |
else{ | |
/* I am assuming that param is a plain xml string. | |
* This code is return just to cater to dispatch sync call where we are using | |
* serviceClient. | |
*/ | |
env = soapfactory.getDefaultEnvelope(); | |
env.getHeader().detach(); | |
SOAPBody body = env.getBody(); | |
body.addChild(toOMElement()); | |
} | |
}catch(Exception e){ | |
throw new WebServiceException(e.getMessage()); | |
} | |
return env; | |
} | |
public void fromEnvelope(Mode mode, SOAPEnvelope env) { | |
// TODO Auto-generated method stub | |
if(env == null){ | |
//add validation code here | |
} | |
if(mode == null || mode.equals(Mode.PAYLOAD)){ | |
OMElement om = env.getBody(); | |
fromOM(om.getFirstElement()); | |
} | |
else if(mode.equals(Mode.MESSAGE)){ | |
fromOM(env); | |
} | |
} | |
} |