mybatis代码自动生成Demo

最新推荐文章于 2023-01-10 14:22:19 发布

weixin_33962923

最新推荐文章于 2023-01-10 14:22:19 发布

阅读量101

点赞数

文章标签： java python

原文链接：https://my.oschina.net/sky2008/blog/515725

版权

2019独角兽企业重金招聘Python工程师标准>>>

mybatis代码自动生成Demo

package com.chao.plugins;

import org.mybatis.generator.api.CommentGenerator;
import org.mybatis.generator.api.IntrospectedTable;
import org.mybatis.generator.api.PluginAdapter;
import org.mybatis.generator.api.dom.java.*;
import org.mybatis.generator.api.dom.xml.Attribute;
import org.mybatis.generator.api.dom.xml.Document;
import org.mybatis.generator.api.dom.xml.TextElement;
import org.mybatis.generator.api.dom.xml.XmlElement;

import java.util.List;

/**
* Created by zhaochaochao on 2015/1/8.
*/
public class MySQLPagePlugin extends PluginAdapter {
   @Override
   public boolean modelExampleClassGenerated(TopLevelClass topLevelClass,
           IntrospectedTable introspectedTable) {
       // add field, getter, setter for limit clause
       addPage(topLevelClass, introspectedTable, "page");
       return super.modelExampleClassGenerated(topLevelClass,
               introspectedTable);
   }

   @Override
   public boolean sqlMapDocumentGenerated(Document document,
           IntrospectedTable introspectedTable) {
       XmlElement parentElement = document.getRootElement();

       // 产生分页语句
       XmlElement paginationSuffixElement = new XmlElement("sql");
       paginationSuffixElement.addAttribute(new Attribute("id", "PageSuffix"));
       XmlElement pageEnd = new XmlElement("if");
       pageEnd.addAttribute(new Attribute("test", "page != null"));
       pageEnd.addElement(new TextElement("limit #{page.begin},#{page.length}"));
       paginationSuffixElement.addElement(pageEnd);
       parentElement.addElement(paginationSuffixElement);

       return super.sqlMapDocumentGenerated(document, introspectedTable);
   }

   @Override
   public boolean sqlMapSelectByExampleWithoutBLOBsElementGenerated(
           XmlElement element, IntrospectedTable introspectedTable) {

       XmlElement pageElement = new XmlElement("include"); //$NON-NLS-1$
       pageElement.addAttribute(new Attribute("refid", "PageSuffix"));
       element.getElements().add(pageElement);

       return super.sqlMapUpdateByExampleWithoutBLOBsElementGenerated(element,
               introspectedTable);
   }

   private void addPage(TopLevelClass topLevelClass,
           IntrospectedTable introspectedTable, String name) {
       String pageClass = "com.chao.plugins.Page";
       topLevelClass.addImportedType(new FullyQualifiedJavaType(pageClass));
       CommentGenerator commentGenerator = context.getCommentGenerator();
       Field field = new Field();
       field.setVisibility(JavaVisibility.PROTECTED);
       field.setType(new FullyQualifiedJavaType("Page"));
       field.setName(name);
       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(new FullyQualifiedJavaType(pageClass), name));
       method.addBodyLine("this." + name + "=" + name + ";");
       commentGenerator.addGeneralMethodComment(method, introspectedTable);
       topLevelClass.addMethod(method);
       method = new Method();
       method.setVisibility(JavaVisibility.PUBLIC);
       method.setReturnType(new FullyQualifiedJavaType(pageClass));
       method.setName("get" + camel);
       method.addBodyLine("return " + name + ";");
       commentGenerator.addGeneralMethodComment(method, introspectedTable);
       topLevelClass.addMethod(method);
   }

   /**
   * This plugin is always valid - no properties are required
   */
   public boolean validate(List<String> warnings) {
       return true;
   }
}