6.Mybatis plus 代码生成器，mybatis-generator 插件使用

2019年10月15日 20:13 · 阅读(788) ·

说明

本文代码基于 5.基于 JWT(Json web token) 验证权限 项目代码进行修改，不过基本没关系

开发环境

名称 版本
操作系统 Windows 10 X64
JDK JDK1.8（jdk-8u151-windows-x64）
IntelliJ IDEA IntelliJ IDEA 2018.3
Maven Maven 3.6.0

根据开发文档创建表

1.开发文档

2.pom

  1. <!-- https://mvnrepository.com/artifact/cn.hutool/hutool-all -->
  2. <dependency>
  3.     <groupId>cn.hutool</groupId>
  4.     <artifactId>hutool-all</artifactId>
  5.     <version>5.7.9</version>
  6. </dependency>
  8. <dependency>
  9.     <groupId>org.apache.poi</groupId>
  10.     <artifactId>poi-ooxml</artifactId>
  11.     <version>4.1.0</version>
  12. </dependency>

3.Java 代码

  1. public static void main(String[] args) {
  3.     ExcelReader reader = ExcelUtil
  4.             .getReader(FileUtil.file("C:\\Users\\luoma\\Desktop\\融资贷款\\融资数据表v1.1.xlsx"),
  5.                     4);//配置路径和sheet页
  6.     int startRow = 202;//表名那行
  7.     int endRow = 218;//逻辑删除那行
  8.     List<List<Object>> read = reader.read(startRow-1,endRow-1);
  9.     String text = "CREATE TABLE `" + read.get(0).get(0) + "` ( `id` bigint(20) NOT NULL AUTO_INCREMENT COMMENT '自增主键ID',";
  10.     for (List<Object> o : read){
  11.         if (o.equals(read.get(0)) || o.equals(read.get(1)) || o.equals(read.get(2))){
  12.             continue;
  13.         }
  14.         if ("VARCHAR".equalsIgnoreCase(String.valueOf(o.get(1))) || "text".equalsIgnoreCase(String.valueOf(o.get(1)))){
  15.             String b = "`" + o.get(0) + "`" + " " + o.get(1) + "(" + o.get(2) + ") " + "COLLATE utf8mb4_bin DEFAULT NULL COMMENT " + "'" + o.get(3) + "',";
  16.             text += b;
  17.         }else if ("datetime".equalsIgnoreCase(String.valueOf(o.get(1))) || "date".equalsIgnoreCase(String.valueOf(o.get(1)))){
  18.             String b = "`" + o.get(0) + "`" + " " + o.get(1) + " " + "DEFAULT NULL COMMENT " + "'" + o.get(3) + "',";
  19.             text += b;
  20.         }else {
  21.             String b = "`" + o.get(0) + "`" + " " + o.get(1) + "(" + o.get(2) + ") " + "DEFAULT NULL COMMENT " + "'" + o.get(3) + "',";
  22.             text += b;
  23.         }
  25.     }
  26.     text += "PRIMARY KEY (`id`) USING BTREE,\n"
  27.             + "  KEY `tenant_id` (`tenant_id`) USING BTREE\n"
  28.             + ") ENGINE=InnoDB AUTO_INCREMENT=49 DEFAULT CHARSET=utf8mb4 COLLATE=utf8mb4_bin ROW_FORMAT=DYNAMIC COMMENT='" + read.get(0).get(0) + "';";
  29.     System.err.println(text);
  30. }

Mybatis plus 代码生成器

测试表-t_rbt_test

测试表-t_rbt_test

创建子模块-test-invoice-generator

test-invoice-cloud 上右键，新建模块

左侧面板选择 Maven，不勾选 Create from archetype 选项，如下图，点击 下一个 即可。

ArtifactId test-invoice-common

模块名称 test-invoice-common

创建类和文件

