1 files changed, 291 insertions, 0 deletions
diff --git a/sca-java-2.x/tags/2.0.1-RC1/modules/contribution/src/main/java/org/apache/tuscany/sca/contribution/processor/DefaultStAXAttributeProcessorExtensionPoint.java b/sca-java-2.x/tags/2.0.1-RC1/modules/contribution/src/main/java/org/apache/tuscany/sca/contribution/processor/DefaultStAXAttributeProcessorExtensionPoint.java
new file mode 100644
index 0000000000..36dec73898
--- /dev/null
+++ b/sca-java-2.x/tags/2.0.1-RC1/modules/contribution/src/main/java/org/apache/tuscany/sca/contribution/processor/DefaultStAXAttributeProcessorExtensionPoint.java
@@ -0,0 +1,291 @@
+/*
+ * 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.sca.contribution.processor;
+
+import java.io.IOException;
+import java.lang.reflect.Constructor;
+import java.util.Collection;
+import java.util.Map;
+
+import javax.xml.namespace.QName;
+import javax.xml.stream.XMLInputFactory;
+import javax.xml.stream.XMLOutputFactory;
+import javax.xml.stream.XMLStreamException;
+import javax.xml.stream.XMLStreamReader;
+import javax.xml.stream.XMLStreamWriter;
+
+import org.apache.tuscany.sca.contribution.resolver.ModelResolver;
+import org.apache.tuscany.sca.core.ExtensionPointRegistry;
+import org.apache.tuscany.sca.core.FactoryExtensionPoint;
+import org.apache.tuscany.sca.core.UtilityExtensionPoint;
+import org.apache.tuscany.sca.extensibility.ServiceDeclaration;
+import org.apache.tuscany.sca.extensibility.ServiceDeclarationParser;
+import org.apache.tuscany.sca.monitor.Monitor;
+import org.apache.tuscany.sca.monitor.MonitorFactory;
+import org.apache.tuscany.sca.monitor.Problem;
+import org.apache.tuscany.sca.monitor.Problem.Severity;
+
+/**
+ * The default implementation of an extension point for StAX artifact processors.
+ *
+ * @version $Rev$ $Date$
+ */
+public class DefaultStAXAttributeProcessorExtensionPoint extends
+    DefaultArtifactProcessorExtensionPoint<StAXAttributeProcessor<?>> implements StAXAttributeProcessorExtensionPoint {
+
+    private FactoryExtensionPoint modelFactories;
+    private StAXAttributeProcessor<Object> extensibleStAXAttributeProcessor;
+    private boolean loaded;
+    private Monitor monitor = null;
+
+    /**
+     * Constructs a new extension point.
+     */
+    public DefaultStAXAttributeProcessorExtensionPoint(ExtensionPointRegistry extensionPoints) {
+        super(extensionPoints);
+        this.modelFactories = extensionPoints.getExtensionPoint(FactoryExtensionPoint.class);
+        XMLInputFactory inputFactory = modelFactories.getFactory(XMLInputFactory.class);
+        XMLOutputFactory outputFactory = modelFactories.getFactory(XMLOutputFactory.class);
+        UtilityExtensionPoint utilities = this.registry.getExtensionPoint(UtilityExtensionPoint.class);
+        MonitorFactory monitorFactory = utilities.getUtility(MonitorFactory.class);
+        if (monitorFactory != null) {
+            this.monitor = monitorFactory.createMonitor();
+        }
+        this.extensibleStAXAttributeProcessor = new ExtensibleStAXAttributeProcessor(this, inputFactory, outputFactory);
+    }
+
+    /**
+     * Report a exception.
+     *
+     * @param problems
+     * @param message
+     * @param model
+    */
+    private void error(String message, Object model, Exception ex) {
+        if (monitor != null) {
+            Problem problem =
+                monitor.createProblem(this.getClass().getName(),
+                                      "contribution-validation-messages",
+                                      Severity.ERROR,
+                                      model,
+                                      message,
+                                      ex);
+            monitor.problem(problem);
+        }
+    }
+
+    public void addArtifactProcessor(StAXAttributeProcessor<?> artifactProcessor) {
+        if (artifactProcessor.getArtifactType() != null) {
+            processorsByArtifactType.put((Object)artifactProcessor.getArtifactType(), artifactProcessor);
+        }
+        if (artifactProcessor.getModelType() != null) {
+            processorsByModelType.put(artifactProcessor.getModelType(), artifactProcessor);
+        }
+    }
+
+    public void removeArtifactProcessor(StAXAttributeProcessor<?> artifactProcessor) {
+        if (artifactProcessor.getArtifactType() != null) {
+            processorsByArtifactType.remove((Object)artifactProcessor.getArtifactType());
+        }
+        if (artifactProcessor.getModelType() != null) {
+            processorsByModelType.remove(artifactProcessor.getModelType());
+        }
+    }
+
+    @Override
+    public <T> StAXAttributeProcessor<T> getProcessor(Class<T> modelType) {
+        loadArtifactProcessors();
+        return (StAXAttributeProcessor<T>)super.getProcessor(modelType);
+    }
+
+    @Override
+    public StAXAttributeProcessor<?> getProcessor(Object artifactType) {
+        loadArtifactProcessors();
+        return super.getProcessor(artifactType);
+    }
+
+    /**
+     * Lazily load artifact processors registered in the extension point.
+     */
+    private synchronized void loadArtifactProcessors() {
+        if (loaded)
+            return;
+
+        // Get the processor service declarations
+        Collection<ServiceDeclaration> processorDeclarations;
+        try {
+            processorDeclarations =
+                registry.getServiceDiscovery().getServiceDeclarations(StAXAttributeProcessor.class.getName());
+        } catch (IOException e) {
+            IllegalStateException ie = new IllegalStateException(e);
+            error("IllegalStateException", extensibleStAXAttributeProcessor, ie);
+            throw ie;
+        }
+
+        for (ServiceDeclaration processorDeclaration : processorDeclarations) {
+            Map<String, String> attributes = processorDeclaration.getAttributes();
+
+            // Load a StAX artifact processor
+
+            // Get the model QName
+            QName artifactType = ServiceDeclarationParser.getQName(attributes.get("qname"));
+
+            // Get the model class name
+            String modelTypeName = attributes.get("model");
+
+            // Get the model factory class name
+            String factoryName = attributes.get("factory");
+
+            // Create a processor wrapper and register it
+            StAXAttributeProcessor<?> processor =
+                new LazyStAXAttributeProcessor(artifactType, modelTypeName, factoryName, processorDeclaration,
+                                               registry, modelFactories, extensibleStAXAttributeProcessor);
+            addArtifactProcessor(processor);
+        }
+
+        loaded = true;
+    }
+
+    /**
+     * A wrapper around an Artifact processor class allowing lazy loading and
+     * initialization of artifact processors.
+     */
+    private static class LazyStAXAttributeProcessor implements StAXAttributeProcessor {
+
+        private ExtensionPointRegistry extensionPoints;
+        private QName artifactType;
+        private String modelTypeName;
+        private String factoryName;
+        private ServiceDeclaration processorDeclaration;
+        private StAXAttributeProcessor<?> processor;
+        private Class<?> modelType;
+        private StAXAttributeProcessor<Object> extensionProcessor;
+
+        LazyStAXAttributeProcessor(QName artifactType,
+                                   String modelTypeName,
+                                   String factoryName,
+                                   ServiceDeclaration processorDeclaration,
+                                   ExtensionPointRegistry extensionPoints,
+                                   FactoryExtensionPoint modelFactories,
+                                   StAXAttributeProcessor<Object> extensionProcessor) {
+
+            this.extensionPoints = extensionPoints;
+            this.artifactType = artifactType;
+            this.modelTypeName = modelTypeName;
+            this.factoryName = factoryName;
+            this.processorDeclaration = processorDeclaration;
+            this.extensionProcessor = extensionProcessor;
+        }
+
+        public QName getArtifactType() {
+            return artifactType;
+        }
+
+        @SuppressWarnings("unchecked")
+        private StAXAttributeProcessor getProcessor() {
+            if (processor == null) {
+                FactoryExtensionPoint modelFactories = extensionPoints.getExtensionPoint(FactoryExtensionPoint.class);
+
+                // Load and instantiate the processor class
+                try {
+                    Class<StAXAttributeProcessor> processorClass =
+                        (Class<StAXAttributeProcessor>)processorDeclaration.loadClass();
+                    try {
+                        Constructor<StAXAttributeProcessor> constructor =
+                            processorClass.getConstructor(FactoryExtensionPoint.class);
+                        processor = constructor.newInstance(modelFactories);
+                    } catch (NoSuchMethodException e) {
+                        try {
+                            Constructor<StAXAttributeProcessor> constructor =
+                                processorClass.getConstructor(ExtensionPointRegistry.class);
+                            processor = constructor.newInstance(extensionPoints);
+                        } catch (NoSuchMethodException e1) {
+                            try {
+                                Constructor<StAXAttributeProcessor> constructor =
+                                    processorClass.getConstructor(FactoryExtensionPoint.class,
+                                                                  StAXArtifactProcessor.class);
+                                processor = constructor.newInstance(modelFactories, extensionProcessor);
+                            } catch (NoSuchMethodException e2) {
+                                try {
+                                    Constructor<StAXAttributeProcessor> constructor =
+                                        processorClass.getConstructor(ExtensionPointRegistry.class,
+                                                                      StAXArtifactProcessor.class);
+                                    processor = constructor.newInstance(extensionPoints, extensionProcessor);
+                                } catch (NoSuchMethodException e3) {
+                                    try {
+                                        Constructor<StAXAttributeProcessor> constructor =
+                                            processorClass.getConstructor(FactoryExtensionPoint.class);
+                                        processor = constructor.newInstance(modelFactories);
+                                    } catch (NoSuchMethodException e4) {
+                                        try {
+                                            Constructor<StAXAttributeProcessor> constructor =
+                                                processorClass.getConstructor(ExtensionPointRegistry.class);
+                                            processor = constructor.newInstance(extensionPoints);
+                                        } catch (NoSuchMethodException e4a) {
+
+                                            Constructor<StAXAttributeProcessor> constructor =
+                                                processorClass.getConstructor();
+                                            processor = constructor.newInstance();
+
+                                        }
+                                    }
+                                }
+                            }
+                        }
+                    }
+                } catch (Exception e) {
+                    IllegalStateException ie = new IllegalStateException(e);
+                    throw ie;
+                }
+
+            }
+            return processor;
+        }
+
+        public Object read(QName attributeName, XMLStreamReader inputSource, ProcessorContext context)
+            throws ContributionReadException, XMLStreamException {
+            return getProcessor().read(attributeName, inputSource, context);
+        }
+
+        @SuppressWarnings("unchecked")
+        public void write(Object model, XMLStreamWriter outputSource, ProcessorContext context)
+            throws ContributionWriteException, XMLStreamException {
+            getProcessor().write(model, outputSource, context);
+        }
+
+        public Class<?> getModelType() {
+            if (modelTypeName != null && modelType == null) {
+                try {
+                    modelType = processorDeclaration.loadClass(modelTypeName);
+                } catch (Exception e) {
+                    IllegalStateException ie = new IllegalStateException(e);
+                    throw ie;
+                }
+            }
+            return modelType;
+        }
+
+        @SuppressWarnings("unchecked")
+        public void resolve(Object model, ModelResolver resolver, ProcessorContext context)
+            throws ContributionResolveException {
+            getProcessor().resolve(model, resolver, context);
+        }
+
+    }
+}