1 files changed, 184 insertions, 0 deletions
diff --git a/sdo-java/branches/sdo-java-M2/sdo/impl/src/main/java/org/apache/tuscany/sdo/util/resource/SDOXMLResourceImpl.java b/sdo-java/branches/sdo-java-M2/sdo/impl/src/main/java/org/apache/tuscany/sdo/util/resource/SDOXMLResourceImpl.java
new file mode 100644
index 0000000000..d08264a702
--- /dev/null
+++ b/sdo-java/branches/sdo-java-M2/sdo/impl/src/main/java/org/apache/tuscany/sdo/util/resource/SDOXMLResourceImpl.java
@@ -0,0 +1,184 @@
+/**
+ *
+ *  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.tuscany.sdo.util.resource;
+
+import java.io.IOException;
+import java.security.AccessController;
+import java.security.PrivilegedActionException;
+import java.security.PrivilegedExceptionAction;
+import java.util.HashMap;
+import java.util.Map;
+
+import javax.xml.stream.XMLStreamException;
+import javax.xml.stream.XMLStreamReader;
+
+import org.apache.tuscany.sdo.util.StAX2SAXAdapter;
+import org.eclipse.emf.common.util.URI;
+import org.eclipse.emf.ecore.resource.Resource;
+import org.eclipse.emf.ecore.xmi.XMIException;
+import org.eclipse.emf.ecore.xmi.XMLHelper;
+import org.eclipse.emf.ecore.xmi.XMLLoad;
+import org.eclipse.emf.ecore.xmi.XMLResource;
+import org.eclipse.emf.ecore.xmi.impl.XMLHelperImpl;
+import org.eclipse.emf.ecore.xmi.impl.XMLLoadImpl;
+import org.eclipse.emf.ecore.xmi.impl.XMLResourceImpl;
+import org.xml.sax.ContentHandler;
+import org.xml.sax.SAXException;
+
+public class SDOXMLResourceImpl extends XMLResourceImpl {
+    private XMLStreamReader reader;
+
+    /**
+     * [rfeng] Override the XMLHelperImpl to replace the NamespaceSupport so that it's aware of the NamespaceContext from the XMLStreamReader
+     */
+    public static class SDOXMLHelperImpl extends XMLHelperImpl {
+
+        private static class StreamNamespaceSupport extends XMLHelperImpl.NamespaceSupport {
+            private XMLStreamReader reader;
+
+            public String getPrefix(String uri) {
+                if (reader == null)
+                    return super.getPrefix(uri);
+                String prefix = null;
+                try {
+                    prefix = uri != null ? reader.getNamespaceContext().getPrefix(uri) : null;
+                } catch (Exception e) {
+                    // HACK:
+                    // java.lang.UnsupportedOperationException
+                    // at org.apache.axiom.om.impl.llom.OMStAXWrapper.getNamespaceContext(OMStAXWrapper.java:984)
+
+                    prefix = null;
+                }
+                return prefix != null ? prefix : super.getPrefix(uri);
+            }
+
+            public String getURI(String prefix) {
+                if (reader == null)
+                    return super.getURI(prefix);
+                String uri;
+                try {
+                    uri = prefix != null ? reader.getNamespaceContext().getNamespaceURI(prefix) : null;
+                } catch (Exception e) {
+                    // HACK:
+                    // java.lang.UnsupportedOperationException
+                    // at org.apache.axiom.om.impl.llom.OMStAXWrapper.getNamespaceContext(OMStAXWrapper.java:984)
+
+                    uri = null;
+                }
+                return uri != null && uri.length()!=0 ? uri : super.getURI(prefix);
+            }
+
+            public StreamNamespaceSupport(XMLStreamReader reader) {
+                super();
+                this.reader = reader;
+            }
+
+        }
+
+        public SDOXMLHelperImpl(XMLResource resource, XMLStreamReader reader) {
+            this(reader);
+            setResource(resource);
+        }
+
+        public SDOXMLHelperImpl(XMLStreamReader reader) {
+            super();
+            this.namespaceSupport = new StreamNamespaceSupport(reader);
+        }
+    }
+
+    /**
+     * An EMF XMLLoad that loads a model from a StAX stream
+     */
+    public class SDOXMLLoadImpl extends XMLLoadImpl {
+        public SDOXMLLoadImpl(XMLHelper helper) {
+            super(helper);
+        }
+
+        /**
+         * Start parsing an XMLReader with the default handler.
+         */
+        public void load(XMLResource resource, final XMLStreamReader reader, Map options) throws IOException {
+            this.resource = resource;
+            Map mergedOptions = new HashMap(defaultLoadOptions);
+            if (options != null)
+                mergedOptions.putAll(options);
+
+            this.options = mergedOptions;
+
+            final ContentHandler handler = makeDefaultHandler();
+
+            if (errors != null) {
+                errors.clear();
+            }
+
+            final StAX2SAXAdapter adapter = new StAX2SAXAdapter(true);
+            // Parse the XMLReader and generate SAX events
+            try {
+                AccessController.doPrivileged(new PrivilegedExceptionAction() {
+                    public Object run() throws XMLStreamException, SAXException {
+                        adapter.parse(reader, handler);
+                        return null;
+                    }
+                });
+            } catch (PrivilegedActionException e) {
+                throw new Resource.IOWrappedException(e.getException());
+            }
+
+            helper = null;
+            if (!resource.getErrors().isEmpty()) {
+                Exception error = (Exception) resource.getErrors().get(0);
+                if (error instanceof XMIException) {
+                    XMIException exception = (XMIException) error;
+                    if (exception.getWrappedException() != null) {
+                        throw new Resource.IOWrappedException(exception.getWrappedException());
+                    }
+                }
+                throw new Resource.IOWrappedException(error);
+            }
+        }
+    }
+
+    public SDOXMLResourceImpl(URI uri) {
+        super(uri);
+    }
+
+    protected XMLHelper createXMLHelper() {
+        return new SDOXMLHelperImpl(this, reader);
+    }
+
+    /**
+     * @see org.eclipse.emf.ecore.xmi.impl.XMLResourceImpl#createXMLLoad()
+     */
+    protected XMLLoad createXMLLoad() {
+        return new SDOXMLLoadImpl(createXMLHelper());
+    }
+
+    /**
+     * Loads the resource from a StAX XMLStreamReader.
+     */
+    public void load(XMLStreamReader reader, Map options) throws IOException {
+        this.reader = reader;
+        SDOXMLLoadImpl xmlLoad = (SDOXMLLoadImpl) createXMLLoad();
+        Map mergedOptions = new HashMap(defaultLoadOptions);
+        if (options != null)
+            mergedOptions.putAll(options);
+        xmlLoad.load(this, reader, mergedOptions);
+    }
+}