package com.yuanchu.limslaboratory;
|
|
import com.baomidou.mybatisplus.core.exceptions.MybatisPlusException;
|
import com.baomidou.mybatisplus.core.toolkit.StringUtils;
|
import com.baomidou.mybatisplus.generator.FastAutoGenerator;
|
import com.baomidou.mybatisplus.generator.config.*;
|
import com.baomidou.mybatisplus.generator.config.rules.NamingStrategy;
|
|
import java.util.Collections;
|
import java.util.Scanner;
|
|
// 演示例子,执行 main 方法控制台输入模块表名回车自动生成对应项目目录中
|
public class CodeGenerator {
|
|
public static String database_url = "jdbc:mysql://192.168.110.209:3306/lims_laboratory?useSSL=false&useUnicode=true&characterEncoding=utf-8&serverTimezone=Asia/Shanghai";
|
public static String database_username = "user";
|
public static String database_password= "123456";
|
public static String author = "江苏鵷雏网络科技有限公司";
|
public static String model_name = "/standard-server"; // 如果为分布式填子模块名称,如果不是分布式为空即可
|
public static String setParent = "com.yuanchu.limslaboratory"; // 包路径
|
public static void main(String[] args) {
|
String projectPath = System.getProperty("user.dir");
|
String s = projectPath + "/lims-after" + model_name + "/src/main/java";
|
String s1 = projectPath + "/lims-after" + model_name + "/src/main/resources/mapper";
|
System.out.println();
|
FastAutoGenerator.create(database_url, database_username, database_password)
|
// 全局配置
|
.globalConfig(builder -> {
|
builder.author(author) // 设置作者
|
.commentDate("yyyy-MM-dd hh:mm:ss") //注释日期
|
.outputDir(s) // 指定输出目录
|
.disableOpenDir() //禁止打开输出目录,默认打开
|
;
|
})
|
// 包配置
|
.packageConfig(builder -> {
|
builder.entity("pojo");
|
builder.parent(setParent) // 设置父包名
|
.pathInfo(Collections.singletonMap(OutputFile.xml, s1)); // 设置mapperXml生成路径
|
})
|
// 策略配置
|
.strategyConfig(builder -> {
|
builder.addInclude(scanner("表名,多个空格分割").split(" ")) // 设置需要生成的表名
|
.addTablePrefix("sys_") // 设置过滤表前缀
|
// Entity 策略配置
|
.entityBuilder()
|
.enableLombok() //开启 Lombok
|
.naming(NamingStrategy.underline_to_camel) //数据库表映射到实体的命名策略:下划线转驼峰命
|
.columnNaming(NamingStrategy.underline_to_camel) //数据库表字段映射到实体的命名策略:下划线转驼峰命
|
// Mapper 策略配置
|
.mapperBuilder()
|
.enableFileOverride() // 覆盖已生成文件
|
// Service 策略配置
|
.serviceBuilder()
|
.enableFileOverride() // 覆盖已生成文件
|
.formatServiceFileName("%sService") //格式化 service 接口文件名称,%s进行匹配表名,如 UserService
|
.formatServiceImplFileName("%sServiceImpl") //格式化 service 实现类文件名称,%s进行匹配表名,如 UserServiceImpl
|
// Controller 策略配置
|
.controllerBuilder()
|
.enableFileOverride() // 覆盖已生成文件
|
;
|
})
|
.execute();
|
}
|
|
/**
|
* <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.isNotBlank(ipt)) {
|
return ipt;
|
}
|
}
|
throw new MybatisPlusException("请输入正确的" + tip + "!");
|
}
|
}
|