Java tutorial
/** * * 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.openejb.server.axis.assembler; import com.ibm.wsdl.extensions.PopulatedExtensionRegistry; import com.ibm.wsdl.extensions.schema.SchemaConstants; import org.apache.commons.logging.Log; import org.apache.commons.logging.LogFactory; import org.apache.openejb.OpenEJBException; import org.apache.openejb.server.ServerRuntimeException; import org.apache.ws.commons.schema.XmlSchemaCollection; import org.w3c.dom.Element; import org.xml.sax.InputSource; import javax.wsdl.Definition; import javax.wsdl.Import; import javax.wsdl.Types; import javax.wsdl.WSDLException; import javax.wsdl.extensions.ExtensionRegistry; import javax.wsdl.extensions.UnknownExtensibilityElement; import javax.wsdl.extensions.schema.Schema; import javax.wsdl.factory.WSDLFactory; import javax.wsdl.xml.WSDLLocator; import javax.wsdl.xml.WSDLReader; import javax.xml.namespace.QName; import java.io.IOException; import java.io.InputStream; import java.net.URI; import java.util.ArrayList; import java.util.List; import java.util.Map; import java.util.jar.JarFile; import java.util.zip.ZipEntry; public class CommonsSchemaLoader { private static final Log log = LogFactory.getLog(CommonsSchemaLoader.class); private final URI wsdlUri; private final JarFile moduleFile; private final XmlSchemaCollection xmlSchemaCollection = new XmlSchemaCollection(); public CommonsSchemaLoader(URI wsdlUri, JarFile moduleFile) { this.wsdlUri = wsdlUri; this.moduleFile = moduleFile; } public XmlSchemaCollection loadSchema() throws OpenEJBException { Definition definition = readWsdl(wsdlUri); addImportsFromDefinition(definition); return xmlSchemaCollection; } private void addImportsFromDefinition(Definition definition) throws OpenEJBException { Types types = definition.getTypes(); if (types != null) { for (Object extensibilityElement : types.getExtensibilityElements()) { if (extensibilityElement instanceof Schema) { Schema unknownExtensibilityElement = (Schema) extensibilityElement; QName elementType = unknownExtensibilityElement.getElementType(); if (new QName("http://www.w3.org/2001/XMLSchema", "schema").equals(elementType)) { Element element = unknownExtensibilityElement.getElement(); xmlSchemaCollection.read(element); } } else if (extensibilityElement instanceof UnknownExtensibilityElement) { //This is allegedly obsolete as of axis-wsdl4j-1.2-RC3.jar which includes the Schema extension above. //The change notes imply that imported schemas should end up in Schema elements. They don't, so this is still needed. UnknownExtensibilityElement unknownExtensibilityElement = (UnknownExtensibilityElement) extensibilityElement; Element element = unknownExtensibilityElement.getElement(); String elementNamespace = element.getNamespaceURI(); String elementLocalName = element.getNodeName(); if ("http://www.w3.org/2001/XMLSchema".equals(elementNamespace) && "schema".equals(elementLocalName)) { xmlSchemaCollection.read(element); } } } } //noinspection unchecked Map<String, List<Import>> imports = definition.getImports(); if (imports != null) { for (Map.Entry<String, List<Import>> entry : imports.entrySet()) { String namespaceURI = entry.getKey(); List<Import> importList = entry.getValue(); for (Import anImport : importList) { //according to the 1.1 jwsdl mr shcema imports are supposed to show up here, //but according to the 1.0 spec there is supposed to be no Definition. Definition importedDef = anImport.getDefinition(); if (importedDef != null) { addImportsFromDefinition(importedDef); } else { log.warn("Missing definition in import for namespace " + namespaceURI); } } } } } private Definition readWsdl(URI wsdlURI) throws OpenEJBException { Definition definition; WSDLFactory wsdlFactory; try { wsdlFactory = WSDLFactory.newInstance(); } catch (WSDLException e) { throw new OpenEJBException("Could not create WSDLFactory", e); } WSDLReader wsdlReaderNoImport = wsdlFactory.newWSDLReader(); wsdlReaderNoImport.setFeature("javax.wsdl.importDocuments", false); ExtensionRegistry extensionRegistry = new PopulatedExtensionRegistry(); extensionRegistry.mapExtensionTypes(Types.class, SchemaConstants.Q_ELEM_XSD_1999, UnknownExtensibilityElement.class); extensionRegistry.registerDeserializer(Types.class, SchemaConstants.Q_ELEM_XSD_1999, extensionRegistry.getDefaultDeserializer()); extensionRegistry.registerSerializer(Types.class, SchemaConstants.Q_ELEM_XSD_1999, extensionRegistry.getDefaultSerializer()); extensionRegistry.mapExtensionTypes(Types.class, SchemaConstants.Q_ELEM_XSD_2000, UnknownExtensibilityElement.class); extensionRegistry.registerDeserializer(Types.class, SchemaConstants.Q_ELEM_XSD_2000, extensionRegistry.getDefaultDeserializer()); extensionRegistry.registerSerializer(Types.class, SchemaConstants.Q_ELEM_XSD_2000, extensionRegistry.getDefaultSerializer()); extensionRegistry.mapExtensionTypes(Types.class, SchemaConstants.Q_ELEM_XSD_2001, UnknownExtensibilityElement.class); extensionRegistry.registerDeserializer(Types.class, SchemaConstants.Q_ELEM_XSD_2001, extensionRegistry.getDefaultDeserializer()); extensionRegistry.registerSerializer(Types.class, SchemaConstants.Q_ELEM_XSD_2001, extensionRegistry.getDefaultSerializer()); wsdlReaderNoImport.setExtensionRegistry(extensionRegistry); JarWSDLLocator wsdlLocator = new JarWSDLLocator(wsdlURI); WSDLReader wsdlReader = wsdlFactory.newWSDLReader(); Thread thread = Thread.currentThread(); ClassLoader oldCl = thread.getContextClassLoader(); thread.setContextClassLoader(this.getClass().getClassLoader()); try { try { definition = wsdlReader.readWSDL(wsdlLocator); } catch (WSDLException e) { throw new OpenEJBException("Failed to read wsdl document", e); } catch (RuntimeException e) { throw new OpenEJBException(e.getMessage(), e); } } finally { thread.setContextClassLoader(oldCl); } return definition; } class JarWSDLLocator implements WSDLLocator { private final List<InputStream> streams = new ArrayList<InputStream>(); private final URI wsdlURI; private URI latestImportURI; public JarWSDLLocator(URI wsdlURI) { this.wsdlURI = wsdlURI; } public InputSource getBaseInputSource() { ZipEntry entry = moduleFile.getEntry(wsdlURI.toString()); if (entry == null) { throw new ServerRuntimeException( "The webservices.xml file points to a non-existant WSDL file " + wsdlURI.toString()); } InputStream wsdlInputStream; try { wsdlInputStream = moduleFile.getInputStream(entry); streams.add(wsdlInputStream); } catch (Exception e) { throw new ServerRuntimeException("Could not open stream to wsdl file", e); } return new InputSource(wsdlInputStream); } public String getBaseURI() { return wsdlURI.toString(); } public InputSource getImportInputSource(String parentLocation, String relativeLocation) { URI parentURI = URI.create(parentLocation); latestImportURI = parentURI.resolve(relativeLocation); InputStream importInputStream; try { ZipEntry entry = moduleFile.getEntry(latestImportURI.toString()); importInputStream = moduleFile.getInputStream(entry); streams.add(importInputStream); } catch (Exception e) { throw new ServerRuntimeException("Could not open stream to import file", e); } InputSource inputSource = new InputSource(importInputStream); inputSource.setSystemId(getLatestImportURI()); return inputSource; } public String getLatestImportURI() { return latestImportURI.toString(); } public void close() { for (InputStream inputStream : streams) { try { inputStream.close(); } catch (IOException e) { //ignore } } streams.clear(); } } }