diff --git a/mybatis-flex-spring/src/main/java/com/mybatisflex/spring/FlexSqlSessionFactoryBean.java b/mybatis-flex-spring/src/main/java/com/mybatisflex/spring/FlexSqlSessionFactoryBean.java index 4fc4e699..cbddee20 100644 --- a/mybatis-flex-spring/src/main/java/com/mybatisflex/spring/FlexSqlSessionFactoryBean.java +++ b/mybatis-flex-spring/src/main/java/com/mybatisflex/spring/FlexSqlSessionFactoryBean.java @@ -95,7 +95,7 @@ public class FlexSqlSessionFactoryBean extends SqlSessionFactoryBean private Properties configurationProperties; -// private SqlSessionFactoryBuilder sqlSessionFactoryBuilder = new SqlSessionFactoryBuilder(); + // private SqlSessionFactoryBuilder sqlSessionFactoryBuilder = new SqlSessionFactoryBuilder(); private SqlSessionFactoryBuilder sqlSessionFactoryBuilder = new FlexSqlSessionFactoryBuilder(); private SqlSessionFactory sqlSessionFactory; @@ -335,6 +335,9 @@ public class FlexSqlSessionFactoryBean extends SqlSessionFactoryBean */ @Override public void setConfiguration(Configuration configuration) { + if (!(configuration instanceof FlexConfiguration)) { + throw new IllegalArgumentException("Only support FlexConfiguration."); + } this.configuration = configuration; }