Blame view

src/main/java/com/zteits/oa/util/MybatisGenerateRun.java 1.76 KB
cc51d294   王富生   最初版本提交
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
  package com.zteits.oa.util;
  
  import java.io.File;
  import java.io.IOException;
  import java.sql.SQLException;
  import java.util.ArrayList;
  import java.util.List;
  import java.util.Properties;
  
  import org.mybatis.generator.api.MyBatisGenerator;
  import org.mybatis.generator.config.Configuration;
  import org.mybatis.generator.config.xml.ConfigurationParser;
  import org.mybatis.generator.exception.InvalidConfigurationException;
  import org.mybatis.generator.exception.XMLParserException;
  import org.mybatis.generator.internal.DefaultShellCallback;
  import org.springframework.core.io.support.PropertiesLoaderUtils;
  import org.springframework.util.CollectionUtils;
  /**
   * 生成mybaits相关mapper,domain,dao
   *
   */
  public class MybatisGenerateRun {
  	public static void main(String[] args) throws IOException, XMLParserException, InvalidConfigurationException, SQLException, InterruptedException {
  	   List<String> warnings = new ArrayList<String>();
  	   boolean overwrite = true;
  	   //加载generatorEmallConfig文件
  	   File configFile = new File(MybatisGenerateRun.class.getClassLoader().getResource("generator/generatorConfig.xml").getPath());
  	   //加载数据库设置相关配置文件
  	   Properties extraProperties = PropertiesLoaderUtils.loadAllProperties("application.properties");
  	   ConfigurationParser cp = new ConfigurationParser(extraProperties, warnings);
  	   Configuration config = cp.parseConfiguration(configFile);
  	   DefaultShellCallback callback = new DefaultShellCallback(overwrite);
  	   MyBatisGenerator myBatisGenerator = new MyBatisGenerator(config, callback, warnings);
  	   myBatisGenerator.generate(null);
  	   if(!CollectionUtils.isEmpty(warnings)){
  		   for (String warn : warnings) {
  			System.out.println(warn);
  		}
  	   }
  	   System.out.println("生成成功!");
  	}
  }