/*
* Copyright 2009-2010 the original author or authors.
*
* Licensed under the Apache License, Version 2.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.apache.org/licenses/LICENSE-2.0
*
* Unless required by applicable law or agreed to in writing, software
* distributed under the License i 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 net.paoding.rose.jade.statement;
import java.util.ArrayList;
import java.util.HashMap;
import java.util.List;
import java.util.Map;
import org.apache.commons.lang.ClassUtils;
import org.springframework.dao.DataRetrievalFailureException;
import org.springframework.dao.InvalidDataAccessApiUsageException;
import org.springframework.jdbc.support.GeneratedKeyHolder;
import org.springframework.jdbc.support.KeyHolder;
import net.paoding.rose.jade.annotation.SQLType;
import net.paoding.rose.jade.dataaccess.DataAccess;
import net.paoding.rose.jade.dataaccess.DataAccessFactory;
/**
*
* @author 王志亮 [qieqie.wang@gmail.com]
* @author 廖涵 [in355hz@gmail.com]
*/
public class UpdateQuerier implements Querier {
private final DataAccessFactory dataAccessFactory;
private final Class<?> returnType;
private DynamicReturnGeneratedKeys returnGeneratedKeys;
public UpdateQuerier(DataAccessFactory dataAccessFactory, StatementMetaData metaData) {
this.dataAccessFactory = dataAccessFactory;
// 转换基本类型
Class<?> returnType = metaData.getReturnType();
if (returnType.isPrimitive()) {
returnType = ClassUtils.primitiveToWrapper(returnType);
}
this.returnType = returnType;
this.returnGeneratedKeys = metaData.getReturnGeneratedKeys();
}
@Override
public Object execute(SQLType sqlType, StatementRuntime... runtimes) {
switch (runtimes.length) {
case 1:
return executeSingle(runtimes[0]);
case 0:
return 0;
default:
return executeBatch(runtimes);
}
}
private Object executeSingle(StatementRuntime runtime) {
Number result;
DataAccess dataAccess = dataAccessFactory.getDataAccess(//
runtime.getMetaData(), runtime.getAttributes());
if (returnGeneratedKeys.shouldReturnGerneratedKeys(runtime)) {
ArrayList<Map<String, Object>> keys = new ArrayList<Map<String, Object>>(1);
KeyHolder generatedKeyHolder = new GeneratedKeyHolder(keys);
dataAccess.update(runtime.getSQL(), runtime.getArgs(), generatedKeyHolder);
if (keys.size() > 0) {
result = generatedKeyHolder.getKey();
} else {
result = null;
}
} else {
result = new Integer(dataAccess.update(runtime.getSQL(), runtime.getArgs(), null));
}
//
if (result == null || returnType == void.class) {
return null;
}
if (returnType == result.getClass()) {
return result;
}
// 将结果转成方法的返回类型
if (returnType == Integer.class) {
return result.intValue();
} else if (returnType == Long.class) {
return result.longValue();
} else if (returnType == Boolean.class) {
return result.intValue() > 0 ? Boolean.TRUE : Boolean.FALSE;
} else if (returnType == Double.class) {
return result.doubleValue();
} else if (returnType == Float.class) {
return result.floatValue();
} else if (returnType == Number.class) {
return result;
} else if (returnType == String.class || returnType == CharSequence.class) {
return String.valueOf(result);
} else {
throw new DataRetrievalFailureException(
"The generated key is not of a supported numeric type: " + returnType.getName());
}
}
//TODO: 支持returnGeneratedKeys (因JdbcTemplate不支持且必要性存疑，暂不实现）
private Object executeBatch(StatementRuntime... runtimes) {
int[] updatedArray = new int[runtimes.length];
Map<String, List<StatementRuntime>> batchs = new HashMap<String, List<StatementRuntime>>();
for (int i = 0; i < runtimes.length; i++) {
StatementRuntime runtime = runtimes[i];
List<StatementRuntime> batch = batchs.get(runtime.getSQL());
if (batch == null) {
batch = new ArrayList<StatementRuntime>(runtimes.length);
batchs.put(runtime.getSQL(), batch);
}
runtime.setAttribute("_index_at_batch_", i); // 该runtime在batch中的位置
batch.add(runtime);
}
// TODO: 多个真正的batch可以考虑并行执行(而非顺序执行)~待定
for (Map.Entry<String, List<StatementRuntime>> batch : batchs.entrySet()) {
String sql = batch.getKey();
List<StatementRuntime> batchRuntimes = batch.getValue();
StatementRuntime runtime = batchRuntimes.get(0);
DataAccess dataAccess = dataAccessFactory.getDataAccess(//
runtime.getMetaData(), runtime.getAttributes());
List<Object[]> argsList = new ArrayList<Object[]>(batchRuntimes.size());
for (StatementRuntime batchRuntime : batchRuntimes) {
argsList.add(batchRuntime.getArgs());
}
int[] batchResult = dataAccess.batchUpdate(sql, argsList);
if (batchs.size() == 1) {
updatedArray = batchResult;
} else {
int index_at_sub_batch = 0;
for (StatementRuntime batchRuntime : batchRuntimes) {
Integer _index_at_batch_ = batchRuntime.getAttribute("_index_at_batch_");
updatedArray[_index_at_batch_] = batchResult[index_at_sub_batch++];
}
}
}
if (returnType == void.class) {
return null;
}
if (returnType == int[].class) {
return updatedArray;
}
if (returnType == Integer.class || returnType == Boolean.class) {
int updated = 0;
for (int value : updatedArray) {
updated += value;
}
return returnType == Boolean.class ? updated > 0 : updated;
}
throw new InvalidDataAccessApiUsageException(
"bad return type for batch update: " + runtimes[0].getMetaData().getMethod());
}
@SuppressWarnings("unused")
private Object _executeBatch(StatementRuntime... runtimes) {
int[] updatedArray = new int[runtimes.length];
for (int i = 0; i < updatedArray.length; i++) {
StatementRuntime runtime = runtimes[i];
updatedArray[i] = (Integer) executeSingle(runtime);
}
return updatedArray;
}
}