改mybatis到mybatis-plus的擎天巨坑

Posted

tags:

篇首语:本文由小常识网(cha138.com)小编为大家整理,主要介绍了改mybatis到mybatis-plus的擎天巨坑相关的知识,希望对你有一定的参考价值。

参考技术A

兄弟们!mybatis-plus是号称对mybatis本身无侵入的一种加强版。
但是当你引入它的时候,可能会广泛踩坑。接下来给大家看看我的踩坑历史。

第一坑:当你选择的是 mybatis-spring-boot-starter + mybatis-plus 引入时
pom依赖如下:

这样。其实,你的mybatis还是你曾经的mybatis,只是附加了一些功能。但是,当你开始配置yml文件的时候,诸多小白(如我),可能会直接将一些属性配置直接替换到mybatis-plus之下。
如:

血泪的教训!!!
其实,yml已经很好心地提醒了你,比如打 mybatis-plus 之后没有属性提示信息,属性高亮黄色等等,但是你的配置很可能就是直接CL复制过来的,所以完全不会注意到。

那么此时请记住,现在你的 mybatis-plus 是没有 mybatis 基础文件的大哥们!!!你们是两个包!!!你不是直接引入的 mybatis-plus-boot-starter 的!!!
因此,该配置在mybatis下的,一个都不能少。比如,驼峰映射。
如果你像我这么配置,明明打开了驼峰映射 map-underscore-to-camel-case: true 但是鉴于它不是有效的配置,所以它是不会生效的。
当你纠结为啥你获取到的参数都是null的时候,花费了N个小时之后,终于觉得可能是驼峰映射的问题的时候,一个断点打进来。
哟嚯,发现了!这个配置是关闭的哦!

所以需要在 mybatis 下保留其基础配置。
如下:

但这样就没法直接在yml中操作 mybatis-plus 了。

第二坑:当你觉得附和两个包太过麻烦,而且无法配置mybatis-plus的属性打算直接引入 mybatis-plus-boot-starter 时
请记住一定要排除掉 mybatis-spring-boot-starter 这个包,不然很可能会有jar包版本冲突。
mybatis造成的启动时版本冲突bug示例
因此,我们直接这样导入包即可

然后配置文件就可以全部转到 mybatis-plus 下
如下:

其实yml真的已经尽力了!如果你真的已经引入了starter包,它的提示就是会生效的配置!

mybatis-plus自controller开始一键生成CURD代码

效果展示

以生成employee表对应的CURD为例

  • 生成的文件
  • controller里面长这样

使用方式

  • 第一步:引入下方的给出的相关依赖
  • 第二步:将下方的单元测试类直接贴进去
  • 第三步:运行单元测试即可
    注:本人测试用的是junit5,如果用的是junit4的话,简单改一下即可

相关依赖

<!-- mybatis-plus -->
<dependency>
    <groupId>com.baomidou</groupId>
    <artifactId>mybatis-plus-boot-starter</artifactId>
    <version>3.5.2</version>
</dependency>

<!-- 代码生成器相关 -->
<dependency>
    <groupId>com.baomidou</groupId>
    <artifactId>mybatis-plus-generator</artifactId>
    <version>3.5.2</version>
    <scope>test</scope>
</dependency>

<dependency>
    <groupId>org.freemarker</groupId>
    <artifactId>freemarker</artifactId>
    <version>2.3.30</version>
    <scope>test</scope>
</dependency>

<!--  IOUtil、PathUtil工具类包 -->
<dependency>
    <groupId>com.idea-aedi</groupId>
    <artifactId>common-ds</artifactId>
    <version>2100.5.2</version>
</dependency>

代码生成器单元测试类