pom.xml

  1. <?xml version="1.0" encoding="UTF-8"?>
  2. <project xmlns="http://maven.apache.org/POM/4.0.0"
  3.          xmlns:xsi="http://www.w3.org/2001/XMLSchema-instance"
  4.          xsi:schemaLocation="http://maven.apache.org/POM/4.0.0 http://maven.apache.org/xsd/maven-4.0.0.xsd">
  5.     <parent>
  6.         <artifactId>test-invoice-cloud</artifactId>
  7.         <groupId>com-test-invoice</groupId>
  8.         <version>1.0-SNAPSHOT</version>
  9.     </parent>
  10.     <modelVersion>4.0.0</modelVersion>
  12.     <artifactId>test-invoice-generator</artifactId>
  13.     <name>test-invoice-generator</name>
  15.     <dependencies>
  16.         <dependency>
  17.             <groupId>org.springframework.boot</groupId>
  18.             <artifactId>spring-boot-starter-jdbc</artifactId>
  19.         </dependency>
  20.         <dependency>
  21.             <groupId>org.springframework.boot</groupId>
  22.             <artifactId>spring-boot-starter-web</artifactId>
  23.         </dependency>
  24.         <dependency>
  25.             <groupId>org.springframework.boot</groupId>
  26.             <artifactId>spring-boot-starter-test</artifactId>
  27.             <scope>test</scope>
  28.         </dependency>
  30.         <dependency>
  31.             <groupId>com.baomidou</groupId>
  32.             <artifactId>mybatis-plus-generator</artifactId>
  33.             <version>3.1.0</version>
  34.         </dependency>
  35.         <dependency>
  36.             <groupId>org.springframework.boot</groupId>
  37.             <artifactId>spring-boot-starter-freemarker</artifactId>
  38.             <version>2.1.4.RELEASE</version>
  39.         </dependency>
  40.         <dependency>
  41.             <groupId>org.freemarker</groupId>
  42.             <artifactId>freemarker</artifactId>
  43.             <!--<version>2.3.20</version>-->
  44.             <version>2.3.27-incubating</version>
  45.         </dependency>
  46.         <dependency>
  47.             <groupId>mysql</groupId>
  48.             <artifactId>mysql-connector-java</artifactId>
  49.             <scope>runtime</scope>
  50.         </dependency>
  51.     </dependencies>
  53.     <build>
  54.         <plugins>
  55.             <plugin>
  56.                 <groupId>org.springframework.boot</groupId>
  57.                 <artifactId>spring-boot-maven-plugin</artifactId>
  58.             </plugin>
  59.         </plugins>
  60.     </build>
  61. </project>

