SpringBoot整合Mabatis-plus程式碼生成器（附原始碼：僅修改連線的資料庫可自動生成）
阿新 • • 發佈：2022-03-27
1、原始碼連結：
https://files.cnblogs.com/files/blogs/721495/springboot-mp-generator.zip?t=1648361457
2、Mabatis-plus程式碼生成器可以生成啥？
controller層、entity實體類、mapper（extends BaseMapper<PmsCategory>）、service層（extends IService<PmsCategory>、extends ServiceImpl<PmsCategoryMapper, PmsCategory> implements PmsCategoryService）
3、主類CodeGenerator改動的地方以及相關配置：
只需要改變CodeGenerator主類中資料庫的連線配置，就可以自動生成資料表的實體類以及其他增刪改查操作；
package com.example.springbootmpgenerator;
import java.util.ArrayList;
import java.util.List;
import com.baomidou.mybatisplus.core.toolkit.StringPool;
import com.baomidou.mybatisplus.generator.AutoGenerator;
import com.baomidou.mybatisplus.generator.InjectionConfig;
import com.baomidou.mybatisplus.generator.config.ConstVal;
import com.baomidou.mybatisplus.generator.config.DataSourceConfig;
import com.baomidou.mybatisplus.generator.config.FileOutConfig;
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.po.TableInfo;
import com.baomidou.mybatisplus.generator.config.rules.NamingStrategy;
import com.baomidou.mybatisplus.generator.engine.FreemarkerTemplateEngine;
/**
* 程式碼生成器 ，先修改下面的常量配置引數，然後執行 main方法
*/
public class CodeGenerator {
// 資料庫連線配置
private static final String JDBC_DRIVER = "com.mysql.cj.jdbc.Driver";
private static final String JDBC_URL = "jdbc:mysql://localhost:3306/category?useSSL=false&serverTimezone=Hongkong&useUnicode=true&characterEncoding=utf-8&autoReconnect=true&allowPublicKeyRetrieval=true&tinyInt1isBit=false";
private static final String JDBC_USER_NAME = "root";
private static final String JDBC_PASSOWRD = "123456";
// 包名和模組名
private static final String PACKAGE_NAME = "com.smile";
private static final String MODULE_NAME = "category"; //為啥這邊叫模組名哈
// 表名，多個表使用英文逗號分割 表名原先就有的
private static final String TBL_NAMES = "pms_category";
// 表名的字首，從表生成程式碼時會去掉字首
private static final String TABLE_PREFIX = "category_";
// 生成程式碼入口main方法
public static void main(String[] args) {
// 0.程式碼生成器
AutoGenerator mpg = new AutoGenerator();
// 1.全域性配置
GlobalConfig gc = getGlobalConfig();
mpg.setGlobalConfig(gc);
// 2.資料來源配置
DataSourceConfig dsc = getDataSourceConfig();
mpg.setDataSource(dsc);
// 3.包配置
PackageConfig pc = getPackageConfig();
mpg.setPackageInfo(pc);
// 4.自定義配置
InjectionConfig cfg = getInjectionConfig();
mpg.setCfg(cfg);
// 5.模板配置
TemplateConfig templateConfig = getTemplateConfig();
mpg.setTemplate(templateConfig);
// 使用Freemarker模板引擎
mpg.setTemplateEngine(new FreemarkerTemplateEngine());
// 6.策略配置
StrategyConfig strategy = getStrategyConfig();
mpg.setStrategy(strategy);
// 7.開始生成程式碼
mpg.execute();
}
/** 1.全域性配置 */
private static GlobalConfig getGlobalConfig() {
GlobalConfig gc = new GlobalConfig();
String projectPath = System.getProperty("user.dir"); //E:\techstack\demo\springboot-mp-generator/
gc.setOutputDir(projectPath + "/src/main/java"); //輸出的路徑/src/main/java + 包名模組名
gc.setAuthor("yuwen"); //* @author yuwen
gc.setOpen(false);
// 自定義生成的ServiceName，去掉預設的ServiceName前面的I
gc.setServiceName("%s" + ConstVal.SERVICE); //IRoleService
// gc.setSwagger2(true); 實體屬性 Swagger2 註解
return gc;
}
/** 2.資料來源配置 */
private static DataSourceConfig getDataSourceConfig() {
DataSourceConfig dsc = new DataSourceConfig();
dsc.setDriverName(JDBC_DRIVER);
dsc.setUrl(JDBC_URL);
// dsc.setSchemaName("public");
dsc.setUsername(JDBC_USER_NAME);
dsc.setPassword(JDBC_PASSOWRD);
return dsc;
}
/** 3.包配置 */
private static PackageConfig getPackageConfig() {
PackageConfig pc = new PackageConfig();
// 生成PACKAGE_NAME.MODULE_NAME的包路徑
pc.setParent(PACKAGE_NAME);
pc.setModuleName(MODULE_NAME);
return pc;
}
/** 4.自定義配置 */
private static InjectionConfig getInjectionConfig() {
// 這裡模板引擎使用的是freemarker freemarker是一套前端模板引擎
String templatePath = "/templates/mapper.xml.ftl";
// 自定義輸出配置
List<FileOutConfig> focList = new ArrayList<>();
String projectPath = System.getProperty("user.dir");
// 自定義配置會被優先輸出
focList.add(new FileOutConfig(templatePath) {
@Override
public String outputFile(TableInfo tableInfo) {
// 自定義輸出檔名 ， 如果你 Entity 設定了前後綴、此處注意 xml 的名稱會跟著發生變化
return projectPath + "/src/main/resources/mapper/" + tableInfo.getEntityName() + "Mapper"
+ StringPool.DOT_XML;
}
});
InjectionConfig cfg = new InjectionConfig() {
@Override
public void initMap() {
// to do nothing
}
};
cfg.setFileOutConfigList(focList);
return cfg;
}
/** 5.模板配置 */
private static TemplateConfig getTemplateConfig() {
TemplateConfig templateConfig = new TemplateConfig();
// 配置自定義輸出模板
// 指定自定義模板路徑，注意不要帶上.ftl/.vm, 會根據使用的模板引擎自動識別
// templateConfig.setEntity("templates/entity2.java");
// templateConfig.setService();
// templateConfig.setController();
templateConfig.setXml(null);
return templateConfig;
}
/** 6.策略配置 */
private static StrategyConfig getStrategyConfig() {
StrategyConfig strategy = new StrategyConfig();
// 下劃線轉駝峰命名
strategy.setNaming(NamingStrategy.underline_to_camel);
strategy.setColumnNaming(NamingStrategy.underline_to_camel);
strategy.setEntityLombokModel(false);
strategy.setRestControllerStyle(true);
strategy.setInclude(TBL_NAMES.split(","));
strategy.setControllerMappingHyphenStyle(true);
strategy.setTablePrefix(TABLE_PREFIX);
return strategy;
}
}4、後續在application.properties配置資料庫連線
5、總結
SpringBoot整合Mabatis-plus程式碼生成器，以後再也不用自己去寫最基本的CRUD了；
少年的志向，不應該是房子,他們應該伏案疾書,或為心中的夢想而揮灑汗水，暢想著自己未來光明的人生，少年的夢想，也不應該是生活,他們應該想要集齊七顆龍珠,或者幻想著擁有一顆皮卡丘。歡迎關注微信公眾號<彭曉琪>