|
@@ -0,0 +1,372 @@
|
|
|
+
|
|
|
+ * Copyright 2010-2015 the original author or authors.
|
|
|
+ * <p>
|
|
|
+ * 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
|
|
|
+ * <p>
|
|
|
+ * http:
|
|
|
+ * <p>
|
|
|
+ * 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.hsweb.web.mybatis.dynamic;
|
|
|
+
|
|
|
+import org.apache.ibatis.exceptions.PersistenceException;
|
|
|
+import org.apache.ibatis.executor.BatchResult;
|
|
|
+import org.apache.ibatis.session.*;
|
|
|
+import org.mybatis.spring.MyBatisExceptionTranslator;
|
|
|
+import org.mybatis.spring.SqlSessionTemplate;
|
|
|
+import org.springframework.beans.factory.DisposableBean;
|
|
|
+import org.springframework.dao.support.PersistenceExceptionTranslator;
|
|
|
+
|
|
|
+import java.lang.reflect.InvocationHandler;
|
|
|
+import java.lang.reflect.Method;
|
|
|
+import java.sql.Connection;
|
|
|
+import java.util.List;
|
|
|
+import java.util.Map;
|
|
|
+
|
|
|
+import static java.lang.reflect.Proxy.newProxyInstance;
|
|
|
+import static org.apache.ibatis.reflection.ExceptionUtil.unwrapThrowable;
|
|
|
+import static org.hsweb.web.mybatis.dynamic.DynamicSqlSessionUtils.*;
|
|
|
+import static org.springframework.util.Assert.notNull;
|
|
|
+
|
|
|
+public class DynamicSqlSessionTemplate extends SqlSessionTemplate implements SqlSession, DisposableBean {
|
|
|
+
|
|
|
+ private final SqlSessionFactory sqlSessionFactory;
|
|
|
+
|
|
|
+ private final ExecutorType executorType;
|
|
|
+
|
|
|
+ private final SqlSession sqlSessionProxy;
|
|
|
+
|
|
|
+ private final PersistenceExceptionTranslator exceptionTranslator;
|
|
|
+
|
|
|
+
|
|
|
+ * Constructs a Spring managed SqlSession with the {@code SqlSessionFactory}
|
|
|
+ * provided as an argument.
|
|
|
+ *
|
|
|
+ * @param sqlSessionFactory
|
|
|
+ */
|
|
|
+ public DynamicSqlSessionTemplate(SqlSessionFactory sqlSessionFactory) {
|
|
|
+ this(sqlSessionFactory, sqlSessionFactory.getConfiguration().getDefaultExecutorType());
|
|
|
+ }
|
|
|
+
|
|
|
+
|
|
|
+ * Constructs a Spring managed SqlSession with the {@code SqlSessionFactory}
|
|
|
+ * provided as an argument and the given {@code ExecutorType}
|
|
|
+ * {@code ExecutorType} cannot be changed once the {@code SqlSessionTemplate}
|
|
|
+ * is constructed.
|
|
|
+ *
|
|
|
+ * @param sqlSessionFactory
|
|
|
+ * @param executorType
|
|
|
+ */
|
|
|
+ public DynamicSqlSessionTemplate(SqlSessionFactory sqlSessionFactory, ExecutorType executorType) {
|
|
|
+ this(sqlSessionFactory, executorType,
|
|
|
+ new MyBatisExceptionTranslator(
|
|
|
+ sqlSessionFactory.getConfiguration().getEnvironment().getDataSource(), true));
|
|
|
+ }
|
|
|
+
|
|
|
+
|
|
|
+ * Constructs a Spring managed {@code SqlSession} with the given
|
|
|
+ * {@code SqlSessionFactory} and {@code ExecutorType}.
|
|
|
+ * A custom {@code SQLExceptionTranslator} can be provided as an
|
|
|
+ * argument so any {@code PersistenceException} thrown by MyBatis
|
|
|
+ * can be custom translated to a {@code RuntimeException}
|
|
|
+ * The {@code SQLExceptionTranslator} can also be null and thus no
|
|
|
+ * exception translation will be done and MyBatis exceptions will be
|
|
|
+ * thrown
|
|
|
+ *
|
|
|
+ * @param sqlSessionFactory
|
|
|
+ * @param executorType
|
|
|
+ * @param exceptionTranslator
|
|
|
+ */
|
|
|
+ public DynamicSqlSessionTemplate(SqlSessionFactory sqlSessionFactory, ExecutorType executorType,
|
|
|
+ PersistenceExceptionTranslator exceptionTranslator) {
|
|
|
+ super(sqlSessionFactory, executorType);
|
|
|
+
|
|
|
+ notNull(sqlSessionFactory, "Property 'sqlSessionFactory' is required");
|
|
|
+ notNull(executorType, "Property 'executorType' is required");
|
|
|
+
|
|
|
+ this.sqlSessionFactory = sqlSessionFactory;
|
|
|
+ this.executorType = executorType;
|
|
|
+ this.exceptionTranslator = exceptionTranslator;
|
|
|
+ this.sqlSessionProxy = (SqlSession) newProxyInstance(
|
|
|
+ SqlSessionFactory.class.getClassLoader(),
|
|
|
+ new Class[]{SqlSession.class},
|
|
|
+ new SqlSessionInterceptor());
|
|
|
+ }
|
|
|
+
|
|
|
+ public SqlSessionFactory getSqlSessionFactory() {
|
|
|
+ return this.sqlSessionFactory;
|
|
|
+ }
|
|
|
+
|
|
|
+ public ExecutorType getExecutorType() {
|
|
|
+ return this.executorType;
|
|
|
+ }
|
|
|
+
|
|
|
+ public PersistenceExceptionTranslator getPersistenceExceptionTranslator() {
|
|
|
+ return this.exceptionTranslator;
|
|
|
+ }
|
|
|
+
|
|
|
+
|
|
|
+ * {@inheritDoc}
|
|
|
+ */
|
|
|
+ @Override
|
|
|
+ public <T> T selectOne(String statement) {
|
|
|
+ return this.sqlSessionProxy.<T>selectOne(statement);
|
|
|
+ }
|
|
|
+
|
|
|
+
|
|
|
+ * {@inheritDoc}
|
|
|
+ */
|
|
|
+ @Override
|
|
|
+ public <T> T selectOne(String statement, Object parameter) {
|
|
|
+ return this.sqlSessionProxy.<T>selectOne(statement, parameter);
|
|
|
+ }
|
|
|
+
|
|
|
+
|
|
|
+ * {@inheritDoc}
|
|
|
+ */
|
|
|
+ @Override
|
|
|
+ public <K, V> Map<K, V> selectMap(String statement, String mapKey) {
|
|
|
+ return this.sqlSessionProxy.<K, V>selectMap(statement, mapKey);
|
|
|
+ }
|
|
|
+
|
|
|
+
|
|
|
+ * {@inheritDoc}
|
|
|
+ */
|
|
|
+ @Override
|
|
|
+ public <K, V> Map<K, V> selectMap(String statement, Object parameter, String mapKey) {
|
|
|
+ return this.sqlSessionProxy.<K, V>selectMap(statement, parameter, mapKey);
|
|
|
+ }
|
|
|
+
|
|
|
+
|
|
|
+ * {@inheritDoc}
|
|
|
+ */
|
|
|
+ @Override
|
|
|
+ public <K, V> Map<K, V> selectMap(String statement, Object parameter, String mapKey, RowBounds rowBounds) {
|
|
|
+ return this.sqlSessionProxy.<K, V>selectMap(statement, parameter, mapKey, rowBounds);
|
|
|
+ }
|
|
|
+
|
|
|
+
|
|
|
+ * {@inheritDoc}
|
|
|
+ */
|
|
|
+ @Override
|
|
|
+ public <E> List<E> selectList(String statement) {
|
|
|
+ return this.sqlSessionProxy.<E>selectList(statement);
|
|
|
+ }
|
|
|
+
|
|
|
+
|
|
|
+ * {@inheritDoc}
|
|
|
+ */
|
|
|
+ @Override
|
|
|
+ public <E> List<E> selectList(String statement, Object parameter) {
|
|
|
+ return this.sqlSessionProxy.<E>selectList(statement, parameter);
|
|
|
+ }
|
|
|
+
|
|
|
+
|
|
|
+ * {@inheritDoc}
|
|
|
+ */
|
|
|
+ @Override
|
|
|
+ public <E> List<E> selectList(String statement, Object parameter, RowBounds rowBounds) {
|
|
|
+ return this.sqlSessionProxy.<E>selectList(statement, parameter, rowBounds);
|
|
|
+ }
|
|
|
+
|
|
|
+
|
|
|
+ * {@inheritDoc}
|
|
|
+ */
|
|
|
+ @Override
|
|
|
+ public void select(String statement, ResultHandler handler) {
|
|
|
+ this.sqlSessionProxy.select(statement, handler);
|
|
|
+ }
|
|
|
+
|
|
|
+
|
|
|
+ * {@inheritDoc}
|
|
|
+ */
|
|
|
+ @Override
|
|
|
+ public void select(String statement, Object parameter, ResultHandler handler) {
|
|
|
+ this.sqlSessionProxy.select(statement, parameter, handler);
|
|
|
+ }
|
|
|
+
|
|
|
+
|
|
|
+ * {@inheritDoc}
|
|
|
+ */
|
|
|
+ @Override
|
|
|
+ public void select(String statement, Object parameter, RowBounds rowBounds, ResultHandler handler) {
|
|
|
+ this.sqlSessionProxy.select(statement, parameter, rowBounds, handler);
|
|
|
+ }
|
|
|
+
|
|
|
+
|
|
|
+ * {@inheritDoc}
|
|
|
+ */
|
|
|
+ @Override
|
|
|
+ public int insert(String statement) {
|
|
|
+ return this.sqlSessionProxy.insert(statement);
|
|
|
+ }
|
|
|
+
|
|
|
+
|
|
|
+ * {@inheritDoc}
|
|
|
+ */
|
|
|
+ @Override
|
|
|
+ public int insert(String statement, Object parameter) {
|
|
|
+ return this.sqlSessionProxy.insert(statement, parameter);
|
|
|
+ }
|
|
|
+
|
|
|
+
|
|
|
+ * {@inheritDoc}
|
|
|
+ */
|
|
|
+ @Override
|
|
|
+ public int update(String statement) {
|
|
|
+ return this.sqlSessionProxy.update(statement);
|
|
|
+ }
|
|
|
+
|
|
|
+
|
|
|
+ * {@inheritDoc}
|
|
|
+ */
|
|
|
+ @Override
|
|
|
+ public int update(String statement, Object parameter) {
|
|
|
+ return this.sqlSessionProxy.update(statement, parameter);
|
|
|
+ }
|
|
|
+
|
|
|
+
|
|
|
+ * {@inheritDoc}
|
|
|
+ */
|
|
|
+ @Override
|
|
|
+ public int delete(String statement) {
|
|
|
+ return this.sqlSessionProxy.delete(statement);
|
|
|
+ }
|
|
|
+
|
|
|
+
|
|
|
+ * {@inheritDoc}
|
|
|
+ */
|
|
|
+ @Override
|
|
|
+ public int delete(String statement, Object parameter) {
|
|
|
+ return this.sqlSessionProxy.delete(statement, parameter);
|
|
|
+ }
|
|
|
+
|
|
|
+
|
|
|
+ * {@inheritDoc}
|
|
|
+ */
|
|
|
+ @Override
|
|
|
+ public <T> T getMapper(Class<T> type) {
|
|
|
+ return getConfiguration().getMapper(type, this);
|
|
|
+ }
|
|
|
+
|
|
|
+
|
|
|
+ * {@inheritDoc}
|
|
|
+ */
|
|
|
+ @Override
|
|
|
+ public void commit() {
|
|
|
+ throw new UnsupportedOperationException("Manual commit is not allowed over a Spring managed SqlSession");
|
|
|
+ }
|
|
|
+
|
|
|
+
|
|
|
+ * {@inheritDoc}
|
|
|
+ */
|
|
|
+ @Override
|
|
|
+ public void commit(boolean force) {
|
|
|
+ throw new UnsupportedOperationException("Manual commit is not allowed over a Spring managed SqlSession");
|
|
|
+ }
|
|
|
+
|
|
|
+
|
|
|
+ * {@inheritDoc}
|
|
|
+ */
|
|
|
+ @Override
|
|
|
+ public void rollback() {
|
|
|
+ throw new UnsupportedOperationException("Manual rollback is not allowed over a Spring managed SqlSession");
|
|
|
+ }
|
|
|
+
|
|
|
+
|
|
|
+ * {@inheritDoc}
|
|
|
+ */
|
|
|
+ @Override
|
|
|
+ public void rollback(boolean force) {
|
|
|
+ throw new UnsupportedOperationException("Manual rollback is not allowed over a Spring managed SqlSession");
|
|
|
+ }
|
|
|
+
|
|
|
+
|
|
|
+ * {@inheritDoc}
|
|
|
+ */
|
|
|
+ @Override
|
|
|
+ public void close() {
|
|
|
+ throw new UnsupportedOperationException("Manual close is not allowed over a Spring managed SqlSession");
|
|
|
+ }
|
|
|
+
|
|
|
+
|
|
|
+ * {@inheritDoc}
|
|
|
+ */
|
|
|
+ @Override
|
|
|
+ public void clearCache() {
|
|
|
+ this.sqlSessionProxy.clearCache();
|
|
|
+ }
|
|
|
+
|
|
|
+
|
|
|
+ * {@inheritDoc}
|
|
|
+ */
|
|
|
+ @Override
|
|
|
+ public Configuration getConfiguration() {
|
|
|
+ return this.sqlSessionFactory.getConfiguration();
|
|
|
+ }
|
|
|
+
|
|
|
+
|
|
|
+ * {@inheritDoc}
|
|
|
+ */
|
|
|
+ @Override
|
|
|
+ public Connection getConnection() {
|
|
|
+ return this.sqlSessionProxy.getConnection();
|
|
|
+ }
|
|
|
+
|
|
|
+
|
|
|
+ * {@inheritDoc}
|
|
|
+ *
|
|
|
+ * @since 1.0.2
|
|
|
+ */
|
|
|
+ @Override
|
|
|
+ public List<BatchResult> flushStatements() {
|
|
|
+ return this.sqlSessionProxy.flushStatements();
|
|
|
+ }
|
|
|
+
|
|
|
+ @Override
|
|
|
+ public void destroy() throws Exception {
|
|
|
+ }
|
|
|
+
|
|
|
+ private class SqlSessionInterceptor implements InvocationHandler {
|
|
|
+ @Override
|
|
|
+ public Object invoke(Object proxy, Method method, Object[] args) throws Throwable {
|
|
|
+ SqlSession sqlSession = getSqlSession(
|
|
|
+ DynamicSqlSessionTemplate.this.sqlSessionFactory,
|
|
|
+ DynamicSqlSessionTemplate.this.executorType,
|
|
|
+ DynamicSqlSessionTemplate.this.exceptionTranslator);
|
|
|
+
|
|
|
+ try {
|
|
|
+ Object result = method.invoke(sqlSession, args);
|
|
|
+ if (!isSqlSessionTransactional(sqlSession, DynamicSqlSessionTemplate.this.sqlSessionFactory)) {
|
|
|
+
|
|
|
+
|
|
|
+ sqlSession.commit(true);
|
|
|
+ }
|
|
|
+ return result;
|
|
|
+ } catch (Throwable t) {
|
|
|
+ Throwable unwrapped = unwrapThrowable(t);
|
|
|
+ if (DynamicSqlSessionTemplate.this.exceptionTranslator != null && unwrapped instanceof PersistenceException) {
|
|
|
+
|
|
|
+ closeSqlSession(sqlSession, DynamicSqlSessionTemplate.this.sqlSessionFactory);
|
|
|
+ sqlSession = null;
|
|
|
+ Throwable translated = DynamicSqlSessionTemplate.this.exceptionTranslator.translateExceptionIfPossible((PersistenceException) unwrapped);
|
|
|
+ if (translated != null) {
|
|
|
+ unwrapped = translated;
|
|
|
+ }
|
|
|
+ }
|
|
|
+ throw unwrapped;
|
|
|
+ } finally {
|
|
|
+ if (sqlSession != null) {
|
|
|
+ closeSqlSession(sqlSession, DynamicSqlSessionTemplate.this.sqlSessionFactory);
|
|
|
+ }
|
|
|
+ }
|
|
|
+ }
|
|
|
+ }
|
|
|
+
|
|
|
+}
|