App.java

  1. package com.test.invoice;
  3. import com.baomidou.mybatisplus.core.toolkit.StringPool;
  4. import com.baomidou.mybatisplus.generator.AutoGenerator;
  5. import com.baomidou.mybatisplus.generator.InjectionConfig;
  6. import com.baomidou.mybatisplus.generator.config.*;
  7. import com.baomidou.mybatisplus.generator.config.po.TableInfo;
  8. import com.baomidou.mybatisplus.generator.config.rules.NamingStrategy;
  9. import com.baomidou.mybatisplus.generator.engine.FreemarkerTemplateEngine;
  10. import org.apache.commons.lang.StringUtils;
  12. import java.util.ArrayList;
  13. import java.util.List;
  14. import java.util.Scanner;
  16. /**
  17.  * 代码生成类
  18.  *
  19.  * @author:
  20.  * @version:
  21.  * @date: 2019-10-15 19:43
  22.  */
  23. public class App {
  25.     /*
  26.     代码生成路径
  27.      */
  28.     private static final String outdir="D:\\finance-generator-framework-code\\";
  30.     /**
  31.      * <p>
  32.      * 读取控制台内容
  33.      * </p>
  34.      */
  35.     public static String scanner(String tip) {
  36.         Scanner scanner = new Scanner(System.in);
  37.         StringBuilder help = new StringBuilder();
  38.         help.append("请输入" + tip + "：");
  39.         System.out.println(help.toString());
  40.         if (scanner.hasNext()) {
  41.             String ipt = scanner.next();
  42.             if (StringUtils.isNotEmpty(ipt)) {
  43.                 return ipt;
  44.             }
  45.         }
  46.         throw new RuntimeException("请输入正确的" + tip + "！");
  47.     }
  49.     /**
  50.      * RUN THIS
  51.      */
  52.     public static void main(String[] args) {
  53.         // 代码生成器
  54.         AutoGenerator mpg = new AutoGenerator();
  56.         // 全局配置
  57.         GlobalConfig gc = new GlobalConfig();
  58.         //String projectPath = System.getProperty("user.dir");
  59.         //gc.setOutputDir(projectPath + "/finance-generator-exp/src/main/java");
  60.         gc.setOutputDir(outdir);
  61.         gc.setAuthor("luoma");//作者
  62.         gc.setOpen(false);
  63.         gc.setBaseResultMap(true);
  64.         gc.setBaseColumnList(true);
  65.         gc.setMapperName("%sRepository");
  66.         gc.setServiceName("I%sService");
  67.         gc.setEntityName("%sEntity");
  68.         gc.setEnableCache(false);
  69.         gc.setFileOverride(true);
  71.         mpg.setGlobalConfig(gc);
  73.         // 数据源配置
  74.         DataSourceConfig dsc = new DataSourceConfig();
  75.         dsc.setUrl("jdbc:mysql://localhost:3306/luoma_test?useUnicode=true&useSSL=false&characterEncoding=utf8");
  76.         // dsc.setSchemaName("public");
  77.         dsc.setDriverName("com.mysql.jdbc.Driver");
  78.         dsc.setUsername("root");
  79.         dsc.setPassword("123456");
  80.         mpg.setDataSource(dsc);
  82.         // 包配置
  83.         PackageConfig pc = new PackageConfig();
  84.         pc.setModuleName(scanner("模块名"));
  85.         pc.setParent("com.test");
  86.         mpg.setPackageInfo(pc);
  88.         // 自定义配置
  89.         InjectionConfig cfg = new InjectionConfig() {
  90.             @Override
  91.             public void initMap() {
  92.                 // to do nothing
  93.             }
  94.         };
  95.         List<FileOutConfig> focList = new ArrayList<>();
  96.         focList.add(new FileOutConfig("/templates/mapper.xml.ftl") {
  97.             @Override
  98.             public String outputFile(TableInfo tableInfo) {
  99.                 // 自定义输入文件名称
  100.                /* return projectPath + "/finance-generator-exp/src/main/resources/mapper/" + pc.getModuleName()
  101.                         + "/" + tableInfo.getEntityName() + "Mapper" + StringPool.DOT_XML;*/
  102.                 return outdir +"com/test/invoice/"+pc.getModuleName()+"/mapper"
  103.                         + "/" + tableInfo.getEntityName() + "Mapper" + StringPool.DOT_XML;
  104.             }
  105.         });
  106.         cfg.setFileOutConfigList(focList);
  107.         mpg.setCfg(cfg);
  108.         mpg.setTemplate(new TemplateConfig().setXml(null));
  110.         // 策略配置
  111.         StrategyConfig strategy = new StrategyConfig();
  112.         strategy.setNaming(NamingStrategy.underline_to_camel);// 表名生成策略
  113.         strategy.setColumnNaming(NamingStrategy.underline_to_camel);//字段名生成策略
  114.         strategy.setSuperEntityClass("com.test.invoice.model.base.SuperEntity");// 自定义实体父类
  115.         strategy.setEntityLombokModel(true);
  116.         strategy.setInclude(scanner("表名")); // 需要生成的表
  117.         strategy.setSuperEntityColumns("id");// 自定义实体，公共字段
  118.         strategy.setControllerMappingHyphenStyle(true);
  119.         strategy.setTablePrefix(pc.getModuleName() + "_");// 表前缀
  120.         strategy.setRestControllerStyle(true);
  122.         mpg.setStrategy(strategy);
  123.         // 选择 freemarker 引擎需要指定如下加，注意 pom 依赖必须有！
  124.         mpg.setTemplateEngine(new FreemarkerTemplateEngine());
  125.         mpg.execute();
  126.     }
  127. }

测试

1.运行 App main()