import com.baomidou.mybatisplus.core.toolkit.StringUtils;
import com.baomidou.mybatisplus.generator.FastAutoGenerator;
import com.baomidou.mybatisplus.generator.config.ConstVal;
import com.baomidou.mybatisplus.generator.config.DataSourceConfig;
import com.baomidou.mybatisplus.generator.config.OutputFile;
import com.baomidou.mybatisplus.generator.config.TemplateConfig;
import com.baomidou.mybatisplus.generator.config.builder.Entity;
import com.baomidou.mybatisplus.generator.config.converts.MySqlTypeConvert;
import com.baomidou.mybatisplus.generator.config.po.TableField;
import com.baomidou.mybatisplus.generator.config.po.TableInfo;
import com.baomidou.mybatisplus.generator.config.rules.DbColumnType;
import com.baomidou.mybatisplus.generator.engine.FreemarkerTemplateEngine;
import com.ideaaedi.commonds.io.IOUtil;
import com.ideaaedi.commonds.path.PathUtil;
import lombok.Data;
import lombok.Getter;
import lombok.Setter;
import lombok.ToString;
import org.junit.jupiter.api.Test;
import org.springframework.beans.factory.annotation.Value;
import org.springframework.boot.test.context.SpringBootTest;
import org.springframework.lang.NonNull;
import org.springframework.util.CollectionUtils;

import java.io.File;
import java.io.Serializable;
import java.lang.reflect.Field;
import java.util.ArrayList;
import java.util.Arrays;
import java.util.Collections;
import java.util.LinkedHashSet;
import java.util.List;
import java.util.Map;
import java.util.Set;
import java.util.TreeSet;
import java.util.stream.Collectors;

/**
 * mybatis-plus代码生成 &emsp;
 * <a href="https://baomidou.com/pages/779a6e/#%E4%BD%BF%E7%94%A8">代码生成官网</a>
 * &emsp;
 * <a href="https://baomidou.com/pages/981406/#%E5%9F%BA%E7%A1%80%E9%85%8D%E7%BD%AE">配置说明官网</a>
 *
 * @author JustryDeng
 * @since 2022/6/28 9:26
 */
