IntelliJ IDE 插件开发 | （七）PSI 入门及实战（实现 MyBatis 插件的跳转功能）

2024-03-25 01:20
文章标签 实战 实现 入门 功能 开发 ide 插件 mybatis intellij 跳转 psi

本文主要是介绍IntelliJ IDE 插件开发 | （七）PSI 入门及实战（实现 MyBatis 插件的跳转功能），希望对大家解决编程问题提供一定的参考价值，需要的开发者们随着小编来一起学习吧！

系列文章

  • IntelliJ IDE 插件开发 |（一）快速入门
  • IntelliJ IDE 插件开发 |（二）UI 界面与数据持久化
  • IntelliJ IDE 插件开发 |（三）消息通知与事件监听
  • IntelliJ IDE 插件开发 |（四）来查收你的 IDEA 使用报告吧
  • IntelliJ IDE 插件开发 |（五）VFS 与编辑器
  • IntelliJ IDE 插件开发 |（六）内部模式的使用
  • IntelliJ IDE 插件开发 |（七）PSI 入门及实战（实现 MyBatis 插件的跳转功能）

前言

所谓 PSI(Program Structure Interface)，直译过来是程序结构接口，其实就是 IntelliJ 平台给我们提供用来解析代码文件，简化对各类编程语言（Java、Kotlin、XML）操作的接口。大部分针对编程语言或者框架的便利插件其实就与此相关，本文则会先介绍关于 PSI 的一些基础知识，然后再以一些 Mybatis 插件提供的 Java 方法 和 Mapper XML 文件互相跳转的例子来说明 PSI 的实际应用，最终实现效果如下图，本文涉及的到的完整代码文件也已上传到GitHub。

动画

PSI file(PSI 文件)

在本系列的第五篇文章中介绍了Virtual Files 和 Documents 用于处理文件的 API，而 PSI file 也是用于处理文件的 API，不过也有一些不同，具体如下：

类别层面范围
VF、Document文本文件应用级
PSI编程语言语法树项目级

获取文件的 PSI file 对象的方式主要有以下几种（来自官网）：

ContextAPI
ActionAnActionEvent.getData(CommonDataKeys.PSI_FILE)
DocumentPsiDocumentManager.getPsiFile()
PSI ElementPsiElement.getContainingFile()
Virtual FilePsiManager.findFile(), PsiUtilCore.toPsiFiles()
File NameFilenameIndex.getVirtualFilesByName()

最后一种方式FilenameIndex.getVirtualFilesByName()得到的结果是Virtual File 对象，需要再通过倒数第二行的方式再获取到对应的 PIS file 对象。

不过通过以上方式获取到的 PsiFile 只是顶层的接口，针对不同的编程语言，我们会使用相应的实现类。例如 Java 是 PsiJavaFile，XML 是 XMLFile。

下面通过实际使用来进行介绍，首先是 PsiJavaFile，当然在使用之前需要确保build.gradle文件中将 Java 添加到插件配置中（XML 内置无需添加）：

intellij {// 用到的插件plugins.set(listOf("com.intellij.java"))
}

然后在plugin.xml中加入以下配置：

<depends>com.intellij.modules.java</depends>

如果是 XML 则需要添加如下配置：

<depends>com.intellij.modules.xml</depends>

经过以上配置后，我们就可以使用 PsiJavaFile 和 XMLFile 的相关 API 了。

例如以下代码可以得到 java 文件的所属表名和类中 所有的方法名，然后展示出来：

