基于Mybatis plus 自动代码生成器的实现代码

<!-- mybatis-plus begin -->    <dependency>      <groupId>com.baomidou</groupId>      <artifactId>mybatis-plus</artifactId>      <version>2.2.0</version>    </dependency> 

<dependency>     <groupId>mysql</groupId>     <artifactId>mysql-connector-java</artifactId>     <scope>compile</scope> </dependency> 

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.DbType; import com.baomidou.mybatisplus.generator.config.rules.NamingStrategy; import org.junit.jupiter.api.Test; /**  * mybatis-plus 自动生成代码  *  * @author Terry  * @version 1.0  * @date 2018-05-16 09:35  */ public class SimpleMp {   @Test   public void generateCode() {     //指定包名     String packageName = "com.hciot.hhhh";     //user -> UserService, 设置成true: user -> IUserService     boolean serviceNameStartWithI = false;     //指定生成的表名     String[] tableNames = new String[]{"data_air_sensor_co", "order_product", "relation_device_gateway"};     generateByTables(serviceNameStartWithI, packageName, tableNames);   }   /**    * 根据表自动生成    *    * @param serviceNameStartWithI 默认为false    * @param packageName      包名    * @param tableNames      表名    * @author Terry    */   private void generateByTables(boolean serviceNameStartWithI, String packageName, String... tableNames) {     //配置数据源     DataSourceConfig dataSourceConfig = getDataSourceConfig();     // 策略配置     StrategyConfig strategyConfig = getStrategyConfig(tableNames);     //全局变量配置     GlobalConfig globalConfig = getGlobalConfig(serviceNameStartWithI);     //包名配置     PackageConfig packageConfig = getPackageConfig(packageName);     //自动生成     atuoGenerator(dataSourceConfig, strategyConfig, globalConfig, packageConfig);   }   /**    * 集成    *    * @param dataSourceConfig 配置数据源    * @param strategyConfig  策略配置    * @param config      全局变量配置    * @param packageConfig  包名配置    * @author Terry    */   private void atuoGenerator(DataSourceConfig dataSourceConfig, StrategyConfig strategyConfig, GlobalConfig config, PackageConfig packageConfig) {     new AutoGenerator()         .setGlobalConfig(config)         .setDataSource(dataSourceConfig)         .setStrategy(strategyConfig)         .setPackageInfo(packageConfig)         .execute();   }   /**    * 设置包名    *    * @param packageName 父路径包名    * @return PackageConfig 包名配置    * @author Terry    */   private PackageConfig getPackageConfig(String packageName) {     return new PackageConfig()         .setParent(packageName)         .setXml("mapper")         .setMapper("dao")         .setController("controller")         .setEntity("entity");   }   /**    * 全局配置    *    * @param serviceNameStartWithI false    * @return GlobalConfig    * @author Terry    */   private GlobalConfig getGlobalConfig(boolean serviceNameStartWithI) {     GlobalConfig globalConfig = new GlobalConfig();     globalConfig         .setBaseColumnList(true)         .setBaseResultMap(true)         .setActiveRecord(false)         .setAuthor("Terry")         //设置输出路径         .setOutputDir(getOutputDir("mybatis-plus"))         .setFileOverride(true);     if (!serviceNameStartWithI) {       //设置service名       globalConfig.setServiceName("%sService");     }     return globalConfig;   }   /**    * 返回项目路径    *    * @param projectName 项目名    * @return 项目路径    * @author Terry    */   private String getOutputDir(String projectName) {     String path = this.getClass().getClassLoader().getResource("").getPath();     int index = path.indexOf(projectName);     return path.substring(1, index) + projectName + "/src/main/java/";   }   /**    * 策略配置    *    * @param tableNames 表名    * @return StrategyConfig    * @author Terry    */   private StrategyConfig getStrategyConfig(String... tableNames) {     return new StrategyConfig()         // 全局大写命名 ORACLE 注意         .setCapitalMode(true)         .setEntityLombokModel(false)         // 表名、字段名、是否使用下划线命名（默认 false）         .setDbColumnUnderline(true)         //从数据库表到文件的命名策略         .setNaming(NamingStrategy.underline_to_camel)         //需要生成的的表名，多个表名传数组         .setInclude(tableNames);   }   /**    * 配置数据源    *    * @return 数据源配置 DataSourceConfig    * @author Terry    */   private DataSourceConfig getDataSourceConfig() {     String dbUrl = "jdbc:mysql://localhost:3306/test";     return new DataSourceConfig().setDbType(DbType.MYSQL)         .setUrl(dbUrl)         .setUsername("root")         .setPassword("root")         .setDriverName("com.mysql.jdbc.Driver");   }   /**    * 根据表自动生成    *    * @param packageName 包名    * @param tableNames 表名    * @author Terry    */   @SuppressWarnings("unused")   private void generateByTables(String packageName, String... tableNames) {     generateByTables(true, packageName, tableNames);   } }