MyBatisPlus 自动生成代码

2022-07-12 14:37:24 浏览数 (2)

本篇主要介绍如何使用MybatisPlus自带的generator模块来生成entity、service、serviceImpl、mapper、mapperXml等,以此来提高开发效率。

1、引入相关依赖

代码语言:javascript复制
<!--   数据库驱动依赖 如果你是mysql数据库就需要替换为mysql驱动        -->
<dependency>
    <groupId>org.postgresql</groupId>
    <artifactId>postgresql</artifactId>
    <version>42.3.3</version>
</dependency>
<!--   freemarker驱动依赖         -->
<dependency>
    <groupId>org.freemarker</groupId>
    <artifactId>freemarker</artifactId>
    <version>2.3.31</version>
</dependency>
<!--   mp驱动依赖         -->
<dependency>
    <groupId>com.baomidou</groupId>
    <artifactId>mybatis-plus-boot-starter</artifactId>
    <version>3.5.1</version>
</dependency>
<!--   mp generator驱动依赖         -->
<dependency>
    <groupId>com.baomidou</groupId>
    <artifactId>mybatis-plus-generator</artifactId>
    <version>3.5.1</version>
</dependency>

2、写一个java类,增加一个main方法

代码语言:javascript复制
package com.mybatisplus;

import com.baomidou.mybatisplus.annotation.FieldFill;
import com.baomidou.mybatisplus.annotation.IdType;
import com.baomidou.mybatisplus.generator.FastAutoGenerator;
import com.baomidou.mybatisplus.generator.config.OutputFile;
import com.baomidou.mybatisplus.generator.config.po.LikeTable;
import com.baomidou.mybatisplus.generator.config.rules.NamingStrategy;
import com.baomidou.mybatisplus.generator.engine.FreemarkerTemplateEngine;
import com.baomidou.mybatisplus.generator.fill.Column;

import java.util.Collections;

/**
 * @author lee
 * @version 1.0
 * @description 描述
 * @date 2022/06/13
 */
public class MPGenerateCode {

    public static void main(String[] args) {
        FastAutoGenerator.create("jdbc:postgresql://localhost:3306/dbname", "name", "password")
                .globalConfig(builder -> {
                    builder.author("lee") // 设置作者
                            .fileOverride() // 覆盖已生成文件
                            .outputDir("/Users/yongyongli/work"); // 指定输出目录
                })
                .packageConfig(builder -> {
                    builder.parent("com.xxx.xxx.xx.xx") // 设置父包名
                            .moduleName("dao") // 设置父包模块名
                            .entity("model")
                            .service("repository")
                            .serviceImpl("repository.impl")
                            .mapper("mapper")
                            .xml("mapper.xml")
                            .pathInfo(Collections.singletonMap(OutputFile.mapperXml, "/Users/yongyongli/work/generator")); // 设置mapperXml生成路径
                })
                .templateEngine(new FreemarkerTemplateEngine()) // 使用Freemarker引擎模板,默认的是Velocity引擎模板

                .strategyConfig(builder->{
                    builder.likeTable(new LikeTable("msg_"));// 匹配以msg开头的表
//                            .addInclude("t_simple") // 设置需要生成的表名
//                            .addTablePrefix("t_", "c_"); // 设置过滤表前缀
                    builder.entityBuilder()
//                            .superClass(BaseEntity.class) // 设置父类
                            .disableSerialVersionUID()
//                            .enableChainModel() // 开启驼峰转连字符
                            .enableLombok()
                            .enableRemoveIsPrefix() //开启 Boolean 类型字段移除 is 前缀
                            .enableTableFieldAnnotation()// 开启生成实体时生成字段注解
//                            .versionColumnName("version")// 乐观锁字段名(数据库)
//                            .versionPropertyName("version")// 乐观锁属性名(实体)
                            .logicDeleteColumnName("delete_flag")// 逻辑删除字段名(数据库)
                            .logicDeletePropertyName("deleteFlag")// 逻辑删除属性名(实体)
                            .naming(NamingStrategy.underline_to_camel)// 数据库表映射到实体的命名策略
                            .columnNaming(NamingStrategy.underline_to_camel)// 数据库表字段映射到实体的命名策略
                            .addTableFills(new Column("create_time", FieldFill.INSERT),// 这些都是通用字段
                                    new Column("creator", FieldFill.INSERT),
                                    new Column("delete_flag", FieldFill.INSERT),
                                    new Column("modify_time", FieldFill.INSERT_UPDATE),
                                    new Column("modifier", FieldFill.INSERT_UPDATE)
                            )
                            .idType(IdType.AUTO)
                            .formatFileName("%sDO")
                            .build();
                    builder.serviceBuilder().formatServiceFileName("%sDao")
                            .formatServiceImplFileName("%sDaoImpl")
                            .build();
                    builder.mapperBuilder().formatMapperFileName("%sMapper")
                            .formatXmlFileName("%sXml")
                            .build();
                })
                .execute();
    }
}

欢迎大家尝试使用MP官方推荐的方式来提效。

0 人点赞