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;

/**
*  @author: Mr.He
*  @Date: 2020/7/17
*  @Description: MP自動(dòng)生成
*/
public class CodeGeneration {

   //main函數(shù)
   public static void main(String[] args) {

       AutoGenerator autoGenerator = new AutoGenerator();

       //全局配置
       GlobalConfig gc = new GlobalConfig();

       //得到當(dāng)前項(xiàng)目的路徑
       String oPath = System.getProperty("user.dir");

//        //生成文件輸出根目錄
       gc.setOutputDir(oPath + "/spread-utils/src/main/java");

       //生成完成后不彈出文件框
       gc.setOpen(false);

       //文件覆蓋
       gc.setFileOverride(true);

       // 不需要ActiveRecord特性的請(qǐng)改為false
       gc.setActiveRecord(false);

       // XML 二級(jí)緩存
       gc.setEnableCache(false);

       // XML ResultMap
       gc.setBaseResultMap(true);

       // XML columList
       gc.setBaseColumnList(false);

       // 作者
       gc.setAuthor("Mr.He");

       // 自定義文件命名，注意 %s 會(huì)自動(dòng)填充表實(shí)體屬性夺克！
       gc.setControllerName("%sController");
       gc.setServiceName("%sService");
       gc.setServiceImplName("%sServiceImpl");
       gc.setMapperName("%sMapper");
       gc.setXmlName("%sMapper");
       autoGenerator.setGlobalConfig(gc);

       // 數(shù)據(jù)源配置
       DataSourceConfig dsc = new DataSourceConfig();

       //設(shè)置數(shù)據(jù)庫類型
       dsc.setDbType(DbType.MYSQL);

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

       //用戶名
       dsc.setUsername("root");

       //密碼
       dsc.setPassword("123456");

       //指定數(shù)據(jù)庫
       dsc.setUrl("jdbc:mysql://localhost:3306/spread");
       autoGenerator.setDataSource(dsc);

       // 策略配置
       StrategyConfig strategy = new StrategyConfig();

       // 表名生成策略
       strategy.setNaming(NamingStrategy.underline_to_camel);

       // 需要生成的表
       strategy.setInclude(new String[]{"t_reward_rule"});
       strategy.setTablePrefix("t_");
       strategy.setSuperServiceClass(null);
       strategy.setSuperServiceImplClass(null);
       strategy.setSuperMapperClass(null);
       strategy.setEntityLombokModel(true);

       //去除表前綴
//        strategy.setTablePrefix("an_");
       //去除字段前綴
//        strategy.setFieldPrefix(new String[]);
       autoGenerator.setStrategy(strategy);

       // 包配置
       PackageConfig pc = new PackageConfig();
       //父包路徑
//        pc.setModuleName("spread-utils");
       pc.setParent("com.spread.utils");
       pc.setController("controller");
       pc.setService("service");
       pc.setServiceImpl("service.impl");
       pc.setMapper("mapper");
       pc.setEntity("entity");
       pc.setXml("mapper");
       autoGenerator.setPackageInfo(pc);
       // 執(zhí)行生成
       autoGenerator.execute();
   }
}