zl程序教程

您现在的位置是:首页 >  后端

当前栏目

如何创建MyBatis的工具类

mybatis工具 如何 创建
2023-09-27 14:27:14 时间

目录

一、前言

二、实现过程

1.创建一个ThreadLocal对象

2.初始化SqlSessionFactory

3.获取并存储sqlSession对象

4.关闭sqlSession对象

三、 总代码


一、前言

使用MyBatis框架在持久层操作中,如果多个DML操作都属于一个事务,因为commit()和rollback()都是SqlSession完成的,所以必须保证只使用一个SqlSession,但是不同的DML操作可能在不同类的不同方法中,每个方法都需要获取SqlSession,所以应该如何在多个DML操作中保证使用同一个SqlSession呢?

二、实现过程

1.创建一个ThreadLocal对象

下面给大家带来了一个工具类:ThreadLocal,它可以储存SqlSession对象,并且保证一个线程只使用一个SqlSession。现在来学习一下如何创建MyBatis这个工具类

首先需要new一个ThreadLocal对象,以便存储SqlSession:

 private static ThreadLocal<SqlSession> threadLocal = new ThreadLocal<>();

2.初始化SqlSessionFactory

在一个静态代码块中初始化SqlSessionFactory:

 static {
        //创建SqlSessionFactory
        InputStream is = null;
        try {
            is = Resources.getResourceAsStream("mybatis-cfg.xml");

        } catch (IOException e) {
            e.printStackTrace();
        }
        sqlSessionFactory = new SqlSessionFactoryBuilder().build(is);

    }

3.获取并存储sqlSession对象

接下来就是从ThreadLocal中获取sqlSession对象,并且使用ThreadLocal存储该对象:

tips:存储SqlSession的时候要判断该对象是否为空,要是为空就要获取一个,然后再存到ThreadLocal,返回一个SqlSession。

//获取SqlSession
    public static SqlSession getSqlSession(){
        SqlSession sqlSession = threadLocal.get(); 
        
        if(sqlSession == null){
            sqlSession = sqlSessionFactory.openSession();     
            threadLocal.set(sqlSession);                        
        }

        return sqlSession;
    }

4.关闭sqlSession对象

接下来也可以在该工具类写关闭sqlSession方法

  //关闭sqlSession
    public static void closeSqlSession(){
        SqlSession sqlSession = threadLocal.get();
        if (sqlSession != null){           
            sqlSession.close();
            threadLocal.set(null);
        }
    }

三、 总代码

public class MyBatisUtils {
    private static ThreadLocal<SqlSession> threadLocal = new ThreadLocal<>();
    private static SqlSessionFactory    sqlSessionFactory = null;
    static {
        //创建SqlSessionFactory
        InputStream is = null;
        try {
            is = Resources.getResourceAsStream("mybatis-cfg.xml");

        } catch (IOException e) {
            e.printStackTrace();
        }
        sqlSessionFactory = new SqlSessionFactoryBuilder().build(is);

    }
    //获取SqlSession
    public static SqlSession getSqlSession(){
        SqlSession sqlSession = threadLocal.get();        
        if(sqlSession == null){
            sqlSession = sqlSessionFactory.openSession();       
            threadLocal.set(sqlSession);                       
        }
        return sqlSession;
    }

    //关闭sqlSession
    public static void closeSqlSession(){
        SqlSession sqlSession = threadLocal.get();
        if (sqlSession != null){            
            sqlSession.close();
            threadLocal.set(null);
        }
    }
}