代码部分
package com.generator.demo.CodeGenerator;
import com.baomidou.mybatisplus.core.exceptions.MybatisPlusException;
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.TemplateConfig;
import com.baomidou.mybatisplus.generator.config.rules.DateType;
import com.baomidou.mybatisplus.generator.config.rules.NamingStrategy;
import java.util.Scanner;
import org.apache.commons.lang3.StringUtils;
/**
* @author EDY
*/
// 演示例子,执行 main 方法控制台输入模块表名回车自动生成对应项目目录中
public class CodeGenerator {
//"/src/main/java"
private static final String PROJECT_PATH = "/src/main/java";
//数据源配置类型 1:xx 2:xx
private static final int DATASOURCE_TYPE = 2;
//包配置 根目录名称
private static final String PACKAGE_PATH = "con.your_pro";
/**
* <p>
* 读取控制台内容
* </p>
*/
public static String scanner(String tip) {
Scanner scanner = new Scanner(System.in);
System.out.println("请输入" + tip + ":");
if (scanner.hasNext()) {
String ipt = scanner.next();
if (StringUtils.isNotBlank(ipt)) {
return ipt;
}
}
throw new MybatisPlusException("请输入正确的" + tip + "!");
}
public static void main(String[] args) {
// 代码生成器
AutoGenerator mpg = new AutoGenerator();
// 全局配置
GlobalConfig gc = getGlobalConfig();
// 自定义文件命名,注意 %s 会自动填充表实体属性!
mpg.setGlobalConfig(gc);
// 数据源配置
DataSourceConfig dsc = new DataSourceConfig();
// 配置自定义输出模板
builderDataSource(DATASOURCE_TYPE, dsc, mpg);
//包配置
PackageConfig pc = getPackageConfig();
mpg.setPackageInfo(pc);
// 配置模板
TemplateConfig templateConfig = getTemplateConfig();
//策略配置
StrategyConfig strategy = getStrategyConfig();
mpg.setStrategy(strategy);
//配置模板
mpg.setTemplate(templateConfig);
mpg.execute();
}
/**
* 包配置
* @return
*/
private static PackageConfig getPackageConfig() {
// 包配置
PackageConfig pc = new PackageConfig();
// pc.setModuleName(scanner("模块名"));
// pc.setParent("huaSuZL");
pc.setParent(PACKAGE_PATH);
pc.setMapper("mapper");
pc.setXml("mapper.xml");
pc.setEntity("entity");
pc.setService("service");
pc.setServiceImpl("service.impl");
pc.setController("controller");
return pc;
}
/**
* 全局配置
* @return
*/
private static GlobalConfig getGlobalConfig() {
// 全局配置
GlobalConfig gc = new GlobalConfig();
String projectPath = System.getProperty("user.dir");
//输出文件路径
gc.setOutputDir(projectPath + PROJECT_PATH);
//是否覆盖以前文件
gc.setFileOverride(true);
// 不需要ActiveRecord特性的请改为false
gc.setActiveRecord(false);
// XML 二级缓存
gc.setEnableCache(false);
// XML ResultMap
gc.setBaseResultMap(true);
// XML columList
gc.setBaseColumnList(true);
// 作者
gc.setAuthor("yangLong");
// 设置 service 接口的名字是否为 %sIService, 默认 true, 修改为 false 即可
gc.setServiceName("%sService");
// 设置日期字段的类型为java.util.Date
gc.setDateType(DateType.ONLY_DATE);
gc.setOpen(false);
return gc;
}
/**
* 策略配置
* @return
*/
private static StrategyConfig getStrategyConfig() {
// 策略配置
StrategyConfig strategy = new StrategyConfig();
// strategy.setTablePrefix(new String[]{"tb_"});// 此处可以修改为您的表前缀
// 表名生成策略,下划线转驼峰命名
strategy.setNaming(NamingStrategy.underline_to_camel);
//数据库表字段映射到实体的命名策略
strategy.setColumnNaming(NamingStrategy.underline_to_camel);
//lombok 模型 @Accessors(chain = true) setter链式操作
strategy.setEntityLombokModel(true);
//restful api风格控制器
strategy.setRestControllerStyle(true);
//url中驼峰转连字符
strategy.setControllerMappingHyphenStyle(true);
//设置逻辑删除字段
strategy.setLogicDeleteFieldName("status");
//strategy.setEntityTableFieldAnnotationEnable(true); // 开启生成实体时生成字段注解
//strategy.setSuperControllerClass("BaseController");
//自定义继承的Service类全称,带包名
strategy.setSuperServiceClass(null);
//自定义继承的ServiceImpl类全称,带包名
strategy.setSuperServiceImplClass(null);
//自定义继承的Mapper类全称,带包名
strategy.setSuperMapperClass(null);
strategy.setInclude(scanner("表名,多个英文逗号分割").split(","));
//乐观锁
strategy.setVersionFieldName("version");
//url中驼峰转连字符
strategy.setControllerMappingHyphenStyle(true);
return strategy;
}
private static TemplateConfig getTemplateConfig() {
TemplateConfig templateConfig = new TemplateConfig();
//生成controller层
templateConfig.setController("/templates/controller.java");
// templateConfig.setController(null);//不生成controller层
//生成xml层
templateConfig.setXml("/templates/mapper.xml");
//生成service层
templateConfig.setService("/templates/service.java");
//生成serviceImpl层
templateConfig.setServiceImpl("/templates/serviceImpl.java");
//生成entity层
templateConfig.setEntity("/templates/entity.java");
//生成mapper层
templateConfig.setMapper("/templates/mapper.java");
//不生成xml层
// templateConfig.setXml(null);
return templateConfig;
}
private static void builderDataSource(int type, DataSourceConfig dsc, AutoGenerator mpg) {
switch (type) {
case 1:
// 设置数据库连接地址
dsc.setUrl("jdbc:mysql://your_url:3306/xx_name?useUnicode=true&characterEncoding=utf-8&serverTimezone=GMT%2B8");
// 设置驱动类
dsc.setDriverName("com.mysql.cj.jdbc.Driver");
// 设置用户名
dsc.setUsername("");
// 设置密码
dsc.setPassword("");
// 设置数据源
mpg.setDataSource(dsc);
break;
default:
// 设置数据库连接地址
dsc.setUrl("jdbc:mysql://your_url:3306/xx_name?useUnicode=true&characterEncoding=utf-8&serverTimezone=GMT%2B8");
// 设置驱动类
dsc.setDriverName("com.mysql.cj.jdbc.Driver");
// 设置用户名
dsc.setUsername("");
// 设置密码
dsc.setPassword("");
// 设置数据源
mpg.setDataSource(dsc);
break;
}
}
}
依赖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 https://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.2.10.RELEASE</version>
<relativePath/> <!-- lookup parent from repository -->
</parent>
<groupId>your_pro_name</groupId>
<artifactId>your_pro_name</artifactId>
<version>0.0.1-SNAPSHOT</version>
<name>your_pro_name</name>
<description>your_pro_name</description>
<properties>
<maven.compiler.source>8</maven.compiler.source>
<maven.compiler.target>8</maven.compiler.target>
<project.build.sourceEncoding>UTF-8</project.build.sourceEncoding>
<qcloud.cos.version>4.4</qcloud.cos.version>
</properties>
<dependencies>
<dependency>
<groupId>org.springframework.boot</groupId>
<artifactId>spring-boot-starter-web</artifactId>
</dependency>
<dependency>
<groupId>org.springframework.boot</groupId>
<artifactId>spring-boot-starter-test</artifactId>
<scope>test</scope>
</dependency>
<dependency>
<groupId>org.projectlombok</groupId>
<artifactId>lombok</artifactId>
</dependency>
<dependency>
<groupId>com.qcloud</groupId>
<artifactId>cos_api</artifactId>
<version>${qcloud.cos.version}</version>
<exclusions>
<exclusion>
<groupId>org.slf4j</groupId>
<artifactId>slf4j-log4j12</artifactId>
</exclusion>
</exclusions>
</dependency>
<!-- mysql 驱动 -->
<dependency>
<groupId>mysql</groupId>
<artifactId>mysql-connector-java</artifactId>
<version>8.0.22</version>
</dependency>
<!--代码生成器配置-->
<!-- generate -->
<dependency>
<groupId>com.baomidou</groupId>
<artifactId>mybatis-plus-generator</artifactId>
<version>3.3.1.tmp</version>
</dependency>
<dependency>
<groupId>mysql</groupId>
<artifactId>mysql-connector-java</artifactId>
<version>8.0.33</version>
</dependency>
<dependency>
<groupId>com.baomidou</groupId>
<artifactId>mybatis-plus-boot-starter</artifactId>
<version>3.3.1.tmp</version>
</dependency>
<!-- 模板引擎 -->
<!--freemarker 依赖-->
<dependency>
<groupId>org.freemarker</groupId>
<artifactId>freemarker</artifactId>
</dependency>
<!--velocity 依赖-->
<dependency>
<groupId>org.apache.velocity</groupId>
<artifactId>velocity-engine-core</artifactId>
<version>2.3</version>
</dependency>
<dependency>
<groupId>org.junit.platform</groupId>
<artifactId>junit-platform-commons</artifactId>
<version>1.8.2</version>
</dependency>
<dependency>
<groupId>com.baomidou</groupId>
<artifactId>mybatis-plus</artifactId>
<version>3.5.3.1</version>
</dependency>
<dependency>
<groupId>com.baomidou</groupId>
<artifactId>mybatis-plus-generate</artifactId>
<version>2.3.3</version>
</dependency>
<dependency>
<groupId>com.baomidou</groupId>
<artifactId>mybatis-plus-annotation</artifactId>
<version>3.5.3.1</version>
</dependency>
<dependency>
<groupId>com.baomidou</groupId>
<artifactId>mybatis-plus-core</artifactId>
<version>3.5.3.1</version>
</dependency>
<dependency>
<groupId>io.swagger</groupId>
<artifactId>swagger-annotations</artifactId>
<version>1.5.21</version>
</dependency>
<dependency>
<groupId>com.databank</groupId>
<artifactId>political-nail-service</artifactId>
<version>0.0.1-SNAPSHOT</version>
</dependency>
<dependency>
<groupId>org.nutz</groupId>
<artifactId>nutz</artifactId>
<version>1.r.62</version>
</dependency>
<dependency>
<groupId>com.jfinal</groupId>
<artifactId>jfinal</artifactId>
<version>2.2</version>
</dependency>
<!-- 代码生成器配置 end -->
<!-- 导入导出 -->
<dependency>
<groupId>org.apache.poi</groupId>
<artifactId>poi</artifactId>
<version>4.1.2</version>
</dependency>
<!-- https://mvnrepository.com/artifact/org.apache.poi/poi-ooxml -->
<dependency>
<groupId>org.apache.poi</groupId>
<artifactId>poi-ooxml</artifactId>
<version>4.1.2</version>
</dependency>
<!-- https://mvnrepository.com/artifact/com.alibaba.fastjson2/fastjson2 -->
<dependency>
<groupId>com.alibaba</groupId>
<artifactId>fastjson</artifactId>
<version>2.0.21</version>
</dependency>
<dependency>
<groupId>org.apache.commons</groupId>
<artifactId>commons-lang3</artifactId>
<version>3.12.0</version>
</dependency>
<!-- https://mvnrepository.com/artifact/net.sourceforge.javacsv/javacsv -->
<dependency>
<groupId>net.sourceforge.javacsv</groupId>
<artifactId>javacsv</artifactId>
<version>2.0</version>
</dependency>
<dependency>
<artifactId>mybatis-flex-core</artifactId>
<groupId>com.mybatis-flex</groupId>
<version>1.8.2</version>
</dependency>
<dependency>
<artifactId>enjoy</artifactId>
<groupId>com.jfinal</groupId>
<version>5.1.3</version>
</dependency>
<dependency>
<artifactId>mybatis-flex-codegen</artifactId>
<groupId>com.mybatis-flex</groupId>
<version>1.8.2</version>
</dependency>
</dependencies>
<build>
<plugins>
<plugin>
<groupId>org.springframework.boot</groupId>
<artifactId>spring-boot-maven-plugin</artifactId>
</plugin>
</plugins>
</build>
</project>