springboot整合mybatis-plus逆向工程

時間 2019-12-11

標籤 springboot 整合 mybatis 逆向工程欄目 Spring 简体版

原文原文鏈接

MyBatis-Plus（簡稱 MP）是一個 MyBatis 的加強工具，在 MyBatis 的基礎上只作加強不作改變，爲簡化開發、提升效率而生。官方文檔html

代碼生成器

AutoGenerator 是 MyBatis-Plus 的代碼生成器，經過 AutoGenerator 能夠快速生成 Entity、Mapper、Mapper XML、Service、Controller 等各個模塊的代碼，極大的提高了開發效率。java

mybatis-plus是根據數據庫表來生成對應的實體類，首先咱們建立數據庫表Usermysql

id	name	age	email
1	Jone	18	test1@baomidou.com
2	Jack	20	test2@baomidou.com
3	Tom	28	test3@baomidou.com
4	Sandy	21	test4@baomidou.com
5	Billie	24	test5@baomidou.com

其對應的數據庫 Schema 腳本以下：git

DROP TABLE IF EXISTS user;

CREATE TABLE user
(
    id BIGINT(20) NOT NULL COMMENT '主鍵ID',
    name VARCHAR(30) NULL DEFAULT NULL COMMENT '姓名',
    age INT(11) NULL DEFAULT NULL COMMENT '年齡',
    email VARCHAR(50) NULL DEFAULT NULL COMMENT '郵箱',
    PRIMARY KEY (id)
);

其對應的數據庫 Data 腳本以下：github

DELETE FROM user;

INSERT INTO user (id, name, age, email) VALUES
(1, 'Jone', 18, 'test1@baomidou.com'),
(2, 'Jack', 20, 'test2@baomidou.com'),
(3, 'Tom', 28, 'test3@baomidou.com'),
(4, 'Sandy', 21, 'test4@baomidou.com'),
(5, 'Billie', 24, 'test5@baomidou.com');

初始化springboot工程

其中mpconfig就是咱們逆向工程配置文件web

基本依賴以下：spring

<dependencies>
        <dependency>
            <groupId>org.springframework.boot</groupId>
            <artifactId>spring-boot-starter-web</artifactId>
        </dependency>

        <dependency>
            <groupId>org.springframework.boot</groupId>
            <artifactId>spring-boot-starter-test</artifactId>
            <scope>test</scope>
        </dependency>
        <dependency>
            <groupId>mysql</groupId>
            <artifactId>mysql-connector-java</artifactId>
            <scope>runtime</scope>
        </dependency>
      <dependency>
            <groupId>org.projectlombok</groupId>
            <artifactId>lombok</artifactId>
            <optional>true</optional>
        </dependency>
    </dependencies>

下面開始引入逆向工程依賴sql

MyBatis-Plus 從 3.0.3 以後移除了代碼生成器與模板引擎的默認依賴，須要手動添加相關依賴：數據庫

<dependency>
            <groupId>com.baomidou</groupId>
            <artifactId>mybatis-plus-boot-starter</artifactId>
            <version>3.1.1</version>
        </dependency>
        <!--添加 代碼生成器 依賴-->
        <dependency>
            <groupId>com.baomidou</groupId>
            <artifactId>mybatis-plus-generator</artifactId>
            <version>3.1.1</version>
        </dependency>

添加模板引擎依賴，MyBatis-Plus 支持 Velocity（默認）、Freemarker、Beetl，用戶能夠選擇本身熟悉的模板引擎，若是都不知足您的要求，能夠採用自定義模板引擎。
Velocity（默認）：apache

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

Freemarker：

<dependency>
    <groupId>org.freemarker</groupId>
    <artifactId>freemarker</artifactId>
    <version>2.3.28</version>
</dependency>

這裏我選擇Freemarker
注意！若是您選擇了非默認引擎，須要在 AutoGenerator 中設置模板引擎

所有依賴以下：

<dependencies>
        <dependency>
            <groupId>org.springframework.boot</groupId>
            <artifactId>spring-boot-starter-web</artifactId>
        </dependency>

        <dependency>
            <groupId>org.springframework.boot</groupId>
            <artifactId>spring-boot-starter-test</artifactId>
            <scope>test</scope>
        </dependency>

        <!-- freemarker 模板引擎 -->
        <dependency>
            <groupId>org.freemarker</groupId>
            <artifactId>freemarker</artifactId>
            <version>2.3.23</version>
        </dependency>
        <dependency>
            <groupId>com.baomidou</groupId>
            <artifactId>mybatis-plus-boot-starter</artifactId>
            <version>3.1.1</version>
        </dependency>
        <!--添加 代碼生成器 依賴-->
        <dependency>
            <groupId>com.baomidou</groupId>
            <artifactId>mybatis-plus-generator</artifactId>
            <version>3.1.1</version>
        </dependency>
        <dependency>
            <groupId>org.projectlombok</groupId>
            <artifactId>lombok</artifactId>
            <optional>true</optional>
        </dependency>
        <dependency>
            <groupId>mysql</groupId>
            <artifactId>mysql-connector-java</artifactId>
            <scope>runtime</scope>
        </dependency>
    </dependencies>

