38 lines
1.2 KiB
Java
38 lines
1.2 KiB
Java
|
package com.realtime.protection.configuration.utils;
|
||
|
|
|
||
|
|
import org.apache.ibatis.session.ExecutorType;
|
||
|
|
import org.apache.ibatis.session.SqlSession;
|
||
|
|
import org.apache.ibatis.session.SqlSessionFactory;
|
||
|
|
import org.springframework.stereotype.Component;
|
||
|
|
|
||
|
|
import java.util.function.Function;
|
||
|
|
|
||
|
|
@Component
|
||
|
|
public class SqlSessionWrapper {
|
||
|
|
|
||
|
|
private final SqlSessionFactory sqlSessionFactory;
|
||
|
|
|
||
|
|
public SqlSessionWrapper(SqlSessionFactory sqlSessionFactory) {
|
||
|
|
this.sqlSessionFactory = sqlSessionFactory;
|
||
|
|
}
|
||
|
|
|
||
|
|
public <M, I, O> O startBatchSession(Class<M> mapperClass,
|
||
|
|
Function<M, Function<I, O>> wrappedFunction,
|
||
|
|
I arguments) {
|
||
|
|
SqlSession sqlSession = sqlSessionFactory.openSession(ExecutorType.BATCH, false);
|
||
|
|
M mapper = sqlSession.getMapper(mapperClass);
|
||
|
|
try {
|
||
|
|
O result = wrappedFunction.apply(mapper).apply(arguments);
|
||
|
|
sqlSession.commit();
|
||
|
|
sqlSession.clearCache();
|
||
|
|
|
||
|
|
return result;
|
||
|
|
} catch (Exception e) {
|
||
|
|
sqlSession.rollback();
|
||
|
|
throw e;
|
||
|
|
} finally {
|
||
|
|
sqlSession.close();
|
||
|
|
}
|
||
|
|
}
|
||
|
|
}
|