JDBC系列：（7）使用Connection操作事務

1、使用事務

package com.rk.db.g_transaction;

import java.sql.Connection;
import java.sql.PreparedStatement;
import java.sql.SQLException;

import com.rk.db.utils.JDBCUtil;

/**
 * // 轉賬，使用事務
 * @author RK
 *
 */
public class Demo01
{
	public static void main(String[] args)
	{
		Connection conn = null;
		try
		{
			conn = JDBCUtil.getConnection();
			// 1、設置事務為手動提交
			conn.setAutoCommit(false);

			boolean flag = true; //表示是否出現SQL異常
			transferMoney(conn, 100, "張三", "李四",flag);
		}
		catch (SQLException e)
		{
			System.out.println("轉賬失敗！");
			try
			{
				// 2、 出現異常，需要回滾事務
				conn.rollback();
				System.out.println("回滾操作成功！！！");
			}
			catch (SQLException ex)
			{
				ex.printStackTrace();
			}
		}
		finally
		{
			// 3、所有的操作執行成功, 提交事務
			try
			{
				conn.commit();
				System.out.println("執行完畢！");
			}
			catch (SQLException e)
			{
				e.printStackTrace();
			}
			JDBCUtil.closeQuietly(conn);
		}
	}

	/**
	 * 模擬銀行轉賬
	 * @param conn 數據庫連接
	 * @param moneyNum 轉賬的金額
	 * @param userAdd 收到Money的用戶
	 * @param userSub 支出Money的用戶
	 * @param flag 是否模擬SQL Exception異常，true表示出現，false表示不出現
	 * @throws SQLException
	 */
	private static void transferMoney(Connection conn, 
			long moneyNum, String userAdd, String userSub,
			boolean flag) throws SQLException
	{
		PreparedStatement pstmtAdd = null;
		PreparedStatement pstmtSub = null;

		try
		{
			String sqlAddMoney = "update T_Bank set money=money+? where username=?";
			pstmtAdd = conn.prepareStatement(sqlAddMoney);
			pstmtAdd.setLong(1, moneyNum);
			pstmtAdd.setString(2, userAdd);
			pstmtAdd.executeUpdate();
			
			if(flag)
			{
				throw new SQLException("模擬SQL執行出錯");
			}

			String sqlSubMoney = "update T_Bank set money=money-? where username=?";
			pstmtSub = conn.prepareStatement(sqlSubMoney);
			pstmtSub.setLong(1, moneyNum);
			pstmtSub.setString(2, userSub);
			pstmtSub.executeUpdate();
		}
		finally
		{
			JDBCUtil.closeQuietly(pstmtAdd);
			JDBCUtil.closeQuietly(pstmtSub);
		}

	}
}

2、使用事務，回滾到指定的代碼段

package com.rk.db.g_transaction;

import java.sql.Connection;
import java.sql.PreparedStatement;
import java.sql.SQLException;
import java.sql.Savepoint;

import com.rk.db.utils.JDBCUtil;

/**
 * // 轉賬，使用事務， 回滾到指定的代碼段
 * @author RK
 *
 */
public class Demo02
{
	public static void main(String[] args)
	{
		Connection conn = null;
		Savepoint sp = null;
		try
		{
			conn = JDBCUtil.getConnection();
			// 1、設置事務為手動提交
			conn.setAutoCommit(false);

			transferMoney(conn, 1000, "李四", "張三",false);
			
			// 如果失敗，回滾到這個位置
			sp = conn.setSavepoint();
			
			boolean flag = true; //表示是否出現SQL異常
			transferMoney(conn, 500, "張三", "李四",flag);
		}
		catch (SQLException e)
		{
			System.out.println("轉賬失敗！");
			try
			{
				// 2、 出現異常，需要回滾 (回滾到指定的代碼段)
				conn.rollback(sp);
				System.out.println("回滾到指定位置操作成功！！！");
			}
			catch (SQLException ex)
			{
				ex.printStackTrace();
			}
		}
		finally
		{
			// 3、所有的操作執行成功, 提交事務
			try
			{
				conn.commit();
				System.out.println("執行完畢！");
			}
			catch (SQLException e)
			{
				e.printStackTrace();
			}
			JDBCUtil.closeQuietly(conn);
		}
	}

	/**
	 * 模擬銀行轉賬
	 * @param conn 數據庫連接
	 * @param moneyNum 轉賬的金額
	 * @param userAdd 收到Money的用戶
	 * @param userSub 支出Money的用戶
	 * @param flag 是否模擬SQL Exception異常，true表示出現，false表示不出現
	 * @throws SQLException
	 */
	private static void transferMoney(Connection conn, 
			long moneyNum, String userAdd, String userSub,
			boolean flag) throws SQLException
	{
		PreparedStatement pstmtAdd = null;
		PreparedStatement pstmtSub = null;

		try
		{
			String sqlAddMoney = "update T_Bank set money=money+? where username=?";
			pstmtAdd = conn.prepareStatement(sqlAddMoney);
			pstmtAdd.setLong(1, moneyNum);
			pstmtAdd.setString(2, userAdd);
			pstmtAdd.executeUpdate();
			
			if(flag)
			{
				throw new SQLException("模擬SQL執行出錯");
			}

			String sqlSubMoney = "update T_Bank set money=money-? where username=?";
			pstmtSub = conn.prepareStatement(sqlSubMoney);
			pstmtSub.setLong(1, moneyNum);
			pstmtSub.setString(2, userSub);
			pstmtSub.executeUpdate();
		}
		finally
		{
			JDBCUtil.closeQuietly(pstmtAdd);
			JDBCUtil.closeQuietly(pstmtSub);
		}

	}
}