io.yawp.driver.postgresql.configuration.XmlLoader Maven / Gradle / Ivy

Go to download

Show more of this group Show more artifacts with this name
Show all versions of yawp-postgresql Show documentation

YAWP! postgresql

There is a newer version: 2.08alpha

package io.yawp.driver.postgresql.configuration;

import org.w3c.dom.Node;
import org.w3c.dom.NodeList;
import org.xml.sax.SAXException;

import javax.xml.parsers.DocumentBuilder;
import javax.xml.parsers.DocumentBuilderFactory;
import javax.xml.parsers.ParserConfigurationException;
import javax.xml.xpath.XPath;
import javax.xml.xpath.XPathConstants;
import javax.xml.xpath.XPathExpressionException;
import javax.xml.xpath.XPathFactory;
import java.io.IOException;
import java.util.ArrayList;
import java.util.List;

public class XmlLoader {

    private Node item;

    public XmlLoader(String path) {
        load(path);
    }

    public XmlLoader(Node item) {
        this.item = item;
    }

    public void load(String path) {
        try {
            DocumentBuilderFactory dbf = createDocumentFactory();
            DocumentBuilder db = dbf.newDocumentBuilder();
            item = db.parse(path).getDocumentElement();
        } catch (ParserConfigurationException | SAXException | IOException e) {
            throw new RuntimeException(e);
        }
    }

    private DocumentBuilderFactory createDocumentFactory() throws ParserConfigurationException {
        DocumentBuilderFactory dbf = DocumentBuilderFactory.newInstance();
        dbf.setValidating(false);
        dbf.setNamespaceAware(true);
        dbf.setFeature("http://xml.org/sax/features/namespaces", false);
        dbf.setFeature("http://xml.org/sax/features/validation", false);
        dbf.setFeature("http://apache.org/xml/features/nonvalidating/load-dtd-grammar", false);
        dbf.setFeature("http://apache.org/xml/features/nonvalidating/load-external-dtd", false);
        return dbf;
    }

    public List find(String xpath) {
        try {
            XPath xPath = XPathFactory.newInstance().newXPath();
            NodeList nodes = (NodeList) xPath.evaluate(xpath, item, XPathConstants.NODESET);
            List result = new ArrayList();
            for (int i = 0; i < nodes.getLength(); i++) {
                result.add(new XmlLoader(nodes.item(i)));
            }
            return result;
        } catch (XPathExpressionException e) {
            throw new RuntimeException(e);
        }
    }

    public String getTextContent() {
        return item.getTextContent();
    }

    public String getAttributeText(String attr) {
        return item.getAttributes().getNamedItem(attr).getNodeValue();
    }
}