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实现
There is a newer version: 1.0.2
package com.github.developframework.kite.dom4j;

import com.github.developframework.kite.core.AbstractProducer;
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.Template;
import com.github.developframework.kite.core.exception.KiteException;
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;
import java.util.Optional;

/**
 * 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, false);
    }

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

    @Override
    public String produce(boolean pretty) {
        final Document document = buildXmlDocument();
        if (document == null) return "";
        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(document);
            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 Document document = buildXmlDocument();
        if (document == null) return;
        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(document);
            xmlWriter.close();
        } catch (IOException e) {
            throw new KiteException("构建xml失败", e.getMessage());
        }
    }

    private Document buildXmlDocument() {
        final Template template = context.extractTemplate(namespace, templateId);
        final DataDefinition dataDefinition = template.getContentAttributes().dataDefinition;
        Object rootValue = null;
        if (dataDefinition != DataDefinition.EMPTY) {
            Optional