MyBatis 自动关闭 session

在某些小项目中，需要单独使用到 mybatis，但是网上常见工具类方法获取的 session 通常要么需要手动关闭，这样即麻烦，而且有时又容易出错，要么要需要结合使用spring，但是我们只想写个简单的增删改，不想引入太多框架。

而下面将要介绍的SqlSessionManager所获取的session以及mapper就无需关心连接关闭的事情了。

环境搭建

数据库与实体类：

工具类：

public class MybatisUtils {
	
	private static SqlSessionManager sessionManager = null;

	static {
		try {
			InputStream inputStream = MybatisUtils.class.getClassLoader()
                                                 .getResourceAsStream("mybatis-config1.xml");
			sessionManager = SqlSessionManager.newInstance(inputStream);
		} catch (Exception e) {
			throw new RuntimeException("the Configuration of Mybatis is not exist!");
		}
	}
	
	/**
	 * 获取 Mapper
	 * @param <T>
	 * @param clazz
	 * @return
	 */
	public static <T> T getMapper(Class<T> clazz) {
		return sessionManager.getMapper(clazz);
	}
	
	/**
	 * 获取 SqlSessionManager 
	 * @return  
	 */
	public static SqlSessionManager getSessionManager() {
		return sessionManager;
	}
}

mapper：

public interface AccountMapper {

	@Select("select * from account")
	public List<Account> selectAccountList();
	
	@Select("select * from account where id = #{id}")
	public Account selectAccountById(Long id);
	
	@Update("update account set money = #{money} where id = #{userId}")
	public void updateAccount(Account account);
}

原理

它的原理就是使用了JDK的动态代理，先看看SqlSessionManager的构造方法

public class SqlSessionManager implements SqlSessionFactory, SqlSession {

  private final SqlSessionFactory sqlSessionFactory;
  private final SqlSession sqlSessionProxy;

  private final ThreadLocal<SqlSession> localSqlSession = new ThreadLocal<>();

  private SqlSessionManager(SqlSessionFactory sqlSessionFactory) {
    this.sqlSessionFactory = sqlSessionFactory;
    this.sqlSessionProxy = (SqlSession) Proxy.newProxyInstance(
        SqlSessionFactory.class.getClassLoader(),
        new Class[]{SqlSession.class},
        new SqlSessionInterceptor());
  }

从上面可以看出SqlSessionManager使用了Proxy.newProxyInstance()即JDK动态代理对默认的session进行了增强，增强类为SqlSessionInterceptor。

下面来结合具体的例子来看一下增强方法

@Test
public void test1() {
    AccountMapper mapper = MybatisUtils.getMapper(AccountMapper.class);
    List<Account> list = mapper.selectAccountList();
    System.out.println(list);
}

上面是一个很简单的查询方法，

然后我们在源码画线处打一个断点