Generator.java 1.3 KB

1234567891011121314151617181920212223242526272829303132333435363738
  1. package com;
  2. import org.mybatis.generator.api.MyBatisGenerator;
  3. import org.mybatis.generator.config.Configuration;
  4. import org.mybatis.generator.config.xml.ConfigurationParser;
  5. import org.mybatis.generator.internal.DefaultShellCallback;
  6. import java.io.InputStream;
  7. import java.util.ArrayList;
  8. import java.util.List;
  9. /**
  10. * 用于生产MBG的代码
  11. * Created by macro on 2018/4/26.
  12. */
  13. public class Generator {
  14. public static void main(String[] args) throws Exception {
  15. //MBG 执行过程中的警告信息
  16. List<String> warnings = new ArrayList<String>();
  17. //当生成的代码重复时,覆盖原代码
  18. boolean overwrite = true;
  19. //读取我们的 MBG 配置文件
  20. InputStream is = Generator.class.getResourceAsStream("/generatorConfig.xml");
  21. ConfigurationParser cp = new ConfigurationParser(warnings);
  22. Configuration config = cp.parseConfiguration(is);
  23. is.close();
  24. DefaultShellCallback callback = new DefaultShellCallback(overwrite);
  25. //创建 MBG
  26. MyBatisGenerator myBatisGenerator = new MyBatisGenerator(config, callback, warnings);
  27. //执行生成代码
  28. myBatisGenerator.generate(null);
  29. //输出警告信息
  30. for (String warning : warnings) {
  31. System.out.println(warning);
  32. }
  33. }
  34. }