|
/*
* Created on 2003-11-16
*
*/
package com.javaeye.crm;
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;
import net.sf.hibernate.tool.hbm2ddl.SchemaExport;
/**
*
* 獲取Session連接工廠類
*
* @author Robbin Fan
*
*/
public class HibernateSession {
private static final ThreadLocal sessionThread = new ThreadLocal();
private static final ThreadLocal transactionThread = new ThreadLocal();
private static SessionFactory sf = null;
/**
* 獲取當前線程使用的Session
*
* @return Session
* @throws HibernateException
*/
public static Session currentSession() throws HibernateException {
Session s = (Session) sessionThread.get();
if (s == null) {
if (sf == null) sf = new Configuration().configure().buildSessionFactory();
s = sf.openSession();
sessionThread.set(s);
}
return s;
}
/**
* 獲取一個新的Session
*
* @return Session
* @throws HibernateException
*/
public static Session newSession() throws HibernateException {
if (sf == null) sf = new Configuration().configure().buildSessionFactory();
Session s = sf.openSession();
return s;
}
/**
* 啟動或者加入當前Session的Transaction
*
* @return Transaction
* @throws HibernateException
*/
public static Transaction currentTransaction() throws HibernateException {
Transaction tx = (Transaction) transactionThread.get();
if (tx == null) {
tx = currentSession().beginTransaction();
transactionThread.set(tx);
}
return tx;
}
/**
* 提交當前Session的Transaction
*
* @throws HibernateException
*/
public static void commitTransaction() throws HibernateException {
Transaction tx = (Transaction) transactionThread.get();
transactionThread.set(null);
if (tx != null) tx.commit();
}
/**
* 回滾當前事務
*
* @throws HibernateException
*/
public static void rollbackTransaction() throws HibernateException {
Transaction tx = (Transaction) transactionThread.get();
transactionThread.set(null);
if (tx != null) tx.rollback();
}
/**
* 關閉當前線程使用的Session
*
* @throws HibernateException
*/
public static void closeSession() throws HibernateException {
Session s = (Session) sessionThread.get();
sessionThread.set(null);
if (s != null) s.close();
}
/**
* 根據映射文件和持久對象生成數據庫DDL，生成文件為create_table.sql
*
* @param args 參數
*/
public static void main(String[] args) {
try {
String conf = "hibernate.cfg.xml";
if (args.length != 0) conf = args[0];
Configuration cfg = new Configuration().configure("/" + conf);
SchemaExport se = new SchemaExport(cfg);
//se.setOutputFile("create_table.sql");
se.create(true,true);
} catch (HibernateException e) {
System.out.println(e.getMessage());
}
}
}
|