STXM-11 Add unregister methods

jbossas · Jan 6, 2012 · 8f05c65 · 8f05c65
1 parent 6bb456f
commit 8f05c65
Show file tree

Hide file tree

Showing 2 changed files with 25 additions and 1 deletion.
diff --git a/src/main/java/org/jboss/staxmapper/XMLMapper.java b/src/main/java/org/jboss/staxmapper/XMLMapper.java
@@ -42,6 +42,13 @@ public interface XMLMapper {
      */
     void registerRootElement(QName name, XMLElementReader<?> reader);
 
+    /**
+     * Removes a {@link #registerRootElement(QName, XMLElementReader) previously registered root element}.
+     *
+     * @param name the element name
+     */
+    void unregisterRootElement(QName name);
+
     /**
      * Add a known root attribute which can be read by {@link XMLExtendedStreamReader#handleAttribute(Object, int)}.
      *
@@ -50,6 +57,13 @@ public interface XMLMapper {
      */
     void registerRootAttribute(QName name, XMLAttributeReader<?> reader);
 
+    /**
+     * Removes a {@link #registerRootAttribute(QName, XMLAttributeReader) previously registered root attribute}.
+     *
+     * @param name the element name
+     */
+    void unregisterRootAttribute(QName name);
+
     /**
      * Parse a document.  The document must have a known, registered root element which can accept the given root object.
      *
@@ -78,7 +92,7 @@ public interface XMLMapper {
      */
     @Deprecated
     void deparseDocument(XMLContentWriter contentWriter, XMLStreamWriter streamWriter) throws XMLStreamException;
-    
+
     /**
      * A factory for creating an instance of {@link XMLMapper}.
      */

diff --git a/src/main/java/org/jboss/staxmapper/XMLMapperImpl.java b/src/main/java/org/jboss/staxmapper/XMLMapperImpl.java
@@ -45,12 +45,22 @@ public void registerRootElement(QName name, XMLElementReader<?> reader) {
         }
     }
 
+    @Override
+    public void unregisterRootElement(QName name) {
+        rootElements.remove(name);
+    }
+
     public void registerRootAttribute(QName name, XMLAttributeReader<?> reader) {
         if (rootAttributes.putIfAbsent(name, reader) != null) {
             throw new IllegalArgumentException("Root attribute for " + name + " already registered");
         }
     }
 
+    @Override
+    public void unregisterRootAttribute(QName name) {
+        rootAttributes.remove(name);
+    }
+
     public void parseDocument(Object rootObject, XMLStreamReader reader) throws XMLStreamException {
         try {
             reader.require(START_DOCUMENT, null, null);