019 例外と例外処理（トランザクションとロールバック） 031 解答例

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

// カスタム例外クラス
class InsufficientFundsException extends Exception {
    public InsufficientFundsException(String message) {
        super(message);
    }
}

public class BankTransactionExample {
    private static final String DB_URL = "jdbc:mysql://localhost:3306/bank";
    private static final String DB_USER = "your_username";
    private static final String DB_PASSWORD = "your_password";

    // 銀行取引メソッド
    public static void transferFunds(int fromAccount, int toAccount, double amount) throws SQLException, InsufficientFundsException {
        // データベースへの接続
        try (Connection connection = DriverManager.getConnection(DB_URL, DB_USER, DB_PASSWORD)) {
            // トランザクションの開始
            connection.setAutoCommit(false);

            // 送金元の残高取得
            double balance = getAccountBalance(connection, fromAccount);

            // 残高が十分か確認
            if (balance < amount) {
                // 残高不足の場合、トランザクションをロールバックし例外をスロー
                connection.rollback();
                throw new InsufficientFundsException("送金元アカウントの残高が不足しています。");
            }

            // 送金元アカウントから引き落とし
            updateAccountBalance(connection, fromAccount, balance - amount);

            // 送金先アカウントに入金
            updateAccountBalance(connection, toAccount, getAccountBalance(connection, toAccount) + amount);

            // トランザクションをコミット
            connection.commit();
        }
    }

    // 指定されたアカウントの残高を取得するメソッド
    private static double getAccountBalance(Connection connection, int accountNumber) throws SQLException {
        String query = "SELECT balance FROM accounts WHERE account_number = ?";
        try (PreparedStatement preparedStatement = connection.prepareStatement(query)) {
            preparedStatement.setInt(1, accountNumber);
            var resultSet = preparedStatement.executeQuery();
            if (resultSet.next()) {
                return resultSet.getDouble("balance");
            }
            throw new SQLException("アカウントが見つかりません。");
        }
    }

    // 指定されたアカウントの残高を更新するメソッド
    private static void updateAccountBalance(Connection connection, int accountNumber, double newBalance) throws SQLException {
        String query = "UPDATE accounts SET balance = ? WHERE account_number = ?";
        try (PreparedStatement preparedStatement = connection.prepareStatement(query)) {
            preparedStatement.setDouble(1, newBalance);
            preparedStatement.setInt(2, accountNumber);
            int affectedRows = preparedStatement.executeUpdate();
            if (affectedRows == 0) {
                throw new SQLException("アカウントが見つかりません。");
            }
        }
    }

    public static void main(String[] args) {
        try {
            // ユーザーがアカウント間で資金を移動する例
            transferFunds(123456, 789012, 1000.0);
            System.out.println("取引が成功しました。");
        } catch (SQLException | InsufficientFundsException e) {
            System.err.println("取引エラー: " + e.getMessage());
        }
    }
}

try (Connection connection = DriverManager.getConnection(DB_URL, DB_USER, DB_PASSWORD)) {
    // トランザクションの開始
    connection.setAutoCommit(false);

    // 送金元の残高取得
    double balance = getAccountBalance(connection, fromAccount);

    // 残高が十分か確認
    if (balance < amount) {
        // 残高不足の場合、トランザクションをロールバックし例外をスロー
        connection.rollback();
        throw new InsufficientFundsException("送金元アカウントの残高が不足しています。");
    }

    // ...（以下略）
    
    // トランザクションをコミット
    connection.commit();
}