不給文檔，不同意加群只能自己打了

package com.example.wms.common;

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 {

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.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")+"/wns";

gc.setOutputDir(projectPath + "/src/main/java");//設(shè)置代碼生成路徑

gc.setAuthor("wns");//設(shè)置項目作者名稱

gc.setOpen(false);

gc.setSwagger2(true);

gc.setBaseResultMap(true);//生成基本ResultMap

gc.setBaseColumnList(true);//生成基本ColumnList

gc.setServiceName("%sService");//去掉服務(wù)默認(rèn)前綴

mpg.setGlobalConfig(gc);

DataSourceConfig dsc = new DataSourceConfig();

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

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

dsc.setUsername("root");

dsc.setPassword("123456");

mpg.setDataSource(dsc);

PackageConfig pc = new PackageConfig();

pc.setParent("com.example.wns");

pc.setEntity("entity");

pc.setMapper("mapper");

pc.setService("service");

pc.setServiceImpl("service.impl");

pc.setController("controller");

mpg.setPackageInfo(pc);

InjectionConfig cfg=new InjectionConfig() {

@Override

public void initMap() {

//to do nothing

}

} ;

String templatePath="template/mapper.xml.ftl";

List<FileOutConfig> focList = new ArrayList<>();

focList.add(new FileOutConfig(templatePath) {

@Override

public String outputFile(TableInfo tableInfo) {// 自定義輸出文件名 ， 如果你 Entity 設(shè)置了前后綴、此處注意 xml 的名稱會跟著發(fā)生變化?。?/p>

return projectPath + "/src/main/resources/mapper/" + tableInfo.getEntityName() + "Mapper" + StringPool.DOT_XML;

}

});

cfg.setFileOutConfigList(focList);

mpg.setCfg(cfg);

TemplateConfig templateConfig = new TemplateConfig();

templateConfig.setXml(null);

mpg.setTemplate(templateConfig);

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

mpg.setStrategy(strategy);

mpg.setTemplateEngine(new FreemarkerTemplateEngine());

mpg.execute();

}

}