mybatis逆向工程的增强插件

最新推荐文章于 2024-07-19 16:56:54 发布
最新推荐文章于 2024-07-19 16:56:54 发布
阅读量504 收藏
点赞数
分类专栏： java
版权声明：本文为博主原创文章，遵循 CC 4.0 BY-SA 版权协议，转载请附上原文出处链接和本声明。
本文链接：https://blog.csdn.net/zhenshanren/article/details/93365095
版权

6月22日，主要就是再写一个mybatis逆向工程的插件，原本希望能实现自定义查询字段，后来使用插件发现不太好实现，所以我就用插件实现了一些别的功能，除了分页以外，还有，在实体类中加一个附加属性用于保存可能需要的东西，mapper文件里边加两个自定义的查询方法，并且在接口中也加上这两个对应的方法.要实现自定义查询字段的功能，最好是通过mybatis拦截器，但是由于我对这个不熟悉，以后有空了再研究


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.*;
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;

/**
 * <pre>
 * 张某松 2019年6月22日修改
 * add paginationusing mysql limit.2
 * This class isonly used in ibator code generator.
 *[generatorConfiguration]
 *     [context id="context1"]
 *            [plugintype="com.epublic.xzk.db.mybatis.plugin.PaginationPlugin" /]
 *
 * </pre>
 */
public class PaginationPluginGenerate extends PluginAdapter {
    @Override
    public boolean modelExampleClassGenerated(TopLevelClass topLevelClass ,IntrospectedTable introspectedTable ) {
        // addfield, getter, setter for limit clause
        addLimit( topLevelClass , introspectedTable , "limitStart",FullyQualifiedJavaType.getIntInstance() );
        addLimit( topLevelClass , introspectedTable , "limitEnd",FullyQualifiedJavaType.getIntInstance() );
        addLimit( topLevelClass , introspectedTable , "group" ,FullyQualifiedJavaType.getStringInstance());
        addLimit( topLevelClass , introspectedTable , "fields"  ,FullyQualifiedJavaType.getStringInstance());
        return super .modelExampleClassGenerated( topLevelClass , introspectedTable );
    }

    @Override
    public boolean sqlMapResultMapWithoutBLOBsElementGenerated(XmlElement element, IntrospectedTable introspectedTable) {
        return super.sqlMapResultMapWithoutBLOBsElementGenerated(element, introspectedTable);
    }

