package javamxj.hibernate.util;

import org.apache.commons.logging.Log;
import org.apache.commons.logging.LogFactory;

import net.sf.hibernate.HibernateException;
import net.sf.hibernate.Session;
import net.sf.hibernate.SessionFactory;
import net.sf.hibernate.Transaction;
import net.sf.hibernate.cfg.Configuration;

public class HibernateUtil {

	private static Log log = LogFactory.getLog(HibernateUtil.class);

	private static SessionFactory sessionFactory;

	private static final ThreadLocal threadSession = new ThreadLocal();

	private static final ThreadLocal threadTransaction = new ThreadLocal();

	public static SessionFactory getSessionFactory() {
		if (sessionFactory == null) {
			try {
				// Create the SessionFactory
				sessionFactory = new Configuration().configure()
						.buildSessionFactory();
			} catch (HibernateException ex) {
				ex.printStackTrace();
				throw new RuntimeException("Configuration problem: "
						+ ex.getMessage(), ex);
			}
		}
		return sessionFactory;
	}

	public static Session currentSession() throws HibernateException {
		Session s = (Session) threadSession.get();
		// Open a new Session, if this Thread has none yet
		if (s == null) {
			s = getSessionFactory().openSession();
			log.debug("###Opening new Session for this thread:" + s);
			threadSession.set(s);
		} else {
			log.debug("###Session was existed:" + s);
		}
		return s;
	}

	public static void closeSession() throws HibernateException {
		Session s = (Session) threadSession.get();
		threadSession.set(null);
		if (s != null) {
			log.debug("###Closing Session of this thread. " + s);
			s.close();
		}
	}

	public static void beginTransaction() throws HibernateException {
		Transaction tx = (Transaction) threadTransaction.get();
		try {
			if (tx == null) {
				tx = currentSession().beginTransaction();
				log.debug("###Starting new database transaction in this thread:"+ tx);
				threadTransaction.set(tx);
			} else {
				log.debug("###Tx was existed:" + tx);
			}
		} catch (HibernateException ex) {
			throw ex;
		}
	}

	public static void commitTransaction() throws HibernateException {
		Transaction tx = (Transaction) threadTransaction.get();
		try {
			if (tx != null && !tx.wasCommitted() && !tx.wasRolledBack()) {
				log.debug("###Committing database transaction of this thread.");
				tx.commit();
			}
			threadTransaction.set(null);
		} catch (HibernateException ex) {
			rollbackTransaction();
			throw ex;
		}
	}

	public static void rollbackTransaction() throws HibernateException {
		Transaction tx = (Transaction) threadTransaction.get();
		try {
			threadTransaction.set(null);
			if (tx != null && !tx.wasCommitted() && !tx.wasRolledBack()) {
				log.debug("###Tyring to rollback database transaction of this thread.");
				tx.rollback();
			}
		} catch (HibernateException ex) {
			throw ex;
		} finally {
			closeSession();
		}
	}

}

package javamxj.hibernate.component;

import java.util.Iterator;
import java.util.List;

import javamxj.hibernate.util.HibernateUtil;


import net.sf.hibernate.HibernateException;
import net.sf.hibernate.Session;

public class ComponentDemo {
	public static void main(String[] args) {
		try {
			new ComponentDemo();
		} catch (HibernateException he) {
			he.printStackTrace();
		}
	}

	public ComponentDemo() throws HibernateException {

		Session sess = HibernateUtil.currentSession();

		Person p = new Person();
		p.setAddress(new Address("中国", "上海", "普陀", "200055"));
		p.setUsername("JavaMXJ");
		sess.save(p);

		p = new Person();
		p.setAddress(new Address("中国", "北京", "海淀", "100086"));
		p.setUsername("张三");
		sess.save(p);

		List animals = sess.find("from " + Person.class.getName());
		for (Iterator it = animals.iterator(); it.hasNext();) {
			Person person = (Person) it.next();
			System.out.println(person.getUsername() + ":" + person.getAddress());
		}
		
		HibernateUtil.closeSession();

	}
}