springBoot项目实战—mybatisPlus逆向工程

168 阅读2分钟

在构建了搭建了项目的情况下,结构如下:

屏幕快照 2021-12-28 上午6.46.25.png

这里的逆向工程主要使用mybatisPlus生成,大概目的就是不用我们关注项目项目的各个包之间的文件创建,可以快速专注业务开发。我单独创建一个聚合项目专门用于代码生成,主要是依赖和代码生成器两个文件组成。
所需依赖(pom.xml):

 <dependencies>
<!--        web依赖-->
        <dependency>
            <groupId>org.springframework.boot</groupId>
            <artifactId>spring-boot-starter-web</artifactId>
            <version>2.5.3</version>
        </dependency>
<!--        mybatis-plus依赖-->
        <dependency>
            <groupId>com.baomidou</groupId>
            <artifactId>mybatis-plus-boot-starter</artifactId>
            <version>3.4.3</version>
        </dependency>
<!--        mybatis-plus代码生成器依赖-->
        <dependency>
            <groupId>com.baomidou</groupId>
            <artifactId>mybatis-plus-generator</artifactId>
            <version>3.4.1</version>
        </dependency>
<!--        freemarker依赖-->
        <dependency>
            <groupId>org.freemarker</groupId>
            <artifactId>freemarker</artifactId>
            <version>2.3.31</version>
        </dependency>
<!--        mysql依赖-->
        <dependency>
            <groupId>mysql</groupId>
            <artifactId>mysql-connector-java</artifactId>
            <version>8.0.25</version>
        </dependency>
        <dependency>
            <groupId>com.baomidou</groupId>
            <artifactId>mybatis-plus-core</artifactId>
            <version>3.4.3</version>
            <scope>compile</scope>
        </dependency>
    </dependencies>

mybatisPlus逆向工程生成器代码(CodeGenerator):

public class CodeGenerator {
    public static String scanner(String tip){
        Scanner scanner=new Scanner(System.in);
        StringBuilder help=new StringBuilder();
        help.append("输入"+tip+":");
        System.out.println(help.toString());
        if(scanner.hasNext()){
            String ipt=scanner.next();
            if(ipt != null){
                return ipt;
            }
        }
        throw new MybatisPlusException("请输入正确的"+tip+"!");
    }

    public static void main(String[] args) {
//        代码生成器
        AutoGenerator mpg=new AutoGenerator();

//        全局配置
        GlobalConfig gc=new GlobalConfig();
        String projectPath=System.getProperty("user.dir");
        gc.setOutputDir(projectPath+"/yeb-generator/src/main/java");
//        作者
        gc.setAuthor("hhk");
//        打开输出目录
        gc.setOpen(false);
//        xml开启BaseResultMap
        gc.setBaseResultMap(true);
//        xml开启BaseColumnList
        gc.setBaseColumnList(true);
//        实体属性Swagger2注解
        gc.setSwagger2(true);
        mpg.setGlobalConfig(gc);

//        数据源配置
        DataSourceConfig dsc=new DataSourceConfig();
        dsc.setUrl("jdbc:mysql://localhost:3306/数据库名?useUnicode=true&characterEncoding=UTF-8&serverTimezone=Asia"+"/Shanghai");
        dsc.setDriverName("com.mysql.cj.jdbc.Driver");
        dsc.setUsername("root");
        dsc.setPassword("root");
        mpg.setDataSource(dsc);

//     包配置
        PackageConfig pc=new PackageConfig();
        pc.setParent("com.hhk")
                .setEntity("pojo")
                .setMapper("mapper")
                .setService("service")
                .setServiceImpl("service.impl")
                .setController("controller");
        mpg.setPackageInfo(pc);
//        自定义配置
        InjectionConfig cfg=new InjectionConfig() {
            @Override
            public void initMap() {
//                to do something
            }
        };
//        如果模板引擎是freemarker
        String templatePath="/templates/mapper.xml.ftl";
//        如果模板引擎是velocity
//        String templatePath="/templates/mapper.xml.vm";
//        自定义输出配置
        List<FileOutConfig> focList=new ArrayList<>();
//        自定义输出会被优先输出
        focList.add(new FileOutConfig(templatePath) {
            @Override
            public String outputFile(TableInfo tableInfo) {
//                自定义输出文件名,如果你Entity设置了前后缀,此处注意xml的名称会跟着发生变化!!
                return projectPath+"/yeb-generator/src/main/resources/mapper"+tableInfo.getEntityName()+ "Mapper"+ StringPool.DOT_XML;
            }
        });
        cfg.setFileOutConfigList(focList);
        mpg.setCfg(cfg);

//        配置模板
        TemplateConfig templateConfig=new TemplateConfig();
        templateConfig.setXml(null);
        mpg.setTemplate(templateConfig);

//        策略配置
        StrategyConfig strategy=new StrategyConfig();
//        数据库映射到实体的命名策略
        strategy.setNaming(NamingStrategy.underline_to_camel);
//        数据库表映射到实体的命名策略
        strategy.setColumnNaming(NamingStrategy.no_change);
//        lombok模型
        strategy.setEntityLombokModel(true);
        strategy.setRestControllerStyle(true);
        strategy.setInclude(scanner("表名,多个英文逗号隔开").split(","));
        strategy.setControllerMappingHyphenStyle(true);
//        表前缀
        strategy.setTablePrefix("t_");
        mpg.setStrategy(strategy);
        mpg.setTemplateEngine(new FreemarkerTemplateEngine());
        mpg.execute();
    }
}

然后就可以运行CodeGenerator文件,根据数据库生成项目所需要的各个包了!!!