mybatis结合generator进行分页插件PluginAdapter开发

使用org.mybatis.generator生成UserExample时，无法进行分页，使用下面这个类运行generator便可以生成分页相关的属性了

package org.mybatis.generator.plugin;

import java.util.List;

import org.mybatis.generator.api.CommentGenerator;

import org.mybatis.generator.api.IntrospectedTable;

import org.mybatis.generator.api.PluginAdapter;

import org.mybatis.generator.api.ShellRunner;

import org.mybatis.generator.api.dom.java.Field;

import org.mybatis.generator.api.dom.java.FullyQualifiedJavaType;

import org.mybatis.generator.api.dom.java.JavaVisibility;

import org.mybatis.generator.api.dom.java.Method;

import org.mybatis.generator.api.dom.java.Parameter;

import org.mybatis.generator.api.dom.java.TopLevelClass;

import org.mybatis.generator.api.dom.xml.Attribute;

import org.mybatis.generator.api.dom.xml.TextElement;

import org.mybatis.generator.api.dom.xml.XmlElement;

public class PaginationMysqlPlugin extends PluginAdapter {

@Override

public boolean modelExampleClassGenerated(TopLevelClass topLevelClass, IntrospectedTable introspectedTable) {

addLimit(topLevelClass, introspectedTable, "limitStart");

addLimit(topLevelClass, introspectedTable, "limitEnd");

return super.modelExampleClassGenerated(topLevelClass, introspectedTable);

}

@Override

public boolean sqlMapSelectByExampleWithoutBLOBsElementGenerated(XmlElement element, IntrospectedTable introspectedTable) {

XmlElement isNotNullElement = new XmlElement("if");

isNotNullElement.addAttribute(new Attribute("test", "limitStart >= 0"));

isNotNullElement.addElement(new TextElement(" limit ${limitStart} , ${limitEnd}"));

element.addElement(isNotNullElement);

return super.sqlMapUpdateByExampleWithoutBLOBsElementGenerated(element, introspectedTable);

}

private void addLimit(TopLevelClass topLevelClass, IntrospectedTable introspectedTable, String name) {

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);

}

public boolean validate(List<String> warnings) {

return true;

}

public static void generate() {

String config = PaginationMysqlPlugin.class.getClassLoader().getResource("generatorConfig.xml").getFile();

String[] arg = { "-configfile", config, "-overwrite" };

ShellRunner.main(arg);

}

public static void main(String[] args) {

generate();

}

}

项目代码下载：https://github.com/vincentduan/Medical.git