io.starter.ignite.generator.MyBatisIgnitePluginAdapter 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.util.List;

import org.mybatis.generator.api.IntrospectedColumn;
import org.mybatis.generator.api.IntrospectedTable;
import org.mybatis.generator.api.Plugin;
import org.mybatis.generator.api.PluginAdapter;
import org.mybatis.generator.api.dom.java.Field;
import org.mybatis.generator.api.dom.java.FullyQualifiedJavaType;
import org.mybatis.generator.api.dom.java.JavaVisibility;
import org.mybatis.generator.api.dom.java.Method;
import org.mybatis.generator.api.dom.java.TopLevelClass;
import org.slf4j.Logger;
import org.slf4j.LoggerFactory;

import io.starter.toolkit.StringTool;

/**
 * @author John McMahon (@TechnoCharms) mcmahon
 *
 */
public class MyBatisIgnitePluginAdapter extends PluginAdapter
		implements Configuration {

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

	public MyBatisIgnitePluginAdapter() {
		logger.error("Instantiating MyBatisIgnitePluginAdapter...");
	}

	@Override
	public boolean modelBaseRecordClassGenerated(TopLevelClass topLevelClass, IntrospectedTable introspectedTable) {
		String scn = getSuperClassName(topLevelClass);
		Field f = new Field();
		f.setName("delegate");
		f.setVisibility(JavaVisibility.PUBLIC);
		f.setInitializationString(" new " + scn + "()");
		f.setType(new TopLevelClass(scn).getType());
		topLevelClass.addField(f);
		topLevelClass.addImportedType(scn);

		FullyQualifiedJavaType dtx = new FullyQualifiedJavaType(
				"java.util.Date");
		topLevelClass.addImportedType(dtx);

		FullyQualifiedJavaType dtd = new FullyQualifiedJavaType(
				"java.math.BigDecimal");
		topLevelClass.addImportedType(dtd);

		FullyQualifiedJavaType dti = new FullyQualifiedJavaType(
				"java.time.Instant");
		topLevelClass.addImportedType(dti);

		FullyQualifiedJavaType dto = new FullyQualifiedJavaType(
				"java.time.OffsetDateTime");
		topLevelClass.addImportedType(dto);

		FullyQualifiedJavaType dtz = new FullyQualifiedJavaType(
				"java.time.ZoneOffset");
		topLevelClass.addImportedType(dtz);

		// add custom methods
		Method tojson = new Method();
		tojson.addAnnotation("@Override");
		tojson.addBodyLine("return delegate.toJSON();");
		tojson.setName("toJSON");
		FullyQualifiedJavaType returnType = new FullyQualifiedJavaType(
				"java.lang.String");
		JavaVisibility visibility = JavaVisibility.PUBLIC;
		tojson.setVisibility(visibility);
		tojson.setReturnType(returnType);
		topLevelClass.addMethod(tojson);

		Method getdelegate = new Method();
		getdelegate.addAnnotation("@Override");
		getdelegate.addBodyLine("return delegate;");
		getdelegate.setName("getDelegate");
		
		FullyQualifiedJavaType ret = new FullyQualifiedJavaType(
				"io.starter.ignite.model.DataModelObject");
		topLevelClass.addImportedType(ret);
		getdelegate.setVisibility(visibility);
		getdelegate.setReturnType(ret);
		topLevelClass.addMethod(getdelegate);

		
		return true;
	}

	/**
	 * get the superclass name
	 * 
	 * @param topLevelClass
	 * @return
	 */
	private String getSuperClassName(TopLevelClass topLevelClass) {

		String cn = topLevelClass.getType().getFullyQualifiedName();

		// handle stripping out the Schema name in the delegate
		// class
		// if (cn.toLowerCase().contains(schemaName)) {
		// the schema is always lowercase, so adjust it for
		// Camelcase
		String ccsn = "dao." + schemaName;

		cn = cn.replace(ccsn, "");

		// replace package with actual delegate model package
		cn = cn.replace(MODEL_DAO_PACKAGE, MODEL_PACKAGE);

		if (cn.contains("..")) {
			throw new IllegalStateException(
					"Could not get getSuperClassName due to package collision: "
							+ cn + ". Change value of schemaName: "
							+ schemaName);
		}
		// }

		logger.info("SuperClass Name MYBATIS member: " + cn);
		return cn;
	}

	@Override
	public boolean modelGetterMethodGenerated(Method method, TopLevelClass topLevelClass, IntrospectedColumn introspectedColumn, IntrospectedTable introspectedTable, Plugin.ModelClassType modelClassType) {
		String secn = introspectedColumn.getRemarks();

		List ln = method.getBodyLines();
		for (String l : ln) {
			method.getBodyLines().remove(l);
			String cnm = introspectedColumn.getJdbcTypeName().toUpperCase();
			l = l.replace("return ", "return delegate.");
			if (cnm.contains("DATE") || cnm.contains("TIMESTAMP")) {
				l = l.replace(";", ".toInstant().getNano() / 1000000);");
				l = l.replace("return delegate.", "return new Date(delegate.");
			} else if (secn.contains(MYBATIS_COL_ENUM_FLAG)) {
				l = l.replace(";", ".getValue();");
			}
			method.addBodyLine(l);
		}
		return super.modelGetterMethodGenerated(method, topLevelClass, introspectedColumn, introspectedTable, modelClassType);
	}

	@Override
	public boolean modelSetterMethodGenerated(Method method, TopLevelClass topLevelClass, IntrospectedColumn introspectedColumn, IntrospectedTable introspectedTable, Plugin.ModelClassType modelClassType) {

		List ln = method.getBodyLines();

		String cnm = introspectedColumn.getJdbcTypeName().toUpperCase();
		String secn = introspectedColumn.getRemarks();

		List it = ln.subList(0, ln.size());
		for (String l : it) {
			method.getBodyLines().remove(l);
			l = l.replace("this.", "delegate.");
			String membername = l.substring(0, l.indexOf("="));
			String varname = membername.substring(membername.indexOf(".") + 1);
			if (cnm.contains("DATE") || cnm.contains("TIMESTAMP")) {

				method.addBodyLine("Instant instant = " + varname
						+ ".toInstant();");
				method.addBodyLine("String TIMEZONE_OFFSET = \""
						+ TIMEZONE_OFFSET + "\";");
				method.addBodyLine("ZoneOffset offset = ZoneOffset.of(TIMEZONE_OFFSET);");

				String bl = "OffsetDateTime odt = OffsetDateTime.ofInstant(instant, offset);";
				method.addBodyLine(bl);
				String stx = membername + " = odt;";
				method.addBodyLine(stx);

			} else if (secn.contains(MYBATIS_COL_ENUM_FLAG)) {
				method.addBodyLine(getEnumHandling(varname, introspectedTable
						.getAliasedFullyQualifiedTableNameAtRuntime()));
			} else {
				method.addBodyLine(l);
			}

		}

		return super.modelGetterMethodGenerated(method, topLevelClass, introspectedColumn, introspectedTable, modelClassType);
	}

	@Override
	public boolean modelFieldGenerated(Field field, TopLevelClass topLevelClass, IntrospectedColumn introspectedColumn, IntrospectedTable introspectedTable, ModelClassType modelClassType) {

		logger.info("MyBatisIgnitePluginAdapter Generating: " + field + " name:"
				+ field.getName() + LINE_FEED + " class:"
				+ field.getType().getShortName());

		field.setVisibility(JavaVisibility.PROTECTED);

		/** TODO: add special annotations to field
		if (ANNOTATAION_CLASS != null) {
			topLevelClass.addImportedType(new FullyQualifiedJavaType(
					ANNOTATAION_CLASS));
		}
		
		if (field.getType().getFullyQualifiedName()
				.equals("java.lang.String")) {
			field.addAnnotation("@" + ANNOTATAION_CLASS);
		} else if (field.getType().equals("java.util.Date")) {
			logger.warn("MyBatisIgnitePluginAdapter SecureField TODO: handle dates: "
					+ field);
		}*/

		return false;
	}

	private String getEnumHandling(String enumName, String tableName) {
		tableName = tableName.substring(TABLE_NAME_PREFIX.length());
		tableName = DBGen.camelize(tableName);
		tableName = StringTool.getUpperCaseFirstLetter(tableName.trim());

		enumName = enumName.trim();

		String setEnumValMethod = StringTool.getSetMethodNameFromVar(enumName);

		String fullEnumName = tableName + "."
				+ StringTool.getUpperCaseFirstLetter(enumName.trim());;
		String enumcode = "delegate." + setEnumValMethod.trim() + "("
				+ fullEnumName + "Enum.fromValue(" + enumName
				+ " == null ? null : " + enumName + ".trim()));";
		return enumcode;
	}

	@Override
	public boolean validate(List warnings) {
		logger.info("MyBatis Warnings: ");
		for (String w : warnings) {
			logger.info(w);
		}
		logger.info("End MyBatis Warnings");
		return true;
	}
}