cn.featherfly.component.sorm.operate.AbstractQueryOperate Maven / Gradle / Ivy

Go to download

package cn.featherfly.component.sorm.operate;

import java.sql.ResultSet;

import org.springframework.jdbc.core.JdbcTemplate;

import cn.featherfly.common.bean.BeanUtils;
import cn.featherfly.common.db.JdbcUtils;
import cn.featherfly.common.lang.LangUtils;
import cn.featherfly.component.sorm.mapping.PropertyMapping;

/**
 * 
 * 数据库操作的抽象类
 * 
 * @param  对象类型
 * @author 钟冀
 * @since 1.0
 * @version 1.0
 */
public abstract class AbstractQueryOperate extends AbstractOperate{
	
	/**
	 * 使用给定数据源以及给定对象生成其相应的操作.
	 * @param type 类
	 * @param jdbcTemplate jdbcTemplate
	 */
	public AbstractQueryOperate(Class type, JdbcTemplate jdbcTemplate) {
		super(type, jdbcTemplate);
	}
	/**
	 * 使用给定数据源以及给定对象生成其相应的操作.
	 * @param type 类
	 * @param jdbcTemplate jdbcTemplate
	 * @param dataBase 具体库
	 */
	public AbstractQueryOperate(Class type, JdbcTemplate jdbcTemplate, String dataBase) {
		super(type, jdbcTemplate, dataBase);
	}

	/**
	 * 
	 * 每条记录映射为对象.
	 * 
	 * @param rs 结果集
	 * @param rowNumber 行数
	 * @return 映射后的对象
	 */
	protected T mapRow(ResultSet rs, int rowNumber) {
		@SuppressWarnings("unchecked")
		T mappedObject = (T) BeanUtils.instantiateClass(classMapping.getType());
		int index = 1;
		for (PropertyMapping propertyMapping : classMapping.getPropertyMappings()) {
			Object value = getColumnValue(rs, index, propertyMapping.getFinalPropertyType());
			if (logger.isDebugEnabled() && rowNumber == 0) {
				logger.debug("Mapping column '{}' to property '{}' of type {}", 
					new Object[]{propertyMapping.getColumnName(), 
						propertyMapping.getFinalPropertyName(), propertyMapping.getFinalPropertyType()
					}
				);
			}
			BeanUtils.setProperty(mappedObject, propertyMapping.getFinalPropertyName(), value);
			index++;
		}
		return mappedObject;
	}
	
	/**
	 * {@inheritDoc}
	 */
	@Override
	protected void initSql() {
		initSelectSql();
		StringBuilder getSql = new StringBuilder();		
		getSql.append(getSelectSql());
		String condition = initCondition();
		if (LangUtils.isNotEmpty(condition)) {
		    getSql.append(" where ")
            .append(condition);
		}
		this.sql = getSql.toString();
		logger.debug("sql: {}" , this.sql);
	}
	
	protected abstract String initCondition();
	
	// ********************************************************************
	//	
	// ********************************************************************
	
	private Object getColumnValue(ResultSet rs, int index, Class propertyType) {
		return JdbcUtils.getResultSetValue(rs, index, propertyType);
	}
	
	private void initSelectSql() {
		StringBuilder selectSql = new StringBuilder();
		selectSql.append("select ");
		int columnNum = 0;
		for (PropertyMapping propertyMapping : classMapping.getPropertyMappings()) {
			selectSql.append(propertyMapping.getColumnName())
				.append(" as ")
				.append(propertyMapping.getPropertyName())
				.append(",");
			columnNum++ ;
		}
		if (columnNum > 0) {
			selectSql.deleteCharAt(selectSql.length() - 1);
		}
		selectSql.append(" from ")
			.append(classMapping.getTableName());
		this.selectSql = selectSql.toString();		
	}
	
	// ********************************************************************
	//	
	// ********************************************************************

	private String selectSql;
	
	/**
	 * 返回selectSql
	 * @return selectSql
	 */
	public String getSelectSql() {
		return selectSql;
	}
}