如何解析mybatis的xml文件获取所有delete语句

如何解析mybatis的xml文件获取所有delete语句

如何解析mybatis的xml文件获取所有delete语句

背景：
项目以前是用oracle数据库，所以sql中有人会用到很多特性函数和一下特性写法，要改造成适配(oracle、db2、mysql)，其中有一种语法问题我遇到了所以写下来记录一下以后好找。

需求：
删除语句oracle和db2支持delete后面直接跟表名，比如：delete t_user where username=‘admin’，但是mysql不支持必须要加上from关键字。

解决方案：
1.比较笨的方法就是对所有的mybatis文件都检查一遍，但是如果文件比较多的情况就很麻烦，比如我的项目中就有几千个文件，如果一个一个检查不得累死。

2.解析所有mybatis文件，获取所有delete节点，然后拿到sql的内容，再判断内容中是否包含关键字from，如果没有则输出语句id和文件名，根据文件名和id找到语句改掉(当时还有个想法，找出来后想直接把语句中的delete替换成delete from，但是如果包含了if标签这种就会有问题，子节点也会被覆盖掉，这会没时间以后有空再试试)。

用到的jar包:
jdom-1.0.jar(文章里传不了jar包可以自己百度下一个，我的资源里也上传了一个“包名.rar”大家可以下载，不用积分)

代码：

st;import java.io.File;
import java.io.FileInputStream;
import java.io.InputStream;
import java.util.List;import org.jdom.Attribute;
import org.jdom.Document;
import org.jdom.Element;
import org.jdom.input.SAXBuilder;public class JDOMGetDeleteSql {private static int count = 0;public static void main(String[] args) throws Exception {File file = new File("D:\mybatis");File[] fileList = file.listFiles();for (int i = 0; i < fileList.length; i++) {printDeleteSql(fileList[i].getPath());}System.out.println("异常文件数:" + count);}public static void printDeleteSql(String filepath) throws Exception{String namespace = "";//mybatis命名空间//创建一个解析器SAXBuilder builder = new SAXBuilder();//创建一个文件输入流InputStream in = new FileInputStream(filepath);//将流加载到解析器中。Document document = builder.build(in);//获取文档的根节点Element rootElement = RootElement();List<Attribute> list0 = Attributes();for(Attribute attr : list0){if("namespace".Name())) {namespace = Value() + ".";}}//将根节点的所有子节点获取放到一个集合中List<Element> list = Children();//循环遍历所有子节点for(Element element : list){String id = "";//sqlidString type = Name();//sql类型String content = Text();//sql内容if(!"delete".equals(type)) {continue;//此处我只拿delete语句，其他不管}//获取所有的属性,输出sqlidList<Attribute> list1 = Attributes();for(Attribute attr : list1){if("id".Name())) {id = Value();}}//如果delete语句并且没找到from关键字则输出if(content.indexOf("from") == -1 && content.indexOf("FROM") == -1) {System.out.println("检查文件路径:" + filepath);System.out.println("id=" + namespace +id);System.out.println("内容 =" + content);//直接修改内容，但是如果有子节点会把子节点也替换content = placeAll("delete", "delete from");content = placeAll("DELETE", "DELETE FROM");System.out.println("修改后内容=" + content);/*element.setText(content);String xmlFileData = new XMLOutputter().outputString(document);FileWriter fileWriter = new FileWriter(filepath);fileWriter.write(xmlFileData);fileWriter.close();*/count++;System.out.println("====================================");}}}
}

控制台输出：
检查文件路径:D:l
id=com.batis.public.invoice.deleteInvoiceRecord
内容 =
delete inv_rgst_reco where invnum = #{invnum}

修改后内容=
delete from inv_rgst_reco where invnum = #{invnum}

====================================
异常文件数:1