526互联

Mybatis-Plus代码生产器

发布时间 2023-04-03 18:28:31作者: 勇敢-的心

【问题描述】

在微服务的开发过程中，每次新创建一个微服务项目的时候，都需要手工添加controller、service、entity和mapper文件夹和文件时会比较麻烦，并且增加不少的工作量。

是否有一个模块，可以自动生成对应的目录和文件呢？真的有这样的代码生产器文件：

一、使用之前先确认POM文件中下面的依赖已经存在；

<dependency>
<groupId>com.baomidou</groupId>
<artifactId>mybatis-plus-boot-starter</artifactId>
<version>3.3.1.tmp</version>
</dependency>

<dependency>

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

<dependency>
<groupId>org.apache.velocity</groupId>
<artifactId>velocity-engine-core</artifactId>
<version>2.2</version>
</dependency>

 

二、然后，再把下面的文件拷贝到工程目录下面，运行main方式即可生产；

package com.cscecnf.construction.site;

import com.baomidou.mybatisplus.annotation.DbType;

import com.baomidou.mybatisplus.annotation.IdType;

import com.baomidou.mybatisplus.generator.AutoGenerator;

import com.baomidou.mybatisplus.generator.config.*;

import com.baomidou.mybatisplus.generator.config.rules.DateType;

import com.baomidou.mybatisplus.generator.config.rules.NamingStrategy;

// 演示例子，执行 main 方法控制台输入模块表名回车自动生成对应项目目录中

public class CodeGenerator {

/**
*     * <p>
* <p>
*     * 读取控制台内容
* <p>
*     * </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 + "！");

    }*/
public static void main(String[] args) {

// 代码生成器

AutoGenerator mpg = new AutoGenerator();

// 全局配置

GlobalConfig gc = new GlobalConfig();

String projectPath = System.getProperty("user.dir");

gc.setOutputDir("G:\\WorkSpace\\achievement\\smart-construction-site\\src\\main\\java\\com\\cscecnf\\construction\\site\\distribution\\box");

gc.setFileOverride(false);

gc.setServiceName("%sService");

gc.setIdType(IdType.ID_WORKER_STR);

gc.setDateType(DateType.ONLY_DATE);

gc.setSwagger2(true);

gc.setAuthor("long");

gc.setOpen(false);

// gc.setSwagger2(true); 实体属性 Swagger2 注解

mpg.setGlobalConfig(gc);

// 数据源配置

DataSourceConfig dsc = new DataSourceConfig();

dsc.setUrl("jdbc:mysql://192.168.98.168:3306/sicm?useUnicode=true&characterEncoding=utf8&allowMultiQueries=true&serverTimezone=GMT%2B8");

// dsc.setSchemaName("public");

dsc.setDriverName("com.mysql.cj.jdbc.Driver");

dsc.setUsername("root");

dsc.setPassword("123456");

dsc.setDbType(DbType.MYSQL);

mpg.setDataSource(dsc);

// 包配置

PackageConfig pc = new PackageConfig();

pc.setModuleName("mybatisplus");

pc.setParent("com.cscecnf.construction.site.distribution.box");

pc.setController("controller");

pc.setEntity("entity");

pc.setService("service");

pc.setMapper("mapper");

mpg.setPackageInfo(pc);

// 策略配置

StrategyConfig strategy = new StrategyConfig();

// strategy.setInclude("edu_teacher");

strategy.setNaming(NamingStrategy.underline_to_camel); // 数据库表映射到实体的命名策略

strategy.setTablePrefix(pc.getModuleName() + "_"); // 生成实体时去掉表前缀

strategy.setColumnNaming(NamingStrategy.underline_to_camel); // 数据库表字段映射到实体的命名策略

strategy.setEntityLombokModel(true); // lombok 模型 @Accessors(chain = true) setter链式操作

strategy.setRestControllerStyle(true); // restful api风格控制器

strategy.setControllerMappingHyphenStyle(true); // url中驼峰转连字符

mpg.setStrategy(strategy);

// 执行

mpg.execute();

}

}