com.github.developframework.kite.dom4j.Dom4jProducer Maven / Gradle / Ivy

Go to download
Show more of this group Show more artifacts with this name
Show all versions of kite-dom4j Show documentation
Dom4j实现
The newest version!
package com.github.developframework.kite.dom4j;

import com.github.developframework.kite.core.AbstractProducer;
import com.github.developframework.kite.core.AssembleContext;
import com.github.developframework.kite.core.KiteConfiguration;
import com.github.developframework.kite.core.data.DataDefinition;
import com.github.developframework.kite.core.data.DataModel;
import com.github.developframework.kite.core.element.ArrayKiteElement;
import com.github.developframework.kite.core.element.Template;
import com.github.developframework.kite.core.exception.KiteException;
import com.github.developframework.kite.core.node.NodeProxy;
import com.github.developframework.kite.core.node.ObjectNodeProxy;
import com.github.developframework.kite.core.structs.TemplatePackage;
import com.github.developframework.kite.core.utils.KiteUtils;
import org.dom4j.Document;
import org.dom4j.DocumentHelper;
import org.dom4j.Element;
import org.dom4j.io.OutputFormat;
import org.dom4j.io.XMLWriter;

import java.io.*;
import java.nio.charset.Charset;
import java.nio.charset.StandardCharsets;
import java.util.List;

/**
 * Dom4j的xml生成器
 *
 * @author qiushui on 2021-06-23.
 */
public final class Dom4jProducer extends AbstractProducer {

    public Dom4jProducer(KiteConfiguration configuration, DataModel dataModel, String namespace, String templateId) {
        super(configuration, dataModel, namespace, templateId);
    }

    public Dom4jProducer(KiteConfiguration configuration, DataModel dataModel, List templatePackages) {
        super(configuration, dataModel, templatePackages);
    }

    @Override
    protected AssembleContext buildAssembleContext(DataModel dataModel) {
        return new Dom4jAssembleContext(configuration, dataModel);
    }

    @Override
    public String produce(boolean pretty) {
        final NodeProxy rootNodeProxy = buildRootNodeProxy();
        if (rootNodeProxy == null) {
            return "";
        } else {
            try {
                OutputFormat format = OutputFormat.createCompactFormat();
                format.setSuppressDeclaration(configuration.getOptions().getXml().isSuppressDeclaration());
                format.setEncoding(StandardCharsets.UTF_8.name());
                format.setIndent(pretty);
                format.setNewlines(pretty);
                Writer writer = new StringWriter();
                XMLWriter xmlWriter = new XMLWriter(writer, format);
                xmlWriter.setEscapeText(true);
                xmlWriter.write(rootNodeProxy.getNode());
                xmlWriter.close();
                return writer.toString();
            } catch (IOException e) {
                throw new KiteException("构建xml失败", e.getMessage());
            }
        }
    }

    @Override
    public void output(OutputStream outputStream, Charset charset, boolean pretty) {
        final NodeProxy rootNodeProxy = buildRootNodeProxy();
        if (rootNodeProxy != null) {
            try {
                OutputFormat format = OutputFormat.createCompactFormat();
                format.setSuppressDeclaration(configuration.getOptions().getXml().isSuppressDeclaration());
                format.setEncoding(charset == null ? StandardCharsets.UTF_8.name() : charset.name());
                format.setIndent(pretty);
                format.setNewlines(pretty);
                Writer writer = new OutputStreamWriter(outputStream);
                XMLWriter xmlWriter = new XMLWriter(writer, format);
                xmlWriter.setEscapeText(true);
                xmlWriter.write(rootNodeProxy.getNode());
                xmlWriter.close();
            } catch (IOException e) {
                throw new KiteException("构建xml失败", e.getMessage());
            }
        }
    }

    @Override
    protected NodeProxy buildRootNodeProxy() {
        final Template template = context.extractTemplate(namespace, templateId);
        final DataDefinition dataDefinition = template.getContentAttributes().dataDefinition;
        Object rootValue = null;
        if (dataDefinition != DataDefinition.EMPTY) {
            rootValue = context.dataModel.getData(dataDefinition.getExpression()).orElse(null);
            if (rootValue == null) {
                return null;
            }
        }
        context.valueStack.push(context.dataModel);
        final Document document = DocumentHelper.createDocument();
        final ObjectNodeProxy rootNodeProxy = new Dom4jObjectNodeProxy(document.addElement(template.getXmlRoot()));
        if (KiteUtils.objectIsArray(rootValue)) {
            // 以数组为根节点
            Dom4jArrayNodeProxy arrayNodeProxy = new Dom4jArrayNodeProxy((Element) rootNodeProxy.getNode());
            ArrayKiteElement innerArrayKiteElement = template.getInnerArrayKiteElement();
            innerArrayKiteElement.assembleArrayItems(context, innerArrayKiteElement.getArrayAttributes(), rootValue, arrayNodeProxy);
        } else {
            // 以对象为根节点
            context.prepareNextOnlyNode(rootNodeProxy, template::assemble);
        }
        return rootNodeProxy;
    }
}