使用Mybatis plus 如何實現(xiàn)一個自動代碼生成器

使用Mybatis plus 如何實現(xiàn)一個自動代碼生成器？相信很多沒有經(jīng)驗的人對此束手無策，為此本文總結(jié)了問題出現(xiàn)的原因和解決方法，通過這篇文章希望你能解決這個問題。

1.使用的是maven項目，添加依賴

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

還有數(shù)據(jù)庫的連接

<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); 
  } 
  /** 
   * 根據(jù)表自動生成 
   * 
   * @param serviceNameStartWithI 默認為false 
   * @param packageName      包名 
   * @param tableNames      表名 
   * @author Terry 
   */ 
  private void generateByTables(boolean serviceNameStartWithI, String packageName, String... tableNames) { 
    //配置數(shù)據(jù)源 
    DataSourceConfig dataSourceConfig = getDataSourceConfig(); 
    // 策略配置 
    StrategyConfig strategyConfig = getStrategyConfig(tableNames); 
    //全局變量配置 
    GlobalConfig globalConfig = getGlobalConfig(serviceNameStartWithI); 
    //包名配置 
    PackageConfig packageConfig = getPackageConfig(packageName); 
    //自動生成 
    atuoGenerator(dataSourceConfig, strategyConfig, globalConfig, packageConfig); 
  } 
  /** 
   * 集成 
   * 
   * @param dataSourceConfig 配置數(shù)據(jù)源 
   * @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) 
        //從數(shù)據(jù)庫表到文件的命名策略 
        .setNaming(NamingStrategy.underline_to_camel) 
        //需要生成的的表名，多個表名傳數(shù)組 
        .setInclude(tableNames); 
  } 
  /** 
   * 配置數(shù)據(jù)源 
   * 
   * @return 數(shù)據(jù)源配置 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"); 
  } 
  /** 
   * 根據(jù)表自動生成 
   * 
   * @param packageName 包名 
   * @param tableNames 表名 
   * @author Terry 
   */ 
  @SuppressWarnings("unused") 
  private void generateByTables(String packageName, String... tableNames) { 
    generateByTables(true, packageName, tableNames); 
  } 
}

看完上述內(nèi)容，你們掌握使用Mybatis plus 如何實現(xiàn)一個自動代碼生成器的方法了嗎？如果還想學到更多技能或想了解更多相關內(nèi)容，歡迎關注億速云行業(yè)資訊頻道，感謝各位的閱讀！