下面開始：建立逆向工程配置類mpconfig

package com.jiangfeixiang.mpdemo.mpconfig;

import com.baomidou.mybatisplus.annotation.DbType;
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.converts.MySqlTypeConvert;
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.*;

/**
 * @ProjectName: mybatis-plus
 * @Package: com.jiangfeixiang.mybatisplus.mpconfig
 * @ClassName: CodeGenerator
 * @Author: jiangfeixiang
 * @email: 1016767658@qq.com
 * @Description: 代碼生成器
 * @Date: 2019/5/10/0010 21:41
 */
public class CodeGenerator {

    /**
     * 讀取控制檯內容
     */
    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 globalConfig = new GlobalConfig();
        //生成文件的輸出目錄
        String projectPath = System.getProperty("user.dir");
        globalConfig.setOutputDir(projectPath + "/src/main/java");
        //Author設置做者
        globalConfig.setAuthor("姜飛祥");
        //是否覆蓋文件
        globalConfig.setFileOverride(true);
        //生成後打開文件
        globalConfig.setOpen(false);
        mpg.setGlobalConfig(globalConfig);

        /**
         * 數據源配置
          */
        DataSourceConfig dataSourceConfig = new DataSourceConfig();
        // 數據庫類型,默認MYSQL
        dataSourceConfig.setDbType(DbType.MYSQL);
        //自定義數據類型轉換
        dataSourceConfig.setTypeConvert(new MySqlTypeConvert());
        dataSourceConfig.setUrl("jdbc:mysql://localhost:3306/mp?characterEncoding=utf-8&serverTimezone=GMT%2B8&useSSL=false");
        dataSourceConfig.setDriverName("com.mysql.jdbc.Driver");
        dataSourceConfig.setUsername("root");
        dataSourceConfig.setPassword("1234");
        mpg.setDataSource(dataSourceConfig);

        /**
         * 包配置
         */
        PackageConfig pc = new PackageConfig();
        pc.setModuleName(scanner("模塊名"));
        //父包名。若是爲空，將下面子包名必須寫所有， 不然就只需寫子包名
        pc.setParent("com.jiangfeixiang.mpdemo");
        mpg.setPackageInfo(pc);


        /**
         * 自定義配置
         */
        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/"+ pc.getModuleName()
                        + "/" + tableInfo.getEntityName() + "Mapper" + StringPool.DOT_XML;
            }
        });
        cfg.setFileOutConfigList(focList);
        mpg.setCfg(cfg);

        /**
         * 配置模板
         */
        TemplateConfig templateConfig = new TemplateConfig();

        // 配置自定義輸出模板
        //指定自定義模板路徑，注意不要帶上.ftl/.vm, 會根據使用的模板引擎自動識別
        // templateConfig.setEntity("templates/entity2.java");
        // templateConfig.setService();
        // templateConfig.setController();

        templateConfig.setXml(null);
        mpg.setTemplate(templateConfig);

        /**
         * 策略配置
         */
        StrategyConfig strategy = new StrategyConfig();
        //設置命名格式
        strategy.setNaming(NamingStrategy.underline_to_camel);
        strategy.setColumnNaming(NamingStrategy.underline_to_camel);
        strategy.setInclude(scanner("表名,多個英文逗號分割").split(","));
        //實體是否爲lombok模型（默認 false）
        strategy.setEntityLombokModel(true);
        //生成 @RestController 控制器
        strategy.setRestControllerStyle(true);
        //設置自定義繼承的Entity類全稱，帶包名
        //strategy.setSuperEntityClass("com.jiangfeixiang.mpdemo.BaseEntity");
        //設置自定義繼承的Controller類全稱，帶包名
        //strategy.setSuperControllerClass("com.jiangfeixiang.mpdemo.BaseController");
        //設置自定義基礎的Entity類，公共字段
        strategy.setSuperEntityColumns("id");
        //駝峯轉連字符
        strategy.setControllerMappingHyphenStyle(true);
        //表名前綴
        strategy.setTablePrefix(pc.getModuleName() + "_");
        mpg.setStrategy(strategy);
        mpg.setTemplateEngine(new FreemarkerTemplateEngine());
        mpg.execute();
    }
}

拆分詳解以下：

/**
     * 讀取控制檯內容
     */
    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 + "！");
    }

讀取控制檯內容無需更改，由於稍後啓動main方法只會須要在控制檯輸入模塊名以及數據庫表名。官網參考
接下來是main方法，這個也是主程序，逆向工程啓動方法。下面看一下配置

AutoGenerator mpg = new AutoGenerator();

代碼生成器，全部的配置都須要set進去

全局配置：