@SpringBootTest
public class MybatisPlusGeneratorHelper 
    
    private static final String templateDir = "mybatis-plus-templates";
    
    @Value("$spring.datasource.url")
    private String url;
    
    @Value("$spring.datasource.username")
    private String username;
    
    @Value("$spring.datasource.password")
    private String password;
    
    /**
     * true-使用自定义模板; false-使用默认模板
     */
    private static final boolean useCustomTemplate = true;
    
    /*
     * 自动生成模板文件
     */
    static 
        if (useCustomTemplate) 
            final String customTemplatesDirRePath = "/src/test/resources/" + templateDir + "/";
            String customTemplatesDirPath = PathUtil.getProjectRootDir(MybatisPlusGeneratorHelper.class)
                    .replace("/target/test-classes/", customTemplatesDirRePath);
            File customTemplatesDir = new File(customTemplatesDirPath);
            if (!customTemplatesDir.exists()) 
                System.err.println(MybatisPlusGeneratorHelper.class.getSimpleName() + ":自定义的模板文件不存在, 自动创建开始. 创建至:" + customTemplatesDirRePath);
                // TODO 将这三个类拷贝到自己的项目下,然后将这里的BaseDTO.class、BasePageDTO.class、 PageInfo.class换成自己的即可
                final String baseDTOName = YourBaseClass.BaseDTO.class.getName().replace("$", ".");
                final String basePageDTOName = YourBaseClass.BasePageDTO.class.getName().replace("$", ".");
                final String PageInfoName = YourBaseClass.PageInfo.class.getName().replace("$", ".");
                for (JustryDengTemplates templateInfo : JustryDengTemplates.values()) 
                    IOUtil.writeContentToFile(
                            templateInfo.getContent()
                                    .replace("JustryDeng_Placeholder.BaseDTO", baseDTOName)
                                    .replace("JustryDeng_Placeholder.BasePageDTO", basePageDTOName)
                                    .replace("JustryDeng_Placeholder.PageInfo", PageInfoName),
                            new File(customTemplatesDir, templateInfo.name));
                
                System.err.println(MybatisPlusGeneratorHelper.class.getSimpleName() + ":自定义的模板文件创建完成. 请重新运行单元测试以生成相关文件.");
                System.exit(-1);
            
        
    
    
    @Test
    void generate() 
        String outputRootDir = PathUtil.getProjectRootDir(MybatisPlusGeneratorHelper.class)
                .replace("/target/test-classes/", "/src/test/java/");
        String packagePath = MybatisPlusGeneratorHelper.class.getPackage().getName() + ".generator";
        // 先清空./generator/目录
        IOUtil.delete(new File(outputRootDir + packagePath.replace(".", "/")));
        
        
        // basic settings
        final FastAutoGenerator fastAutoGenerator = FastAutoGenerator
                .create(
                        new DataSourceConfig
                                .Builder(url, username, password)
                        ///.typeConvert((globalConfig, fieldType) -> 
                        ///    String columeType = fieldType.toLowerCase();
                        ///    if (columeType.contains("tinyint")) 
                        ///        return DbColumnType.INTEGER;
                        ///    
                        ///    // 其它字段采用默认转换
                        ///    return new MySqlTypeConvert().processTypeConvert(globalConfig, fieldType);
                        ///)
                )
                .globalConfig(builder -> builder
                                // 设置作者
                                .author("mybatis-plus generator")
                                .commentDate("yyyy-MM-dd HH:mm:ss")
                                // 指定输出目录
                                .outputDir(outputRootDir)
                        // 开启swagger模式
                        //.enableSwagger()
                )
                // 包配置
                .packageConfig(builder -> builder
                        // 指定生成的类的所属包
                        .parent(packagePath)
                        .entity("entity.po")
                        .service("service")
                        .serviceImpl("service.impl")
                        .mapper("mapper")
                        .xml("mapper.xml")
                        .controller("controller")
                        .other("other"))
                // 策略配置
                .strategyConfig(builder -> builder
                        // => 表配置
                        // 设置需要生成代码的表名(不设置则全库生成) TODO
                        ///                       .addInclude(
                        ///                                "user_face_rec_info"
                        ///                        )
                        // 设置过滤表前缀
                        // .addTablePrefix("tmp_")
                        // 设置过滤表后缀
                        // .addTableSuffix("_tmp")
                        
                        // => entity配置
                        .entityBuilder()
                        .convertFileName((entityName) -> entityName + "PO")
                        /// 给po设置公共父类
                        ///.superClass(PoSupperEntity.class)
                        /// 设置生成的PO里面的对应字段有填充标识或者逻辑删除标识
                        /// // 逻辑删除字段名
                        /// .logicDeletePropertyName("del")
                        /// // 自动填充
                        /// .addTableFills(new Property("createdBy", FieldFill.INSERT))
                        /// .addTableFills(new Property("createdAt", FieldFill.INSERT))
                        /// .addTableFills(new Property("updatedBy", FieldFill.UPDATE))
                        /// .addTableFills(new Property("updatedAt", FieldFill.UPDATE))
                        // 启用TableField等相关注解
                        .enableTableFieldAnnotation()
                        .enableLombok()
                        
                        // => controller配置
                        .controllerBuilder()
                        .enableRestStyle()
                        .convertFileName((entityName) -> entityName + "Controller")
                        // .superClass(BaseController.class)
                        
                        // => service配置
                        .serviceBuilder()
                        .convertServiceFileName((entityName) -> entityName + "Service")
                        .convertServiceImplFileName((entityName) -> entityName + "ServiceImpl")
                        
                        // => mapper配置
                        .mapperBuilder()
                        .enableMapperAnnotation()
                        .convertMapperFileName((entityName) -> entityName + "Mapper")
                        .convertXmlFileName((entityName) -> entityName + "Mapper"))
                .templateEngine(useCustomTemplate ? new EnhanceFreemarkerTemplateEngine() : new FreemarkerTemplateEngine());
        
        // entity、service、serviceImpl、controller、mapper、xml,使用指定模板生成
        fastAutoGenerator.templateConfig((TemplateConfig.Builder builder) -> 
            if (useCustomTemplate) 
                /*
                 * 使用自定义的模板
                 * 注:参考mybatis-plus-generator包下的默认模板进行编写即可
                 * 注:自定义的模板放在classpath下的对应位置即可
                 * 注:指定模板时,无需带对应的模板后缀名(如:这里不带.ftl)
                 */
                builder
                        .entity("/" + templateDir + "/" + "custom_entity")
                        .service("/" + templateDir + "/" + "custom_service")
                        .serviceImpl("/" + templateDir + "/" + "custom_serviceImpl")
                        .controller("/" + templateDir + "/" + "custom_controller")
                        // 不改的话,用默认的即可
                        .mapper(ConstVal.TEMPLATE_MAPPER)
                        .xml(ConstVal.TEMPLATE_XML);
             else 
                // 使用mybatis-plus-generator包下的默认模板
                builder
                        .entity(ConstVal.TEMPLATE_ENTITY_JAVA)
                        .mapper(ConstVal.TEMPLATE_MAPPER)
                        .xml(ConstVal.TEMPLATE_XML)
                        .service(ConstVal.TEMPLATE_SERVICE)
                        .serviceImpl(ConstVal.TEMPLATE_SERVICE_IMPL)
                        .controller(ConstVal.TEMPLATE_CONTROLLER);
            
        );
        
        fastAutoGenerator.execute();
    
    
    
    /**
     * 您的项目中需要这三种类<br />
     */
    public static class YourBaseClass 
        /**
         * 基础模型
         *
         * @author JustryDeng
         * @since 2022/6/24 10:00
         */
        @Setter
        @Getter
        @ToString
        public static class BaseDTO implements Serializable 
            
            private static final long serialVersionUID = -1;
        
        
        /**
         * 基础模型
         *
         * @author JustryDeng
         * @since 2022/6/24 10:00
         */
        @Setter
        @Getter
        @ToString
        @SuppressWarnings("AlibabaPojoNoDefaultValue", "AlibabaPojoMustUsePrimitiveField")
        public static class BasePageDTO extends BaseDTO 
            
            /** 页码 */
            private int pageNum = 1;
            
            /** 每页条数 */
            private int pageSize = 10;
        
        
        /**
         * 分页信息模型
         *
         * @author kuoyi
         * @since 1.0.0
         */
        @Data
        
        public static class PageInfo<T> 
            
            /** 总条数 */
            private long total;
            
            /** 页码 */
            private int pageNum;
            
            /** 每页条数 */
            private int pageSize;
            
            /** 数据集 */
            private List<T> dataList;
            
            /**
             * 快速构造
             */
            @SuppressWarnings("unused")
            public static <T> PageInfo<T> of(long total, int pageNum, int pageSize) 
                return PageInfo.of(total, pageNum, pageSize, Collections.emptyList());
            
            
            /**
             * 快速构造
             */
            public static <T> PageInfo<T> of(long total, int pageNum, int pageSize, List<T> dataList) 
                PageInfo<T> pageInfo = new PageInfo<>();
                pageInfo.setTotal(total);
                pageInfo.setPageNum(pageNum);
                pageInfo.setPageSize(pageSize);
                pageInfo.setDataList(dataList);
                return pageInfo;
            
        
        
    
    
    /* --------------------------------------------------------------- 以下部分,可以不关心 --------------------------------------------------------------- */
    
    
    /**
     * 扩展FreemarkerTemplateEngine,生成更多相关文件
     */
    public static class EnhanceFreemarkerTemplateEngine extends FreemarkerTemplateEngine 
        
        /**
         * (non-javadoc)
         *
         * @param customFile
         *         自定义文件; key-生成的文件; value-生成该文件使用的(classpath下的带模板文件后缀名的)模板文件
         * @param tableInfo
         *         表信息
         * @param objectMap
         *         模板可使用的占位符即对应变量值(即:objectMap里的key可以在模板文件中直接引用)
 

以上是关于改mybatis到mybatis-plus的擎天巨坑的主要内容,如果未能解决你的问题,请参考以下文章

mybatis-plus自controller开始一键生成CURD代码

mybatis-plus自controller开始一键生成CURD代码

检索20万颗星星只需10秒!这款AI产品能“擎天”

#yyds干货盘点#MyBatis-plus 从入门到入土 !

mybatis-plus 代码生成

Mybatis-Plus:了解Mybatis-Plus快速开始(Mybatis + Mybatis-Plus,Mybatis-Plus自动做了属性映射)