titan.lightbatis.mybatis.QueryDslBuilder 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
/**
 * 
 */
package titan.lightbatis.mybatis;

import com.querydsl.core.DefaultQueryMetadata;
import com.querydsl.core.JoinType;
import com.querydsl.core.types.*;
import com.querydsl.sql.RelationalPath;
import com.querydsl.sql.SQLTemplates;
import com.querydsl.sql.SQLTemplatesRegistry;
import lombok.extern.slf4j.Slf4j;
import org.apache.ibatis.mapping.ParameterMapping;
import org.apache.ibatis.scripting.xmltags.DynamicContext;
import org.apache.ibatis.session.Configuration;
import titan.lightbatis.mybatis.meta.QEntity;

import javax.sql.DataSource;
import java.sql.Connection;
import java.sql.SQLException;
import java.util.List;
import java.util.Map;

/**
 * @author [email protected]
 *
 */
@Slf4j
public class QueryDslBuilder {

	private Configuration configuration = null;
	private SQLTemplatesRegistry templatesRegistry = new SQLTemplatesRegistry();
	private SQLTemplates sqlTemplates;
	private DefaultQueryMetadata queryMetadata = null;

	/**
	 * 
	 */
	public QueryDslBuilder(Configuration configuration) {
		this.configuration = configuration;
		this.queryMetadata = new DefaultQueryMetadata();
		DataSource ds = this.configuration.getEnvironment().getDataSource();
		try {
			Connection connection = ds.getConnection();
			sqlTemplates = templatesRegistry.getTemplates(connection.getMetaData());
			connection.close();
		} catch (SQLException e) {
			e.printStackTrace();
		}
	}

	public QueryDslBuilder from(QEntity queryEntity) {
		queryMetadata.addJoin(JoinType.DEFAULT,queryEntity);
		return this;
	}
	public QueryDslBuilder where(Predicate... predicates) {
		for (Predicate predicate: predicates) {
			queryMetadata.addWhere(predicate);
		}
		return this;
	}
	public void buildDeleteSQL(DynamicContext context, List parameterMappings, RelationalPath queryEntry) {
		LightbatisSQLSerializer sqlSerializer = new LightbatisSQLSerializer(new com.querydsl.sql.Configuration(sqlTemplates));
		sqlSerializer.serializeDelete(queryMetadata, queryEntry);
		String sql = sqlSerializer.toString();
		buildSQL(context, sql, sqlSerializer, parameterMappings);
	}
	public void buildUpdateSQL (DynamicContext context, List parameterMappings, RelationalPath queryEntry, Map, Expression> updates) {
		LightbatisSQLSerializer sqlSerializer = new LightbatisSQLSerializer(new com.querydsl.sql.Configuration(sqlTemplates));
		//sqlSerializer.serialize(queryMetadata,false);
		sqlSerializer.serializeUpdate(queryMetadata, queryEntry, updates);
		String sql = sqlSerializer.toString();
		buildSQL(context, sql, sqlSerializer, parameterMappings);
	}

	public void buildQuerySQL(DynamicContext context,  List parameterMappings, boolean forCountRow){
		LightbatisSQLSerializer sqlSerializer = new LightbatisSQLSerializer(new com.querydsl.sql.Configuration(sqlTemplates));
		sqlSerializer.serialize(queryMetadata,forCountRow);
		String sql = sqlSerializer.toString();
		//context.appendSql(sql);
		buildSQL(context, sql, sqlSerializer, parameterMappings);
//		List