接着昨天的Mybatis扩展——IDaoTemplate接口。
扩展9：批量执行
1、明确什么是批量执行
首先说明一下，这里的批量执行不是利用<foreach>标签生成一长串的sql字符串——这并不是真正的JDBC批量执行，我这里所说的批量是指在内核调用JDBC的addBatch、executeBatch等方法来实现的。类似下面的示例：
private void batchUpdate(Connection conn) throws SQLException {
PreparedStatement ps = null;
try{
String sql = " INSERT INTO BF_PARAM_ENUM_DATA "
+ " (PARAM_CODE,DATA_CODE,DATA_TEXT,DATA_PARAM,SEQNO,DES) "
+ " VALUES(?,?,?,?,?,?) ";
ps = conn.prepareStatement(sql);
for(int i = 0; i < 10; i++){
String random = RandomStringUtils.randomAlphabetic(8);
ps.setString(1, "TEST");//PARAM_CODE
ps.setString(2, random);//DATA_CODE
ps.setString(3, "数据" + random);//DATA_TEXT
ps.setString(4, "参数" + random);//DATA_PARAM
ps.setInt(5, i);//SEQNO
ps.setString(6, "");//MEMO
ps.addBatch();
}
int[] rs = ps.executeBatch();
System.out.print("batch insert : ");
for(int r : rs){
System.out.print(r+",");
}
System.out.println();
}finally{
try{
ps.close();
}catch(Exception e){}
}
}
2、批量执行的API
明白了批量执行的含义，接着看批量执行的API，再看一次IDaoTemplate接口中批量执行相关方法：
/**
* 执行批量：一个SQL执行多次
* @param sqlId SQL-ID
* @param parameters 参数对象数组
* @return 批量执行的影响记录数组
*/
public int[] executeBatch(String sqlId, List<?> parameters); /**
* 执行批量：一次执行多个SQL
* @param sqlIds 要执行的一组SQL-ID
* @return 批量执行的影响记录数组
*/
public int[] executeBatch(List<String> sqlIds); /**
* 执行批量：一次执行多个SQL
* @param sqlIds 要执行的一组SQL-ID
* @param parameters 参数对象数组
* @return 批量执行的影响记录数组
*/
public int[] executeBatch(List<String> sqlIds, List<?> parameters);
简单的理解，批量执行就是需要执行一组操作，要么是相同SQL不同的参数，要么就是不同的SQL（参数相同或不同都可以）。至于这些接口怎么实现，都比较简单，根据SqlSessionFactory，创建批量执行的会话，大概如下：
this.batchSqlSession = new SqlSessionTemplate(sqlSessionFactory, ExecutorType.BATCH);
然后和普通会话一样去执行update/insert/delete等操作，完了再调用批量会话的flushBatch方法返回结果即可。
这里有几个问题：
- 批量执行的事务：批量执行怎么保证事务，是否通过添加Transactional注解即可？事实上，事务注解也只是一个注解，它本身不会对程序运行起任何影响，只有解析这个注解的程序，根据注解属性执行不同的业务逻辑，才会真正体现注解的作用。而另外一方面，Spring注解必须要求调用方法是容器管理的Bean方法，而IDaoTemplate接口的实现类，并不一定是Spring容器中的bean。因此，这里通过事务注解并不能生效，而需要使用编程式事务。相关代码大致如下：
SqlSession batchSqlSession = getExecuteSqlSession();
PlatformTransactionManager txManager = this.getTransactinManager(dataSource);
if(null == txManager){
for(int i = 0, s = sqlIds.size(); i < s; i++){
batchSqlSession.update(SqlManager.getExecuteSqlId(sqlIds.get(i)), parameters==null?null : parameters.get(i));
}
List<BatchResult> r = batchSqlSession.flushStatements();
return resolveBatchResult(r);
}else{
DefaultTransactionDefinition def = new DefaultTransactionDefinition();
def.setIsolationLevel(TransactionDefinition.ISOLATION_READ_COMMITTED);
def.setPropagationBehavior(TransactionDefinition.PROPAGATION_REQUIRED);
TransactionStatus status = txManager.getTransaction(def);
try {
for(int i = 0, s = sqlIds.size(); i < s; i++){
batchSqlSession.update(SqlManager.getExecuteSqlId(sqlIds.get(i)), parameters==null?null : parameters.get(i));
}
List<BatchResult> r = batchSqlSession.flushStatements();
txManager.commit(status);
return resolveBatchResult(r);
} catch (Throwable e) {
txManager.rollback(status);
throw Throw.createRuntimeException(e);
}
}
- 执行方式的混合：如果在一个服务方法中，既有一般方式的dao访问，也有批量执行的dao访问，同时还处于一个事务中。由于Mybatis自身的限制，不能在同一个事务中切换执行方法，因此会抛出异常。那怎么办呢？我是利用ThreadLocal变量保存当前线程是否开启了批量执行模式，然后提供了如下三个管理方法作为批量执行的支持接口：
3、批量执行模式管理API
/**
* 打开批量执行模式
*/
public void openBatchType(); /**
* 恢复打开批量执行模式之前的执行模式
*/
public void resetBatchType(); /**
* 获取批量执行结果
* @return
*/
public int[] flushBatch();
这样，在前面提到的情况下，可以在服务方法开始，就调用openBatchType打开批量执行方式，然后依次执行各sql脚本，调用flushBatch()获取批量执行结果，在finally快再将当前线程的执行方式切换为调用openBatchType方法之前的执行方式。
批量执行先说到这里，下一次再结合Dao接口的方法签名、Dao接口的自动代理子类等详细说明怎么使用自动代理调用批量执行。
扩展10：merge
在实际业务中，常常会有存在更新、不存在插入的数据库操作——merge，这种操作在Oracle数据库中可以直接利用SQL语法去更新，但是这种方法并不通用，更好的方法就是在Java中添加这个接口方法，但是添加接口方法也面临一个问题：怎么保证事务性？
其实在看过前面的批量执行的事务处理之后，就没什么秘密可言了，类似的使用编程式事务就可以了。
遗憾的是——由于平台组评审过于谨慎，这个接口并没有添加到IDaoTemplate中：
/**
* 存在更新、不存在插入
* @param insertSqlId 插入的SQL-ID
* @param updateSqlId 更新的SQL-ID
* @param parameter 参数对象
* @return 影响的记录条数
*/
public int merge(String insertSqlId, String updateSqlId, Object parameter);
类似于merge，还可进一步做更多的扩展，只是通用性比较低，这里举一个例子来说明一下：
/**
* 根据条件执行不同sql
* <p>
* <ul>
* <li> 将参数对象parameter作为根对象，对条件表达式caseExpression求值
* <li> 根据求值结果，从sqlId映射中查找对应的sqlId，如果没有找到，就返回默认的sqlId
* <li> 根据计算的sqlId和参数对象，执行数据库操作，并返回结果
* </ul>
* </p>
* @param caseExpression 条件表达式，可为OGNL/SpEL
* @param sqlIdMapping sqlId映射
* @param defaultSqlId 默认的sqlId
* @param parameter 参数对象
* @return 影响的记录条数
*/
public int switchExecute(String caseExpression, Map<String, String> sqlIdMapping, String defaultSqlId, Object parameter);
今天到此为止。