public class IconUtils {
private static ImageIcon methodIcon = new ImageIcon(IconUtils.class.getClassLoader().getResource("icon/mybatis.png"));
private static ImageIcon xmlIcon = new ImageIcon(IconUtils.class.getClassLoader().getResource("icon/mybatis-ns.png"));
public static Icon useMyBatisIcon() {
return methodIcon;
}
public static Icon useXmlIcon() {
return xmlIcon;
}
}
package com.ccnode.codegenerator.view;
import com.ccnode.codegenerator.util.IconUtils;
import com.intellij.codeInsight.daemon.RelatedItemLineMarkerInfo;
import com.intellij.codeInsight.daemon.RelatedItemLineMarkerProvider;
import com.intellij.codeInsight.navigation.NavigationGutterIconBuilder;
import com.intellij.openapi.editor.markup.GutterIconRenderer;
import com.intellij.openapi.module.Module;
import com.intellij.openapi.module.ModuleUtilCore;
import com.intellij.psi.PsiClass;
import com.intellij.psi.PsiElement;
import com.intellij.psi.PsiFile;
import com.intellij.psi.PsiMethod;
import com.intellij.psi.search.GlobalSearchScope;
import com.intellij.psi.search.PsiShortNamesCache;
import com.intellij.psi.xml.XmlAttribute;
import com.intellij.psi.xml.XmlFile;
import com.intellij.psi.xml.XmlTag;
import org.jetbrains.annotations.NotNull;
import java.util.Collection;
import java.util.HashSet;
import java.util.Set;
/**
* Created by bruce.ge on 2016/12/8.
*/
public class MybatisXmlLineMarkerProvider extends RelatedItemLineMarkerProvider {
private static Set<String> tagNameSet = new HashSet<String>() {{
add("select");
add("insert");
add("update");
add("delete");
}};
@Override
protected void collectNavigationMarkers(@NotNull PsiElement element, Collection<? super RelatedItemLineMarkerInfo> result) {
if (!(element instanceof XmlTag))
return;
PsiFile psiFile = element.getContainingFile();
if (!(psiFile instanceof XmlFile))
return;
XmlFile xmlFile = (XmlFile) psiFile;
if (!xmlFile.getRootTag().getName().equals("mapper")) {
return;
}
XmlTag tag = (XmlTag) element;
if (!tagNameSet.contains(tag.getName())) {
return;
}
XmlAttribute namespaceAttribute = xmlFile.getRootTag().getAttribute("namespace");
if (namespaceAttribute == null || namespaceAttribute.getValue() == null) {
return;
}
String namespace = namespaceAttribute.getValue();
XmlAttribute id = tag.getAttribute("id");
if (id == null || id.getValue() == null) {
return;
}
String[] split = namespace.split("\.");
String className = split[split.length - 1];
Module moduleForPsiElement = ModuleUtilCore.findModuleForPsiElement(element);
if (moduleForPsiElement == null) {
return;
}
PsiClass[] classesByName = PsiShortNamesCache.getInstance(element.getProject()).getClassesByName(className, GlobalSearchScope.moduleScope(moduleForPsiElement));
PsiClass realClass = null;
for (PsiClass psiClass : classesByName) {
if (psiClass.isInterface() && psiClass.getQualifiedName().equals(namespace)) {
//
realClass = psiClass;
break;
}
}
if (realClass == null) {
return;
}
String xmlMethodName = id.getValue();
PsiElement findedMethod = null;
PsiMethod[] allMethods = realClass.getAllMethods();
for (PsiMethod classMethod : allMethods) {
if (xmlMethodName.equals(classMethod.getName())) {
findedMethod = classMethod;
break;
}
}
if (findedMethod == null) {
return;
}
result.add(NavigationGutterIconBuilder.create(IconUtils.useXmlIcon()).setAlignment(GutterIconRenderer.Alignment.CENTER)
.setTarget(findedMethod).setTooltipTitle("navigation to mapper class").createLineMarkerInfo(element));
//means we find the class. then get it all method and get it.
}
}
package com.ccnode.codegenerator.view;
import com.ccnode.codegenerator.util.IconUtils;
import com.ccnode.codegenerator.util.MyPsiXmlUtils;
import com.intellij.codeInsight.daemon.RelatedItemLineMarkerInfo;
import com.intellij.codeInsight.daemon.RelatedItemLineMarkerProvider;
import com.intellij.codeInsight.navigation.NavigationGutterIconBuilder;
import com.intellij.openapi.components.ServiceManager;
import com.intellij.openapi.editor.markup.GutterIconRenderer;
import com.intellij.openapi.module.Module;
import com.intellij.openapi.module.ModuleUtilCore;
import com.intellij.openapi.project.Project;
import com.intellij.psi.PsiClass;
import com.intellij.psi.PsiElement;
import com.intellij.psi.PsiFile;
import com.intellij.psi.PsiMethod;
import com.intellij.psi.search.GlobalSearchScope;
import com.intellij.psi.search.PsiNonJavaFileReferenceProcessor;
import com.intellij.psi.search.PsiSearchHelper;
import com.intellij.psi.search.PsiShortNamesCache;
import com.intellij.psi.xml.XmlAttribute;
import com.intellij.psi.xml.XmlFile;
import org.jetbrains.annotations.NotNull;
import java.util.ArrayList;
import java.util.Collection;
import java.util.List;
/**
* Created by bruce.ge on 2016/12/8.
*/
public class MybatisJavaLineMarkerProvider extends RelatedItemLineMarkerProvider {
@Override
protected void collectNavigationMarkers(@NotNull PsiElement element, Collection<? super RelatedItemLineMarkerInfo> result) {
if (element instanceof PsiMethod) {
PsiMethod method = (PsiMethod) element;
PsiClass containingClass = method.getContainingClass();
if (!containingClass.isInterface()) {
return;
}
// if it's interface, then find it in the xml file to check if it contain the name.
//shall be mapper then go to find to corresponding xml file.
Project project = element.getProject();
String qualifiedName = containingClass.getQualifiedName();
PsiElement methodElement = null;
PsiFile[] filesByName = PsiShortNamesCache.getInstance(project).getFilesByName(containingClass.getName() + ".xml");
//first search with samename xml.
if (filesByName.length == 0) {
methodElement = handleWithFileNotFound(method, project, qualifiedName, result);
} else {
for (PsiFile file : filesByName) {
if (file instanceof XmlFile) {
XmlFile xmlFile = (XmlFile) file;
XmlAttribute namespace = xmlFile.getRootTag().getAttribute("namespace");
if (namespace == null || !namespace.getValue().equals(qualifiedName)) {
continue;
}
//say we find the xml file.
PsiElement psiElement = MyPsiXmlUtils.findTagForMethodName(xmlFile, method.getName());
if (psiElement != null) {
methodElement = psiElement;
break;
}
}
}
if (methodElement == null) {
methodElement = handleWithFileNotFound(method, project, qualifiedName, result);
}
}
if (methodElement != null) {
result.add(NavigationGutterIconBuilder.create(IconUtils.useMyBatisIcon()).setAlignment(GutterIconRenderer.Alignment.CENTER)
.setTarget(methodElement).setTooltipTitle("navigation to mapper xml").createLineMarkerInfo(method.getNameIdentifier()));
}
//只进行method的判断 进行控制 其他的不管
}
}
private PsiElement handleWithFileNotFound(@NotNull PsiMethod method, Project project, final String qualifiedName, Collection<? super RelatedItemLineMarkerInfo> result) {
PsiSearchHelper searchService = ServiceManager.getService(project, PsiSearchHelper.class);
List<XmlFile> xmlFiles = new ArrayList<XmlFile>();
Module moduleForPsiElement = ModuleUtilCore.findModuleForPsiElement(method);
if (moduleForPsiElement == null) {
return null;
}
searchService.processUsagesInNonJavaFiles("mapper", new PsiNonJavaFileReferenceProcessor() {
@Override
public boolean process(PsiFile file, int startOffset, int endOffset) {
if (file instanceof XmlFile) {
XmlFile xmlFile = (XmlFile) file;
if (xmlFile.getRootTag() != null) {
XmlAttribute namespace = xmlFile.getRootTag().getAttribute("namespace");
if (namespace != null && namespace.getValue().equals(qualifiedName)) {
xmlFiles.add(xmlFile);
return false;
}
}
}
return true;
}
}, GlobalSearchScope.moduleScope(moduleForPsiElement));
if (xmlFiles.size() == 0) {
return null;
}
return MyPsiXmlUtils.findTagForMethodName(xmlFiles.get(0), method.getName());
}
}
<idea-plugin version="2">
<id>com.ccnode.codegenerator.MyBatisCodeHelper</id>
<name>MyBatisCodeHelper</name>
<version>1.5.0-SNAPSHOT</version>
<vendor email="gejun123456@gmail.com">MyBatisCodeHelper</vendor>
<description><![CDATA[
<div>
<p><a href="https://github.com/gejun123456/MyBatisCodeHelper">GitHub</a>
| <a href="https://github.com/gejun123456/MyBatisCodeHelper/issues">Issues</a></p>
<p>特性</p>
<ul>
<li>支持mysql和oracle</li>
<li><p>根据java对象生成mybatis crud代码和建表sql</p></li>
<li><p><b>根据mybatis接口中的方法名生成mybatis的sql
支持find,update,delete,count方法 只需定义一个方法名就可以得到完整mybatis xml代码 可生成大部分单表操作sql 极大提升效率</b></p></li>
<li><p>mybatis接口文件的mapper xml之间的相互跳转</p></li>
<li><p>数据库对象更新后一键更新对应的Sql和mybatis xml文件</p></li>
<li><p>默认提供insert，insertSelective，insertList，update。因为delete可以由方法名生成 默认不提供</p></li>
<li><p>提供mybatis接口方法名的重构</p></li>
<li>refid, resultMap跳转到到定义，支持重命名</li>
<li>refid,resultMap,keyProperty,property的自动补全</li>
<li><p>mybatis mapper xml文件sql的自动补全</p></li>
</ul>
<p>使用方法</p>
<ul>
<li><p>在数据库对象上使用alt+insert (generate mybatis files)来生成crud代码和建表sql(mac上使用ctrl+N)</p></li>
<li><p>数据库对象添加字段后使用alt+insert (generate mybatis files) 来生成更新sql,mapper xml中的字段</p></li>
<li><p>在mybatis接口的方法名上使用alt+enter来生成对应的mapper sql</p></li>
<li><p>详细配置: <a href="https://github.com/gejun123456/MyBatisCodeHelper">https://github.com/gejun123456/MyBatisCodeHelper</a></p></li>
<li>qq群：542735959</li>
</ul>
<p>Features</p>
<ul>
<li>Generate mybatis crud and create table sql according to domain class</li>
<li><p><b>Generate mybatis sql based on mybatis interface method name, with this, you don't have to write most sql for
non join query</b>,support with method name start with find,update,delete,count</p></li>
<li><p>Jump from mybatis dao interface to mapper xml each other</p></li>
<li><p>generate files provide insert, insertSelective,insertList,update. others can be generated by methodName</p></li>
<li><p>refactor for mybatis interface method name</p></li>
<li><p>jump from refid resultMap to their definition, refactor their name as well</li>
<li><p>mybatis mapper sql auto completion</p></li>
<li>refid,resultMap,keyProperty,property auto complete</li>
<li><p>Support java + MySQL and Oracle and later will support more DB.</p></li>
<li><a href="https://github.com/gejun123456/MyBatisCodeHelper">https://github.com/gejun123456/MyBatisCodeHelper</a> to learn more.</li>
</ul>
<p>How to use</p>
<ul>
<li>alt+insert (generate mybatis files) on domain class to generate mybatis files (ctrl+N on mac)</li>
<li>alt+insert (generate mybatis files) on domain class to update mybatis files when domain class add field (ctrl+N on mac)</li>
<li>alt+enter on dao interface method to generate mybatis mapper sql</li>
<li>view more on <a href="https://github.com/gejun123456/MyBatisCodeHelper">https://github.com/gejun123456/MyBatisCodeHelper</a></li>
</ul>
</div>
]]></description>
<change-notes><![CDATA[
<p><h4>1.4.5</h4>
<ul>
<li>support with enum type</li>
</ul>
</p>
<p><h4>1.4.4</h4>
<ul>
<li>fix exception when start up</li>
</ul>
</p>
<p><h4>1.4.3</h4>
<ul>
<li> support multiple method xml generate</li>
<li> support for mybatis plus</li>
<li> support domain class with protected field and static field</li>
<li> support for small resolution</li>
<li>bugfix - fix can't generate to path</li>
</ul>
</p>
<p><h4>1.4.2</h4>
<ul>
<li>bugfix - fix Could not initialize class com.ccnode.codegenerator.freemarker.TemplateUtil</li>
</ul>
</p>
<p><h4>1.4.1</h4>
<ul>
<li>bugfix - oralce insertList</li>
<li>bugfix - import issue</li>
</ul>
</p>
<p><h4>1.4</h4>
<ul>
<li>add support for oracle</li>
<li>config use generate key</li>
<li>auto complete for resultMap,refid,keyProperty,property</li>
<li>support with java.sql.Timestamp java.sql.Date java.sql.Time Enum LocalDateTime LocalDate</li>
<li>support update field with insertSelective</li>
<li>add new icon of mybatis jump to xml ect</li>
</ul>
</p>
<p><h4>1.3</h4>
<ul>
<li>add index column and hasDefault column when generate mybatis files</li>
<li>jump from refid resultMap to their definition in mybatis xml. could refactor as well</li>
<li>could refactor method name in xml</li>
<li>add insertSelective when generate mybatis files</li>
<li>generate for greaterThanOrEqualTo and lessThanOrEqualTo and betweenOrEqualTo</li>
<li>add configuration to use with @Mapper</li>
<li>fix not null issue for find module - bug fix</li>
</ul>
</p>
<p><h4>1.2</h4>
<ul>
<li>add support for unsigned type, small int.</li>
<li>check for using object type instead of primitive type</li>
<li>support more auto completion for sql</li>
<li>double type support - bug fix</li>
<li>update field exception -bug fix</li>
</ul>
</p>
<p> 1.1 fix doc</p>
<p>1.0 use gradle, fix bugs.</p>
]]>
</change-notes>
<actions>
<group id="com.ccnode.codegenerator.MybatisCodeHelper" text="MybatisCodeHelper" description="MybatisCodeHelper"
popup="true">
<separator/>
<group id="MybatisCodeHelperSubGroup"/>
<add-to-group group-id="ToolsMenu" relative-to-action="GenerateJavadoc" anchor="before"/>
<action id="MybatisCodeHelper.LearnMore" class="com.ccnode.codegenerator.view.ShowLearnMoreAction"
text="Learn More"
description="Learn_More"/>
<!--<action id="mybatis.enterLicense" class="com.ccnode.codegenerator.view.EnterLicenseAction" text="Register" description="License"/>-->
<!--<action id="mybatis.generator" class="com.ccnode.codegenerator.view.GenCodeAction" text="genCode" description="A test menu item"/>-->
</group>
<group id="MybatisCodeHelpergeneratesqlgroup">
<add-to-group group-id="GenerateGroup" relative-to-action="GenerateGetter" anchor="after"/>
<action class="com.ccnode.codegenerator.view.GenCodeActionUsingAlt" id="MybatisCodeHelpergencodeusingalt"
text="generate mybatis files"
description="generate sql"/>
</group>
<group id="MyBatisCodeHelperEditorGroup">
<add-to-group group-id="EditorPopupMenu"></add-to-group>
<action class="com.ccnode.codegenerator.view.MultipleMethodGenerateAction"
id="MybatisMutlipleMethodGenerate"
text="generate mybatis xml" description="generate mybatis xml"/>
</group>
<!--<group id="com.ccnode.codegenerator.mybatis.generator" text="Gen" description="CodeGenerator">-->
<!--<add-to-group group-id="TestActionSubGroup" />-->
<!--<action id="mybatis.generator_1" class="com.ccnode.codegenerator.view.GenCodeAction" text="genCode1" description="A test menu item1"/>-->
<!--<action id="mybatis.generator_2" class="com.ccnode.codegenerator.view.GenCodeAction" text="genCode2" description="A test menu item2"/>-->
<!--</group>-->
</actions>
<!-- please see http://www.jetbrains.org/intellij/sdk/docs/basics/getting_started/build_number_ranges.html for description -->
<idea-version since-build="141.0"/>
<!-- please see http://www.jetbrains.org/intellij/sdk/docs/basics/getting_started/plugin_compatibility.html
on how to target different products -->
<!-- uncomment to enable plugin in all products
<depends>com.intellij.modules.lang</depends>
-->
<application-components>
<component>
<implementation-class>com.ccnode.codegenerator.myconfigurable.MyBatisCodeHelperApplicationComponent
</implementation-class>
</component>
</application-components>
<project-components>
<component>
<implementation-class>com.ccnode.codegenerator.view.datasource.MyBatisDatasourceComponent</implementation-class>
</component>
</project-components>
<extensions defaultExtensionNs="com.intellij">
<!-- Add your extensions here -->
<codeInsight.lineMarkerProvider language="JAVA"
implementationClass="com.ccnode.codegenerator.view.MybatisJavaLineMarkerProvider"/>
<codeInsight.lineMarkerProvider language="XML"
implementationClass="com.ccnode.codegenerator.view.MybatisXmlLineMarkerProvider"/>
<completion.contributor language="JAVA"
implementationClass="com.ccnode.codegenerator.view.completion.MethodNameCompletionContributor"/>
<completion.contributor language="XML"
implementationClass="com.ccnode.codegenerator.view.completion.MapperSqlCompletionContributor"/>
<completion.contributor language="XML"
implementationClass="com.ccnode.codegenerator.view.completion.MapperXmlTagElementCompletionContributor"/>
<inspectionToolProvider implementation="com.ccnode.codegenerator.view.inspection.MyBatisMethodNotFoundInXmlProvider"/>
<psi.referenceContributor
implementation="com.ccnode.codegenerator.view.MyBatisJavaToXmlPsiReferenceContributor"/>
<psi.referenceContributor
implementation="com.ccnode.codegenerator.view.MyBatisXmlAttributeReferenceContributor"/>
<applicationConfigurable instance="com.ccnode.codegenerator.myconfigurable.MyConfigurable"
id="MyBatisCodeHelper"
displayName="MyBatisCodeHelper"/>
<toolWindow id="Mybatis datasource" secondary="true" anchor="right" factoryClass="com.ccnode.codegenerator.view.datasource.MyBatisDatasourceToolWindow"/>
<!--<renamePsiElementProcessor implementation="com.ccnode.codegenerator.view.rename.RenameResultMapSqlProcessor" order="first"/>-->
<intentionAction>
<className>com.ccnode.codegenerator.view.GenerateMethodXmlAction</className>
<!--<category>Conditional Operator</category>-->
<!--<descriptionDirectoryName>ConditionalOperatorConvertor</descriptionDirectoryName>-->
<!--<bundleName>Generate daoxml</bundleName>-->
</intentionAction>
<projectService serviceInterface="com.ccnode.codegenerator.view.completion.MysqlCompleteCacheInteface" serviceImplementation="com.ccnode.codegenerator.view.completion.MyBatisSqlCompleteCache"/>
<!--<intentionAction>-->
<!--<className>com.ccnode.codegenerator.view.GenerateDaoXmlFixAction</className>-->
<!--<category>Conditional Operator</category>-->
<!--<descriptionDirectoryName>ConditionalOperatorConvertor</descriptionDirectoryName>-->
<!--</intentionAction>-->
</extensions>
<!--<project-components>-->
<!--<component>-->
<!--<!–<implementation-class>com.intellij.codeInsight.intention.Condition</implementation-class>–>-->
<!--</component>-->
<!--</project-components>-->
<actions>
<!-- Add your actions here -->
</actions>
</idea-plugin>