package com.orizone.oa.extra.service;

import java.sql.Connection;
import java.sql.ResultSet;
import java.sql.SQLException;
import java.sql.Statement;

import net.orizone.oa.common.ConnectionPoolBean;

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

import com.orizone.comm.util.BusinessException;

public class TransactionUtil
{
 private final static ThreadLocal local = new ThreadLocal();
 private static Log log = LogFactory.getLog(TransactionUtil.class);
 
 /**
  * 開啟事務(wù)
  */
 public static void startTransaction()throws  BusinessException
 {
  Transaction tran = (Transaction)local.get();
  //判斷此事務(wù)是否屬于一個(gè)頂層事務(wù)。
  if(tran == null)
  {
   tran = new Transaction();
   //設(shè)置本地線程的connection
   Connection con = ConnectionPoolBean.getConnection();
   try
   {
    con.setAutoCommit(false);
   }catch(SQLException e)
   {
    e.printStackTrace();
    throw new BusinessException(e, "開啟事務(wù)失?。?);
   }
   tran.setConnection(con);
   tran.setCommitCount(0);
   tran.setTransCount(1);
   tran.setTransDeep(1);
   
   local.set(tran);
  }else
  {
   //事務(wù)已經(jīng)開啟，將嵌套層次深度加一，將事務(wù)次數(shù)加一
   tran.setTransCount(tran.getTransCount() + 1);
   tran.setTransDeep(tran.getTransDeep() + 1);
  }  
 }
 
 /**
  * 提交事務(wù)
  *
  */
 public static void commitTransaction()throws  BusinessException
 {
  Transaction tran = (Transaction)local.get();
  //如果事務(wù)屬于嵌套，則不提交數(shù)據(jù)，直接將層次數(shù)減一。
  if(tran.getTransDeep() > 1)
  {
   tran.setTransDeep(tran.getTransDeep() - 1);
   tran.setCommitCount(tran.getCommitCount() + 1);
   return;
  }
  
  Connection con = tran.getConnection();
  try
  {
   if(tran.hasFullExecute())
   {
    con.commit();   
   }
  }catch(SQLException e)
  {
   log.error(e);   
   throw new BusinessException(e, "提交事務(wù)失敗!");
  }
 }
 
 /**
  * 結(jié)束事務(wù)
  *
  */
 public static void endTransaction()throws  BusinessException
 {
  Transaction tran = (Transaction)local.get();
  //如果事務(wù)屬于嵌套，則不關(guān)閉連接，直接將層次數(shù)減一。
  if(tran.getTransDeep() > 1)
  {
   tran.setTransDeep(tran.getTransDeep() - 1);
   return;
  }
  
  //當(dāng)前事務(wù)已經(jīng)結(jié)束，清空ThreadLocal變量，防止下一次操作拿到已經(jīng)關(guān)閉的Connection對象。
  local.set(null);
  Connection con = tran.getConnection();
  try
  {
   if(!tran.hasFullExecute())
   {
    con.rollback();
   }
  }catch(SQLException e)
  {
   log.error(e);
   throw new BusinessException(e, "事務(wù)回滾失敗!");
  }finally
  {
   try
   {
    con.close();
   }catch(SQLException se)
   {
    log.error(se);
    throw new BusinessException(se, "關(guān)閉事務(wù)失敗!");
   }
  }
  
 }
 
 /**
  * 獲取當(dāng)前事務(wù)的數(shù)據(jù)庫連接。
  * @return
  */
 public static Connection getConnection()
 {
  Transaction tran = (Transaction)local.get();
  Connection  con = tran.getConnection();
  if(con == null)
  {
   con = ConnectionPoolBean.getConnection();   
  }
  return con;
 }
 
 /**
  * 測試代碼
  * @param args
  * @throws Exception
  */
 public static void main(String args[])throws Exception
 {
  test();
  test();
 }
 
 /**
  * 測試代碼
  *
  */
 private static void test()
 {
  TransactionUtil.startTransaction();
  try
  {
   Connection con = TransactionUtil.getConnection();
   Statement stmt = con.createStatement(ResultSet.TYPE_SCROLL_SENSITIVE, ResultSet.CONCUR_UPDATABLE);
   stmt.executeUpdate("INSERT INTO bb(bb) values(‘bb‘)");
   stmt.executeUpdate("INSERT INTO aa(aa) values(‘a(chǎn)a‘)");
   //if(true) throw new Exception("");
   TransactionUtil.commitTransaction();
  }catch(Exception e)
  {
   e.printStackTrace();
   //throw new BusinessException("");
  }finally
  {
   TransactionUtil.endTransaction();
  }
 }
}

package com.orizone.oa.extra.service;

import java.sql.Connection;

public class Transaction
{
 //數(shù)據(jù)庫連接對象
 private Connection connection;
 //事務(wù)次數(shù)
 private int transCount;
 //提交次數(shù)
 private int commitCount;
 //事務(wù)嵌套層次
 private int transDeep;

 int getCommitCount()
 {
  return commitCount;
 }

 void setCommitCount(int commitCount)
 {
  this.commitCount = commitCount;
 }

 Connection getConnection()
 {
  return connection;
 }

 void setConnection(Connection conn)
 {
  this.connection = conn;
 }

 public int getTransCount()
 {
  return transCount;
 }

 void setTransCount(int transCount)
 {
  this.transCount = transCount;
 }

 int getTransDeep()
 {
  return transDeep;
 }

 void setTransDeep(int transDeep)
 {
  this.transDeep = transDeep;
 }
 
 /**
  * 判斷事務(wù)是否完全提交。
  * 通過提交次數(shù)和事務(wù)次數(shù)來判斷事務(wù)是否完全提交。
  * @return
  */
 boolean hasFullExecute()
 {
  return commitCount + 1 == transCount;
 }

}