如何解析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”大家可以下载,不用积分)
代码:
package myproject.com.test;
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 = document.getRootElement();
List<Attribute> list0 = rootElement.getAttributes();
for(Attribute attr : list0){
if("namespace".equals(attr.getName())) {
namespace = attr.getValue() + ".";
}
}
//将根节点的所有子节点获取放到一个集合中
List<Element> list = rootElement.getChildren();
//循环遍历所有子节点
for(Element element : list){
String id = "";//sqlid
String type = element.getName();//sql类型
String content = element.getText();//sql内容
if(!"delete".equals(type)) {
continue;//此处我只拿delete语句,其他不管
}
//获取所有的属性,输出sqlid
List<Attribute> list1 = element.getAttributes();
for(Attribute attr : list1){
if("id".equals(attr.getName())) {
id = attr.getValue();
}
}
//如果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 = content.replaceAll("delete", "delete from");
content = content.replaceAll("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:\mybatis\invoice.mybatis.xml
id=com.sunline.sunfi.mybatis.public.invoice.deleteInvoiceRecord
内容 =
delete inv_rgst_reco where invnum = #{invnum}
修改后内容=
delete from inv_rgst_reco where invnum = #{invnum}
====================================
异常文件数:1