All Downloads are FREE. Search and download functionalities are using the official Maven repository.

org.tinygroup.tinydb.spring.PreparedStatementSetterMapSetter Maven / Gradle / Ivy

There is a newer version: 2.2.3
Show newest version
/**
 *  Copyright (c) 1997-2013, www.tinygroup.org ([email protected]).
 *
 *  Licensed under the GPL, Version 3.0 (the "License");
 *  you may not use this file except in compliance with the License.
 *  You may obtain a copy of the License at
 *
 *       http://www.gnu.org/licenses/gpl.html
 *
 *  Unless required by applicable law or agreed to in writing, software
 *  distributed under the License is distributed on an "AS IS" BASIS,
 *  WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
 *  See the License for the specific language governing permissions and
 *  limitations under the License.
 */
package org.tinygroup.tinydb.spring;

import org.springframework.jdbc.core.PreparedStatementSetter;

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

public class PreparedStatementSetterMapSetter implements PreparedStatementSetter {

	private List parameters;
	private List dataTypes;
	
	public PreparedStatementSetterMapSetter( List parameters,List dataTypes){
		this.parameters = parameters;
		this.dataTypes = dataTypes;
	}
	
	private void dealParams(PreparedStatement ps,
			List parameters, List dataTypes)
			throws SQLException {
		if (parameters != null) {// 20120326 增加非空判断
			for (int i = 0; i < parameters.size(); i++) {
				if (dataTypes.size() > i && dataTypes.get(i) != null){
					setParameter(ps, i + 1, parameters.get(i), dataTypes.get(i));
				}else{
					setParameter(ps, i + 1, parameters.get(i), null);
				}	
			}
		}
	}

	private void setParameter(PreparedStatement ps, int index, Object obj,
			Integer dataType) throws SQLException {
		if (obj == null) {
			ps.setNull(index, dataType);
		} else {
			if (obj instanceof Character || obj.getClass().equals(char.class)) {
				ps.setString(index, obj.toString());
			} else if (obj instanceof String
					|| obj.getClass().equals(String.class)) {
				ps.setString(index, obj.toString());
			} else if (obj instanceof byte[]
					|| obj.getClass().equals(byte[].class)) {
				ps.setBytes(index, (byte[]) obj);
			} else {
				if (dataType == null) {
					ps.setObject(index, obj);
				} else {
					ps.setObject(index, obj, dataType);
				}
			}
		}
	}

	public void setValues(PreparedStatement ps) throws SQLException {
		dealParams(ps, parameters, dataTypes);
	}

}