GlobalConfig globalConfig = new GlobalConfig();
       //生成文件的輸出目錄(下面兩行無需改動)
       String projectPath = System.getProperty("user.dir");
       globalConfig.setOutputDir(projectPath + "/src/main/java");
       //Author設置做者
       globalConfig.setAuthor("姜飛祥");
       //是否覆蓋文件
       globalConfig.setFileOverride(true);
       //生成後打開文件
       globalConfig.setOpen(false);
       //set進去代碼生成器對象中
       mpg.setGlobalConfig(globalConfig);

數據源配置

DataSourceConfig dataSourceConfig = new DataSourceConfig();
        // 數據庫類型,默認MYSQL
        dataSourceConfig.setDbType(DbType.MYSQL);
        //自定義數據類型轉換
        dataSourceConfig.setTypeConvert(new MySqlTypeConvert());
       //驅動，URL，用戶名以及密碼配置，這裏使用的是mysql5.6版本
        dataSourceConfig.setUrl("jdbc:mysql://localhost:3306/mp?characterEncoding=utf-8&serverTimezone=GMT%2B8&useSSL=false");
        dataSourceConfig.setDriverName("com.mysql.jdbc.Driver");
        dataSourceConfig.setUsername("root");
        dataSourceConfig.setPassword("1234");
        //set進去代碼生成器對象中
        mpg.setDataSource(dataSourceConfig);

包配置

PackageConfig pc = new PackageConfig();
        //這裏的模塊名須要在控制檯輸入的，即生成的代碼在哪一個包下
        pc.setModuleName(scanner("模塊名"));
        //父包名。若是爲空子包名必須寫所有， 不然就只需寫子包名
        pc.setParent("com.jiangfeixiang.mpdemo");
        //set進去代碼生成器對象中
        mpg.setPackageInfo(pc);

上面父包名是根據工程路徑來的，以下參考：

自定義配置

InjectionConfig cfg = new InjectionConfig() {
            @Override
            public void initMap() {
                // to do nothing
            }
        };

自定義輸出配置

String templatePath = "/templates/mapper.xml.ftl";
List<FileOutConfig> focList = new ArrayList<>();
        // 自定義配置會被優先輸出
        focList.add(new FileOutConfig(templatePath) {
            @Override
            public String outputFile(TableInfo tableInfo) {
                // 自定義輸出文件名 ， 若是你 Entity 設置了先後綴、此處注意 xml 的名稱會跟着發生變化！！
                return projectPath + "/src/main/resources/mapper/"+ pc.getModuleName()
                        + "/" + tableInfo.getEntityName() + "Mapper" + StringPool.DOT_XML;
            }
        });
        //這塊是set到上面自定義配置中
        cfg.setFileOutConfigList(focList);
        //set進去代碼生成器對象中
        mpg.setCfg(cfg);

最後是策略配置

StrategyConfig strategy = new StrategyConfig();
        //設置命名格式
        strategy.setNaming(NamingStrategy.underline_to_camel);
        strategy.setColumnNaming(NamingStrategy.underline_to_camel);
        strategy.setInclude(scanner("表名,多個英文逗號分割").split(","));
        //實體是否爲lombok模型（默認 false）
        strategy.setEntityLombokModel(true);
        //生成 @RestController 控制器
        strategy.setRestControllerStyle(true);
        //設置自定義繼承的Entity類全稱，帶包名
        //strategy.setSuperEntityClass("com.jiangfeixiang.mpdemo.BaseEntity");
        //設置自定義繼承的Controller類全稱，帶包名
        //strategy.setSuperControllerClass("com.jiangfeixiang.mpdemo.BaseController");
        //設置自定義基礎的Entity類，公共字段
        strategy.setSuperEntityColumns("id");
        //駝峯轉連字符
        strategy.setControllerMappingHyphenStyle(true);
        //表名前綴
        strategy.setTablePrefix(pc.getModuleName() + "_");
        mpg.setStrategy(strategy);
        mpg.setTemplateEngine(new FreemarkerTemplateEngine());
        mpg.execute();

以上所有配置好以後直接啓動main方法，以後進入控制檯

個人模塊名是springbootmp，由於我有兩張表，輸入兩個表的名稱回車便可生成對應的代碼，所生成的代碼在模塊名springbootmp下

正確執行控制檯輸出以下

而後看一下模塊：

xxxmapper.xml文件是空的：

實體類已經加上@Data註解省略了get/set方法並序列化

mapper接口繼承了BaseMapper

接口中沒有數據的增刪改查方法，那麼咱們直接在UserController類中注入IUserService接口，查詢全部user看看有沒有輸出：

@RestController
@RequestMapping("/springbootmp/user")
public class UserController {

    @Autowired
    private IUserService iUserService;

    /**
     * 獲取全部User
     * @return
     */
    @RequestMapping("/getAllUser")
    public List<User> getAllUser(){
        List<User> list = iUserService.list();
        return list;
    }

}

項目運行直接報錯以下：

緣由是由於主程序中沒有加入@MapperScan("com.jiangfeixiang.mpdemo.springbootmp.mapper")

引入便可。以後從新運行啓動成功控制檯以下圖：

還有mybatisplus是否是很漂亮。
調用接口測試以下

源碼