MyBatis Gen如何将MySQL分页插件改写为长尾？

本文共计553个文字，预计阅读时间需要3分钟。

javapackage org.mybatis.generator.api;

import java.util.List;import org.mybatis.generator.api.IntrospectedTable;import org.mybatis.generator.api.PluginAdapter;import org.mybatis.generator.api.dom.java.Field;

public class MyBatisGenPlugin extends PluginAdapter {

@Override public boolean modelFieldGenerated(Field field, TopLevelClass topLevelClass, IntrospectedTable introspectedTable) { // Add pagination field field.setFullyQualifiedJavaType(List.class); field.setName(list); field.addAnnotation(@ApiModelProperty(value=\列表数据\)); return true; }}

mybatis gen MySQL分页插件

package org.mybatis.generator.api; import java.util.List; import org.mybatis.generator.api.IntrospectedTable; import org.mybatis.generator.api.PluginAdapter; import org.mybatis.generator.api.dom.java.Field; import org.mybatis.generator.api.dom.java.FullyQualifiedJavaType; import org.mybatis.generator.api.dom.java.InnerClass; import org.mybatis.generator.api.dom.java.Interface; 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.PrimitiveTypeWrapper; 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 MySQLLimitPlugin extends PluginAdapter { @Override public boolean validate(List list) { return true; } @Override public boolean clientGenerated(Interface interfaze, TopLevelClass topLevelClass, IntrospectedTable introspectedTable) { String superInterface="io.zzdc.mapper.base.BaseMapper<"+ introspectedTable.getBaseRecordType()+","+introspectedTable.getExampleType()+">"; interfaze.addImportedType(new FullyQualifiedJavaType(introspectedTable.getBaseRecordType())); interfaze.addImportedType(new FullyQualifiedJavaType(introspectedTable.getExampleType())); interfaze.addSuperInterface(new FullyQualifiedJavaType(superInterface)); interfaze.addImportedType(new FullyQualifiedJavaType("io.zzdc.mapper.base.BaseMapper")); return super.clientGenerated(interfaze, topLevelClass, introspectedTable); } /** * 为每个Example类添加limit和offset属性已经set、get方法 */ @Override public boolean modelExampleClassGenerated(TopLevelClass topLevelClass, IntrospectedTable introspectedTable) { PrimitiveTypeWrapper integerWrapper = FullyQualifiedJavaType.getIntInstance().getPrimitiveTypeWrapper(); Field limit = new Field(); limit.setName("limit"); limit.setVisibility(JavaVisibility.PRIVATE); limit.setType(integerWrapper); topLevelClass.addField(limit); Method setLimit = new Method(); setLimit.setVisibility(JavaVisibility.PUBLIC); setLimit.setName("setLimit"); setLimit.addParameter(new Parameter(integerWrapper, "limit")); setLimit.addBodyLine("this.limit = limit;"); topLevelClass.addMethod(setLimit); Method getLimit = new Method(); getLimit.setVisibility(JavaVisibility.PUBLIC); getLimit.setReturnType(integerWrapper); getLimit.setName("getLimit"); getLimit.addBodyLine("return limit;"); topLevelClass.addMethod(getLimit); Field offset = new Field(); offset.setName("offset"); offset.setVisibility(JavaVisibility.PRIVATE); offset.setType(integerWrapper); topLevelClass.addField(offset); Method setOffset = new Method(); setOffset.setVisibility(JavaVisibility.PUBLIC); setOffset.setName("setOffset"); setOffset.addParameter(new Parameter(integerWrapper, "offset")); setOffset.addBodyLine("this.offset = offset;"); topLevelClass.addMethod(setOffset); Method getOffset = new Method(); getOffset.setVisibility(JavaVisibility.PUBLIC); getOffset.setReturnType(integerWrapper); getOffset.setName("getOffset"); getOffset.addBodyLine("return offset;"); topLevelClass.addMethod(getOffset); List classes=topLevelClass.getInnerClasses(); /** * 修改内部类 */ for(InnerClass innerClass:classes){ if(innerClass.getType().getShortName().equals("GeneratedCriteria")){ innerClass.setVisibility(JavaVisibility.PUBLIC); String basecri="io.zzdc.entity.po.base.BaseCriteria"; FullyQualifiedJavaType fullyQualifiedJavaType=new FullyQualifiedJavaType(basecri); topLevelClass.addImportedType(fullyQualifiedJavaType); innerClass.addSuperInterface(fullyQualifiedJavaType); List methods= innerClass.getMethods(); for(Method method:methods){ method.setVisibility(JavaVisibility.PUBLIC); } } } FullyQualifiedJavaType fullyQualifiedJavaType=new FullyQualifiedJavaType("io.zzdc.entity.po.base.BaseExample"); topLevelClass.addSuperInterface(fullyQualifiedJavaType); return true; } /** * 为Mapper.xml的selectByExample添加limit */ @Override public boolean sqlMapSelectByExampleWithoutBLOBsElementGenerated(XmlElement element, IntrospectedTable introspectedTable) { XmlElement ifLimitNotNullElement = new XmlElement("if"); ifLimitNotNullElement.addAttribute(new Attribute("test", "limit != null")); XmlElement ifOffsetNotNullElement = new XmlElement("if"); ifOffsetNotNullElement.addAttribute(new Attribute("test", "offset != null")); ifOffsetNotNullElement.addElement(new TextElement("limit ${offset}, ${limit}")); ifLimitNotNullElement.addElement(ifOffsetNotNullElement); XmlElement ifOffsetNullElement = new XmlElement("if"); ifOffsetNullElement.addAttribute(new Attribute("test", "offset == null")); ifOffsetNullElement.addElement(new TextElement("limit ${limit}")); ifLimitNotNullElement.addElement(ifOffsetNullElement); element.addElement(ifLimitNotNullElement); return true; } }