package com.example.demo;

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.internal.DefaultShellCallback;
import org.springframework.core.io.ClassPathResource;
import org.springframework.core.io.Resource;

/**
 * MyBatis generator通过main方法生成示例
 * @author 程就人生
 * @date 2019年9月6日
 */
public class TestGenerator {

	public  static void main(String[] agro){
		try{
			//可对配置文件里的参数做动态替换
    	    //Properties prop = new Properties();
    	    
			//收集错误信息
			List<String> warnings = new ArrayList<String>();
			//重写标识，此标识为true时，实体类、dao已存在会在后面追加；
			//设置为false时，实体类、dao不会被追加，会生成一个新的java类类名加数字，但xml文件依旧会被追加；
			boolean overwrite = false;
			//读取配置文件，三种写法
			//File configFile = new File(TableInfoController.class.getResource("generatorConfig.xml").getFile());  
			//ResourceUtils.getURL("classpath:generatorConfig.xml")
			Resource resource = new ClassPathResource("generatorConfig.xml"); 
			
			//动态参数的替换加载
			//ConfigurationParser cp = new ConfigurationParser(prop, warnings);
			ConfigurationParser cp = new ConfigurationParser(warnings);
			
			Configuration config = cp.parseConfiguration(resource.getFile());
			
			DefaultShellCallback callback = new DefaultShellCallback(overwrite);
			
			MyBatisGenerator myBatisGenerator = new MyBatisGenerator(config, callback, warnings);
			
			myBatisGenerator.generate(null);
			//输出错误信息,一定要输出，不然出错了，会找不到错误信息
			if(warnings.size() > 0){
				System.out.println("有错误信息，请跟进错误提示进行修改！");
				warnings.forEach(System.out::println);
			}else{
				System.out.println("文件已经生成,刷新项目后查看！");
			}
		}catch(Exception e){
			e.printStackTrace();
		}
	}
}