    /** 如果在接口中不能添加这些方法，可以通过sql会话模板进行操作
     * @param document
     * @param introspectedTable
     * @return
     */
    @Override
    public boolean sqlMapDocumentGenerated(Document document, IntrospectedTable introspectedTable) {
        System.out.println(document);
        XmlElement selectCustomiz=new XmlElement("select");
        selectCustomiz.addAttribute(new Attribute("id","selectCustomiz"));
//        String tbName=introspectedTable.get
//        String exampleType=introspectedTable.getContext() .getJavaModelGeneratorConfiguration().getTargetPackage() + "." + interfaze.getType().getShortName().replace("Mapper", "Example");
        String exampleType=introspectedTable.getExampleType();//.getBaseRecordType()
        selectCustomiz.addAttribute(new Attribute("parameterType",exampleType));
        selectCustomiz.addAttribute(new Attribute("resultMap","map"));
        selectCustomiz.addElement(new TextElement("select \n") );
        XmlElement choose=new XmlElement("choose");
        XmlElement when=new XmlElement("when");
        when.addAttribute(new Attribute("test","fields !=null"));
        when.addElement(new TextElement("${fields}"));
        XmlElement otherwise=new XmlElement("otherwise");
        otherwise.addElement(new TextElement(" * "));
        choose.addElement(when);
        choose.addElement(otherwise);
        selectCustomiz.addElement(choose);
        selectCustomiz.addElement(new TextElement(" from  "+introspectedTable.getFullyQualifiedTableNameAtRuntime()));
        XmlElement where_Clause = new XmlElement( "if" ); //$NON-NLS-1$
        where_Clause .addAttribute( new Attribute( "test" , "_parameter != null  " ));
        XmlElement include = new XmlElement( "include" );
        include.addAttribute(new Attribute("refid","Example_Where_Clause"));
        where_Clause.addElement(include);
        selectCustomiz.addElement(where_Clause);
        XmlElement group = new XmlElement( "if" ); //$NON-NLS-1$
        group .addAttribute( new Attribute( "test" , "group != null  " )); //$NON-NLS-1$ //$NON-NLS-2$
        group .addElement( new TextElement( "group by ${group}  " ));
        selectCustomiz.addElement(group);
        XmlElement limit = new XmlElement( "if" ); //$NON-NLS-1$
        limit .addAttribute( new Attribute( "test" , "limitStart != null and limitStart >=0" )); //$NON-NLS-1$ //$NON-NLS-2$
        limit .addElement( new TextElement( "limit ${limitStart} , ${limitEnd}" ));
        selectCustomiz.addElement(limit);
        XmlElement selectBySql=new XmlElement("select");
        selectBySql.addAttribute(new Attribute("id","selectBySql"));
        selectBySql.addAttribute(new Attribute("parameterType","string"));
        XmlElement sql  = new XmlElement( "if" ); //$NON-NLS-1$
        sql  .addAttribute( new Attribute( "test" , "sql .trim() .toLowerCase().startsWith('select')  " )); //$NON-NLS-1$ //$NON-NLS-2$
        sql.addElement(new TextElement("${sql }"));//比较危险的方法 可查mysql.user密码
        selectBySql.addElement(sql);
        document.getRootElement().addElement(selectCustomiz);
        document.getRootElement().addElement(selectBySql);
        return super.sqlMapDocumentGenerated(document, introspectedTable);
    }

    @Override
    public boolean clientGenerated(Interface interfaze, TopLevelClass topLevelClass, IntrospectedTable introspectedTable) {
        System.out.println(interfaze);
//        topLevelClass.
        FullyQualifiedJavaType exampleType = new FullyQualifiedJavaType(introspectedTable.getExampleType());
//        FullyQualifiedJavaType exampleType = new FullyQualifiedJavaType(context.getJavaModelGeneratorConfiguration().getTargetPackage() + "." + interfaze.getType().getShortName().replace("Mapper", "Example"));
        if(exampleType==null)
             exampleType = new FullyQualifiedJavaType(context.getJavaModelGeneratorConfiguration().getTargetPackage() + "." + interfaze.getType().getShortName().replace("Mapper", "Example"));
        interfaze.addImportedType(FullyQualifiedJavaType.getNewListInstance());
//        interfaze.addImportedType(new FullyQualifiedJavaType(interfaze.getType()));
        interfaze.addImportedType(exampleType);
        Method selectCustomiz=new Method("selectCustomiz");
        selectCustomiz.setVisibility(JavaVisibility.PUBLIC);
        selectCustomiz.setReturnType(FullyQualifiedJavaType.getNewListInstance());
//        selectCustomiz.addParameter(new Parameter(FullyQualifiedJavaType.getStringInstance()," " ));
//        selectCustomiz.addParameter(new Parameter(FullyQualifiedJavaType.getCriteriaInstance(),"example"));
        selectCustomiz.addParameter(new Parameter(exampleType,"example"));
        Method selectBySql=new Method("selectBySql");
        selectBySql.setVisibility(JavaVisibility.PUBLIC);
        selectBySql.setReturnType(FullyQualifiedJavaType.getNewListInstance());
        selectBySql.addParameter(new Parameter(FullyQualifiedJavaType.getStringInstance(),"sql" ));
//        selectBySql.addParameter(new Parameter(FullyQualifiedJavaType.getCriteriaInstance(),"example"));
        interfaze.addMethod(selectCustomiz);
        interfaze.addMethod(selectBySql);
        return super.clientGenerated(interfaze, topLevelClass, introspectedTable);
    }

    @Override
    public boolean providerGenerated(TopLevelClass topLevelClass, IntrospectedTable introspectedTable) {
        System.out.println(topLevelClass);
        return super.providerGenerated(topLevelClass, introspectedTable);
    }

