mybatis-plus代码自动生成

mybatis-plus代码自动生成

这里使用的是3.0.4版的jar包（目前发现最新版本的jar包没有generator包）

package com.example.autocode;

import com.baomidou.mybatisplus.annotation.DbType;
import com.baomidou.mybatisplus.annotation.FieldFill;
import com.baomidou.mybatisplus.annotation.IdType;
import com.baomidou.mybatisplus.annotation.TableField;
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.po.TableFill;
import com.baomidou.mybatisplus.generator.config.rules.DateType;
import com.baomidou.mybatisplus.generator.config.rules.NamingStrategy;

import java.util.ArrayList;

//代码自动生成
public class AutoCode {
    public static void main(String[] args) {
        //需要构建一个代码自动生成对象
        AutoGenerator autoGenerator = new AutoGenerator();
        //配置策略
        //1.全局配置
        GlobalConfig config = new GlobalConfig();
        String projectPath = System.getProperty("user.dir");
        config.setOutputDir(projectPath + "/src/main/java");
        config.setAuthor("阿杰")
              .setOpen(false)
              .setFileOverride(false)  //是否覆盖
              .setServiceName("")
              .setIdType(IdType.ID_WORKER)
              .setDateType(DateType.ONLY_DATE)
              .setSwagger2(true);
        autoGenerator.setGlobalConfig(config);
        //设置数据源
        DataSourceConfig dsc = new DataSourceConfig();
        dsc.setUrl("jdbc:mysql://localhost:3306/数据库名称?useSSL=trueserverTimezone=GMT%2B8")
            .setDriverName("com.mysql.jdbc.Driver")
            .setUsername("root")
            .setPassword("123456")
            .setDbType(DbType.MYSQL);
        autoGenerator.setDataSource(dsc);
        //包的配置
        PackageConfig packageConfig = new PackageConfig();
        packageConfig.setEntity("pojo");
        packageConfig.setModuleName("project");
        packageConfig.setParent("com.lit");
        packageConfig.setMapper("mapper");
        packageConfig.setService("service");
        packageConfig.setController("controller");
        autoGenerator.setPackageInfo(packageConfig);
        //策略配置
        StrategyConfig strategyConfig = new StrategyConfig();
        strategyConfig.setInclude("user");//映射的表名
        strategyConfig.setNaming(NamingStrategy.underline_to_camel);
        strategyConfig.setColumnNaming(NamingStrategy.underline_to_camel);
        strategyConfig.setEntityLombokModel(true);
        strategyConfig.setLogicDeleteFieldName("deleted"); //逻辑删除配置
        //自动填充配置
        TableFill created_time = new TableFill("created_time", FieldFill.INSERT);
        TableFill update_time = new TableFill("update_time", FieldFill.INSERT_UPDATE);
        ArrayList<TableFill> list = new ArrayList<>();
        list.add(created_time);
        list.add(update_time);
        strategyConfig.setTableFillList(list);
        //乐观锁
        strategyConfig.setVersionFieldName("version");
        strategyConfig.setRestControllerStyle(true);
        strategyConfig.setControllerMappingHyphenStyle(true); //xxxx:8080/hello_id_x 形式
        autoGenerator.setStrategy(strategyConfig);
        autoGenerator.execute();
    }
}