AutoGenerate.java 4.2 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899
  1. package com.fdkankan.scene.generate;
  2. import com.baomidou.mybatisplus.core.mapper.BaseMapper;
  3. import com.baomidou.mybatisplus.generator.FastAutoGenerator;
  4. import com.baomidou.mybatisplus.generator.config.OutputFile;
  5. import com.baomidou.mybatisplus.generator.config.rules.DateType;
  6. import com.fdkankan.db.base.BaseEntity;
  7. import java.util.ArrayList;
  8. import java.util.Arrays;
  9. import java.util.Collections;
  10. import java.util.List;
  11. public class AutoGenerate {
  12. public static void main(String[] args) {
  13. String path =System.getProperty("user.dir") + "\\4dkankan-center-scene";
  14. generate(path,"scene", getTables(new String[]{
  15. "t_scene_repair_log"
  16. }));
  17. // generate(path,"goods", getTables(new String[]{
  18. // "t_camera","t_camera_detail","t_camera_out","t_camera_space","t_camera_version",
  19. // "t_company","t_goods","t_goods_sku","t_cart","t_goods_spec",
  20. // "t_goods_spec_value","t_goods_spu_spec","t_sn_code"
  21. // }));
  22. //
  23. // generate(path,"order", getTables(new String[]{
  24. // "t_increment_order","t_invoice","t_order","t_order_item",
  25. // "t_pre_sale","t_space_sdk","t_trade_log","t_commerce_order","t_download_order","t_expansion_order"
  26. // }));
  27. // generate(path,"order", getTables(new String[]{
  28. // "t_virtual_order"
  29. // }));
  30. //
  31. // generate(path,"user", getTables(new String[]{
  32. // "t_user","t_user_increment","t_manager","t_province","t_increment_type","t_intercom_message","t_receiver_info"
  33. // }));
  34. }
  35. public static List<String> getTables(String [] tableNames){
  36. return new ArrayList<>(Arrays.asList(tableNames));
  37. }
  38. public static void generate(String path,String moduleName, List<String> tables){
  39. FastAutoGenerator.create("jdbc:mysql://120.24.144.164:3306/4dkankan-center-scene",
  40. "root","4Dage@4Dage#@168")
  41. .globalConfig(builder -> {
  42. builder.author("") //作者
  43. .outputDir(path+"\\src\\main\\java") //输出路径(写到java目录)
  44. //.enableSwagger() //开启swagger
  45. .commentDate("yyyy-MM-dd")
  46. .dateType(DateType.ONLY_DATE)
  47. .fileOverride(); //开启覆盖之前生成的文件
  48. })
  49. .packageConfig(builder -> {
  50. builder.parent("com.fdkankan")
  51. .moduleName(moduleName)
  52. .entity("entity")
  53. .service("service")
  54. .serviceImpl("service.impl")
  55. .controller("controller")
  56. .mapper("mapper")
  57. .xml("test.mapper")
  58. .pathInfo(Collections.singletonMap(OutputFile.mapperXml,path+"\\src\\main\\resources\\mapper\\"+moduleName));
  59. })
  60. .strategyConfig(builder -> {
  61. builder.addInclude(tables)
  62. .addTablePrefix("t_")
  63. .serviceBuilder()
  64. .formatServiceFileName("I%sService")
  65. .formatServiceImplFileName("%sServiceImpl")
  66. .entityBuilder()
  67. .enableLombok()
  68. .logicDeleteColumnName("tb_status")
  69. .enableTableFieldAnnotation()
  70. // .superClass(BaseEntity.class)
  71. .controllerBuilder()
  72. .formatFileName("%sController")
  73. .enableRestStyle()
  74. .mapperBuilder()
  75. .superClass(BaseMapper.class)
  76. .formatMapperFileName("I%sMapper")
  77. .enableMapperAnnotation()
  78. .formatXmlFileName("%sMapper");
  79. })
  80. // .templateEngine(new FreemarkerTemplateEngine()) // 使用Freemarker引擎模板,默认的是Velocity引擎模板
  81. .execute();
  82. }
  83. }