    @Override
    public boolean sqlMapSelectByExampleWithoutBLOBsElementGenerated(XmlElement element ,
                                                                     IntrospectedTable introspectedTable ) {
        System.out.println(element);
//        element.getElements()
        // LIMIT5,10; // 检索记录行 6-15
        XmlElement isNotNullElement = new XmlElement( "if" ); //$NON-NLS-1$
        isNotNullElement .addAttribute( new Attribute( "test" , "limitStart != null and limitStart >=0" )); //$NON-NLS-1$ //$NON-NLS-2$
        isNotNullElement .addElement( new TextElement( "limit ${limitStart} , ${limitEnd}" ));
        element .addElement( isNotNullElement );
        // LIMIT 5;// 检索前 5 个记录行

        return super .sqlMapUpdateByExampleWithoutBLOBsElementGenerated( element , introspectedTable );
    }

    //添加一个map/object用于放可能的其他属性
    @Override
    public boolean modelBaseRecordClassGenerated(TopLevelClass topLevelClass, IntrospectedTable introspectedTable) {
        Field additional = new Field("additional", FullyQualifiedJavaType.getObjectInstance());
        additional.setVisibility(JavaVisibility.PUBLIC);
        topLevelClass.addField(additional);
        Method get=new Method("getAdditional");
        get.setVisibility(JavaVisibility.PUBLIC);
        get.setReturnType(FullyQualifiedJavaType.getObjectInstance());
        get.addBodyLine("return this.additional;");
        Method set=new Method("setAdditional");
        set.setVisibility(JavaVisibility.PUBLIC);
//        get.setReturnType(FullyQualifiedJavaType.void));
        set.addParameter(new Parameter(FullyQualifiedJavaType.getObjectInstance(),"additional"));
        set.addBodyLine("  this.additional= additional;");
        topLevelClass.addMethod(get);
        topLevelClass.addMethod(set);
        return super.modelBaseRecordClassGenerated(topLevelClass, introspectedTable);
    }

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

