defaultsessionowner.java

来自「Enjoy Web Dev With Tapestry 一书的源代码」· Java 代码 · 共 62 行

package com.ttdev.customerCRUD;

import org.apache.hivemind.*;
import org.hibernate.*;

public class DefaultSessionOwner implements SessionOwner, Discardable {
	private SessionCreator creator;

	private Session session;

	private Transaction tx;

	private boolean isToRollback;

	public Session getSession() {
		if (session == null) {
			session = creator.createSession();
			if (tx == null) {
				tx = session.beginTransaction();
				isToRollback = false;
			}
		}
		return session;
	}

	public void threadDidDiscardService() {
		if (session != null) {
			try {
				endTransaction();
			} finally {
				session.close();
				session = null;
			}
		}
	}

	public void setSessionCreator(SessionCreator creator) {
		this.creator = creator;
	}

	public void setToRollback() {
		isToRollback = true;
	}

	public void endTransaction() {
		if (tx != null) {
			try {
				if (isToRollback) {
					tx.rollback();
				} else {
					tx.commit();
				}
			} catch (RuntimeException e) {
				tx.rollback();
				throw e;
			} finally {
				tx = null;
			}
		}
	}
}