titan.lightbatis.mybatis.script.MybatisScriptFactory Maven / Gradle / Ivy

Go to download
Show more of this group Show more artifacts with this name
Show all versions of lightbatis-core Show documentation
通用数据访问构架
There is a newer version: 1.2.0
/**
 *
 * 联系作者扫描以下二维码：
 *
 █████████████████████████████████████
 █████████████████████████████████████
 ████ ▄▄▄▄▄ █▀█ █▄▄▀▀ ▀▄█ █ ▄▄▄▄▄ ████
 ████ █   █ █▀▀▀█ ▀▀ ████▄█ █   █ ████
 ████ █▄▄▄█ █▀ █▀▀▄▀▀▄ ▀█ █ █▄▄▄█ ████
 ████▄▄▄▄▄▄▄█▄▀ ▀▄█▄▀▄█ ▀ █▄▄▄▄▄▄▄████
 ████ ▄ ▄ ▀▄  ▄▀▄▀▄ █ █▀ █ ▀ ▀▄█▄▀████
 ████▄ ▄   ▄▄██▄█▀▄  ▄▄▀█ ▄▀  ▀█▀█████
 ████ ▀▄▄█ ▄▄▄ ▄█▄▄▀▄▄█▀ ▀▀▀▀▀▄▄█▀████
 █████ ▀ ▄ ▄▄█▀  ▄██ █▄▄▀  ▄ ▀▄▄▀█████
 ████▀▄  ▄▀▄▄█▄▀▄▀█▄▀▀ ▄ ▀▀▀ ▀▄ █▀████
 ████ ██▄▄▄▄█▀▄▀█▀█▀▄▀█ ▀▄▄█▀██▄▀█████
 ████▄███▄█▄█▀▄ █▄▀▄▄▀▄██ ▄▄▄ ▀   ████
 ████ ▄▄▄▄▄ █▄█▄ ▄▄  ██▄  █▄█ ▄▄▀█████
 ████ █   █ █ ▀█▄ ▀ ▄▄▀▀█ ▄▄▄▄▀ ▀ ████
 ████ █▄▄▄█ █ ▄▀███▀▄▄▄▄▄ █▄▀  ▄ █████
 ████▄▄▄▄▄▄▄█▄███▄█▄▄▄▄▄██▄█▄▄▄▄██████
 █████████████████████████████████████
 █████████████████████████████████████
 *
 * 基于 MyBatis 扩展的数据访问统一层
 */
package titan.lightbatis.mybatis.script;

import com.github.jknack.handlebars.Handlebars;
import com.github.jknack.handlebars.Template;
import com.github.jknack.handlebars.helper.*;
import titan.lightbatis.mybatis.meta.ColumnMeta;

import java.io.IOException;
import java.util.*;

/**
 * @author [email protected]
 * SQL 的脚本编辑处理
 */
public class MybatisScriptFactory {
	private static Handlebars handlebars = new Handlebars();

	static {
		handlebars.getLoader().setPrefix("/" + MybatisScriptFactory.class.getPackage().getName().replace(".", "/"));
		handlebars.getLoader().setSuffix(".mustache");
		handlebars.registerHelpers(new MustacheHelper());
		handlebars.registerHelper(IfHelper.NAME ,IfHelper.INSTANCE);
		handlebars.registerHelper(ConditionalHelpers.eq.name(), ConditionalHelpers.eq);
		handlebars.registerHelper(ConditionalHelpers.gt.name(), ConditionalHelpers.gt);
		handlebars.registerHelper(ConditionalHelpers.neq.name(), ConditionalHelpers.neq);
		handlebars.registerHelper(UnlessHelper.NAME, UnlessHelper.INSTANCE);
		handlebars.registerHelper(EachHelper.NAME, EachHelper.INSTANCE);
		handlebars.registerHelper(LogHelper.NAME, LogHelper.INSTANCE);
	}
	/**
	 * 
	 */
	public MybatisScriptFactory() {

	}

	public static String buildDynamicSelectSQL(String table, String mappedStatementId) throws IOException {
		Map param = new HashMap<>();
		param.put("table", table);
		param.put("mappedStatementId", mappedStatementId);
		Template template = handlebars.compile("dynamicSelectSQL");
		String script = template.apply(param);
		return script;
	}

	/**
	 * 生成查询的SQL
	 * @param mappedStatementId
	 * @param table
	 * @param columns
	 * @param orderColumns
	 * @return
	 * @throws IOException
	 */
	public static String buildSelectSQL(String mappedStatementId, String table,Set columns,Set orderColumns) throws IOException {
		Map param = new HashMap<>();
		param.put("table", table);
		param.put("columns", columns);
		param.put("orderColumns", orderColumns);
		param.put("mappedStatementId", mappedStatementId);
		Template template = handlebars.compile("selectSQL");
		String script = template.apply(param);
		return script;
	}

	public static String updateByPrimaryKey(String table, Set columns, Set pkColumns) throws IOException {
		Map param = new HashMap<>();
		param.put("table", table);
		param.put("columns", columns);
		param.put("pkColumns", pkColumns);
		Template template = handlebars.compile("updateByPrimaryKey");
		String script = template.apply(param);
		return script;
	}

	/**
	 * 生成按主键删除数据的SQL
	 *
	 * @param table
	 * @param pkColumns
	 * @return
	 * @throws IOException
	 */
	public static String deleteByPrimaryKey(String table, Set pkColumns) throws IOException {
		Map param = new HashMap<>();
		param.put("table", table);
		param.put("pkColumns", pkColumns);
		Template template = handlebars.compile("deleteByPrimaryKey");
		String script = template.apply(param);
		return script;
	}

	public static String removeByPrimaryKey(String table, ColumnMeta logicColumn, Set pkColumns)  throws IOException {
		Map param = new HashMap<>();
		param.put("table", table);
		param.put("logicColumn", logicColumn);
		param.put("pkColumns", pkColumns);
		Template template = handlebars.compile("removeByPrimaryKey");
		String script = template.apply(param);
		return script;
	}

	public static String buildInsert (String table, Set columns, Set values) throws IOException {
		Map param = new HashMap<>();
		param.put("table", table);
		param.put("columns", columns);
		param.put("values", values);
		Template template = handlebars.compile("insertSQL");
		String script = template.apply(param);
		return script;
	}

	public static String buildSave (String table, Set columns, Set values,Set updateColumns, Boolean updateOn) throws IOException {
		Map param = new HashMap<>();
		param.put("table", table);
		param.put("columns", columns);
		param.put("values", values);
		param.put("updateColumns", updateColumns);
		param.put("updateOn",updateOn);
		Template template = handlebars.compile("saveSQL");
		String script = template.apply(param);
		return script;
	}

}