mybatisplus配置

本文介绍了如何使用Mybatis-Plus的FastAutoGenerator进行数据库表结构的自动化代码生成,包括配置数据源、全局参数、包结构、模板策略和定制化模板。重点讲解了自定义模板引擎及交互式生成过程。
摘要由CSDN通过智能技术生成

package com.baomidou.mybatisplus.samples.mysql.auto;

import com.baomidou.mybatisplus.annotation.FieldFill;
import com.baomidou.mybatisplus.annotation.IdType;
import com.baomidou.mybatisplus.core.mapper.BaseMapper;
import com.baomidou.mybatisplus.generator.FastAutoGenerator;
import com.baomidou.mybatisplus.generator.config.*;
import com.baomidou.mybatisplus.generator.config.converts.MySqlTypeConvert;
import com.baomidou.mybatisplus.generator.config.po.LikeTable;
import com.baomidou.mybatisplus.generator.config.po.TableInfo;
import com.baomidou.mybatisplus.generator.config.querys.MySqlQuery;
import com.baomidou.mybatisplus.generator.config.rules.DateType;
import com.baomidou.mybatisplus.generator.config.rules.NamingStrategy;
import com.baomidou.mybatisplus.generator.engine.FreemarkerTemplateEngine;
import com.baomidou.mybatisplus.generator.fill.Column;
import com.baomidou.mybatisplus.generator.fill.Property;
import com.baomidou.mybatisplus.generator.keywords.MySqlKeyWordsHandler;
import lombok.var;
import org.jetbrains.annotations.NotNull;

import java.io.File;
import java.util.Arrays;
import java.util.Collections;
import java.util.List;
import java.util.Map;

import static javax.management.MBeanServerFactory.builder;

public class AutoCode {

    private static final DataSourceConfig.Builder DATA_SOURCE_CONFIG = null;

