利用mybatis-plus怎么實現一個代碼自動生成功能

本篇文章為大家展示了利用mybatis-plus怎么實現一個代碼自動生成功能，內容簡明扼要并且容易理解，絕對能使你眼前一亮，通過這篇文章的詳細介紹希望你能有所收獲。

使用 mybatis-plus 工具自動給我們生成 Controller、Service、Entity、Mapper、Mapper.xml 層代碼

要求:

① 生成的Controller類，需要繼承 BaseController

② 生成的Entity 類，需要繼承 BaseEntity

③ 生成的 Service，默認名稱下是以 I 開頭的接口, 在生成Service層代碼中需要把這個 I 去掉

二、實現步驟

① 在數據庫中創建好 數據庫 與 要生成代碼對應的表

這里拿 user 表舉例

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)
);
 
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 項目，其中 pom.xml 內容如下:

<?xml version="1.0" encoding="UTF-8"?>
<project xmlns="http://maven.apache.org/POM/4.0.0" xmlns:xsi="http://www.w3.org/2001/XMLSchema-instance"
     xsi:schemaLocation="http://maven.apache.org/POM/4.0.0 https://maven.apache.org/xsd/maven-4.0.0.xsd">
  <modelVersion>4.0.0</modelVersion>
  <parent>
    <groupId>org.springframework.boot</groupId>
    <artifactId>spring-boot-starter-parent</artifactId>
    <version>2.1.11.RELEASE</version>
    <relativePath/> <!-- lookup parent from repository -->
  </parent>
  <groupId>com.lcy</groupId>
  <artifactId>mybatis-plus-demo</artifactId>
  <version>0.0.1-SNAPSHOT</version>
  <name>mybatis-plus-demo</name>
  <description>Demo project for Spring Boot</description>
 
  <properties>
    <java.version>1.8</java.version>
  </properties>
 
  <dependencies>
    <dependency>
      <groupId>org.springframework.boot</groupId>
      <artifactId>spring-boot-starter-web</artifactId>
    </dependency>
 
    <dependency>
      <groupId>org.projectlombok</groupId>
      <artifactId>lombok</artifactId>
      <optional>true</optional>
    </dependency>
    <dependency>
      <groupId>org.springframework.boot</groupId>
      <artifactId>spring-boot-starter-test</artifactId>
      <scope>test</scope>
    </dependency>
 
    <dependency>
      <groupId>com.baomidou</groupId>
      <artifactId>mybatis-plus-boot-starter</artifactId>
      <version>3.3.0</version>
    </dependency>
 
    <dependency>
      <groupId>mysql</groupId>
      <artifactId>mysql-connector-java</artifactId>
      <version>5.1.35</version>
    </dependency>
 
    <dependency>
      <groupId>com.baomidou</groupId>
      <artifactId>mybatis-plus-generator</artifactId>
      <version>3.3.0</version>
    </dependency>
    <dependency>
      <groupId>org.apache.velocity</groupId>
      <artifactId>velocity-engine-core</artifactId>
      <version>2.1</version>
    </dependency>
  </dependencies>
 
  <build>
    <plugins>
      <plugin>
        <groupId>org.springframework.boot</groupId>
        <artifactId>spring-boot-maven-plugin</artifactId>
      </plugin>
    </plugins>
  </build>
 
</project>

③ 創建代碼生成類

package com.lcy.demo.generator;
 
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.NamingStrategy;
 
public class MysqlGenerator {
  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.setAuthor("liangcy");  // 作者
    gc.setOpen(false);   //生成代碼后是否打開文件夾
    gc.setServiceName("%sService"); // 設置Service接口生成名稱,這樣生成接口前面就不會有 I
    mpg.setGlobalConfig(gc);
 
    DataSourceConfig dsc = new DataSourceConfig();
    dsc.setUrl("jdbc:mysql://localhost:3306/mybatis-plus-demo?useUnicode=true&serverTimezone=GMT&useSSL=false&characterEncoding=utf8");
    dsc.setDriverName("com.mysql.jdbc.Driver");
    dsc.setUsername("root");
    dsc.setPassword("123456");
    mpg.setDataSource(dsc);
 
    // 包配置
    PackageConfig pc = new PackageConfig();
    pc.setModuleName("sys"); // 模塊名稱, 這里可以根據不同模塊來寫
    pc.setParent("com.lcy.demo"); // 父包名
    mpg.setPackageInfo(pc);
 
    // 策略配置
    StrategyConfig strategy = new StrategyConfig();
    strategy.setNaming(NamingStrategy.underline_to_camel);
    strategy.setColumnNaming(NamingStrategy.underline_to_camel);
    strategy.setSuperControllerClass("com.lcy.demo.sys.controller.BaseController");
    strategy.setSuperEntityClass("com.lcy.demo.sys.entity.BaseEntity");
    strategy.setEntityLombokModel(true);
    strategy.setInclude("user"); // 如果要生成多個,這里可以傳入String[]
    mpg.setStrategy(strategy);
    mpg.execute();
  }
 
}

補充說明:

• 代碼生成過程中用到了 Lombok 插件, 因此需要在IDEA 中安裝好 Lombok插件

• 如果我們刪除了生成的代碼, 包名可能會飄紅, 這時我們需要 更新一下maven工程,飄紅就會消失

• 代碼將生成在我們設置的 父包名 + 模塊名中
  

上述內容就是利用mybatis-plus怎么實現一個代碼自動生成功能，你們學到知識或技能了嗎？如果還想學到更多技能或者豐富自己的知識儲備，歡迎關注億速云行業資訊頻道。