如何解析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”大家可以下载,不用积分)

代码:

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

  • 0
    点赞
  • 0
    收藏
    觉得还不错? 一键收藏
  • 0
    评论

“相关推荐”对你有帮助么?

  • 非常没帮助
  • 没帮助
  • 一般
  • 有帮助
  • 非常有帮助
提交
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值