2.输入模块名称和表名称

  1. "D:\Program Files\Java\jdk1.8.0_151\bin\java.exe"
  2. 请输入模块名：
  3. invoice
  4. 请输入表名：
  5. t_rbt_test
  6. 20:23:40.415 [main] DEBUG com.baomidou.mybatisplus.generator.AutoGenerator - ==========================准备生成文件...==========================
  7. 20:23:41.411 [main] DEBUG com.baomidou.mybatisplus.generator.engine.AbstractTemplateEngine - 创建目录： [D:\finance-generator-framework-code\com\test\invoice\entity]
  8. 20:23:41.425 [main] DEBUG com.baomidou.mybatisplus.generator.engine.AbstractTemplateEngine - 创建目录： [D:\finance-generator-framework-code\com\test\invoice\controller]
  9. 20:23:41.426 [main] DEBUG com.baomidou.mybatisplus.generator.engine.AbstractTemplateEngine - 创建目录： [D:\finance-generator-framework-code\com\test\invoice\service]
  10. 20:23:41.426 [main] DEBUG com.baomidou.mybatisplus.generator.engine.AbstractTemplateEngine - 创建目录： [D:\finance-generator-framework-code\com\test\invoice\mapper]
  11. 20:23:41.426 [main] DEBUG com.baomidou.mybatisplus.generator.engine.AbstractTemplateEngine - 创建目录： [D:\finance-generator-framework-code\com\test\invoice\service\impl]
  12. 20:23:41.524 [main] DEBUG com.baomidou.mybatisplus.generator.engine.AbstractTemplateEngine - 模板:/templates/mapper.xml.ftl;  文件:D:\finance-generator-framework-code\com/test/invoice/invoice/mapper/TRbtTestEntityMapper.xml
  13. 20:23:41.615 [main] DEBUG com.baomidou.mybatisplus.generator.engine.AbstractTemplateEngine - 模板:/templates/entity.java.ftl;  文件:D:\finance-generator-framework-code\com\test\invoice\entity\TRbtTestEntity.java
  14. 20:23:41.620 [main] DEBUG com.baomidou.mybatisplus.generator.engine.AbstractTemplateEngine - 模板:/templates/mapper.java.ftl;  文件:D:\finance-generator-framework-code\com\test\invoice\mapper\TRbtTestRepository.java
  15. 20:23:41.625 [main] DEBUG com.baomidou.mybatisplus.generator.engine.AbstractTemplateEngine - 模板:/templates/service.java.ftl;  文件:D:\finance-generator-framework-code\com\test\invoice\service\ITRbtTestService.java
  16. 20:23:41.635 [main] DEBUG com.baomidou.mybatisplus.generator.engine.AbstractTemplateEngine - 模板:/templates/serviceImpl.java.ftl;  文件:D:\finance-generator-framework-code\com\test\invoice\service\impl\TRbtTestServiceImpl.java
  17. 20:23:41.640 [main] DEBUG com.baomidou.mybatisplus.generator.engine.AbstractTemplateEngine - 模板:/templates/controller.java.ftl;  文件:D:\finance-generator-framework-code\com\test\invoice\controller\TRbtTestController.java
  18. 20:23:41.640 [main] DEBUG com.baomidou.mybatisplus.generator.AutoGenerator - ==========================文件生成完成！！！==========================
  20. 进程已结束，退出代码 0

3.查看生成的文件

D:\finance-generator-framework-code\com\test\invoice

(1)TRbtTestController.java

D:\finance-generator-framework-code\com\test\invoice\controller

  1. package com.test.invoice.controller;
  4. import org.springframework.web.bind.annotation.RequestMapping;
  6. import org.springframework.web.bind.annotation.RestController;
  8. /**
  9.  * <p>
  10.  *  前端控制器
  11.  * </p>
  12.  *
  13.  * @author luoma
  14.  * @since 2019-10-15
  15.  */
  16. @RestController
  17. @RequestMapping("/invoice/t-rbt-test-entity")
  18. public class TRbtTestController {
  20. }

(2)TRbtTestEntity.java

D:\finance-generator-framework-code\com\test\invoice\entity

  1. package com.test.invoice.entity;
  3. import com.baomidou.mybatisplus.annotation.TableName;
  4. import java.time.LocalDateTime;
  5. import com.test.invoice.model.base.SuperEntity;
  6. import lombok.Data;
  7. import lombok.EqualsAndHashCode;
  8. import lombok.experimental.Accessors;
  10. /**
  11.  * <p>
  12.  * 
  13.  * </p>
  14.  *
  15.  * @author luoma
  16.  * @since 2019-10-15
  17.  */
  18. @Data
  19. @EqualsAndHashCode(callSuper = true)
  20. @Accessors(chain = true)
  21. @TableName("t_rbt_test")
  22. public class TRbtTestEntity extends SuperEntity {
  24.     private static final long serialVersionUID = 1L;
  26.     /**
  27.      * 名称
  28.      */
  29.     private String name;
  31.     /**
  32.      * 版本号
  33.      */
  34.     private String version;
  36.     /**
  37.      * 创建人
  38.      */
  39.     private String createBy;
  41.     /**
  42.      * 创建时间
  43.      */
  44.     private LocalDateTime createTime;
  46.     /**
  47.      * 修改人
  48.      */
  49.     private String updateBy;
  51.     /**
  52.      * 修改时间
  53.      */
  54.     private LocalDateTime updateTime;
  56.     /**
  57.      * 是否删除
  58.      */
  59.     private Integer isDelete;
  61.     /**
  62.      * 系统时间戳
  63.      */
  64.     private LocalDateTime bizTime;
  67. }

(3)ITRbtTestService.java

D:\finance-generator-framework-code\com\test\invoice\service

  1. package com.test.invoice.service;
  3. import com.test.invoice.entity.TRbtTestEntity;
  4. import com.baomidou.mybatisplus.extension.service.IService;
  6. /**
  7.  * <p>
  8.  *  服务类
  9.  * </p>
  10.  *
  11.  * @author luoma
  12.  * @since 2019-10-15
  13.  */
  14. public interface ITRbtTestService extends IService<TRbtTestEntity> {
  16. }

