io.starter.ignite.generator.MyBatisGen Maven / Gradle / Ivy

Go to download
Show more of this group Show more artifacts with this name
Show all versions of stackgen Show documentation
Starter StackGen CORE Service Generator
There is a newer version: 0.9.25
package io.starter.ignite.generator;

import java.io.File;
import java.io.IOException;
import java.lang.reflect.Field;
import java.net.URL;
import java.net.URLClassLoader;
import java.util.ArrayList;
import java.util.List;
import java.util.Map;

import org.jdom.Document;
import org.jdom.Element;
import org.jdom.JDOMException;
import org.mybatis.generator.api.MyBatisGenerator;
import org.mybatis.generator.api.ProgressCallback;
import org.mybatis.generator.config.xml.ConfigurationParser;
import org.mybatis.generator.internal.DefaultShellCallback;
import org.slf4j.Logger;
import org.slf4j.LoggerFactory;

import com.squareup.javapoet.FieldSpec;
import com.squareup.javapoet.MethodSpec;

import io.starter.ignite.generator.DMLgenerator.Table;
import io.starter.ignite.util.DOMEditor;
import io.starter.ignite.util.FileUtil;
import io.starter.toolkit.StringTool;

/**
 * responsible for generating MyBatis config
 * 
 * @author John McMahon (@TechnoCharms)
 *
 */
public class MyBatisGen extends Gen implements Generator {

	protected static final Logger	logger			= LoggerFactory
			.getLogger(MyBatisGen.class);

	List					alreadyAdded	= new ArrayList();	// dedupe

	public static Map createMyBatis(Class c, MyBatisGen gen) throws Exception {

		// projectDir.mkdirs();
		logger.info("Generate MyBatis...");

		Map classesToGenerate = gen.processClasses(c, null, gen);

		logger.info("Write updated XML...");

		return classesToGenerate;

	}

	/**
	 * feed it an api class and it will attempt to sanitize and map to MyBatis artifact name
	 * 
	 * @param apiClassName
	 * @return
	 */
	public static String getMyBatisModelClassName(String apiClassName) {
		String apibn = getBaseJavaName(apiClassName);
		return schemaName + apibn;
	}

	/**
	 * strips the package if any
	 * 
	 * @param n
	 * @return
	 */
	static String getBaseJavaName(String n) {
		if (n.length() <= 0)
			return n;

		if (n.contains(".")) {
			n = n.substring(n.lastIndexOf(".") + 1);
		}
		return n;
	}

	/**
	 * generate the MyBatis generator config
	 */
	public static void generate() throws Exception {

		List warnings = new ArrayList();
		boolean overwrite = true;
		File configFile = new File(MYBATIS_GEN_CONFIG_OUT);

		ConfigurationParser cp = new ConfigurationParser(warnings);
		org.mybatis.generator.config.Configuration config = cp
				.parseConfiguration(configFile);

		DefaultShellCallback callback = new DefaultShellCallback(overwrite);

		MyBatisGenerator myBatisGenerator = new MyBatisGenerator(config,
				callback, warnings);

		ProgressCallback cb = null;

		myBatisGenerator.generate(cb);
		for (String warning : warnings) {
			logger.warn("WARNING: MyBatis Generation: " + warning);
		}
	}

	@Override
	public Object createSetter(Field f) {
		// TODO Auto-generated method stub
		return null;
	}

	@Override
	public Object createAccessor(Field f) {
		// TODO Auto-generated method stub
		return null;
	}

	@Override
	public Object createMember(Field f) {
		return null;
	}

	private Document	jdx;
	private Document	jdt;

