mybatis-plus使用generator实现逆向工程

<!-- mybatis-plus begin -->

< dependency >

         < groupId >com.baomidou</ groupId >

         < artifactId >mybatis-plus-boot-starter</ artifactId >

         < version >3.4.1</ version >

</ dependency >

< 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 >5.1.44</ version >

</ dependency >

<!-- mybatis-plus 默认是vm引擎 -->

< dependency >

         < groupId >org.apache.velocity</ groupId >

         < artifactId >velocity-engine-core</ artifactId >

         < version >2.0</ version >

</ dependency >

<!-- mybatis-plus end -->

public static void main(String[] args) {

     // 构建一个代码生成器对象

     AutoGenerator mpg = new AutoGenerator();

     // 配置执行策略

     // 1.全局配置

     GlobalConfig gc = new GlobalConfig();

     // 当前项目路径

     String proPath = System.getProperty( "user.dir" );

     // 设置代码生成路径

     gc.setOutputDir(proPath + "/src/main/java" );

     // 生成的类的注释中作者信息

     gc.setAuthor( "curry" );

     // 生成后是否打开文件夹

     gc.setOpen( false );

     // 是否覆盖

     gc.setFileOverride( true );

     // 生成service类的后缀

     gc.setServiceName( "%sService" );

     // 主键生成策略 和数据库id生成策略一致

     gc.setIdType(IdType.AUTO);

     // 设置日期类型

     gc.setDateType(DateType.ONLY_DATE);

     // 是否生成Swagger

     gc.setSwagger2( false );

     // 生成entity类的后缀

     gc.setEntityName( "%sEntity" );

     mpg.setGlobalConfig(gc);

     // 2.设置数据源

     DataSourceConfig dsc = new DataSourceConfig();

     dsc.setUrl( "jdbc:mysql://127.0.0.1:3306/test_db" );

     dsc.setDriverName( "com.mysql.jdbc.Driver" );

     dsc.setUsername( "root" );

     dsc.setPassword( "123456" );

     dsc.setDbType(DbType.MYSQL);

     mpg.setDataSource(dsc);

     // 3.配置生成包的路径

     PackageConfig pc = new PackageConfig();

     // 设置模块存放位置

     pc.setParent( "com." );

     // 设置该模块包的路径

     pc.setModuleName( "dao" );

     pc.setEntity( "entity" );

     pc.setMapper( "mapper" );

     pc.setService( "service" );

     pc.setController( "controller" );

     mpg.setPackageInfo(pc);

     // 4.策略配置

     StrategyConfig strategy= new StrategyConfig();

     // 设置要映射的表名 不配置默认处理全部表

     // strategy.setInclude("user");

     // 表名中下划线转驼峰命名

     strategy.setNaming(NamingStrategy.underline_to_camel);

     // 表中字段如果有下划线，转驼峰命名

     strategy.setColumnNaming(NamingStrategy.underline_to_camel);

     // strategy.setEntityLombokModel(true);//自动生成Lombok

     // strategy.setRestControllerStyle(true);//开启 RestFul 风格

     // strategy.setControllerMappingHyphenStyle(true);

     // 对表中的字段 设置逻辑删除 生成的dao层代码会添加@TableLogic

     strategy.setLogicDeleteFieldName( "delete_flag" );

     // 5.自动填充 (表中如果有创建时间、修改时间话，可以使用自动填充)

     TableFill createTime = new TableFill( "created_date" , FieldFill.INSERT);

     TableFill updateTime = new TableFill( "modified_date" , FieldFill.INSERT_UPDATE);

     ArrayList<TableFill> tableFills = new ArrayList<>();

     tableFills.add(createTime);

     tableFills.add(updateTime);

     strategy.setTableFillList(tableFills);

     // 乐观锁配置

     // strategy.setVersionFieldName("version");

     mpg.setStrategy(strategy);

     // 6.配置实体类模板

     TemplateConfig templateConfig = new TemplateConfig();

     // 如果setXxxxx(null) 不会生成Xxxx实体类相关代码

     // 因此如果只生成dao层代码

     // 可以在这里控制

     templateConfig.setController( null );

     templateConfig.setMapper( null );

     templateConfig.setService( null );

     templateConfig.setServiceImpl( null );

     templateConfig.setXml( null );

     mpg.setTemplate(templateConfig);

     // 7.执行代码生成操作

     mpg.execute();

}

// 3.配置生成包的路径

PackageConfig pc = new PackageConfig();

// 设置模块存放位置

pc.setParent( "com." );

// 设置该模块包的路径

pc.setModuleName( "dao" );

pc.setEntity( "entity" );

pc.setMapper( "mapper" );

pc.setService( "service" );

pc.setController( "controller" );

mpg.setPackageInfo(pc);

templateConfig.setXml( null );

// 自定义配置

InjectionConfig cfg = new InjectionConfig() {

         @Override

         public void initMap() {

                 // to do nothing

         }

};

// 如果模板引擎是 freemarker

// String templatePath = "/templates/mapper.xml.ftl";

// 如果模板引擎是 velocity

String templatePath = "/templates/mapper.xml.vm" ;

// 自定义输出配置

List<FileOutConfig> focList = new ArrayList<>();

// 自定义配置会被优先输出

// 这里自定义配置的是*Mapper.xml文件

// 所以templatePath = "/templates/mapper.xml.vm";

// 如果你想自定义配置其它 修改templatePath即可

focList.add( new FileOutConfig(templatePath) {

         @Override

         public String outputFile(TableInfo tableInfo) {

                 // 自定义输出文件名 如果你 Entity 设置了前后缀

                 String entityName = tableInfo.getEntityName();

                 int length = entityName.length();

                 entityName = entityName.substring( 0 , length - 6 );

                 return proPath + "/src/main/resources/mapper/" +

                         entityName + "Mapper" + StringPool.DOT_XML;

         }

});

cfg.setFileOutConfigList(focList);

mpg.setCfg(cfg);