mybatis plus 代码自动生成

1.需要引入的依赖

<!-- 数据库连接 -->
<dependency>
    <groupId>mysql</groupId>
    <artifactId>mysql-connector-java</artifactId>
    <scope>runtime</scope>
</dependency>
<!-- lombok -->
<dependency>
    <groupId>org.projectlombok</groupId>
    <artifactId>lombok</artifactId>
    <optional>true</optional>
</dependency>
<!-- mybatis-plus 启动器-->
<dependency>
   	<groupId>com.baomidou</groupId>
   	<artifactId>mybatis-plus-boot-starter</artifactId>
   	<version>3.3.1</version>
</dependency>
<!-- mybatis-plus 生成器-->
<dependency>
   	<groupId>com.baomidou</groupId>
   	<artifactId>mybatis-plus-generator</artifactId>
   	<version>3.3.1</version>
</dependency>
<!-- mybatis-plus 默认模板引擎-->
<dependency>
   	<groupId>org.apache.velocity</groupId>
   	<artifactId>velocity-engine-core</artifactId>
   	<version>2.2</version>
</dependency>

2.代码

package com.shuo;

import com.baomidou.mybatisplus.annotation.IdType;
import com.baomidou.mybatisplus.generator.AutoGenerator;
import com.baomidou.mybatisplus.generator.config.DataSourceConfig;
import com.baomidou.mybatisplus.generator.config.GlobalConfig;
import com.baomidou.mybatisplus.generator.config.PackageConfig;
import com.baomidou.mybatisplus.generator.config.StrategyConfig;
import com.baomidou.mybatisplus.generator.config.builder.ConfigBuilder;
import com.baomidou.mybatisplus.generator.config.po.TableInfo;
import com.baomidou.mybatisplus.generator.config.rules.NamingStrategy;

import java.util.List;

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

        // 全局配置
        GlobalConfig gc = new GlobalConfig();
        String projectPath = System.getProperty("user.dir");
        final String mapperPath = projectPath + "/src/main/resources/mapper/";
        gc.setOutputDir(projectPath + "/src/main/java");
        gc.setAuthor("作者");
        // 生成代码后是否打开文件夹 默认打开
        gc.setOpen(false);
        // gc.setSwagger2(true); 实体属性 Swagger2 注解
        // 设置Service接口生成名称,这样生成接口前面就不会有 I
        gc.setServiceName("%sService");
        // id 生成策略
        gc.setIdType(IdType.AUTO);
        // gc.setFileOverride(true);
        mpg.setGlobalConfig(gc);

        // 数据源配置
        DataSourceConfig dsc = new DataSourceConfig();
        dsc.setUrl("jdbc:mysql://localhost:3306/库?useUnicode=true&serverTimezone=GMT&useSSL=false&characterEncoding=utf8");
        dsc.setDriverName("com.mysql.cj.jdbc.Driver");
        dsc.setUsername("账号");
        dsc.setPassword("密码");
        mpg.setDataSource(dsc);

        // 包配置
        PackageConfig pc = new PackageConfig();
        // 当前表所要生成的目录
        pc.setModuleName("test");
        // 公共目录
        pc.setParent("com.shuo");
        mpg.setPackageInfo(pc);

        // 策略配置
        StrategyConfig strategy = new StrategyConfig();
        // 去掉下划线 并转成驼峰命名
        strategy.setNaming(NamingStrategy.underline_to_camel);
        strategy.setColumnNaming(NamingStrategy.underline_to_camel);
        // strategy.setSuperEntityClass("你自己的父类实体,没有就不用设置!");
        strategy.setEntityLombokModel(true);
        strategy.setRestControllerStyle(true);
        // 公共父类
        // strategy.setSuperControllerClass("你自己的父类控制器,没有就不用设置!");
        // 如果要生成多个,这里可以传入String[]
        strategy.setInclude("o2o_repository");
        strategy.setTablePrefix("o2o_"); // 去掉表前缀
        mpg.setStrategy(strategy);

        // 重新命名
        mpg.setConfig(new ConfigBuilder(mpg.getPackageInfo(), mpg.getDataSource(), mpg.getStrategy(), mpg.getTemplate(), mpg.getGlobalConfig()));
        List<TableInfo> tableList = mpg.getConfig().getTableInfoList();
        for (TableInfo table : tableList) {
            table.setEntityName(table.getEntityName() + "Mp");
            table.setMapperName(table.getEntityName() + "Mapper");
            table.setXmlName(table.getEntityName() + "Mapper");
            table.setServiceName(table.getEntityName() + "Service");
            table.setServiceImplName(table.getEntityName() + "ServiceImpl");
        }

        mpg.execute();
    }
}

3. 结果图