Springboot Mybatis Plus自動生成工具類詳解代碼
前言
代碼生成器,也叫逆向工程,是根據數據庫裡的表結構,自動生成對應的實體類、映射文件和接口。
看到很多小夥伴在為數據庫生成實體類發愁,現分享給大傢,提高開發效率。
一、pom依賴
<dependency> <groupId>com.baomidou</groupId> <artifactId>mybatis-plus-boot-starter</artifactId> <version>3.4.1</version> </dependency> <dependency> <groupId>com.baomidou</groupId> <artifactId>mybatis-plus-generator</artifactId> <version>3.4.1</version> </dependency> <dependency> <groupId>org.freemarker</groupId> <artifactId>freemarker</artifactId> <version>2.3.30</version> </dependency>
二、工具類
package com.his.utils; 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.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.Scanner; /** * Mybatis plus代碼自動生成 */ public class MybatisPlusUtil { /** 作者 */ public static final String AUTHOR = "dd"; /** 類命名 */ /** * Entity命名 */ public static final String FILE_NAME_ENTITY = "%sEntity"; /** * MAPPER命名 */ public static final String FILE_NAME_MAPPER = "%sMapper"; /** * xml命名 */ public static final String FILE_NAME_XML = "%sMapper"; /** * Service命名 */ public static final String FILE_NAME_SERVICE = "%sService"; /** * ServiceImpl命名 */ public static final String FILE_NAME_SERVICE_IMPL = "%sDO"; /** * Controller命名 */ public static final String FILE_NAME_CONTROLLER = "%sController"; /** 包命名,可以根據自己的項目情況自定義生成後的存放路徑 entity默認路徑為父目錄.entity mapper默認路徑為父目錄.mapper service默認路徑為父目錄.service serviceImpl默認路徑為父目錄.service.impl controller默認路徑為父目錄.controller */ /** * PARENT命名 */ public static final String PACKAGE_NAME_PARENT = "com.his"; /** * Entity命名 */ public static final String PACKAGE_NAME_ENTITY = "repository.entity.control"; /** * MAPPER命名 */ public static final String PACKAGE_NAME_MAPPER = "repository.mapper.control"; /** * xml命名 */ public static final String PACKAGE_NAME_XML = "sys"; /** * Service命名 */ public static final String PACKAGE_NAME_SERVICE = "domain.control"; /** * ServiceImpl命名 */ public static final String PACKAGE_NAME_SERVICE_IMPL = "domain.control"; /** * Controller命名 */ public static final String PACKAGE_NAME_CONTROLLER = "facade.controller.control"; /** * 讀取控制臺內容 */ private 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 + "!"); } /** * 運行這個main方法進行代碼生成 */ public static void main(String[] args) { // 代碼生成器 AutoGenerator mpg = new AutoGenerator(); // 全局配置 GlobalConfig gc = new GlobalConfig(); String projectPath = System.getProperty("user.dir"); gc.setOutputDir(projectPath + "/src/main/java"); gc.setFileOverride(true); gc.setAuthor(AUTHOR); gc.setOpen(false); gc.setActiveRecord(false);// 不需要ActiveRecord特性的請改為false gc.setEnableCache(false);// XML 二級緩存 gc.setSwagger2(true); // 實體屬性 Swagger2 註解 gc.setBaseResultMap(true); gc.setBaseColumnList(true); gc.setEntityName(FILE_NAME_ENTITY); gc.setMapperName(FILE_NAME_MAPPER); gc.setXmlName(FILE_NAME_XML); gc.setServiceName(FILE_NAME_SERVICE); gc.setServiceImplName(FILE_NAME_SERVICE_IMPL); gc.setControllerName(FILE_NAME_CONTROLLER); mpg.setGlobalConfig(gc); // 數據源配置 DataSourceConfig dsc = new DataSourceConfig(); dsc.setUrl("jdbc:oracle:thin:@ip:port/test"); dsc.setDriverName("oracle.jdbc.OracleDriver"); dsc.setUsername("user"); dsc.setPassword("pass"); mpg.setDataSource(dsc); //包配置 PackageConfig pc = new PackageConfig(); pc.setModuleName(null); pc.setParent(PACKAGE_NAME_PARENT); pc.setController(PACKAGE_NAME_CONTROLLER); pc.setService(PACKAGE_NAME_SERVICE); pc.setServiceImpl(PACKAGE_NAME_SERVICE_IMPL); pc.setMapper(PACKAGE_NAME_MAPPER); pc.setEntity(PACKAGE_NAME_ENTITY); pc.setXml(PACKAGE_NAME_XML); mpg.setPackageInfo(pc); // 策略配置 StrategyConfig strategy = new StrategyConfig(); strategy.setNaming(NamingStrategy.underline_to_camel); strategy.setColumnNaming(NamingStrategy.underline_to_camel); //strategy.setEntityLombokModel(true); strategy.setRestControllerStyle(true); strategy.setInclude(scanner("表名,多個英文逗號分割").split(",")); strategy.setControllerMappingHyphenStyle(true); // 設置表前綴 strategy.setTablePrefix("IEMR_"); mpg.setStrategy(strategy); mpg.setTemplateEngine(new FreemarkerTemplateEngine()); // 自定義配置 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 + "/src/main/resources/mapper/" + "/" + tableInfo.getMapperName() + StringPool.DOT_XML; } }); cfg.setFileOutConfigList(focList); mpg.setCfg(cfg); // 配置模板 TemplateConfig templateConfig = new TemplateConfig(); templateConfig.setXml(null); mpg.setTemplate(templateConfig); mpg.execute(); } }
結尾
感謝大傢的耐心閱讀,如有建議請私信或評論留言。如有收獲,勞煩支持,關註、點贊、評論、收藏均可,博主會經常更新,與大傢共同進步
到此這篇關於Springboot Mybatis Plus自動生成工具類詳解代碼的文章就介紹到這瞭,更多相關Springboot Mybatis Plus 生成工具類內容請搜索WalkonNet以前的文章或繼續瀏覽下面的相關文章希望大傢以後多多支持WalkonNet!
推薦閱讀:
- SpringBoot項目使用mybatis-plus逆向自動生成全套代碼
- mybatis plus generator 根據數據庫自動生成實體類的實現示例
- 使用代碼生成器自定義Entity的部分註解
- Mybatis-Plus開發提速器mybatis-plus-generator-ui詳解
- 一篇超詳細的SpringBoot整合MybatisPlus的文章