intellij idea 插件开发--快速定位到mybatis mapper文件中的sql

    intellij idea 提供了openApi,经过openApi咱们能够本身开发插件,提升工做效率。这边直接贴个连接,能够搭个入门的demo;http://www.jianshu.com/p/2427e4cfd3e9,也能够本身网上找,文章比较多。。本人用的intellij idea 2017,jdk须要1.8以上。java

    第一步:new project,选择IntelliJ Plaltform Plugin 如图,sql

而后点击下一步,填写项目名,点击finish,项目结构以下:api

plugin.xml是插件的配置文件;mybatis

第二步:鼠标点击src文件夹,而后Alt+Insert 快捷键,选择action,如图:app

而后填写相关信息,这边分组到JavaGenerateGroup1,快捷键为Ctrl+Shift+X(即此快捷键执行这个action的代码)编辑器

点击ok后,将建立名为GoToMapperAction的java类,以及plugin.xml中出现对应配置信息,以下图:ide

到此,执行快捷键Ctrl+Shift+X,就会执行actionPerformed方法了;接下来就是写具体逻辑代码了;测试

第三步:方法actionPerformed 中的参数AnActionEvent颇有用,携带了交互的当前上下文信息,要检索活动project、选中的文件、编辑器中选中状态等IDE的当前状态信息,可使用AnActionEvent.getData()方法。DataKeys类中定义了能够传递给此方法的不一样数据键值;idea

1.获取方法名:插件

PsiElement psiElement = e.getData(PlatformDataKeys.PSI_ELEMENT);//鼠标所在的元素,这里就是方法
if (psiElement == null) {
   return;
}
String methodName = psiElement.toString().replace("PsiMethod:", "");//获取到方法名

2.获取方法所在的类名:

PsiElement psiElementParent = psiElement.getParent();//获取方法的父元素
if (psiElementParent == null) {
   return;
}
PsiFile containingFile = psiElementParent.getContainingFile();//获取到文件,这里是java类
String className = containingFile.getName();//获取到类名

3.到这里,咱们已经获取到了类名,方法名,,那就能够肯定对应的mybatis mapper文件的id=“方法名”的sql位置;个人项目中命名比较有规律,因此对应mapper文件的名称能够这样肯定:

String mapperName ;
if (className.endsWith("Service.java")) {
    mapperName = className.replace("Service.java", "Mapper.xml");
}
else if (className.endsWith("Dao.java")) {
    mapperName = className.replace("Dao.java", "Mapper.xml");
}
else {
    return;
}

固然,还能够经过mapper的命名空间找到;

4.打开xml

Project project = e.getProject();
//查找名称为mapperName的文件
PsiFile[] files = PsiShortNamesCache.getInstance(project).getFilesByName(mapperName);
if (files.length == 1) {
    XmlFile xmlFile = (XmlFile) files[0];
    String xml = xmlFile.getDocument().getText();//获取mapper xml字符串
    //判断mapper是否存id="methodName"的sql,存在就打开对应的mapper xml
    //这里判断比较简单,不严谨。能够经过XmlFile遍历节点判断是否存在
    if (StringUtil.isNotEmpty(xml) && xml.contains("id=\"" + methodName + "\"")) {
        toMapper(project, methodName, files[0].getVirtualFile(), xml);
    }
}
/**
 * 进入mapper
 * @param project
 * @param methodName
 * @param mapperFile
 * @param xml
 */
private void toMapper(Project project, String methodName, VirtualFile mapperFile, String xml) {
    //打开xml文件
    OpenFileDescriptor openFileDescriptor = new OpenFileDescriptor(project, mapperFile);
    Editor editor = FileEditorManager.getInstance(project).openTextEditor(openFileDescriptor, true);
    
    //获取sql所在的行数,这里用了比较笨的方法。api找了好久没找到有什么方法能够获取行号,但愿有大神指点
    String[] split = xml.split("\n");
    int lineNumber = 0;
    for (int i = 0; i < split.length; i++) {
        String line = split[i];
        if (StringUtil.isNotEmpty(line) && line.contains(methodName)) {
            lineNumber = i;
            break;
        }
    }
    //定位到对应的sql
    CaretModel caretModel = editor.getCaretModel();
    LogicalPosition logicalPosition = caretModel.getLogicalPosition();
    logicalPosition.leanForward(true);
    LogicalPosition logical = new LogicalPosition(lineNumber, logicalPosition.column);
    caretModel.moveToLogicalPosition(logical);
    SelectionModel selectionModel = editor.getSelectionModel();
    selectionModel.selectLineAtCaret();
}

5.至此,代码完成,如今能够测试下了,点击右上角运行,会打开一个新的idea进行调试

而后在新的idea中打开项目,把鼠标聚焦在你要找的方法中,而后快捷键Ctrl+Shift+X,跳到xml,这样就是成功了。

6.发布,生成jar文件,而后就能够安装到idea使用了

 

最后:这里只是说明了大致逻辑,你还能够判断xml是否有这个sql,没有就自动建立一个tag等等,openapi有不少概念等等,你们能够自行百度或去官网查看文档。

下图是本人最近写的插件功能,只适用本身的项目,就不上代码了,更多有意思功能能够本身开发。

欢迎加入qq群讨论:461964997

相关文章
相关标签/搜索