使用java程式碼啟動generator生成相關的檔案

喬巴14發表於2017-06-29
package com.utils.generator;

import org.mybatis.generator.api.MyBatisGenerator;
import org.mybatis.generator.api.ProgressCallback;
import org.mybatis.generator.api.VerboseProgressCallback;
import org.mybatis.generator.config.Configuration;
import org.mybatis.generator.config.xml.ConfigurationParser;
import org.mybatis.generator.internal.DefaultShellCallback;

import java.io.File;
import java.util.ArrayList;
import java.util.List;

public class GeneratorUtils {

   public static void main(String[] args) {
           shell();
       }

       private static void shell() {
           List<String> warnings = new ArrayList<String>();
           try {
               String configFilePath = System.getProperty("user.dir").concat("/src/main/resources/generator/generatorConfig.xml");
               System.out.println("載入配置檔案===" + configFilePath);
               boolean overwrite = true;
               File configFile = new File(configFilePath);
               ConfigurationParser cp = new ConfigurationParser(warnings);
               Configuration config = cp.parseConfiguration(configFile);
               DefaultShellCallback callback = new DefaultShellCallback(overwrite);
               MyBatisGenerator myBatisGenerator = new MyBatisGenerator(config, callback, warnings);
               ProgressCallback progressCallback = new VerboseProgressCallback();
               // myBatisGenerator.generate(null);
               myBatisGenerator.generate(progressCallback);
           }
           catch (Exception e) {
               e.printStackTrace();
           }
           for (String warning : warnings) {
               System.out.println(warning);
           }
       }
}

相關文章