mybatis generator插件开发

mybatis现在普遍使用的每一个人DAO框架。mybatis generator它可以基于数据库中的表结构，生成自己主动mybatis代码和配置文件，方便使用，当然，实际的使用过程中。generator当然，也有很多不尽人意的地方，幸运的是，他提供了一种机制插头，来帮我们做扩大。解说的插件开发依赖下面mybatis版本号：

<dependency>
<groupId>org.mybatis</groupId>
<artifactId>mybatis</artifactId>
<version>3.2.2</version>
</dependency>
<dependency>
<groupId>org.mybatis.generator</groupId>
<artifactId>mybatis-generator-maven-plugin</artifactId>
<version>1.3.2</version>
</dependency>

首先我们的插件类须要扩展PluginAdapter类，例如以下：

public class PaginationPlugin extends PluginAdapter {

pluginAdapter提供了全部的接口，方便我们在插件进行到某个操作的时候。做自己定义的改动。这里我们主要介绍下面几类方法：

public boolean modelExampleClassGenerated(TopLevelClass topLevelClass,
IntrospectedTable introspectedTable)

public boolean sqlMapSelectByExampleWithoutBLOBsElementGenerated(XmlElement element, IntrospectedTable introspectedTable) 

modelExampleClassGenerated方法顾名思义。就是生产modelExample类时的扩展，了解generator的都知道。他会帮你生产mapper、model、example三类java对象，当中example类是辅助我们用代码的方式来生产sql的，这个modelExampleClassGenerated方法就是在生产example的时候起效果的。比方。原有的generator1.3.2插件不支持分页，那我们能够首先在modelExampleClassGenerated方法中。对生产的Class加上limit的參数，代码例如以下：

CommentGenerator commentGenerator = context.getCommentGenerator();Field field = new Field();field.setVisibility(JavaVisibility.PROTECTED);field.setType(FullyQualifiedJavaType.getIntInstance());field.setName(name);field.setInitializationString("-1");commentGenerator.addFieldComment(field, introspectedTable);topLevelClass.addField(field);char c = name.charAt(0);String camel = Character.toUpperCase(c) + name.substring(1);Method method = new Method();method.setVisibility(JavaVisibility.PUBLIC);method.setName("set" + camel);method.addParameter(new Parameter(FullyQualifiedJavaType.getIntInstance(), name));method.addBodyLine("this." + name + "=" + name + ";");commentGenerator.addGeneralMethodComment(method, introspectedTable);topLevelClass.addMethod(method);method = new Method();method.setVisibility(JavaVisibility.PUBLIC);method.setReturnType(FullyQualifiedJavaType.getIntInstance());method.setName("get" + camel);method.addBodyLine("return " + name + ";");commentGenerator.addGeneralMethodComment(method, introspectedTable);topLevelClass.addMethod(method);

如代码所看到的，事实上我们就是定义了一个private int limitStart的属性，并加入了get和set方法。加入完example的属性后，我们须要将这个属性加到生成的mapper xml配置文件里。这就用到sqlMapSelectByExampleWithBLOBsElementGenerated方法，当然，注意这种方法的名字，我们想改动哪个xml文件里的sql，就须要找到相应的方法来复写。

XmlElement isNotNullElement = new XmlElement("if"); //$NON-NLS-1$isNotNullElement.addAttribute(new Attribute("test", "limitStart != null and limitStart>-1")); //$NON-NLS-1$ //$NON-NLS-2$isNotNullElement.addElement(new TextElement("limit ${limitStart} , ${limitEnd}"));element.addElement(isNotNullElement);

如上述代码，我们在生成的xml文件里。增加一个if条件推断，推断假设当前example类的limitStart和limitEnd不为-1。注意这个-1是我们在example扩展字段时候加上的默认值。假设不是默认值。那么就加上limit ${limitStart}, ${limitEnd}分页条件。如上所述，就是这么简单。我们能够对生成的mybatis java代码、xml配置做自己的改动。以下看一下，怎样来使用该插件。我们首先须要在使用generator的projectpom文件里增加generator插件，并将我们的插件包依赖加进来，这里我们的样例是mybatis-page-plugin。

<build><defaultGoal>install</defaultGoal><plugins><plugin><groupId>org.mybatis.generator</groupId><artifactId>mybatis-generator-maven-plugin</artifactId><version>1.3.2</version><configuration><configurationFile>generatorConfig.xml</configurationFile><verbose>true</verbose><overwrite>true</overwrite></configuration><dependencies><dependency><groupId>mysql</groupId><artifactId>mysql-connector-java</artifactId><version>5.1.6</version></dependency><dependency><groupId>org.duoku.groom</groupId><artifactId>mybatis-page-plugin</artifactId><version>1.0-SNAPSHOT</version></dependency></dependencies></plugin></plugins></build>

如配置所述，我们的插件配置文件是generatorConfig.xml。我们来看一下这个文件：

<generatorConfiguration><context id="MBG" targetRuntime="MyBatis3" defaultModelType="conditional">	<!--targetRuntime 此属性用于指定生成的代码的执行目标。--><plugin type="org.mybatis.generator.plugins.EqualsHashCodePlugin" /><plugin type="org.duoku.groom.mybatis.plugin.PaginationPlugin"></plugin>

如上图所看到的。我们在plugin中，加上我们的插件，PaginationPlugin。此，我们可以使用这个插件。