    public static void main(String[] args) {
        // 数据库连接配置
        DataSourceConfig root = new DataSourceConfig.Builder("jdbc:mysql://127.0.0.1:3306/mybatis-plus", "root", "123456")
                .dbQuery(new MySqlQuery()) // 数据库查询
                .schema("mybatis-plus") // 数据库schema(部分数据库适用)
                .typeConvert(new MySqlTypeConvert()) // 数据库类型转换器
                .keyWordsHandler(new MySqlKeyWordsHandler()) // 数据库关键字处理器
                .build();
        // 全局配置
        new GlobalConfig.Builder()
                .fileOverride()    // 废弃,覆盖已生成文件
                .outputDir("/opt/baomidou") // 制定输出目录
                .author("baomidou") // 作者名称
                .enableKotlin() //开启kotlin模式
                .enableSwagger()// 开启swagger模式
                .dateType(DateType.TIME_PACK)//时间策略
                .commentDate("yyyy-MM-dd")// 注释日期
                .disableOpenDir() // 禁止打开输出目录
                .build();

        // 包配置
        new PackageConfig.Builder()
                .parent("com.baomidou.mybatisplus.samples.generator") // 父包名
                .moduleName("sys") // 父包模块
                .entity("po")// entity 包
                .service("service")// service包名
                .serviceImpl("service.impl")// serviceImpl包名
                .mapper("mapper") // mapper包名
                .xml("mapper.xml")// mapper xml 包名
                .controller("controller")// controller 包名
                .other("other")  // 自定义文件包名
                .pathInfo(Collections.singletonMap(OutputFile.mapperXml, "D://")) // 路径配置信息
                .build();

        // 模板配置
        new TemplateConfig.Builder()
                .disable(TemplateType.ENTITY)                                   //设置禁用entity模板
                .entity("/templates/entity.java")//设置实体模板路径
                .service("/templates/service.java")//设置service模板路径
                .serviceImpl("/templates/serviceImpl.java")//设置serviceImpl模板路径
                .mapper("/templates/mapper.java")//设置mapper模板路径
                .mapperXml("/templates/mapper.xml")//设置mapperxml模板路径
                .controller("/templates/controller.java")//设置controller模板路径
                .build();//设置

        // 注入配置
        new InjectionConfig.Builder()
                .beforeOutputFile((tableInfo, objectMap) -> {
                    System.out.println("tableInfo: " + tableInfo.getEntityName() + " objectMap: " + objectMap.size());
                })
                .customMap(Collections.singletonMap("test", "baomidou"))
                .customFile(Collections.singletonMap("test.txt", "/templates/test.vm"))
                .build();
        // 策略配置(StrategyConfig)
        new StrategyConfig.Builder()
                .enableCapitalMode() // 开启大写命名
                .enableSkipView()   // 开启跳过视图
                .disableSqlFilter()
                .likeTable(new LikeTable("USER"))
                .addInclude("t_simple")
                .addTablePrefix("t_", "c_")
                .addFieldSuffix("_flag")
                .build();
        // 1. entity 配置
        new StrategyConfig.Builder()
                .entityBuilder()
                .superClass(BaseEntity.class)
                .disableSerialVersionUID()
                .enableChainModel()
                .enableLombok()
                .enableRemoveIsPrefix()
                .enableTableFieldAnnotation()
                .enableActiveRecord()
                .versionColumnName("version")
                .versionPropertyName("version")
                .logicDeleteColumnName("deleted")
                .logicDeletePropertyName("deleteFlag")
                .naming(NamingStrategy.no_change)
                .columnNaming(NamingStrategy.underline_to_camel)
                .addSuperEntityColumns("id", "created_by", "created_time", "updated_by", "updated_time")
                .addIgnoreColumns("age")
                .addTableFills(new Column("create_time", FieldFill.INSERT))
                .addTableFills(new Property("updateTime", FieldFill.INSERT_UPDATE))
                .idType(IdType.AUTO)
                .formatFileName("%sEntity")
                .build();
        // controller 策略配置
        new StrategyConfig.Builder()
                .controllerBuilder()
                .superClass(BaseController.class)
                .enableHyphenStyle()
                .enableRestStyle()
                .formatFileName("%sAction")
                .build();
        // service
        new StrategyConfig.Builder()
                .serviceBuilder()
                .superServiceClass(BaseService.class)
                .superServiceImplClass(BaseServiceImpl.class)
                .formatServiceFileName("%sService")
                .formatServiceImplFileName("%sServiceImp")
                .build();
        // mapper
        new StrategyConfig.Builder()
                .mapperBuilder()
                .superClass(BaseMapper.class)
                .enableMapperAnnotation()
                .enableBaseResultMap()
                .enableBaseColumnList()
                .cache(MyMapperCache.class)
                .formatMapperFileName("%sDao")
                .formatXmlFileName("%sXml")
                .build();

        // 快速生成
        FastAutoGenerator.create("url", "username", "password")
                // 全局配置
                .globalConfig(builder -> {
                    builder.author("超级无敌霸王龙007号") // 设置作者
                            .fileOverride() // 覆盖已生成文件
                            .outputDir("D://") // 指定输出目录
                            .enableKotlin() //开启kotlin模式
                            .enableSwagger()// 开启swagger模式
                            .dateType(DateType.TIME_PACK)//时间策略
                            .commentDate("yyyy-MM-dd")// 注释日期
                            .disableOpenDir() // 禁止打开输出目录
                            .build();
                })
                .packageConfig(builder -> {
                    builder.parent("com.baomidou.mybatisplus.samples.generator") // 设置父包名
                            .moduleName("system") // 设置父包模块名
                            .parent("com.baomidou.mybatisplus.samples.generator") // 父包名
                            .moduleName("sys") // 父包模块
                            .entity("pojo")// entity 包
                            .service("service")// service包名
                            .serviceImpl("service.impl")// serviceImpl包名
                            .mapper("mapper") // mapper包名
                            .xml("mapper.xml")// mapper xml 包名
                            .controller("controller")// controller 包名
                            .other("other")  // 自定义文件包名
                            .pathInfo(Collections.singletonMap(OutputFile.xml, "D://")); // 设置mapperXml生成路径
                })
                .strategyConfig(builder -> {
                    builder.addInclude("t_simple") // 设置需要生成的表名
                            .addTablePrefix("t_", "c_")// 设置过滤表前缀
                            .entityBuilder()
                            .disableSerialVersionUID()
                            .enableChainModel()
                            .enableLombok() // 自动开启lombok
                            .enableRemoveIsPrefix()
                            .enableTableFieldAnnotation()
                            .enableActiveRecord()
                            .versionColumnName("version")
                            .versionPropertyName("version") // 乐观锁
                            .logicDeleteColumnName("deleted") // 逻辑删除
                            .logicDeletePropertyName("deleteFlag")
                            .naming(NamingStrategy.no_change)
                            .columnNaming(NamingStrategy.underline_to_camel)
                            .addSuperEntityColumns("id", "created_by", "created_time", "updated_by", "updated_time")
                            .addIgnoreColumns("age")
                            .addTableFills(new Column("create_time", FieldFill.INSERT))  // 自动填充日期
                            .addTableFills(new Property("updateTime", FieldFill.INSERT_UPDATE))
                            .idType(IdType.AUTO)
                            .formatFileName("%sEntity")
                            .build();
                })
                .templateEngine(new FreemarkerTemplateEngine()) // 使用Freemarker引擎模板,默认的是Velocity引擎模板
                .execute();


        // 交互式生成
        FastAutoGenerator.create(DATA_SOURCE_CONFIG)
                // 全局配置
                .globalConfig((scanner, builder) -> builder.author(scanner.apply("请输入作者名称?")).fileOverride())
                // 包配置
                .packageConfig((scanner, builder) -> builder.parent(scanner.apply("请输入包名?")))
                // 策略配置
                .strategyConfig((scanner, builder) -> builder.addInclude(getTables(scanner.apply("请输入表名,多个英文逗号分隔?所有输入 all")))
                        .controllerBuilder().enableRestStyle().enableHyphenStyle()
                        .entityBuilder().enableLombok().addTableFills(
                                new Column("create_time", FieldFill.INSERT)
                        ).build())
                /*
                    模板引擎配置,默认 Velocity 可选模板引擎 Beetl 或 Freemarker
                   .templateEngine(new BeetlTemplateEngine())
                   .templateEngine(new FreemarkerTemplateEngine())
                 */
                .execute();
    }
    /**
     * 代码生成器支持自定义[DTO\VO等]模版
     */
    public final class EnhanceFreemarkerTemplateEngine extends FreemarkerTemplateEngine {

