/Util/MyBatisUtil.java

最新推荐文章于 2024-03-01 19:52:20 发布

原创最新推荐文章于 2024-03-01 19:52:20 发布 · 338 阅读

0 ·

CC 4.0 BY-SA版权

java 专栏收录该内容

42 篇文章

订阅专栏

本文介绍了一个针对MyBatis框架设计的实用工具类，该类实现了SqlSession的线程安全获取与释放，确保了资源的有效管理和重复利用。通过使用ThreadLocal保证了每个线程内的SqlSession实例唯一性。

摘要生成于 C知道，由 DeepSeek-R1 满血版支持，前往体验 >

package com.Util;

import java.io.IOException;
import java.io.Reader;
import org.apache.ibatis.io.Resources;
import org.apache.ibatis.session.SqlSession;
import org.apache.ibatis.session.SqlSessionFactory;
import org.apache.ibatis.session.SqlSessionFactoryBuilder;

public class MyBatisUtil {
	
	private MyBatisUtil() {}

	/**声明静态的
	* （1）ThreadLocal<org.apache.ibatis.session.SqlSession>
	* （2）org.apache.ibatis.session.SqlSessionFactory
	* 以保证SqlSession在线程（准确的说是一次请求）中的唯一性。
	*/
	private static final ThreadLocal<SqlSession> threadLocal = new ThreadLocal<SqlSession>();
    private static SqlSessionFactory sqlSessionFactory;
    
    
    /**
     *  （1）加载myBatis-config.xml文件
     *  （2）创建org.apache.ibatis.session.SqlSessionFactory对象
     */
	public static void buildSqlSessionFactory() {
		Reader reader=null;
		try {
			reader=Resources.getResourceAsReader("myBatis-config.xml");
			sqlSessionFactory = new SqlSessionFactoryBuilder().build(reader);
		} catch (Exception e) {
			System.err.println("加载myBatis-config.xml文件、创建org.apache.ibatis.session.SqlSessionFactory对象失败");
			e.printStackTrace();
		}finally{
			try {
				reader.close();//关闭字符流
			} catch (IOException e) {
				e.printStackTrace();//输入输出异常
			}
		}
	}
    
	/*应用程序启动时执行*/
	static {
    	try {
    		buildSqlSessionFactory();
		} catch (Exception e) {
			e.printStackTrace();
		}
    }
    
	
	/**从当前线程的threadLocal中获取SqlSession，
	 * 判断SqlSession是否为空，
	 * 如果SqlSession为空，判断SqlSessionFactory是否为空，是则重建SqlSessionFactory，否则创建SqlSession对象，并放入threadLocal
	 * 如果SqlSession不为空，返回SqlSession
	 **/
    public static SqlSession getSqlSession(){
		SqlSession sqlSession = null;
		try {
			sqlSession = (SqlSession) threadLocal.get();
			if (sqlSession == null) {
				if (sqlSessionFactory == null) {
					buildSqlSessionFactory();
				}
				sqlSession = (sqlSessionFactory != null) ? sqlSessionFactory
						.openSession() : null;
				threadLocal.set(sqlSession);
			}
		} catch (Exception e) {
			e.printStackTrace();
			System.out.println("-------------获取sqlSession失败--------------");
		}
		return sqlSession;
    }

	

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

	/**
     *获取SqlSessionFactory
     */
	public static SqlSessionFactory getSessionFactory() {
		return sqlSessionFactory;
	}
}