| | |
|---|
| | | package com.yuanchu.limslaboratory; |
|---|
| | | |
|---|
| | | import com.baomidou.mybatisplus.core.exceptions.MybatisPlusException; |
|---|
| | | import com.baomidou.mybatisplus.core.toolkit.StringPool; |
|---|
| | | import com.baomidou.mybatisplus.core.toolkit.StringUtils; |
|---|
| | | import com.baomidou.mybatisplus.generator.AutoGenerator; |
|---|
| | | import com.baomidou.mybatisplus.generator.InjectionConfig; |
|---|
| | | import com.baomidou.mybatisplus.generator.FastAutoGenerator; |
|---|
| | | 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.List; |
|---|
| | | import java.util.Collections; |
|---|
| | | import java.util.Scanner; |
|---|
| | | |
|---|
| | | // 演示例子，执行 main 方法控制台输入模块表名回车自动生成对应项目目录中 |
|---|
| | | public class CodeGenerator { |
|---|
| | | |
|---|
| | | public static String database_url = "jdbc:mysql://localhost:3306/lims_laboratory?useSSL=false&useUnicode=true&characterEncoding=utf-8&serverTimezone=Asia/Shanghai"; |
|---|
| | | public static String database_driver_name = "com.mysql.cj.jdbc.Driver"; |
|---|
| | | public static String database_username = "root"; |
|---|
| | | 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 = "/inspection-server"; // 如果为分布式填子模块名称，如果不是分布式为空即可 |
|---|
| | | public static String model_name = "/cnas-server"; // 如果为分布式填子模块名称，如果不是分布式为空即可 |
|---|
| | | public static String setParent = "com.yuanchu.limslaboratory"; // 包路径 |
|---|
| | | public static Boolean Override = false; // 是否覆盖原来的文件？ |
|---|
| | | |
|---|
| | | public static void main(String[] args) { |
|---|
| | | // 代码生成器 |
|---|
| | | AutoGenerator mpg = new AutoGenerator(); |
|---|
| | | |
|---|
| | | String projectPath = System.getProperty("user.dir"); |
|---|
| | | System.out.println(projectPath+"==================="); |
|---|
| | | |
|---|
| | | GlobalConfig gc = new GlobalConfig() // 全局配置 |
|---|
| | | .setOutputDir(projectPath + model_name + "/src/main/java") // 输出路径 |
|---|
| | | .setAuthor(author) // 作者注释 |
|---|
| | | .setOpen(false) // 是否打开 |
|---|
| | | .setSwagger2(true) //实体属性 Swagger2 注解 |
|---|
| | | .setServiceName("%sService") // 设置serviceName的名称去大写I |
|---|
| | | .setFileOverride(Override) |
|---|
| | | .setEntityName("pojo");// 是否覆盖已生成文件 |
|---|
| | | mpg.setGlobalConfig(gc); |
|---|
| | | |
|---|
| | | // 数据源配置 数据库名 账号密码 |
|---|
| | | DataSourceConfig dsc = new DataSourceConfig() |
|---|
| | | .setUrl(database_url) |
|---|
| | | .setDriverName(database_driver_name) |
|---|
| | | .setUsername(database_username) |
|---|
| | | .setPassword(database_password); |
|---|
| | | mpg.setDataSource(dsc); |
|---|
| | | |
|---|
| | | |
|---|
| | | // 包配置 |
|---|
| | | PackageConfig pc = new PackageConfig() |
|---|
| | | .setModuleName(null) |
|---|
| | | .setParent(setParent) |
|---|
| | | .setEntity("pojo");// 包路径 |
|---|
| | | mpg.setPackageInfo(pc); |
|---|
| | | |
|---|
| | | // 自定义配置 |
|---|
| | | 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<>(); |
|---|
| | | // 自定义配置会被优先输出 |
|---|
| | | focList.add(new FileOutConfig(templatePath) { |
|---|
| | | @Override |
|---|
| | | public String outputFile(TableInfo tableInfo) { |
|---|
| | | // 自定义输出文件名 ， 如果你 Entity 设置了前后缀、此处注意 xml 的名称会跟着发生变化！ |
|---|
| | | return projectPath + model_name + "/src/main/resources/mapper/" |
|---|
| | | + "/" + tableInfo.getEntityName() + "Mapper" + StringPool.DOT_XML; |
|---|
| | | } |
|---|
| | | }); |
|---|
| | | |
|---|
| | | cfg.setFileOutConfigList(focList); |
|---|
| | | mpg.setCfg(cfg); |
|---|
| | | |
|---|
| | | // 配置模板 |
|---|
| | | TemplateConfig templateConfig = new TemplateConfig() |
|---|
| | | .setXml(null); |
|---|
| | | |
|---|
| | | mpg.setTemplate(templateConfig); |
|---|
| | | |
|---|
| | | // 策略配置 |
|---|
| | | StrategyConfig strategy = new StrategyConfig() |
|---|
| | | .setNaming(NamingStrategy.underline_to_camel) |
|---|
| | | .setColumnNaming(NamingStrategy.underline_to_camel) |
|---|
| | | .setEntityLombokModel(true) |
|---|
| | | .setRestControllerStyle(true) |
|---|
| | | .setInclude(scanner("表名，多个道号分割").split(",")) |
|---|
| | | .setControllerMappingHyphenStyle(true) |
|---|
| | | .setTablePrefix("m_"); |
|---|
| | | mpg.setStrategy(strategy); |
|---|
| | | mpg.setTemplateEngine(new FreemarkerTemplateEngine()); |
|---|
| | | //如果不调用该方法、就会使用MyBatis-Plus默认的文件生成路径和包路径生成文件、但可以使用上面的PackageConfig做一些简单的配置 |
|---|
| | | mpg.execute(); |
|---|
| | | 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(); |
|---|
| | | } |
|---|
| | | |
|---|
| | | /** |
|---|
