BankAccountWithAtomicReference

BankAccountWithAtomicReference.java

package org.example.bankaccount;

import java.util.concurrent.atomic.AtomicReference;

public class BankAccountWithAtomicReference {
    
    private AtomicReference<BankAccount> atomicBalance = new AtomicReference<>(new BankAccount());

    public void deposit(double amount) {
        BankAccount oldBalance;
        BankAccount newBalance;

        do {
            oldBalance = atomicBalance.get();
            var bank = new BankAccount();
            bank.deposit(oldBalance.getBalance() + amount);
            newBalance = bank;

        } while (!this.atomicBalance.compareAndSet(oldBalance, newBalance));

    }

    public void withdraw(double amount) {
        BankAccount oldBalance;
        BankAccount newBalance;
        do {

            oldBalance = atomicBalance.get();
            var bank = new BankAccount();
            bank.deposit(oldBalance.getBalance() - amount);
            newBalance = bank;

        } while (!this.atomicBalance.compareAndSet(oldBalance, newBalance));
    }

    public double getBalance() {
        return this.atomicBalance.get().getBalance();
    }

}