CodeGenerator.java 2.5 KB

1234567891011121314151617181920212223242526272829303132333435363738394041424344454647484950515253545556575859606162636465666768
  1. import com.baomidou.mybatisplus.annotation.IdType;
  2. import com.baomidou.mybatisplus.generator.AutoGenerator;
  3. import com.baomidou.mybatisplus.generator.config.DataSourceConfig;
  4. import com.baomidou.mybatisplus.generator.config.GlobalConfig;
  5. import com.baomidou.mybatisplus.generator.config.PackageConfig;
  6. import com.baomidou.mybatisplus.generator.config.StrategyConfig;
  7. import com.baomidou.mybatisplus.generator.config.rules.NamingStrategy;
  8. public class CodeGenerator {
  9. public static void main(String[] args) {
  10. // 代码生成器
  11. AutoGenerator mpg = new AutoGenerator();
  12. // 全局配置
  13. GlobalConfig gc = new GlobalConfig();
  14. gc.setOutputDir("E:\\fmNew\\fm-business\\fm-sop\\src\\main\\java")
  15. .setAuthor("weijinhao").setIdType(IdType.ASSIGN_UUID)
  16. .setActiveRecord(true)
  17. .setFileOverride(true)
  18. .setServiceName("%sService")
  19. .setBaseResultMap(true).setBaseColumnList(true)
  20. .setOpen(false);
  21. // gc.setSwagger2(true); 实体属性 Swagger2 注解
  22. mpg.setGlobalConfig(gc);
  23. // 数据源配置
  24. DataSourceConfig dsc = new DataSourceConfig();
  25. dsc.setUrl("jdbc:mysql://192.168.100.94:9934/fm_sop?useUnicode=true&useSSL=false&characterEncoding=utf8");
  26. // dsc.setSchemaName("public");
  27. dsc.setDriverName("com.mysql.cj.jdbc.Driver");
  28. dsc.setUsername("root");
  29. dsc.setPassword("persagy@2020");
  30. mpg.setDataSource(dsc);
  31. // 包配置
  32. PackageConfig pc = new PackageConfig();
  33. pc.setParent("com.persagy.fm.sop");
  34. pc.setMapper("dao");
  35. pc.setXml("dao");
  36. //pc.setController();
  37. pc.setService("service");
  38. pc.setServiceImpl("service.impl");
  39. pc.setEntity("model.dto");
  40. mpg.setPackageInfo(pc);
  41. // 策略配置
  42. StrategyConfig strategy = new StrategyConfig();
  43. strategy.setCapitalMode(true);
  44. strategy.setNaming(NamingStrategy.underline_to_camel);
  45. strategy.setColumnNaming(NamingStrategy.underline_to_camel);
  46. strategy.setEntityLombokModel(false);
  47. strategy.setRestControllerStyle(true);
  48. strategy.setControllerMappingHyphenStyle(true);
  49. //strategy.setInclude("sop","sop_sop_rel","sop_obj_rel","sop_equip_model_rel","sop_order_type_rel","sop_label_rel");
  50. strategy.setInclude("sop_edit");
  51. //strategy.setTablePrefix(pc.getModuleName() + "_");
  52. mpg.setStrategy(strategy);
  53. //mpg.setTemplateEngine(new FreemarkerTemplateEngine());
  54. mpg.execute();
  55. }
  56. }