        @Override
        protected void outputCustomFile(@NotNull Map<String, String> customFile, @NotNull TableInfo tableInfo, @NotNull Map<String, Object> objectMap) {
            String entityName = tableInfo.getEntityName();
            String otherPath = this.getPathInfo(OutputFile.other);
            customFile.forEach((key, value) -> {
                String fileName = String.format(otherPath + File.separator + entityName + "%s", key);
                this.outputFile(new File(fileName), objectMap, value, this.getConfigBuilder().getInjectionConfig().isFileOverride());
            });
        }
    }

    // 处理 all 情况
    protected static List<String> getTables(String tables) {
        return "all".equals(tables) ? Collections.emptyList() : Arrays.asList(tables.split(","));
    }


}

application.properties

spring:
  datasource:
    driver-class-name: com.mysql.cj.jdbc.Driver
    url: jdbc:mysql://localhost:3306/mybatis_plus?useSSL=false&useUnicode=true&characterEncoding=UTF-8
    username: root
    password: xxxx

# Mybatis Mate 配置
mybatis-mate:
  cert:
    grant: thisIsTestLicense
    license: as/bsBaSVrsA9FfjC/N77ruEt2/QZDrW+MHETNuEuZBra5mlaXZU+DE1ZvF8UjzlLCpH3TFVH3WPV+Ya7Ugiz1Rx4wSh/FK6Ug9lhos7rnsNaRB/+mR30aXqtlLt4dAmLAOCT56r9mikW+t1DDJY8TVhERWMjEipbqGO9oe1fqYCegCEX8tVCpToKr5J1g1V86mNsNnEGXujnLlEw9jBTrGxAyQroD7Ns1Dhwz1K4Y188mvmRQp9t7OYrpgsC7N9CXq1s1c2GtvfItHArkqHE4oDrhaPjpbMjFWLI5/XqZDtW3D+AVcH7pTcYZn6vzFfDZEmfDFV5fQlT3Rc+GENEg==

# Logger Config
logging:
  level:
    com.baomidou.mybatisplus.samples.mysql: debug

pom.xml

 <dependencies>
        <dependency>
            <groupId>javax.servlet</groupId>
            <artifactId>javax.servlet-api</artifactId>
            <version>3.1.0</version>
        </dependency>
        <dependency>
            <groupId>com.baomidou</groupId>
            <artifactId>mybatis-plus-generator</artifactId>
            <version>3.5.2</version>
        </dependency>
        <dependency>
            <groupId>org.springframework.boot</groupId>
            <artifactId>spring-boot-starter</artifactId>
        </dependency>
        <dependency>
            <groupId>mysql</groupId>
            <artifactId>mysql-connector-java</artifactId>
            <scope>runtime</scope>
        </dependency>
        <dependency>
            <groupId>com.baomidou</groupId>
            <artifactId>mybatis-plus-boot-starter</artifactId>
        </dependency>
        <dependency>
            <groupId>com.baomidou</groupId>
            <artifactId>mybatis-mate-starter</artifactId>
            <version>1.1.0</version>
        </dependency>
    </dependencies>

  • 1
    点赞
  • 0
    收藏
    觉得还不错? 一键收藏
  • 0
    评论
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

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

抵扣说明:

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

余额充值