	@Override
	public void generate(String className, List fieldList, List getters, List setters) throws Exception {

		// create a new JDOM Element
		// generatorConfiguration
		// 
		// 
		// 

		String packageName = null;
		int dotpos = className.lastIndexOf(".");
		packageName = className.substring(0, dotpos);
		packageName = "gen." + packageName;
		className = className.substring(dotpos + 1);

		logger.info("Load MyBatis Generator XML template...");
		File genConfigFile = new File(MYBATIS_GEN_CONFIG_TEMPLATE);
		jdt = createMyBatisXMLGenConfigNodes(jdt, className, genConfigFile);

		FileUtil.ensurePathExists(new File(MYBATIS_GEN_CONFIG_OUT));
		DOMEditor.write(jdt, MYBATIS_GEN_CONFIG_OUT);

		logger.info("Load MyBatis Generator XML template...");
		File configFile = new File(MYBATIS_CONFIG_TEMPLATE);
		jdx = createMyBatisXMLConfigNodes(jdx, className, configFile);
		DOMEditor.write(jdx, MYBATIS_CONFIG_OUT); // for runtime
	}

	// mappers>
	// 
	private Document createMyBatisXMLConfigNodes(Document jdo, String className, File configFile) throws JDOMException, IOException {

		logger.info("Parse MyBatis Template: " + configFile.getAbsolutePath());

		if (jdo == null)
			jdo = DOMEditor.parse("mybatis", configFile.getAbsolutePath());

		Element el = new Element("mapper").setAttribute("resource", MyBatisGen
				.convertToMapperSyntax(className));

		Element rootElement = jdo.getRootElement();
		List listEmpElement = rootElement.getChildren();

		// loop through to add every sqlf mapping element
		for (Element empElement : listEmpElement) {
			if (empElement.getName().equals("mappers"))
				empElement.addContent(el);

		}
		return jdo;
	}

	private static String convertToMapperSyntax(String className) {
		return SQL_MAPS_PATH + schemaName + className
				+ "Mapper.xml";
	}

	private Document createMyBatisXMLGenConfigNodes(Document jdo, String className, File configFile) throws JDOMException, IOException {

		logger.info("Parse MyBatis Template: " + configFile.getAbsolutePath());

		if (jdo == null)
			jdo = DOMEditor.parse("mybatis", configFile.getAbsolutePath());

		// dedupe
		if (!alreadyAdded.contains(className)) {
			alreadyAdded.add(className);
			Element el = new Element("table").setAttribute("schema", schemaName)
					.setAttribute("tableName", Table
							.convertToDBSyntax(className));

			Element el2 = new Element("generatedKey")
					.setAttribute("column", "id")
					.setAttribute("sqlStatement", "JDBC");
			el.addContent(el2);

			Element rootElement = jdo.getRootElement();
			List listEmpElement = rootElement.getChildren();

			// loop through to add every sqlf mapping element
			for (Element empElement : listEmpElement) {
				if (empElement.getName().equals("context"))
					empElement.addContent(el);

				Object pn = empElement.getAttribute("tableName");
				if (pn != null) {
					if (pn.toString().equals("PLACEHOLDER_NODE")) {
						rootElement.removeContent(empElement);
					}
				}
			}
		}
		return jdo;
	}

	@Override
	public String toString() {
		return "MyBatis Generator";
	}

	static void createMyBatisFromModelFolder() throws Exception {
		logger.info("Iterate Swagger Entities and create Tables...");
		File[] modelFiles = Gen
				.getJavaFiles(JAVA_GEN_SRC_FOLDER + "/" + MODEL_PACKAGE_DIR, false);
		MyBatisGen gen = new MyBatisGen();
		for (File mf : modelFiles) {
			String cn = mf.getName().substring(0, mf.getName().indexOf("."));
			cn = IGNITE_MODEL_PACKAGE + "." + cn;
			logger.info("Loading Class from ModelFile: " + cn);
			URLClassLoader classLoader = new URLClassLoader(new URL[] {
					new File(JAVA_GEN_SRC_FOLDER).toURI().toURL() });
			Class loadedClass = classLoader.loadClass(cn);
			createMyBatis(loadedClass, gen);
			classLoader.close();
		}
		logger.info("Generate...");
		generate();
	}

}