All Downloads are FREE. Search and download functionalities are using the official Maven repository.
  • Log In
  • Register

    • org.onetwo.dbm.jdbc.DbmNamedJdbcTemplate Maven / Gradle / Ivy

    The newest version!
    package org.onetwo.dbm.jdbc;

import java.sql.PreparedStatement;
import java.sql.SQLException;
import java.util.List;
import java.util.Map;

import javax.sql.DataSource;

import org.apache.commons.lang3.ArrayUtils;
import org.onetwo.dbm.jdbc.internal.SpringStatementParameterSetter;
import org.onetwo.dbm.jdbc.spi.JdbcStatementParameterSetter;
import org.slf4j.Logger;
import org.slf4j.LoggerFactory;
import org.springframework.dao.DataAccessException;
import org.springframework.jdbc.core.JdbcOperations;
import org.springframework.jdbc.core.PreparedStatementCallback;
import org.springframework.jdbc.core.PreparedStatementCreator;
import org.springframework.jdbc.core.PreparedStatementSetter;
import org.springframework.jdbc.core.SqlParameter;
import org.springframework.jdbc.core.namedparam.MapSqlParameterSource;
import org.springframework.jdbc.core.namedparam.NamedParameterJdbcTemplate;
import org.springframework.jdbc.core.namedparam.NamedParameterUtils;
import org.springframework.jdbc.core.namedparam.ParsedSql;
import org.springframework.jdbc.core.namedparam.SqlParameterSource;

/***
 * 对jdbc查询的扩展，查询的sql里可包含命名的参数
 * @author weishao
 *
 */
public class DbmNamedJdbcTemplate extends NamedParameterJdbcTemplate {
	private final Logger logger = LoggerFactory.getLogger(this.getClass());

	private JdbcStatementParameterSetter jdbcParameterSetter = new SpringStatementParameterSetter();

	public DbmNamedJdbcTemplate(DataSource dataSource) {
		super(dataSource);
	}

	public DbmNamedJdbcTemplate(JdbcOperations classicJdbcTemplate) {
		super(classicJdbcTemplate);
	}

	public void setJdbcParameterSetter(JdbcStatementParameterSetter jdbcParameterSetter) {
		this.jdbcParameterSetter = jdbcParameterSetter;
	}

	public Object execute(String sql, Map paramMap) throws DataAccessException {
		PreparedStatementCreator pstCreator = getPreparedStatementCreator(sql, new MapSqlParameterSource(paramMap));
		final PreparedStatementSetter setter = (PreparedStatementSetter) pstCreator;
//		System.out.println("sql: " + ((SqlProvider)setter).getSql());
		return execute(sql, new MapSqlParameterSource(paramMap), new PreparedStatementCallback(){

			@Override
			public Object doInPreparedStatement(PreparedStatement ps) throws SQLException, DataAccessException {
				setter.setValues(ps);
				return ps.execute();
			}
			
		});
	}
	
	@Override
	protected PreparedStatementCreator getPreparedStatementCreator(String sql, SqlParameterSource paramSource) {
		ParsedSql parsedSql = getParsedSql(sql);
		String sqlToUse = NamedParameterUtils.substituteNamedParameters(parsedSql, paramSource);
		Object[] params = NamedParameterUtils.buildValueArray(parsedSql, paramSource, null);
		List declaredParameters = NamedParameterUtils.buildSqlParameterList(parsedSql, paramSource);
		/*DbmPreparedStatementCreatorFactory pscf = new DbmPreparedStatementCreatorFactory(sqlToUse, declaredParameters);
		return pscf.newPreparedStatementCreator(params);*/
		if(logger.isDebugEnabled()){
			logger.info("sql: {}", sqlToUse);
			logger.info("sql params: {}", ArrayUtils.toString(params));
		}

		/*PreparedStatementCreatorFactory pscf = new PreparedStatementCreatorFactory(sqlToUse, declaredParameters);
		return pscf.newPreparedStatementCreator(params);*/
		
		return this.createPreparedStatementCreator(sqlToUse, params, declaredParameters);
	}
	
	/****
	 * 增加JdbcStatementParameterSetter接口，统一参数设置入口
	 * @param sqlToUse
	 * @param params
	 * @param declaredParameters
	 * @return
	 */
	protected PreparedStatementCreator createPreparedStatementCreator(String sqlToUse, Object[] params, List declaredParameters) {
		DbmPreparedStatementCreatorFactory pscf = new DbmPreparedStatementCreatorFactory(sqlToUse, declaredParameters);
		pscf.setParameterSetter(jdbcParameterSetter);
		return pscf.newPreparedStatementCreator(params);
	}
	
	/*
	@Override
	public  List query(String sql, Map paramMap, RowMapper rowMapper)
			throws DataAccessException {

		return query(sql, new MapSqlParameterSource(paramMap), rowMapper);
	}
	
	static class DbmMapSqlParameterSource extends MapSqlParameterSource {
		
		public MapSqlParameterSource addValue(String paramName, Object value) {
			Object newValue = unwrapSqlParameterValue(value);
			return super.addValue(paramName, newValue);
		}
		
		@Override
		public MapSqlParameterSource addValue(String paramName, Object value, int sqlType) {
			Object newValue = unwrapSqlParameterValue(value);
			return super.addValue(paramName, newValue, sqlType);
		}

		@Override
		public MapSqlParameterSource addValue(String paramName, Object value, int sqlType, String typeName) {
			Object newValue = unwrapSqlParameterValue(value);
			return super.addValue(paramName, newValue, sqlType, typeName);
		}

		@Override
		public DbmMapSqlParameterSource addValues(Map values) {
			if (values != null) {
				values.entrySet().forEach(e->addValue(e.getKey(), e.getValue()));
			}
			return this;
		}

		public static Object unwrapSqlParameterValue(Object value){
			return DbmUtils.unwrapSqlParameterValue(value);
		}
	}*/

}
    
    
    
    

    




    
    
    © 2015 - 2025 Weber Informatics LLC | Privacy Policy