package com.cxd.stock.tools;
import com.baomidou.mybatisplus.annotation.DbType;
import com.baomidou.mybatisplus.annotation.IdType;
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.rules.DateType;
import com.baomidou.mybatisplus.generator.config.rules.NamingStrategy;
import com.baomidou.mybatisplus.generator.engine.FreemarkerTemplateEngine;
import com.cxd.stock.StockSeataApplication;
import org.junit.Test;
import org.junit.runner.RunWith;
import org.springframework.boot.test.context.SpringBootTest;
import org.springframework.test.context.junit4.SpringRunner;
/**
* @创建人: 陈晓东
* @创建时间: 2023/1/10
* @email: 2774398338@qq.com
* @类描述:
*/
//执行 main 方法,控制台输入模块表名,回车自动生成对应项目目录中
@SpringBootTest(classes = StockSeataApplication.class)
@RunWith(SpringRunner.class)
//执行 main 方法,控制台输入模块表名,回车自动生成对应项目目录中
public class MybatisPlusCodeGenerator {
@Test
public void generatorCode() {
//====================配置变量区域=====================//
String author="";//生成文件的作者,可以不填
String rootPackage="com.cxd.stock";//生成的entity、controller、service等包所在的公共上一级包路径全限定名
String moduleName=""; // 生成的文件所在module, 这里为空即可
//数据库配置
String url="jdbc:mysql://localhost:4306/seata_sotck?useSSL=false&characterEncoding=utf8";
String driverClassName="com.mysql.cj.jdbc.Driver";//com.mysql.jdbc.Driver或者com.mysql.cj.jdbc.Driver
String username="root";
String password="123456";
String tableNames="stock_tbl";//表名,多个使用,分隔
//====================配置变量区域=====================//
// 代码生成器
AutoGenerator generator = new AutoGenerator();
// 全局配置
GlobalConfig globalConfig = new GlobalConfig();
String projectPath = System.getProperty("user.dir");
globalConfig.setOutputDir(projectPath +"\\"+moduleName+"\\src\\main\\java");
globalConfig.setFileOverride(false);//是否覆盖已有文件,默认false
globalConfig.setOpen(false);//是否打开输出目录
globalConfig.setAuthor(author);
globalConfig.setServiceName("%sService");//去掉service接口的首字母I
globalConfig.setBaseResultMap(true);//开启 BaseResultMap
globalConfig.setDateType(DateType.ONLY_DATE);//只使用 java.util.date代替
globalConfig.setIdType(IdType.AUTO);//分配ID (主键类型为number或string)
generator.setGlobalConfig(globalConfig);
// 数据源配置
DataSourceConfig dataSourceConfig = new DataSourceConfig();
dataSourceConfig.setUrl(url);
dataSourceConfig.setDbType(DbType.MYSQL);//数据库类型
dataSourceConfig.setDriverName(driverClassName);
dataSourceConfig.setUsername(username);
dataSourceConfig.setPassword(password);
generator.setDataSource(dataSourceConfig);
// 包配置
PackageConfig packageConfig = new PackageConfig();
//packageConfig.setModuleName(scanner("模块名"));
packageConfig.setParent(rootPackage);//例:org.jeecg.modules.xqxy
generator.setPackageInfo(packageConfig);
//注意:模板引擎在mybatisplus依赖中的templates目录下,可以依照此默认模板进行自定义
// 策略配置:配置根据哪张表生成代码
StrategyConfig strategy = new StrategyConfig();
strategy.setInclude(tableNames);//表名,多个英文逗号分割(与exclude二选一配置)
strategy.setNaming(NamingStrategy.underline_to_camel); // 将表名的_转为驼峰命名
strategy.setColumnNaming(NamingStrategy.underline_to_camel); // 将字段转为驼峰命名
//strategy.setSuperEntityClass("你自己的父类实体,没有就不用设置!");
strategy.setEntityLombokModel(true);//lombok模型,@Accessors(chain = true)setter链式操作
strategy.setRestControllerStyle(true);//controller生成@RestController
strategy.setEntityTableFieldAnnotationEnable(true);//是否生成实体时,生成字段注解
generator.setStrategy(strategy);
generator.setTemplateEngine(new FreemarkerTemplateEngine()); // 生成模板
generator.execute();
}
}
mybatis 代码生成器
最新推荐文章于 2024-07-20 23:59:39 发布