Hibernate基于ThreadLocal管理Session过程解析

当我们使用Hibernate进行对象关系映射时，我们常常需要处理Session对象的创建、使用和关闭等生命周期的管理。为了确保线程安全和线程隔离，通常采用ThreadLocal变量来管理Session对象。Hibernate基于ThreadLocal管理Session的过程如下：

1. 创建ThreadLocal对象

我们可以使用如下代码创建一个ThreadLocal变量：

private static final ThreadLocal<Session> threadLocalSession = new ThreadLocal<>();

它是一个static的ThreadLocal对象，定义了一个Session类型的泛型变量，用来保存当前线程所关联的Session对象。

1. 获取Session对象并关联到当前线程

在需要访问数据库时，通过SessionFactory获取Session对象，并将其保存到ThreadLocal变量中，代码如下：

public static Session getSession() {
    Session session = threadLocalSession.get();
    if (session == null || !session.isOpen()) {
        session = sessionFactory.openSession();
        threadLocalSession.set(session);
    }
    return session;
}

在每个线程内，都会保存一个Session对象，不同的线程之间互不干扰。

1. 使用Session对象进行数据库操作

我们可以使用获取到的Session对象来进行insert、update、delete、select等数据库操作：

public void saveEntity(T entity) {
    getSession().save(entity);
}

1. 关闭Session对象并清除ThreadLocal变量

在数据库操作结束后，需要关闭当前的Session对象，同时将ThreadLocal变量中的Session对象清除：

public static void closeSession() {
    Session session = threadLocalSession.get();
    if (session != null) {
        threadLocalSession.remove();
        session.close();
    }
}

这样可以防止Session对象泄露，同时保障线程安全和线程隔离。

示例一：

@Service
public class UserServiceImpl implements UserService {
    @Override
    public void saveUser(User user) {
        Session session = HibernateUtils.getSession();
        Transaction transaction = null;
        try {
            transaction = session.beginTransaction();
            session.save(user);
            transaction.commit();
        } catch (Exception e) {
            if (transaction != null) {
                transaction.rollback();
            }
            e.printStackTrace();
        } finally {
            HibernateUtils.closeSession();
        }
    }
}

示例二：

@Repository
public class UserDaoImpl implements UserDao {
    @Override
    public User getUserById(int id) {
        User user = null;
        Session session = HibernateUtils.getSession();
        Transaction transaction = null;
        try {
            transaction = session.beginTransaction();
            user = session.get(User.class, id);
            transaction.commit();
        } catch (Exception e) {
            if (transaction != null) {
                transaction.rollback();
            }
            e.printStackTrace();
        } finally {
            HibernateUtils.closeSession();
        }
        return user;
    }
}