diff --git a/src/main/java/com/baomidou/plugin/idea/mybatisx/inspection/MapperXmlInspection.java b/src/main/java/com/baomidou/plugin/idea/mybatisx/inspection/MapperXmlInspection.java index 71ba011495d439e0018d3d1fb35c86c90b7485c3..c55b897febcb7c5809e24983119ba696db2244bd 100644 --- a/src/main/java/com/baomidou/plugin/idea/mybatisx/inspection/MapperXmlInspection.java +++ b/src/main/java/com/baomidou/plugin/idea/mybatisx/inspection/MapperXmlInspection.java @@ -1,10 +1,14 @@ package com.baomidou.plugin.idea.mybatisx.inspection; -import com.baomidou.plugin.idea.mybatisx.smartjpa.common.appender.CompositeAppender; +import com.intellij.psi.PsiFile; +import com.intellij.psi.impl.source.xml.XmlFileImpl; +import com.intellij.psi.xml.XmlFile; +import com.intellij.psi.xml.XmlTag; import com.intellij.util.xml.DomElement; import com.intellij.util.xml.highlighting.BasicDomElementsInspection; import com.intellij.util.xml.highlighting.DomElementAnnotationHolder; import com.intellij.util.xml.highlighting.DomHighlightingHelper; +import org.jetbrains.annotations.NotNull; import org.slf4j.Logger; import org.slf4j.LoggerFactory; @@ -39,4 +43,13 @@ public class MapperXmlInspection extends BasicDomElementsInspection public String getStaticDescription() { return "Static MapperXmlInspection"; } + + @Override + public boolean isAvailableForFile(@NotNull PsiFile file) { + if (file instanceof XmlFile) { + XmlTag rootTag = ((XmlFile) file).getRootTag(); + return rootTag != null && rootTag.getName().equals("mapper"); + } + return false; + } }