class PsiJavaAction: AnAction() {override fun actionPerformed(e: AnActionEvent) {// 获取 PsiFile 对象val psiFile = e.getData(CommonDataKeys.PSI_FILE)// 转换为 PsiJavaFileval psiJavaFile = psiFile as PsiJavaFile// 获取类所属包Utils.info("当前类所属包：${psiJavaFile.packageName}")// 遍历获取所有的方法名psiJavaFile.accept(object: JavaRecursiveElementVisitor() {override fun visitMethod(method: PsiMethod) {Utils.info("查找到方法：${method.name}")}})}}

image-20240324153748554

在以上代码中需要注意psiJavaFile.accept()方法，其中 accept 方法是 PsiFile 所提供的方法，方法签名为void accept(@NotNull PsiElementVisitor visitor)，用于遍历 PSI 文件中的各类元素，可以看到上面我们在传参时传递的是JavaRecursiveElementVisitor，这是用于遍历 Java 中各类元素（字段、方法、注解等）的一个实现类，只需要重写对应的方法即可，在上面我们重写了visitMethod方法，其实内部提供了很多方法，大家可以自行尝试，通过方法名也可以看到这里还支持遍历 break 语句，断言语句等等：

image-20240324155416824

下面再说明如何遍历 XML 文件中的元素：

class PsiXMLAction: AnAction() {override fun actionPerformed(e: AnActionEvent) {// 获取 PsiFile 对象val psiFile = e.getData(CommonDataKeys.PSI_FILE)// 转换为 XmlFileval xmlFile = psiFile as XmlFile// 获取根标签名称Utils.info("根标签名称：${xmlFile.rootTag?.name}")// 遍历获取所有的元素信息xmlFile.accept(object: XmlRecursiveElementVisitor() {override fun visitXmlAttribute(attribute: XmlAttribute) {Utils.info("属性名称：${attribute.name}, 属性值：${attribute.value}")}})}}

image-20240324160702331

可以看到这里遍历使用的是XmlRecursiveElementVisitor，是 XML 对于PsiElementVisitor 的一个实现类，用于遍历 XML 文件中的各种元素：

image-20240324161835384

PSI Element(PSI 元素)

在上面介绍 PSI 文件的时候多次提到元素的概念，PSI 文件则正是由一系列的 PSI Element 所组成。和 PSI file 类似，PSI Element 也属于一个顶层接口，针对不同的编程语言，会有多种 PSI 元素。以 Java 为例，有 PsiClass、PSIMethod、PsiField 等对应 Java 语法的各类元素。而 XML 中也有 XmlTag、XmlAttribute 等概念。那我们如何快速知道一个文件中有哪些 PSI 元素？如何快速知道一个我们不熟悉的编程语言中的 PSI 元素？别慌，IntelliJ平台给我提供了工具：

image-20240324163733415

通过 IntelliJ 平台的工具，我们可以很方便地查看当前或者任意一种文件的 PSI 结构，下面分别以 Java 和 XML 文件为例，首先是 Java 文件：

image-20240324164120824

同时点击左下的元素节点，上方还会自动对应到元素位置：

image-20240324164235622

然后是 XML 文件：

image-20240324164626925

当然，除了 Java 和 XML，IntelliJ 支持的编程语言远不止这些，这里展示一部分，剩下的大家可以自行探索：

image-20240324164741314

上面介绍了如何快速查看 PSI 文件中的元素，下面再介绍如何去获取 PSI 元素，以下来自官网：

ContextAPI
ActionAnActionEvent.getData(CommonDataKeys.PSI_ELEMENT)Note: If an editor is currently open and the element under caret is a reference, this will return the result of resolving the reference.
PSI FilePsiFile.findElementAt(offset)
ReferencePsiReference.resolve()

可以看到总共有三种方式：第一种是直接获取当前光标位置的 PSI 元素；第二种是可以自己指定偏移量（如果不熟悉偏移量的概念，可以看本系列第五篇文章中讲解 CaretModel 的部分），获取指定文件指定位置的 PSI 元素；最后一种引用则使用的较少，这里不再展开介绍，大家可以查看官方文档进行了解。

除了获取某个位置的 PSI 元素，我们还可以获取其所属父元素或者子元素，下面以 Java 文件为例讲解如何使用：

class PsiJavaAction: AnAction() {override fun actionPerformed(e: AnActionEvent) {val psiFile = e.getData(CommonDataKeys.PSI_FILE)// 获取光标处 PSI 元素，假定该元素在方法内部val psiElement = e.getData(PlatformDataKeys.EDITOR)?.caretModel?.let { psiFile?.findElementAt(it.offset) }// 获取该元素所属的方法名val psiMethod = PsiTreeUtil.getParentOfType(psiElement, PsiMethod::class.java)// 获取该元素所属的类名val psiClass = PsiTreeUtil.getParentOfType(psiElement, PsiClass::class.java)Utils.info("所属方法名：${psiMethod?.name}")Utils.info("所属类名：${psiClass?.name}")}}

可以看到上面我们使用PsiTreeUtil::getParentOfType可以获取到一个元素的父元素，同时支持跨层级获取，既可以获取元素所属的方法，也可以获取元素所属的类。

效果如下：

image-20240324172523728

相应地我们也可以通过PsiTreeUtil::getChildrenOfTypeAsList去获取某个元素的所有子元素：

class PsiJavaAction: AnAction() {override fun actionPerformed(e: AnActionEvent) {val psiFile = e.getData(CommonDataKeys.PSI_FILE)// 先获取光标所在处元素所属的类val psiElement = e.getData(PlatformDataKeys.EDITOR)?.caretModel?.let { psiFile?.findElementAt(it.offset) }val psiClass = PsiTreeUtil.getParentOfType(psiElement, PsiClass::class.java)// 获取类中所有的方法val psiMethods = PsiTreeUtil.getChildrenOfTypeAsList(psiClass, PsiMethod::class.java)Utils.info("包含的方法：${psiMethods.joinToString(",") { it.name }}")}}

image-20240324174920319

实战

在正式实现前，先介绍一下整体的实现思路，这里只说明如何从 Java 方法跳转到 Mapper XML 文件中的节点，反向参考代码也很好理解，思路如下：

