summaryrefslogtreecommitdiff
path: root/spring-framework/07-bank-jdbc-ds-transaction/src/main/java/com/example/spring/bank/tx/TxAccountManager.java
blob: a5b310d91ae6eedafbb827c27ddac9e71c3e3fcb (plain)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
package com.example.spring.bank.tx;

import java.math.BigDecimal;

import org.springframework.jdbc.datasource.DataSourceTransactionManager;
import org.springframework.transaction.TransactionStatus;
import org.springframework.transaction.support.TransactionCallback;
import org.springframework.transaction.support.TransactionCallbackWithoutResult;
import org.springframework.transaction.support.TransactionTemplate;

import com.example.spring.bank.Account;
import com.example.spring.bank.AccountManager;

/**
 * This class decorates a given AccountManager by adding transaction boundaries.
 * 
 * @author kamal
 *
 */
public class TxAccountManager implements AccountManager {

	private AccountManager accountManager;

	public void setAccountManager(AccountManager accountManager) {
		this.accountManager = accountManager;
	}

	private TransactionTemplate txTemplate;

	public void setTxManager(DataSourceTransactionManager txManager) {
		txTemplate = new TransactionTemplate(txManager);
	}

	@Override
	public Account create() {
		return txTemplate.execute(new TransactionCallback<Account>() {
			@Override
			public Account doInTransaction(TransactionStatus arg0) {
				return accountManager.create();
			}
		});
	}
	
	@Override
	public Account find(int accountNumber) {
		return accountManager.find(accountNumber);
	}

	@Override
	public Account deposit(int accountNumber, BigDecimal amount) {
		return txTemplate.execute(new TransactionCallback<Account>() {

			@Override
			public Account doInTransaction(TransactionStatus arg0) {
				return accountManager.deposit(accountNumber, amount);
			}
		});
	}

	@Override
	public Account withdraw(int accountNumber, BigDecimal amount) {
		return txTemplate.execute(new TransactionCallback<Account>() {

			@Override
			public Account doInTransaction(TransactionStatus arg0) {
				return accountManager.withdraw(accountNumber, amount);
			}
		});
	}

	@Override
	public void delete(int accountNumber) {
		txTemplate.execute(new TransactionCallbackWithoutResult() {

			@Override
			protected void doInTransactionWithoutResult(TransactionStatus arg0) {
				accountManager.delete(accountNumber);
			}
		});
	}

	@Override
	public void chargeForLowBalance(BigDecimal minimumBalance, BigDecimal amount) {
		txTemplate.execute(new TransactionCallbackWithoutResult() {

			@Override
			protected void doInTransactionWithoutResult(TransactionStatus arg0) {
				accountManager.chargeForLowBalance(minimumBalance, amount);
			}
		});
	}

	@Override
	public Account transfer(int accountNumber1, int accountNumber2, BigDecimal amount) {
		return txTemplate.execute(new TransactionCallback<Account>() {

			@Override
			public Account doInTransaction(TransactionStatus status) {
				try {
					return accountManager.transfer(accountNumber1, accountNumber2, amount);
				} catch (Exception e) {
					// Some exception occurred.
					status.setRollbackOnly();
					throw e;
				}
			}
		});
	}
}