Mybatis代码生成器

最新推荐文章于 2024-07-09 08:30:00 发布

林邵晨

最新推荐文章于 2024-07-09 08:30:00 发布

阅读量246

点赞数 1

分类专栏： Mybatis 文章标签： mybatis java 开发语言

本文链接：https://blog.csdn.net/qq_54429571/article/details/126722795

版权

Mybatis 专栏收录该内容

3 篇文章 2 订阅

订阅专栏

Mybatis代码生成器

AutoGenerator 是 MyBatis-Plus 的代码生成器，通过 AutoGenerator 可以快速生成 Entity、Mapper、MapperXML、Service、Controller 等各个模块的代码，极大的提升了开发效率。

创建工程

pom.xml：

<?xml version="1.0" encoding="UTF-8"?>
<project xmlns="http://maven.apache.org/POM/4.0.0"
xmlns:xsi="http://www.w3.org/2001/XMLSchema-instance"
xsi:schemaLocation="http://maven.apache.org/POM/4.0.0
http://maven.apache.org/xsd/maven-4.0.0.xsd">
<modelVersion>4.0.0</modelVersion>

<parent>
<groupId>org.springframework.boot</groupId>
<artifactId>spring-boot-starter-parent</artifactId>
<version>2.1.4.RELEASE</version>
</parent>

<groupId>cn.mp</groupId>
<artifactId>mp-generator</artifactId>
<version>1.0-SNAPSHOT</version>

<dependencies>

<dependency>
<groupId>org.springframework.boot</groupId>
<artifactId>spring-boot-starter-test</artifactId>
<scope>test</scope>
</dependency>

<!--mybatis-plus的springboot支持-->
<dependency>
<groupId>com.baomidou</groupId>
<artifactId>mybatis-plus-boot-starter</artifactId>
<version>3.1.1</version>
</dependency>

<dependency>
<groupId>com.baomidou</groupId>
<artifactId>mybatis-plus-generator</artifactId>
<version>3.1.1</version>
</dependency>

<dependency>
<groupId>org.springframework.boot</groupId>
<artifactId>spring-boot-starter-freemarker</artifactId>
</dependency>

<!--mysql驱动-->
<dependency>
<groupId>mysql</groupId>
<artifactId>mysql-connector-java</artifactId>
<version>5.1.47</version>
</dependency>

<dependency>
<groupId>org.slf4j</groupId>
<artifactId>slf4j-log4j12</artifactId>
</dependency>

</dependencies>

<build>
<plugins>
<plugin>
<groupId>org.springframework.boot</groupId>
<artifactId>spring-boot-maven-plugin</artifactId>
</plugin>
</plugins>
</build>
</project>

代码

/**
* <p>
* mysql 代码生成器演示例子
* </p>
*/
public class MysqlGenerator {
/**
* <p>
* 读取控制台内容
* </p>
*/
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 (StringUtils.isNotEmpty(ipt)) {
		return ipt;
	}
	}
	throw new MybatisPlusException("请输入正确的" + tip + "！");
}
/**
* RUN THIS
*/
public static void main(String[] args) {
	// 代码生成器
	AutoGenerator mpg = new AutoGenerator();
	// 全局配置
	GlobalConfig gc = new GlobalConfig();
	String projectPath = System.getProperty("user.dir");
	gc.setOutputDir(projectPath + "/src/main/java");
	gc.setAuthor("XXX");
	gc.setOpen(false);
	mpg.setGlobalConfig(gc);
	// 数据源配置
	DataSourceConfig dsc = new DataSourceConfig();
	dsc.setUrl("jdbc:mysql://127.0.0.1:3306/mp?
	useUnicode=true&useSSL=false&characterEncoding=utf8");
	// dsc.setSchemaName("public");
	dsc.setDriverName("com.mysql.jdbc.Driver");
	dsc.setUsername("root");
	dsc.setPassword("root");
	mpg.setDataSource(dsc);
	// 包配置
	PackageConfig pc = new PackageConfig();
	pc.setModuleName(scanner("模块名"));
	pc.setParent("cn.mp.generator");
	mpg.setPackageInfo(pc);
	// 自定义配置
	InjectionConfig cfg = new InjectionConfig() {
		@Override
		public void initMap() {
		 	// to do nothing
		}
		};
		List<FileOutConfig> focList = new ArrayList<>();
		focList.add(new FileOutConfig("/templates/mapper.xml.ftl") {
			@Override
			public String outputFile(TableInfo tableInfo) {
			// 自定义输入文件名称
			return projectPath + "/mpgenerator/src/main/resources/mapper/" + pc.getModuleName()
			+ "/" + tableInfo.getEntityName() + "Mapper" +
			StringPool.DOT_XML;
			}
	});
	cfg.setFileOutConfigList(focList);
	mpg.setCfg(cfg);
	mpg.setTemplate(new TemplateConfig().setXml(null));
	// 策略配置
	StrategyConfig strategy = new StrategyConfig();
	strategy.setNaming(NamingStrategy.underline_to_camel);
	strategy.setColumnNaming(NamingStrategy.underline_to_camel);
	//
	strategy.setSuperEntityClass("com.baomidou.mybatisplus.samples.generator.common.BaseE
	ntity");
	strategy.setEntityLombokModel(true);
	//
	strategy.setSuperControllerClass("com.baomidou.mybatisplus.samples.generator.common.B
	aseController");
	strategy.setInclude(scanner("表名"));
	strategy.setSuperEntityColumns("id");
	strategy.setControllerMappingHyphenStyle(true);
	strategy.setTablePrefix(pc.getModuleName() + "_");
	mpg.setStrategy(strategy);
	// 选择 freemarker 引擎需要指定如下加，注意 pom 依赖必须有！
	mpg.setTemplateEngine(new FreemarkerTemplateEngine());
	mpg.execute();
	}
}