mybatis-plus插件快速生成代碼

需要的依賴：

? ? <groupId>com.baomidou</groupId>

? ? <artifactId>mybatis-plus-boot-starter</artifactId>

? ? <version>3.4.0</version>

</dependency>

? ? <groupId>com.baomidou</groupId>

? ? <artifactId>mybatis-plus-generator</artifactId>

? ? <version>3.4.0</version>

</dependency>

? ? <groupId>org.freemarker</groupId>

? ? <artifactId>freemarker</artifactId>

? ? <version>2.3.30</version>

</dependency>

創(chuàng)建代碼生成類：

package com.example;

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;

public class CodeGenerator{

? ? /**

? ? *

? ? * 讀取控制臺內(nèi)容

? ? *

? ? */

? ? public static Stringscanner(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 +"评矩！");

? ? }

? ? 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("jobob");

? ? ? ? gc.setOpen(false);

? ? ? ? // gc.setSwagger2(true); 實體屬性 Swagger2 注解

? ? ? ? mpg.setGlobalConfig(gc);

? ? ? ? // 數(shù)據(jù)源配置

? ? ? ? DataSourceConfig dsc =new DataSourceConfig();

? ? ? ? dsc.setUrl("jdbc:mysql://localhost:3306/test?useUnicode=true&characterEncoding=utf-8&useSSL=true&serverTimezone=GMT%2B8&nullCatalogMeansCurrent=true");

? ? ? ? // dsc.setSchemaName("public");

? ? ? ? dsc.setDriverName("com.mysql.cj.jdbc.Driver");

? ? ? ? dsc.setUsername("root");

? ? ? ? dsc.setPassword("123456");

? ? ? ? mpg.setDataSource(dsc);

? ? ? ? // 包配置

? ? ? ? PackageConfig pc =new PackageConfig();

? ? ? ? pc.setModuleName(scanner("模塊名"));

? ? ? ? pc.setParent("com.baomidou.ant");

? ? ? ? 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<>();

? ? ? ? // 自定義配置會被優(yōu)先輸出

? ? ? ? boolean add = focList.add(new FileOutConfig(templatePath) {

? ? ? ? ? ? @Override

? ? ? ? ? ? public StringoutputFile(TableInfo tableInfo) {

? ? ? ? ? ? ? ? // 自定義輸出文件名饺藤，如果你 Entity 設(shè)置了前后綴剑辫、此處注意 xml 的名稱會跟著發(fā)生變化８删伞！

? ? ? ? ? ? ? ? return projectPath +"/src/main/resources/mapper/" +pc.getModuleName()

? ? ? ? ? ? ? ? ? ? ? ? +"/" + tableInfo.getEntityName() +"Mapper" + StringPool.DOT_XML;

? ? ? ? ? ? }

? ? ? ? });

? ? ? ? /*

cfg.setFileCreate(new IFileCreate() {

@Override

public boolean isCreate(ConfigBuilder configBuilder, FileType fileType, String filePath) {

// 判斷自定義文件夾是否需要創(chuàng)建

checkDir("調(diào)用默認(rèn)方法創(chuàng)建的目錄妹蔽，自定義目錄用");

if (fileType == FileType.MAPPER) {

// 已經(jīng)生成 mapper 文件判斷存在椎眯，不想重新生成返回 false

return !new File(filePath).exists();

}

// 允許生成模板文件

return true;

}

});

? ? ? ? cfg.setFileOutConfigList(focList);

? ? ? ? mpg.setCfg(cfg);

? ? ? ? // 配置模板

? ? ? ? TemplateConfig templateConfig =new TemplateConfig();

? ? ? ? // 配置自定義輸出模板

//指定自定義模板路徑挠将，注意不要帶上.ftl/.vm, 會根據(jù)使用的模板引擎自動識別

// 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.setSuperEntityClass("你自己的父類實體,沒有就不用設(shè)置!");

? ? ? ? strategy.setEntityLombokModel(true);

? ? ? ? strategy.setRestControllerStyle(true);

? ? ? ? // 公共父類

? ? ? ? strategy.setSuperControllerClass("你自己的父類控制器,沒有就不用設(shè)置!");

? ? ? ? // 寫于父類中的公共字段

? ? ? ? strategy.setSuperEntityColumns("id");

? ? ? ? strategy.setInclude(scanner("表名，多個英文逗號分割").split(","));

? ? ? ? strategy.setControllerMappingHyphenStyle(true);

? ? ? ? strategy.setTablePrefix(pc.getModuleName() +"_");

? ? ? ? mpg.setStrategy(strategy);

? ? ? ? mpg.setTemplateEngine(new FreemarkerTemplateEngine());

? ? ? ? mpg.execute();

? ? }

}

直接復(fù)制到項目中编整，修改對應(yīng)的package直接運行舔稀。在控制臺根據(jù)提示輸入即可。此代碼為官方實例掌测。不明白的可以在官方網(wǎng)站查看内贮。