pers.clare.hisql.method.SQLInjector Maven / Gradle / Ivy

Go to download
Show more of this group Show more artifacts with this name
Show all versions of hi-sql Show documentation
Have fun using SQL
The newest version!
package pers.clare.hisql.method;

import org.apache.logging.log4j.LogManager;
import org.apache.logging.log4j.Logger;
import org.springframework.util.StringUtils;
import org.w3c.dom.Document;
import org.w3c.dom.Node;
import org.w3c.dom.NodeList;

import javax.xml.parsers.DocumentBuilder;
import javax.xml.parsers.DocumentBuilderFactory;
import javax.xml.parsers.ParserConfigurationException;
import java.io.InputStream;
import java.util.HashMap;
import java.util.Map;

public class SQLInjector {
    private static final Logger log = LogManager.getLogger();

    private static final DocumentBuilder documentBuilder;

    static {
        try {
            documentBuilder = DocumentBuilderFactory.newInstance().newDocumentBuilder();
        } catch (ParserConfigurationException e) {
            throw new Error(e.getMessage(), e);
        }
    }

    private SQLInjector() {
    }

    public static Map getContents(String root, Class clazz) {
        String xmlPath = toPath(root, clazz);
        Map map = new HashMap<>();
        try (InputStream is = clazz.getClassLoader().getResourceAsStream(xmlPath)) {
            if (is == null) return map;
            log.debug("load {}", xmlPath);
            Document doc = documentBuilder.parse(is);
            NodeList nodeList = doc.getDocumentElement().getChildNodes();
            Node node;
            String content;
            for (int i = 0; i < nodeList.getLength(); i++) {
                node = nodeList.item(i);
                if (Node.ELEMENT_NODE != node.getNodeType()) continue;
                content = node.getTextContent();
                if (!StringUtils.hasLength(content)) continue;
                map.put(node.getNodeName(), content);
            }
        } catch (Exception e) {
            log.error(e.getMessage(), e);
        }
        return map;
    }

    private static String toPath(String root, Class clazz) {
        StringBuilder path = new StringBuilder(root);
        if (!root.endsWith("/")) {
            path.append("/");
        }
        return path.append(clazz.getSimpleName())
                .append(".xml")
                .toString();
    }
}