mybatis plus代码生成器。

参考博客：https://blog.csdn.net/weixin_43770545/article/details/90483900

 

pom.xml

    
            com.baomidou
            mybatis-plus-boot-starter
            3.1.0
        
        
            com.baomidou
            mybatis-plus-generator
            3.2.0
        
        
            org.freemarker
            freemarker
            2.3.28
        
        
            mysql
            mysql-connector-java
            8.0.16
        

 

生成代码之后将下面的注释掉。： 

 
            com.baomidou
            mybatis-plus-generator
            3.2.0
        
        
            org.freemarker
            freemarker
            2.3.28
        

完整的pom.xml

	
		org.springframework.boot
		spring-boot-starter-parent
		2.1.6.RELEASE
	
	4.0.0

	xiaoming
	jar

	测试模块
	
		
			cn.hutool
			hutool-all
			4.5.16
		
		
			javax.servlet
			javax.servlet-api
		
		
			org.springframework.boot
			spring-boot-starter-validation
		
		
			org.springframework.boot
			spring-boot-starter-json
		
		
		
			io.springfox
			springfox-swagger2
			2.9.2
		
		
			io.springfox
			springfox-swagger-ui
			2.9.2
		

		
			org.projectlombok
			lombok
			provided
		
		
			org.springframework.boot
			spring-boot-starter-test
			test
		
		
		
			com.baomidou
			mybatis-plus-boot-starter
			3.1.0
		

		
			mysql
			mysql-connector-java
			8.0.16
		
		
		
			com.alibaba
			druid-spring-boot-starter
			1.1.16
		
		
			org.springframework.boot
			spring-boot-starter-web
		
		
			org.springframework
			spring-test
			5.1.8.RELEASE
			compile
		
		
			org.springframework.boot
			spring-boot-starter-aop
		
	

	
		
			
				org.springframework.boot
				spring-boot-maven-plugin
			
			
				com.spotify
				docker-maven-plugin
			
		
	

 

代码如下：

package com.xiaoming.demo;
import com.baomidou.mybatisplus.annotation.DbType;
import com.baomidou.mybatisplus.core.toolkit.StringPool;
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 com.baomidou.mybatisplus.generator.engine.FreemarkerTemplateEngine;

import java.util.ArrayList;
import java.util.Arrays;
import java.util.List;





public class MapGenerator {


	public static void main(String[] args) {
// 只用一个字典表做测试，如果多个表，就咋下面都填上
		List listA = Arrays.asList(
				"dict"
		);

		//List listA = Arrays.asList("zydescription","fileinfos","labelzy","zytype");
for(String one : listA){
		// 代码生成器
		AutoGenerator mpg = new AutoGenerator();

		// 全局配置
		GlobalConfig gc = new GlobalConfig();
		//获得程序当前路径E:\project
		//String projectPath = System.getProperty("user.dir");
	String projectPath = "E:\\project";
		//生成文件的输出目录
		gc.setOutputDir(projectPath + "/src/main/java");
		gc.setAuthor("xiaoming");
		//是否展开输出目录
		gc.setOpen(false);
		//实体属性 Swagger2 注解
		//gc.setSwagger2(true);
		gc.setBaseResultMap(true);
		gc.setBaseColumnList(true);
		gc.setEntityName("%sEntity");
		gc.setControllerName("%sController");
		gc.setServiceName("%sService");
		gc.setServiceImplName("%sServiceImpl");
		gc.setMapperName("%sMapper");
		gc.setXmlName("%sMapper");
		//设置全局配置
		mpg.setGlobalConfig(gc);

		// 数据源配置
		DataSourceConfig dsc = new DataSourceConfig();
		dsc.setUrl("jdbc:mysql://localhost:3306/demo?characterEncoding=utf8&zeroDateTimeBehavior=convertToNull&useSSL=false&useJDBCCompliantTimezoneShift=true&useLegacyDatetimeCode=false&serverTimezone=GMT%2B8");
		dsc.setDriverName("com.mysql.cj.jdbc.Driver");
		dsc.setUsername("root");
		dsc.setPassword("root");
		dsc.setDbType(DbType.MYSQL);
		mpg.setDataSource(dsc);

		// 包配置
		PackageConfig pc = new PackageConfig();
		pc.setModuleName("demo");
		pc.setParent("com.xiaoming");
		pc.setController("controller");
		pc.setEntity("entity");
		pc.setMapper("mapper");
		pc.setXml("mapper");
		pc.setService("service");
		pc.setServiceImpl("service.impl");
		mpg.setPackageInfo(pc);

		// 自定义配置  调整 xml 生成目录
		InjectionConfig cfg = new InjectionConfig() {
			@Override
			public void initMap() {

			}
		};
		// 如果模板引擎是 freemarker
		String templatePath = "/templates/mapper.xml.ftl";

		// 自定义输出配置
		List focList = new ArrayList<>();
		// 自定义配置会被优先输出
		focList.add(new FileOutConfig(templatePath) {
			@Override
			public String outputFile(TableInfo tableInfo) {
				// 自定义输出文件名 ， 如果你 Entity 设置了前后缀、此处注意 xml 的名称会跟着发生变化！！
				return projectPath + "/src/main/resources/mapper/" + pc.getModuleName()
						+ "/" + tableInfo.getEntityName() + "Mapper" + StringPool.DOT_XML;
			}
		});
		cfg.setFileOutConfigList(focList);
		mpg.setCfg(cfg);

		// 配置模板
		TemplateConfig templateConfig = new TemplateConfig();
		// 关闭 默认 xml文件生成
		templateConfig.setXml(null);
		mpg.setTemplate(templateConfig);

		// 策略配置
		StrategyConfig strategy = new StrategyConfig();
		//表名生成策略  下划线转驼峰
		strategy.setNaming(NamingStrategy.underline_to_camel);
		//字段名生成策略  下划线转驼峰
		strategy.setColumnNaming(NamingStrategy.underline_to_camel);
		//启动lombok
		strategy.setEntityLombokModel(true);
		//实体类添加注解
		//strategy.entityTableFieldAnnotationEnable(true);
		//生成 @RestController 控制器
		strategy.setRestControllerStyle(true);
		//写于父类中的公共字段
		//strategy.setSuperEntityColumns("id");
		//表名，多个英文逗号分割
		strategy.setInclude(one);
		//驼峰转连字符
		strategy.setControllerMappingHyphenStyle(true);
		//设置表前缀
		//strategy.setTablePrefix(pc.getModuleName() + "_");
		mpg.setStrategy(strategy);
		//设置模板引擎
		mpg.setTemplateEngine(new FreemarkerTemplateEngine());
		mpg.execute();

	}


	}



}