package com.hx.auto;
|
|
import org.w3c.dom.Document;
|
import org.w3c.dom.NamedNodeMap;
|
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.transform.OutputKeys;
|
import javax.xml.transform.Transformer;
|
import javax.xml.transform.TransformerException;
|
import javax.xml.transform.TransformerFactory;
|
import javax.xml.transform.dom.DOMSource;
|
import javax.xml.transform.stream.StreamResult;
|
import java.io.*;
|
import java.util.*;
|
|
/**
|
* 自动生成Service and ServiceImpl
|
*
|
* @author chenjiahe 2020-06-28
|
*
|
*/
|
public class GeneratorReadXmlUtil {
|
|
/**
|
*
|
* @param targetFile 文件路径
|
* @return 返回自定义的sql
|
*/
|
public static String readMapperXml(String targetFile) throws ParserConfigurationException, IOException, SAXException {
|
String temp = "";
|
|
//映射文件的文件夹
|
File file = new File(targetFile);
|
if(!file.exists()){
|
return "false";
|
}
|
|
DocumentBuilderFactory factory = DocumentBuilderFactory.newInstance();
|
DocumentBuilder builder = factory.newDocumentBuilder();
|
Document doc = builder.parse(file);
|
Node Node= doc.getLastChild();
|
NodeList nodeList = Node.getChildNodes();
|
|
String comment = "";
|
for(int i=0;i<nodeList.getLength();i++) {
|
//从节点集中获取i个book节点
|
Node childNode = nodeList.item(i);
|
//获取子节点内的文本内容
|
String content = childNode.getTextContent();
|
//获取节点名
|
String name = childNode.getNodeName();
|
//System.out.println("content:"+content+";name:"+name);
|
//System.out.println("name....:"+name);
|
if(!"#text".equals(name)&&!"#comment".equals(name)){
|
//获取第一个节点内的所有属性
|
NamedNodeMap nameNodeMap = childNode.getAttributes();
|
//获取节点内名为id的属性的值
|
String id = nameNodeMap.getNamedItem("id").getTextContent();
|
if(isCustom(nameNodeMap.getNamedItem("id").getTextContent())){
|
String con = NodetoString(childNode);
|
//去掉附带的内容
|
con = con.replace("<?xml version=\"1.0\" encoding=\"UTF-8\"?>","");
|
String con2 = con;
|
con = con.replaceAll("\t","").replaceAll("\n","");
|
if(con.length()>3){
|
con= con2;
|
}
|
temp+=comment+"\n\t"+con;
|
}
|
comment = "";
|
}else{
|
String con = NodetoString(childNode);
|
//去掉附带的内容
|
con = con.replace("<?xml version=\"1.0\" encoding=\"UTF-8\"?>","");
|
String con2 = con;
|
//去掉多余的
|
con = con.replaceAll("\t","").replaceAll("\n","");
|
if(con.length()>5){
|
comment+= con2;
|
}
|
}
|
}
|
System.out.println("temp:...:"+temp);
|
return temp;
|
}
|
|
public static boolean isCustom(String id){
|
boolean custom = true;
|
Set<String> noCustomIds = new HashSet<String>();
|
noCustomIds.add("BaseResultMap");
|
noCustomIds.add("Entity_Id");
|
noCustomIds.add("Table_Id");
|
noCustomIds.add("Table_Name");
|
noCustomIds.add("Base_Column_List");
|
noCustomIds.add("Blob_Column_List");
|
noCustomIds.add("Insert_Column_All");
|
noCustomIds.add("Update_Column_All");
|
noCustomIds.add("Update_Column_NoNull");
|
noCustomIds.add("keyFind");
|
noCustomIds.add("insert");
|
noCustomIds.add("insertById");
|
noCustomIds.add("selectList");
|
noCustomIds.add("selectListBlob");
|
noCustomIds.add("selectOne");
|
noCustomIds.add("selectOneBlob");
|
noCustomIds.add("selectOneByKey");
|
noCustomIds.add("selectOneByKeyBlob");
|
noCustomIds.add("updateWhere");
|
noCustomIds.add("updateAll");
|
noCustomIds.add("updateByNoNull");
|
noCustomIds.add("deleteWhere");
|
noCustomIds.add("deleteById");
|
noCustomIds.add("selectCount");
|
noCustomIds.add("Insert_Values_All");
|
noCustomIds.add("selectListMap");
|
noCustomIds.add("selectOneMap");
|
noCustomIds.add("selectCountSql");
|
if (noCustomIds.contains(id)){
|
custom = false;
|
}
|
return custom;
|
}
|
|
|
|
/*
|
* 把dom文件转换为xml字符串
|
*/
|
public static String toStringFromDoc(Document document) throws IOException {
|
String result = null;
|
|
if (document != null) {
|
StringWriter strWtr = new StringWriter();
|
StreamResult strResult = new StreamResult(strWtr);
|
TransformerFactory tfac = TransformerFactory.newInstance();
|
try {
|
javax.xml.transform.Transformer t = tfac.newTransformer();
|
t.setOutputProperty(OutputKeys.ENCODING, "UTF-8");
|
t.setOutputProperty(OutputKeys.INDENT, "yes");
|
t.setOutputProperty(OutputKeys.METHOD, "xml"); // xml, html,
|
// text
|
t.setOutputProperty(
|
"{http://xml.apache.org/xslt}indent-amount", "4");
|
t.transform(new DOMSource(document.getDocumentElement()),
|
strResult);
|
} catch (Exception e) {
|
System.err.println("XML.toString(Document): " + e);
|
}
|
result = strResult.getWriter().toString();
|
strWtr.close();
|
}
|
return result;
|
}
|
|
|
/**
|
* 将传入的一个DOM Node对象输出成字符串。如果失败则返回一个空字符串""。
|
*
|
* @param node
|
* DOM Node 对象。
|
* @return a XML String from node
|
*/
|
public static String NodetoString(Node node) {
|
Transformer transformer = null;
|
String result = null;
|
if (node == null) {
|
throw new IllegalArgumentException();
|
}
|
try {
|
transformer = TransformerFactory.newInstance().newTransformer();
|
} catch (Exception e) {
|
throw new RuntimeException(e.getMessage());
|
}
|
if (transformer != null) {
|
try {
|
StringWriter sw = new StringWriter();
|
transformer.transform(new DOMSource(node), new StreamResult(sw));
|
return sw.toString();
|
} catch (TransformerException te) {
|
throw new RuntimeException(te.getMessage());
|
}
|
}
|
return result;
|
}
|
|
|
}
|
