org.apache.cxf.xmlbeans.XmlBeansSchemaInitializer Maven / Gradle / Ivy
Go to download
Show more of this group Show more artifacts with this name
Show all versions of cxf-bundle-minimal Show documentation
Show all versions of cxf-bundle-minimal Show documentation
Apache CXF Minimal Bundle Jar
/**
* 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.cxf.xmlbeans;
import java.io.InputStream;
import java.lang.reflect.Field;
import java.math.BigDecimal;
import java.math.BigInteger;
import java.net.URI;
import java.net.URISyntaxException;
import java.util.Calendar;
import java.util.Date;
import java.util.HashMap;
import java.util.Map;
import java.util.logging.Level;
import java.util.logging.Logger;
import javax.xml.namespace.QName;
import org.w3c.dom.Document;
import org.w3c.dom.Element;
import org.w3c.dom.NodeList;
import org.xml.sax.InputSource;
import org.apache.cxf.common.WSDLConstants;
import org.apache.cxf.common.logging.LogUtils;
import org.apache.cxf.common.util.StringUtils;
import org.apache.cxf.common.xmlschema.SchemaCollection;
import org.apache.cxf.helpers.DOMUtils;
import org.apache.cxf.helpers.XMLUtils;
import org.apache.cxf.helpers.XPathUtils;
import org.apache.cxf.service.ServiceModelVisitor;
import org.apache.cxf.service.model.MessagePartInfo;
import org.apache.cxf.service.model.ServiceInfo;
import org.apache.ws.commons.schema.XmlSchema;
import org.apache.ws.commons.schema.XmlSchemaElement;
import org.apache.ws.commons.schema.XmlSchemaException;
import org.apache.ws.commons.schema.XmlSchemaType;
import org.apache.ws.commons.schema.resolver.URIResolver;
import org.apache.xmlbeans.SchemaType;
import org.apache.xmlbeans.SchemaTypeSystem;
import org.apache.xmlbeans.XmlAnySimpleType;
import org.apache.xmlbeans.XmlObject;
import org.apache.xmlbeans.impl.schema.BuiltinSchemaTypeSystem;
import org.apache.xmlbeans.impl.schema.SchemaTypeSystemImpl;
/**
* Walks the service model and sets up the element/type names.
*/
class XmlBeansSchemaInitializer extends ServiceModelVisitor {
public static final String XML_BEANS_SCHEMA_PREFIX = "schema"
+ SchemaTypeSystemImpl.METADATA_PACKAGE_GEN + "/src/";
private static final Logger LOG = LogUtils.getLogger(XmlBeansSchemaInitializer.class);
private static final Map, Class> CLASS_MAP
= new HashMap, Class>();
private SchemaCollection schemas;
private XmlBeansDataBinding dataBinding;
private Map schemaMap
= new HashMap();
private URIResolver schemaResolver;
static {
CLASS_MAP.put(String.class, org.apache.xmlbeans.XmlString.class);
CLASS_MAP.put(Integer.class, org.apache.xmlbeans.XmlInt.class);
CLASS_MAP.put(Integer.TYPE, org.apache.xmlbeans.XmlInt.class);
CLASS_MAP.put(Short.class, org.apache.xmlbeans.XmlShort.class);
CLASS_MAP.put(Short.TYPE, org.apache.xmlbeans.XmlShort.class);
CLASS_MAP.put(Byte.class, org.apache.xmlbeans.XmlByte.class);
CLASS_MAP.put(Byte.TYPE, org.apache.xmlbeans.XmlByte.class);
CLASS_MAP.put(Float.class, org.apache.xmlbeans.XmlFloat.class);
CLASS_MAP.put(Float.TYPE, org.apache.xmlbeans.XmlFloat.class);
CLASS_MAP.put(Double.class, org.apache.xmlbeans.XmlDouble.class);
CLASS_MAP.put(Double.TYPE, org.apache.xmlbeans.XmlDouble.class);
CLASS_MAP.put(Long.class, org.apache.xmlbeans.XmlLong.class);
CLASS_MAP.put(Long.TYPE, org.apache.xmlbeans.XmlLong.class);
CLASS_MAP.put(Boolean.class, org.apache.xmlbeans.XmlBoolean.class);
CLASS_MAP.put(Boolean.TYPE, org.apache.xmlbeans.XmlBoolean.class);
CLASS_MAP.put(BigDecimal.class, org.apache.xmlbeans.XmlDecimal.class);
CLASS_MAP.put(BigInteger.class, org.apache.xmlbeans.XmlInteger.class);
CLASS_MAP.put(Date.class, org.apache.xmlbeans.XmlDate.class);
CLASS_MAP.put(Calendar.class, org.apache.xmlbeans.XmlDate.class);
CLASS_MAP.put(byte[].class, org.apache.xmlbeans.XmlBase64Binary.class);
}
public XmlBeansSchemaInitializer(ServiceInfo serviceInfo,
SchemaCollection col,
XmlBeansDataBinding db) {
super(serviceInfo);
schemas = col;
dataBinding = db;
schemaResolver = serviceInfo.getXmlSchemaCollection().getXmlSchemaCollection().getSchemaResolver();
}
public class XMLSchemaResolver implements URIResolver {
final SchemaTypeSystem sts;
public XMLSchemaResolver(SchemaTypeSystem sts) {
this.sts = sts;
}
public InputSource resolveEntity(String targetNamespace, String schemaLocation, String baseUri) {
InputStream ins = sts.getSourceAsStream(schemaLocation);
if (ins != null) {
return new InputSource(ins);
}
return null;
}
}
XmlSchema addSchemaElement(SchemaTypeSystem sts, Document doc,
Element elem, String file) throws URISyntaxException {
doc.appendChild(elem);
elem = DOMUtils.getFirstElement(elem);
while (elem != null) {
if (elem.getLocalName().equals("import")) {
URI uri = new URI(file);
String loc = elem.getAttribute("schemaLocation");
if (!StringUtils.isEmpty(loc)) {
URI locUri = uri.resolve(loc);
String newLoc = locUri.toString();
getSchema(sts, newLoc);
}
}
elem = DOMUtils.getNextElement(elem);
}
XmlSchema schema = dataBinding.addSchemaDocument(serviceInfo,
schemas,
doc,
file);
doc.removeChild(doc.getDocumentElement());
schemaMap.put(file, schema);
return schema;
}
protected XmlSchema getSchema(SchemaTypeSystem sts, String file) {
if (schemaMap.containsKey(file)) {
return schemaMap.get(file);
}
try {
InputSource fileSource = schemaResolver.resolveEntity(null,
file,
null);
String systemId = removePrefix(fileSource.getSystemId(),
XML_BEANS_SCHEMA_PREFIX);
return getSchemaInternal(sts, systemId);
} catch (XmlSchemaException e) {
if (LOG.isLoggable(Level.FINEST)) {
LOG.log(Level.FINEST,
"The XML catalog is not configured to map the file [" + file + "] ", e);
}
}
return getSchemaInternal(sts, file);
}
protected XmlSchema getSchemaInternal(SchemaTypeSystem sts, String file) {
InputStream ins = sts.getSourceAsStream(file);
if (ins == null) {
return null;
}
try {
//temporary marker to make sure recursive imports don't blow up
schemaMap.put(file, null);
Document doc = XMLUtils.parse(ins);
Element elem = doc.getDocumentElement();
doc.removeChild(elem);
if ("schema".equals(elem.getLocalName())
&& "http://www.w3.org/2001/XMLSchema".equals(elem.getNamespaceURI())) {
return addSchemaElement(sts, doc, elem, file);
}
Map ns = new HashMap();
ns.put("wsdl", WSDLConstants.NS_WSDL11);
ns.put("xsd", WSDLConstants.NS_SCHEMA_XSD);
XPathUtils xpath = new XPathUtils(ns);
NodeList list = xpath.getValueList("/wsdl:definitions/wsdl:types/xsd:schema", elem);
for (int x = 0; x < list.getLength(); x++) {
addSchemaElement(sts, doc, (Element)list.item(x), file + "#1");
}
return null;
} catch (Exception e) {
throw new RuntimeException("Failed to find schema for: " + file, e);
}
}
/**
* Removes the prefix ending with the given suffix. For instance, the value
* XYZ where the prefix is Y, the result will be Z. The removed string is XY.
*
* @param value the value from where the returned string is extracted
* @param prefixSuffix the prefix
* @return the rest of the string
*/
protected String removePrefix(String value, String prefixSuffix) {
return value.substring(value.indexOf(prefixSuffix) + prefixSuffix.length());
}
@Override
public void begin(MessagePartInfo part) {
LOG.finest(part.getName().toString());
// Check to see if the WSDL information has been filled in for us.
if (part.getTypeQName() != null || part.getElementQName() != null) {
checkForExistence(part);
return;
}
Class clazz = part.getTypeClass();
if (clazz == null) {
return;
}
boolean isFromWrapper = part.getMessageInfo().getOperation().isUnwrapped();
if (isFromWrapper && clazz.isArray() && !Byte.TYPE.equals(clazz.getComponentType())) {
clazz = clazz.getComponentType();
}
mapClass(part, clazz);
}
private void mapClass(MessagePartInfo part, Class clazz) {
if (!XmlObject.class.isAssignableFrom(clazz)) {
Class type = CLASS_MAP.get(clazz);
if (type == null) {
LOG.log(Level.SEVERE, clazz.getName() + " was not found in class map");
return;
}
SchemaTypeSystem sts = BuiltinSchemaTypeSystem.get();
SchemaType st2 = sts.typeForClassname(type.getName());
part.setProperty(SchemaType.class.getName(), st2);
part.setProperty(XmlAnySimpleType.class.getName(), type);
part.setTypeQName(st2.getName());
XmlSchemaType xmlSchema = schemas.getTypeByQName(st2.getName());
part.setXmlSchema(xmlSchema);
return;
}
try {
Field field = clazz.getField("type");
SchemaType st = (SchemaType)field.get(null);
part.setProperty(SchemaType.class.getName(), st);
SchemaTypeSystem sts = st.getTypeSystem();
schemas.getXmlSchemaCollection().setSchemaResolver(new XMLSchemaResolver(sts));
String sourceName = st.getSourceName();
XmlSchema schema = getSchema(sts, sourceName);
if (schema != null) {
if (st.isDocumentType()) {
XmlSchemaElement sct = schema.getElementByName(st.getDocumentElementName());
part.setXmlSchema(sct);
part.setElement(true);
part.setElementQName(st.getDocumentElementName());
part.setConcreteName(st.getDocumentElementName());
} else if (st.getComponentType() == SchemaType.ELEMENT) {
XmlSchemaElement sct = schema.getElementByName(st.getName());
part.setXmlSchema(sct);
part.setElement(true);
} else {
XmlSchemaType sct = schema.getTypeByName(st.getName());
part.setTypeQName(st.getName());
part.setXmlSchema(sct);
part.setElement(false);
}
} else {
if (st.isDocumentType()) {
part.setElement(true);
part.setElementQName(st.getDocumentElementName());
part.setConcreteName(st.getDocumentElementName());
part.setXmlSchema(schemas.getElementByQName(st.getDocumentElementName()));
} else if (st.getComponentType() == SchemaType.ELEMENT) {
part.setElement(true);
part.setElementQName(st.getName());
part.setConcreteName(st.getName());
part.setXmlSchema(schemas.getElementByQName(st.getName()));
} else {
part.setTypeQName(st.getName());
part.setElement(false);
part.setXmlSchema(schemas.getTypeByQName(st.getName()));
}
}
} catch (RuntimeException ex) {
throw ex;
} catch (Exception ex) {
throw new RuntimeException(ex);
}
}
public void checkForExistence(MessagePartInfo part) {
QName qn = part.getElementQName();
if (qn != null) {
XmlSchemaElement el = schemas.getElementByQName(qn);
if (el == null) {
Class clazz = part.getTypeClass();
if (clazz == null) {
return;
}
boolean isFromWrapper = part.getMessageInfo().getOperation().isUnwrapped();
if (isFromWrapper && clazz.isArray() && !Byte.TYPE.equals(clazz.getComponentType())) {
clazz = clazz.getComponentType();
}
mapClass(part, clazz);
}
}
}
}