net.sinofool.wechat.base.OneLevelOnlyXML Maven / Gradle / Ivy

Go to download
Show more of this group Show more artifacts with this name
Show all versions of wechat-java-sdk Show documentation
WeChat open platform SDK
The newest version!
package net.sinofool.wechat.base;

import net.sinofool.wechat.WeChatException;
import org.w3c.dom.Document;
import org.w3c.dom.Element;

import javax.xml.parsers.DocumentBuilder;
import javax.xml.parsers.DocumentBuilderFactory;
import javax.xml.transform.Transformer;
import javax.xml.transform.TransformerFactory;
import javax.xml.transform.dom.DOMSource;
import javax.xml.transform.stream.StreamResult;
import java.io.ByteArrayOutputStream;

public class OneLevelOnlyXML {
    private static final org.slf4j.Logger LOG = org.slf4j.LoggerFactory.getLogger(OneLevelOnlyXML.class);

    private Document doc;
    private Element root;

    public void createRootElement(String tag) {
        try {
            DocumentBuilderFactory factory = DocumentBuilderFactory.newInstance();
            DocumentBuilder builder = factory.newDocumentBuilder();
            doc = builder.newDocument();
            root = doc.createElement(tag);
            doc.appendChild(root);
        } catch (RuntimeException e) {
            LOG.warn("Cannot create XML document:", e);
            throw new WeChatException(e);
        } catch (Exception e) {
            LOG.warn("Cannot create XML document", e);
            throw new WeChatException(e);
        }
    }

    public void createChild(String tag, String value) {
        if (doc == null || root == null) {
            return;
        }
        Element encrypt = doc.createElement(tag);
        encrypt.appendChild(doc.createTextNode(value));
        root.appendChild(encrypt);
    }

    public void createChild(String tag, int value) {
        createChild(tag, String.valueOf(value));
    }

    public String toXMLString() {
        try {
            TransformerFactory transformerFactory = TransformerFactory.newInstance();
            Transformer transformer = transformerFactory.newTransformer();
            ByteArrayOutputStream out = new ByteArrayOutputStream();
            transformer.transform(new DOMSource(doc), new StreamResult(out));
            return out.toString("utf-8");
        } catch (RuntimeException e) {
            LOG.warn("Cannot serialize to XML:", e);
            throw new WeChatException(e);
        } catch (Exception e) {
            LOG.warn("Cannot serialize to XML", e);
            throw new WeChatException(e);
        }
    }
}