        XmlElement isNotNullOrderElement = new XmlElement( "if" ); //$NON-NLS-1$
        isNotNullOrderElement .addAttribute( new Attribute( "test" , "orderByClause != null " )); //$NON-NLS-1$ //$NON-NLS-2$
        isNotNullOrderElement .addElement( new TextElement( "order by ${orderByClause}  " ));
        element.addElement(isNotNullOrderElement);
        XmlElement isNotNullElement = new XmlElement( "if" ); //$NON-NLS-1$
        isNotNullElement .addAttribute( new Attribute( "test" , "limitStart != null and limitStart >=0" )); //$NON-NLS-1$ //$NON-NLS-2$
        isNotNullElement .addElement( new TextElement( "limit ${limitStart}  " ));
        element .addElement( isNotNullElement );
        return super.sqlMapUpdateByExampleSelectiveElementGenerated( element , introspectedTable );
    }
    @Override
    public boolean sqlMapUpdateByExampleWithoutBLOBsElementGenerated(XmlElement element, IntrospectedTable introspectedTable) {

        XmlElement isNotNullOrderElement = new XmlElement( "if" ); //$NON-NLS-1$
        isNotNullOrderElement .addAttribute( new Attribute( "test" , "orderByClause != null " )); //$NON-NLS-1$ //$NON-NLS-2$
        isNotNullOrderElement .addElement( new TextElement( "order by ${orderByClause}  " ));
        element.addElement(isNotNullOrderElement);
        XmlElement isNotNullElement = new XmlElement( "if" ); //$NON-NLS-1$
        isNotNullElement .addAttribute( new Attribute( "test" , "limitStart != null and limitStart >=0" )); //$NON-NLS-1$ //$NON-NLS-2$
        isNotNullElement .addElement( new TextElement( "limit ${limitStart}  " ));
        element .addElement( isNotNullElement );
        return super.sqlMapUpdateByExampleWithoutBLOBsElementGenerated( element , introspectedTable );
    }

    private void addLimit(TopLevelClass topLevelClass , IntrospectedTable introspectedTable , String name ,FullyQualifiedJavaType type){
        CommentGenerator commentGenerator = context .getCommentGenerator();
        Field field = new Field();
        field .setVisibility(JavaVisibility. PROTECTED );
        field .setType(type );
        field .setName( name );
        //默认值
        if(FullyQualifiedJavaType.getIntInstance().equals(type))
            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(type , name ));
        method .addBodyLine( "this." + name + "=" + name + ";" );
        commentGenerator .addGeneralMethodComment( method , introspectedTable );
        topLevelClass .addMethod( method );
        method = new Method();
        method .setVisibility(JavaVisibility. PUBLIC );
        method .setReturnType(type );
        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 ;
    }

    public static void generate() {
        String config = PaginationPluginGenerate. class .getClassLoader().getResource( "generatorConfig.xml" ).getFile();
        String[] arg = { "-configfile" , config , "-overwrite" };
        ShellRunner.main( arg );
    }

    public static void main(String[] args ) {
        com.mysql.jdbc.Driver driver=null;
        generate();
    }
}
zhenshanrenhao
关注
专栏目录
【mybatis笔记-07】MyBatis逆向工程&分页插件
tulingxueyuanXM的博客
05-12 196
1、分页插件 自定义分页插件 分页插件使用 PageHelper 原理 2、mybatis逆向工程 1、分页插件 MyBatis 通过提供插件机制，让我们可以根据自己的需要去增强MyBatis 的功能。需要注意的是，如果没有完全理解MyBatis 的运行原理和插件的工作方式，最好不要使用插件，因为它会改变系底层的工作逻辑，给系统带来很大的影响。 MyBatis 的插件可以在不修改原来的代码的情况下，通过拦截的方式，改变四大核心 对象的行为，比如处理参数，处理SQL，处理结
mybatis逆向工程所需jar包和文档
11-30
MyBatis逆向工程是一种快速构建数据库模型及Mapper接口、XML配置文件的方法，极大地提高了开发效率。本资源包包含了进行MyBatis逆向工程所需的jar包以及相关文档，适用于1.3.5版本。 首先，我们需要理解MyBatis逆向...
MyBatis逆向工程官方插件
10-12
当前文件压缩包为mybatisDAO层框架逆向工程插件安装包，具体使用方法可在博主的博文中检索查询！
mybatis 逆向工程插件
阿沐的博客
12-06 587
编译插件 &lt;build&gt;         &lt;plugins&gt;             &lt;!--编译--&gt;             &lt;plugin&gt;                 &lt;groupId&gt;org.springframework.boot&lt;/groupId&gt;                 &lt;artifact...
IDEA逆向生成插件-模板记录
最新发布
stillfantasy77的博客
07-19 1001
VO专注于视图展示，封装展示层需要的数据。DTO专注于数据的跨层或跨系统传输，减少数据冗余，提高传输效率。DOMAIN专注于业务逻辑的实现，封装业务领域的知识和规则。在实际项目中，应根据具体需求和场景来选择合适的类型，以达到最佳的开发效率和系统性能。我惊喜地发现，将这些基础知识系统地记录下来，不仅是对自我学习的一种巩固，也是一件非常有意义且值得分享的事情。您也完全可以这样做，让我们一起在基础中深耕，发现更多可能。
mybatis---逆向工程（插件）
fly的博客
07-31 1475
文件名必须是：generatorConfig.xml。出现BUILD SUCCESS就是创建成功了。
idea插件之一一逆向工程
余生一个帆的博客
02-29 6068
前言 之前公司使用的正向工程（正向工程：用Java代码生成---->数据库表）在Java代码中写好表字段和表结构去映射到数据库，并生成表结构，但是在使用的过程中发现了一些不足。 所以我就在想，能不能先把数据库建好，然后生成代码呢？带着这个疑问，于是我便趁这个疫情期间学习到了mybatis的逆向工程（逆向工程：数据库---->Java代码，可以针对单表自动生成实体类等代码（mapper....
逆向工程 -- generator插件
寿春
11-06 274
放置Resources文件下 <?xml version="1.0" encoding="UTF-8"?> <!DOCTYPE generatorConfiguration PUBLIC "-//mybatis.org//DTD MyBatis Generator Configuration 1.0//EN" "http://mybatis.or...
mybatis逆向工程插件
01-07
MyBatis逆向工程插件，也被称为MyBatis Generator（MBG），是MyBatis框架的一个强大工具，主要用于自动生成MyBatis的Mapper接口、XML映射文件以及实体类，极大地提高了开发效率。这个插件能够根据数据库中的表结构...
Mybatis逆向工程增强版 generatorSqlmap-increase
12-22
Mybatis逆向工程增强版generatorSqlmap-increase是一款基于传智播客的Mybatis-generator的扩展工具，旨在提供更高效、灵活的代码生成方案。它不仅保留了原版的逆向工程特性，还添加了一些实用的功能，如将Mapper接口...
mybatis逆向工程升级版工具
08-18
传统的MyBatis逆向工程可能需要手动添加分页插件或编写分页逻辑，而升级版工具已经内置了分页支持，可以自动为每个Mapper接口生成分页查询的方法，简化了开发流程。 再者，注释功能的增强使得代码更易于理解和维护...
eclipse的mybatis逆向工程生成插件
01-05
用于mybatis逆向生成实体类,解压后放入dropins文件夹即可
IDEA+Mybatis实现逆向工程
10-13
IDEA+Mybatis实现逆向工程,IDEA+Mybatis实现逆向工程,IDEA+Mybatis实现逆向工程;
MyBatis逆向工程插件
Janson_Lin
06-15 234
pom.xml <dependency> <groupId>org.mybatis.generator</groupId> <artifactId>mybatis-generator-core</artifactId> <version>1.3.2</version> </dependency> <plugin> <groupId
idea插件：逆向工程插件--Easy Code
程序前行者的博客
07-13 1424
easy code是基于idea的插件，他可以根据你的需求，对应你所选择的数据库，直接对数据的表生成entity,controller,service,dao,mapper,无需任何编码，简单而强大。 安装的步骤很简单： 在idea打开:设置–>插件–>Marketplace–>搜索 easy code–>安装–>应用确定。这样子就安装成功了。...
Mybatis 逆向工程 maven 插件
weixin_38408945的博客
04-06 591
maven pom.xml配置 <!-- mybatis逆向工程 --> <dependency> <groupId>org.mybatis.generator</groupId> <artifactId>mybatis-generator-core</artifa...
逆向工程-----Mybatis插件generator
weixin_48314726的博客
07-20 895
- mysql依赖-->-- mybatis逆向工程插件-->-- 插件的依赖-->-- 允许覆盖-->3.在generatorCond.xml中写如下代码。--生成Mapper接口的包名和位置-->--生成sql映射文件的包名和位置-->-- 不生成注释-->-- 连接数据库-->--生成pojo类的包名和位置-->--表名和对应的实体类名-->
eclipse中使用逆向工程的插件安装
飘摇中的风雨
09-23 318
1. Help-->install new software: 2. Work with选择：Indigo - http://download.eclipse.org/releases/Indigo 根据你的版本不同，Indigo可能是Helios，Juno等... 3. 过滤框输入 dali，然后过滤出来的几个勾上，下一步、确定之类的一路下去就行了...
MyBatis逆向工程完全指南
"详解MyBatis逆向工程" MyBatis逆向工程是一种自动化工具，它能够根据数据库中的表结构自动生成对应的MyBatis映射文件（mapper.xml）、Mapper接口（mapper.java）以及实体类（pojo）等代码，极大地减少了开发人员...
评论
添加红包

请填写红包祝福语或标题

个

红包个数最小为10个

元

红包金额最低5元

当前余额3.43元 前往充值 >
需支付：10.00元
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付元
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明：

1.余额是钱包充值的虚拟货币，按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载，可以购买VIP、付费专栏及课程。

余额充值