(4)TRbtTestServiceImpl.java

D:\finance-generator-framework-code\com\test\invoice\service\impl

  1. package com.test.invoice.service.impl;
  3. import com.test.invoice.entity.TRbtTestEntity;
  4. import com.test.invoice.mapper.TRbtTestRepository;
  5. import com.test.invoice.service.ITRbtTestService;
  6. import com.baomidou.mybatisplus.extension.service.impl.ServiceImpl;
  7. import org.springframework.stereotype.Service;
  9. /**
  10.  * <p>
  11.  *  服务实现类
  12.  * </p>
  13.  *
  14.  * @author luoma
  15.  * @since 2019-10-15
  16.  */
  17. @Service
  18. public class TRbtTestServiceImpl extends ServiceImpl<TRbtTestRepository, TRbtTestEntity> implements ITRbtTestService {
  20. }

(5)TRbtTestRepository.java

D:\finance-generator-framework-code\com\test\invoice\mapper

  1. package com.test.invoice.mapper;
  3. import com.test.invoice.entity.TRbtTestEntity;
  4. import com.baomidou.mybatisplus.core.mapper.BaseMapper;
  6. /**
  7.  * <p>
  8.  *  Mapper 接口
  9.  * </p>
  10.  *
  11.  * @author luoma
  12.  * @since 2019-10-15
  13.  */
  14. public interface TRbtTestRepository extends BaseMapper<TRbtTestEntity> {
  16. }

(6)TRbtTestEntityMapper.xml

D:\finance-generator-framework-code\com\test\invoice\invoice\mapper

  1. <?xml version="1.0" encoding="UTF-8"?>
  2. <!DOCTYPE mapper PUBLIC "-//mybatis.org//DTD Mapper 3.0//EN" "http://mybatis.org/dtd/mybatis-3-mapper.dtd">
  3. <mapper namespace="com.test.invoice.mapper.TRbtTestRepository">
  5.     <!-- 通用查询映射结果 -->
  6.     <resultMap id="BaseResultMap" type="com.test.invoice.entity.TRbtTestEntity">
  7.     <result column="id" property="id" />
  8.         <result column="name" property="name" />
  9.         <result column="version" property="version" />
  10.         <result column="create_by" property="createBy" />
  11.         <result column="create_time" property="createTime" />
  12.         <result column="update_by" property="updateBy" />
  13.         <result column="update_time" property="updateTime" />
  14.         <result column="is_delete" property="isDelete" />
  15.         <result column="biz_time" property="bizTime" />
  16.     </resultMap>
  18.     <!-- 通用查询结果列 -->
  19.     <sql id="Base_Column_List">
  20.         id,
  21.         name, version, create_by, create_time, update_by, update_time, is_delete, biz_time
  22.     </sql>
  24. </mapper>

mybatis-generator 插件使用

安装插件

Database 工具操作

连接数据库

这里我连接的是 Oracle

属性 描述
名称 根据业务情况起一个易记的名称
主机 数据库 IP 地址
端口 数据库端口号
SID 服务名称，TNS 中的 SERVICE_NAME
用户 用户名
密码 密码
url jdbc:oracle:thin:@IP:端口号:服务名

找到对应的表，右键 mybatis-generator

选择对应选项

对应目录下生成了代码

Mybatis X 插件

https://baomidou.com/guides/mybatis-x/
MybatisX 是一款专为 IntelliJ IDEA 设计的快速开发插件，旨在提升 MyBatis 与 MyBatis-Plus 框架的开发效率。

IDEA DB 连接 Oracle 错误总结

ORA-12505, TNS:listener does not currently know of SID

参考：idea链接oracle数据库报错：[66000][12505] Listener refused the connection with the following error:

错误描述

  1. [66000][12505] Listener refused the connection with the following error: ORA-12505, TNS:listener does not currently know of SID given in connect descriptor (CONNECTION_ID=Bn8THbnmRYmuHin4mmnWzw==) oracle.net.ns.NetException: Listener refused the connection with the following error: ORA-12505, TNS:listener does not currently know of SID given in connect descriptor (CONNECTION_ID=Bn8THbnmRYmuHin4mmnWzw==).

查询 sid

  1. select INSTANCE_NAME from v$instance;

上图中 SID 选项替换为查询结果即可

ORA-01005: 给出空口令; 登录被拒绝

问题描述

  1. [72000][1005] ORA-01005: 给出空口令; 登录被拒绝

找了半天，妈的原来是密码输入错误了导致的，输入正确的密码即可