  1. 左侧图标行标记符通过实现RelatedItemLineMarkerProvider并重写collectNavigationMarkers方法设置。
  2. 判断代码行的元素类型为 PsiMethod 才进行设置，同时文件类名以 Mapper 结尾。
  3. 根据类名在项目查找同名的 Mapper XML 文件。
  4. 通过 accept 方法遍历 XML 文件所有的属性，将 id 值为对应方法名的标签所对应的元素保存到可跳转的目标。

设置行标记符号，平台给我们提供了 RelatedItemLineMarkerProvider 类进行设置，只需要自定义了自己的行标记类，然后在 plugin.xml 中添加 如下配置即可：

<codeInsight.lineMarkerProvider language="JAVA"implementationClass="cn.butterfly.psi.provider.JavaMapperLineMarkerProvider"/>

代码实现如下：

class JavaMapperLineMarkerProvider: RelatedItemLineMarkerProvider() {override fun collectNavigationMarkers(element: PsiElement,result: MutableCollection<in RelatedItemLineMarkerInfo<*>>) {// 查找类名后缀为 Mapper 内的所有方法if (element !is PsiMethod) {return}val psiClass = PsiTreeUtil.getParentOfType(element, PsiClass::class.java) ?: returnval className = psiClass.name ?: returnif (!className.endsWith("Mapper")) {return}// 查找同名 XML 文件对应的 PSI 文件对象val virtualFile = FileTypeIndex.getFiles(XmlFileType.INSTANCE, GlobalSearchScope.allScope(element.project)).first { it.name.startsWith(className) }val psiFile = PsiManager.getInstance(element.project).findFile(virtualFile)// 遍历 XML 文件中标签 id 节点值等于 Java 方法名的元素, 然后添加可跳转的行标记符psiFile?.accept(object : XmlRecursiveElementVisitor() {override fun visitXmlAttribute(attribute: XmlAttribute) {if (attribute.name == "id" && attribute.value == element.name) {// NavigationGutterIconBuilder 用于创建标识符result.add(NavigationGutterIconBuilder.create(PluginIcons.MAPPER_ICON).setTargets(setOf(attribute.navigationElement)).setTooltipText("Navigation to target in mapper xml").createLineMarkerInfo(element))}}})}}

总结

本文简单介绍了关于 PSI 文件和元素的基础知识，最后以一个 Mybatis 文件跳转的例子去演示了如何去实际运用 PSI，在下一篇文章则会介绍关于 PSI 的进阶知识，敬请期待~~

这篇关于IntelliJ IDE 插件开发 | （七）PSI 入门及实战（实现 MyBatis 插件的跳转功能）的文章就介绍到这儿，希望我们推荐的文章对编程师们有所帮助！


http://www.chinasem.cn/article/843423。

相关文章

网页解析 lxml 库--实战

网页解析 lxml 库--实战

lxml库使用流程 lxml 是 Python 的第三方解析库，完全使用 Python 语言编写，它对 XPath表达式提供了良好的支 持，因此能够了高效地解析 HTML/XML 文档。本节讲解如何通过 lxml 库解析 HTML 文档。 pip install lxml lxm| 库提供了一个 etree 模块，该模块专门用来解析 HTML/XML 文档，下面来介绍一下 lxml 库
阅读更多...
Spring Security 从入门到进阶系列教程

Spring Security 从入门到进阶系列教程

Spring Security 入门系列 《保护 Web 应用的安全》 《Spring-Security-入门（一）：登录与退出》 《Spring-Security-入门（二）：基于数据库验证》 《Spring-Security-入门（三）：密码加密》 《Spring-Security-入门（四）：自定义-Filter》 《Spring-Security-入门（五）：在 Sprin
阅读更多...
mybatis的整体架构

mybatis的整体架构

mybatis的整体架构分为三层： 1.基础支持层 该层包括：数据源模块、事务管理模块、缓存模块、Binding模块、反射模块、类型转换模块、日志模块、资源加载模块、解析器模块 2.核心处理层 该层包括：配置解析、参数映射、SQL解析、SQL执行、结果集映射、插件 3.接口层 该层包括：SqlSession 基础支持层 该层保护mybatis的基础模块，它们为核心处理层提供了良好的支撑。
阅读更多...
这15个Vue指令，让你的项目开发爽到爆

这15个Vue指令，让你的项目开发爽到爆

1. V-Hotkey 仓库地址: github.com/Dafrok/v-ho… Demo: 戳这里 https://dafrok.github.io/v-hotkey 安装: npm install --save v-hotkey 这个指令可以给组件绑定一个或多个快捷键。你想要通过按下 Escape 键后隐藏某个组件，按住 Control 和回车键再显示它吗？小菜一碟： <template
阅读更多...
Hadoop企业开发案例调优场景

Hadoop企业开发案例调优场景

需求 （1）需求：从1G数据中，统计每个单词出现次数。服务器3台，每台配置4G内存，4核CPU，4线程。 （2）需求分析： 1G / 128m = 8个MapTask；1个ReduceTask；1个mrAppMaster 平均每个节点运行10个 / 3台 ≈ 3个任务（4    3    3） HDFS参数调优 （1）修改：hadoop-env.sh export HDFS_NAMENOD
阅读更多...
hdu1043（八数码问题，广搜 + hash（实现状态压缩） ）

hdu1043（八数码问题，广搜 + hash（实现状态压缩） ）

利用康拓展开将一个排列映射成一个自然数，然后就变成了普通的广搜题。 #include<iostream>#include<algorithm>#include<string>#include<stack>#include<queue>#include<map>#include<stdio.h>#include<stdlib.h>#include<ctype.h>#inclu
阅读更多...
性能分析之MySQL索引实战案例

性能分析之MySQL索引实战案例

文章目录 一、前言二、准备三、MySQL索引优化四、MySQL 索引知识回顾五、总结 一、前言 在上一讲性能工具之 JProfiler 简单登录案例分析实战中已经发现SQL没有建立索引问题，本文将一起从代码层去分析为什么没有建立索引？ 开源ERP项目地址：https://gitee.com/jishenghua/JSH_ERP 二、准备 打开IDEA找到登录请求资源路径位置
阅读更多...
嵌入式QT开发：构建高效智能的嵌入式系统

嵌入式QT开发：构建高效智能的嵌入式系统

摘要： 本文深入探讨了嵌入式 QT 相关的各个方面。从 QT 框架的基础架构和核心概念出发，详细阐述了其在嵌入式环境中的优势与特点。文中分析了嵌入式 QT 的开发环境搭建过程，包括交叉编译工具链的配置等关键步骤。进一步探讨了嵌入式 QT 的界面设计与开发，涵盖了从基本控件的使用到复杂界面布局的构建。同时也深入研究了信号与槽机制在嵌入式系统中的应用，以及嵌入式 QT 与硬件设备的交互，包括输入输出设
阅读更多...
OpenHarmony鸿蒙开发（ Beta5.0）无感配网详解

OpenHarmony鸿蒙开发（ Beta5.0）无感配网详解

1、简介 无感配网是指在设备联网过程中无需输入热点相关账号信息，即可快速实现设备配网，是一种兼顾高效性、可靠性和安全性的配网方式。 2、配网原理 2.1 通信原理 手机和智能设备之间的信息传递，利用特有的NAN协议实现。利用手机和智能设备之间的WiFi 感知订阅、发布能力，实现了数字管家应用和设备之间的发现。在完成设备间的认证和响应后，即可发送相关配网数据。同时还支持与常规Sof
阅读更多...
C++11第三弹：lambda表达式 | 新的类功能 | 模板的可变参数

C++11第三弹：lambda表达式 | 新的类功能 | 模板的可变参数

🌈个人主页： 南桥几晴秋 🌈C++专栏： 南桥谈C++ 🌈C语言专栏： C语言学习系列 🌈Linux学习专栏： 南桥谈Linux 🌈数据结构学习专栏： 数据结构杂谈 🌈数据库学习专栏： 南桥谈MySQL 🌈Qt学习专栏： 南桥谈Qt 🌈菜鸡代码练习： 练习随想记录 🌈git学习： 南桥谈Git 🌈🌈🌈🌈🌈🌈🌈🌈🌈🌈🌈🌈🌈�
阅读更多...

Copyright China编程 23002807@qq.com

沪ICP备2023033072号-2