配置mybatis-plus-generator自动生成实体相关代码

最新推荐文章于 2024-04-23 10:00:00 发布

寅灯

最新推荐文章于 2024-04-23 10:00:00 发布

阅读量1.3k

点赞数

分类专栏： Mybatis 文章标签： mybatis intellij-idea spring boot

本文链接：https://blog.csdn.net/nandao158/article/details/127074079

版权

Mybatis 专栏收录该内容

15 篇文章 3 订阅

订阅专栏

配置mybatis-plus-generator自动生成实体相关代码今天分享，mybatis-plus是常用的组件，使用方便，而且可以有自动生成代码的配置，使用方便：

1、pom相关jar引入

<dependency>
            <groupId>com.baomidou</groupId>
            <artifactId>mybatis-plus-generator</artifactId>
            <version>3.4.1</version>
        </dependency>

        <dependency>
            <groupId>mysql</groupId>
            <artifactId>mysql-connector-java</artifactId>
            <version>8.0.28</version>
        </dependency>

        <dependency>
            <groupId>org.freemarker</groupId>
            <artifactId>freemarker</artifactId>
            <version>2.3.28</version>
            <scope>compile</scope>
        </dependency>

        <dependency>
            <groupId>org.apache.velocity</groupId>
            <artifactId>velocity</artifactId>
            <version>1.7</version>
        </dependency>

2、核心工具伪代码

package com.example.demo1.util;

import com.baomidou.mybatisplus.generator.AutoGenerator;
import com.baomidou.mybatisplus.generator.InjectionConfig;
import com.baomidou.mybatisplus.generator.config.*;
import com.baomidou.mybatisplus.generator.config.po.TableInfo;
import com.baomidou.mybatisplus.generator.config.rules.NamingStrategy;

import java.util.ArrayList;
import java.util.HashMap;
import java.util.List;
import java.util.Map;

/**
 * @author wanghuainan
 *  1.注释调的代码自己根据需求参考使用
 *  2.tableNames 添加要生成的表明
 *  3.将repository包下的生成实现中的注解@Service 换成@Repository 方便层级清晰
 */
public class CommonMyBatisPlusGenerator {

    public static void main(String[] args) {
        AutoGenerator autoGenerator = new AutoGenerator();
        //1、全局配置
        GlobalConfig gc = new GlobalConfig();
        gc.setFileOverride(true);
        String projectPath = System.getProperty("user.dir");
        String filePath = projectPath + "/src/main/java";
        String filePathRe = projectPath + "/src/main/resources/";
        gc.setOutputDir(filePath);  //生成路径(一般都是生成在此项目的src/main/java下面)
        gc.setAuthor("nandao@.com"); //设置作者
        gc.setOpen(false);

        // 自定义文件命名，注意 %s 会自动填充表实体属性！
        gc.setFileOverride(true); //第二次生成会把第一次生成的覆盖掉
        gc.setBaseResultMap(true); //生成resultMap
        gc.setControllerName("%sController");
        gc.setServiceName("I"+"%sRepository");
        gc.setServiceImplName("%sRepImpl");
        gc.setMapperName("I"+"%sMapper");
        gc.setXmlName("%sMapper");
        //    gc.setSwagger2(true);
        gc.setEntityName("%sEntity");
        autoGenerator.setGlobalConfig(gc);
        //2、数据源配置
        DataSourceConfig dsc = new DataSourceConfig();
        dsc.setUrl("jdbc:mysql://18.1.0.9:3360/user_test?useUnicode=true&serverTimezone=GMT&useSSL=false" +
                "&characterEncoding=utf8");
        dsc.setDriverName("com.mysql.cj.jdbc.Driver");
        dsc.setUsername("root");
        dsc.setPassword("root");
        autoGenerator.setDataSource(dsc);

        // 3、包配置（包名配置）
        PackageConfig pc = new PackageConfig();
        pc.setParent("com.example.demo1");
        pc.setService("repository");
        pc.setServiceImpl("repository.impl");
        autoGenerator.setPackageInfo(pc);

        InjectionConfig cfg = new InjectionConfig() {
            @Override
            public void initMap() {
                Map<String, Object> map = new HashMap<String, Object>();
                this.setMap(map);
            }
        };
        // 调整 xml 生成目录演示
        List<FileOutConfig> focList = new ArrayList<>();
        focList.add(new FileOutConfig("/templates/mapper.xml.vm") {
            @Override
            public String outputFile(TableInfo tableInfo) {
                return filePathRe + "\\mapper\\" + tableInfo.getEntityName() + "Mapper.xml";
            }
        });
        cfg.setFileOutConfigList(focList);
        autoGenerator.setCfg(cfg);

        // 配置模板
        TemplateConfig templateConfig = new TemplateConfig();
        //控制 不生成 controller
        templateConfig.setController("");
        //templateConfig.setService("");
        //templateConfig.setEntity("");
        //templateConfig.setServiceImpl("");
        //templateConfig.setMapper("");
        //将mapper包  下面生成的xml文件移动到跟目录下的resources/mapper包下，若不用则不需要打开;如果注释调，则两个地方都有xml文件
        templateConfig.setXml("");
        autoGenerator.setTemplate(templateConfig);

        // 4、策略配置
        StrategyConfig strategy = new StrategyConfig();
        strategy.setNaming(NamingStrategy.underline_to_camel);
        strategy.setColumnNaming(NamingStrategy.underline_to_camel);
        // 表名前缀
        strategy.setTablePrefix("t_");
        //使用lombok
        //  strategy.setEntityLombokModel(true);

        // 逆向工程使用的表   如果要生成多个,这里可以传入String[]
        String[] tableNames={"undo_log"};
        strategy.setInclude(tableNames);

        //数据库表映射到实体的命名策略
        //  strategy.setNaming(NamingStrategy.underline_to_camel);
        //数据库表字段映射到实体的命名策略, 未指定按照 naming 执行
        //   strategy.setColumnNaming(NamingStrategy.underline_to_camel);
        //自定义继承的Entity类全称，带包名
        //  strategy.setSuperEntityClass(null);
        //【实体】是否为lombok模型（默认 false）
        //   strategy.setEntityLombokModel(true);
        //生成 @RestController 控制器
        // strategy.setRestControllerStyle(true);
        //自定义继承的Controller类全称，带包名
        //  strategy.setSuperControllerClass(null);
        //表名
       /* if (tableNameArray != null && tableNameArray.length != 0) {
            strategy.setInclude(tableNameArray);
        }*/
        //自定义基础的Entity类，公共字段
        //  strategy.setSuperEntityColumns(null);
        //驼峰转连字符
        //   strategy.setControllerMappingHyphenStyle(true);
        //表前缀
        //   strategy.setTablePrefix(pc.getModuleName() + "_");
        autoGenerator.setStrategy(strategy);
        //5、执行
        autoGenerator.execute();
    }
}

3、执行后生成的目录结构：