org.beanio.internal.parser.format.xml.XmlMarshallingContext Maven / Gradle / Ivy

Go to download
Show more of this group Show more artifacts with this name
Show all versions of beanio Show documentation
A Java un/marshalling library for CSV, XML, delimited and fixed length stream formats.
There is a newer version: 3.1.0
/*
 * Copyright 2011-2012 Kevin Seim
 * 
 * Licensed 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.beanio.internal.parser.format.xml;

import java.io.IOException;

import org.beanio.internal.parser.MarshallingContext;
import org.beanio.internal.util.DomUtil;
import org.beanio.stream.xml.XmlWriter;
import org.w3c.dom.*;

/**
 * A {@link MarshallingContext} for XML records.
 * 
 * @author Kevin Seim
 * @since 2.0
 */
public class XmlMarshallingContext extends MarshallingContext {

    private boolean streaming = false;
    
    protected Document document;
    protected Node parent;
    
    private XmlNode[] groupStack;
    private int groupStackCount = 0;
    
    private int ungroup = 0;
    
    /**
     * Constructs a new XmlMarshallingContext.
     * @param groupDepth the maximum depth of a group in the parser tree
     */
    public XmlMarshallingContext(int groupDepth) {
        groupStack = new XmlNode[groupDepth];
    }
    
    @Override
    public void clear() {
        setDocument(null);
    }
    
    @Override
    public Object getRecordObject() {
        return getDocument();
    }
    
    @Override
    public void writeRecord() throws IOException {
        super.clear();
        
        for (int i=0; i 0) {
                for (int i=groupStackCount-1; i>=0; i--) {
                    XmlNode xml = groupStack[i];
                    
                    Node node = parent.appendChild(document.createElementNS(
                        xml.getNamespace(), xml.getLocalName()));
                    node.setPrefix(xml.getPrefix());
                    node.setUserData(XmlWriter.IS_GROUP_ELEMENT, Boolean.TRUE, null);
                    if (!xml.isNamespaceAware()) {
                        node.setUserData(XmlWriter.IS_NAMESPACE_IGNORED, Boolean.TRUE, null);
                    }
                    else {
                        if ("".equals(xml.getPrefix())) {
                            node.setUserData(XmlWriter.IS_DEFAULT_NAMESPACE, Boolean.TRUE, null);
                        }
                        else {
                            node.setPrefix(xml.getPrefix());
                        }
                    }
                    
                    this.parent = node;
                }
                groupStackCount = 0;
            }
        }
        return parent;
    }

    /**
     * Sets the parent node to append in the document being marshalled.
     * @param parent the parent {@link Node}
     */
    public void setParent(Node parent) {
        this.parent = parent;
    }
    
    @Override
    public Document toDocument(Object record) {
        return (Document)record;
    }
    
    /**
     * Sets whether a stream is being marshalled, versus a single document.
     * @param streaming true if marshalling a stream, false if marshalling single documents
     */
    public void setStreaming(boolean streaming) {
        this.streaming = streaming;
    }

    /**
     * Returns whether a stream is being marshalled, versus a single document.
     * @return true if marshalling a stream, false if marshalling single documents
     */
    public boolean isStreaming() {
        return streaming;
    }
}