diff --git a/Behavioural Patterns/Chain of Responsibility/src/IATMDispenseChain.java b/Behavioural Patterns/Chain of Responsibility/src/IATMDispenseChain.java index 4c7eeb4..5692874 100644 --- a/Behavioural Patterns/Chain of Responsibility/src/IATMDispenseChain.java +++ b/Behavioural Patterns/Chain of Responsibility/src/IATMDispenseChain.java @@ -6,4 +6,12 @@ public void setNextChain(IATMDispenseChain nextChain) { } public abstract void dispense(int amount); + + public void nextChain(int amount) { + if (this.nextChain != null) { + this.nextChain.dispense(amount); + } else { + System.out.println("No notes can be dispensed"); + } + } } diff --git a/Behavioural Patterns/Chain of Responsibility/src/Rupee10DispenseChain.java b/Behavioural Patterns/Chain of Responsibility/src/Rupee10DispenseChain.java index 1a32df6..5b3c97c 100644 --- a/Behavioural Patterns/Chain of Responsibility/src/Rupee10DispenseChain.java +++ b/Behavioural Patterns/Chain of Responsibility/src/Rupee10DispenseChain.java @@ -6,7 +6,7 @@ public void dispense(int amount) { int remainingAmount = amount % 10; System.out.println("Number of 10 Rs notes are " + String.valueOf(notes)); } else if (amount > 0) { - this.nextChain.dispense(amount); + nextChain(amount); } } } diff --git a/Behavioural Patterns/Chain of Responsibility/src/Rupee20DispenseChain.java b/Behavioural Patterns/Chain of Responsibility/src/Rupee20DispenseChain.java index e1bb298..a6f00c7 100644 --- a/Behavioural Patterns/Chain of Responsibility/src/Rupee20DispenseChain.java +++ b/Behavioural Patterns/Chain of Responsibility/src/Rupee20DispenseChain.java @@ -7,7 +7,7 @@ public void dispense(int amount) { System.out.println("Number of 20 Rs notes are " + String.valueOf(notes)); this.nextChain.dispense(remainingAmount); } else if (amount > 0) { - this.nextChain.dispense(amount); + nextChain(amount); } } } diff --git a/Behavioural Patterns/Chain of Responsibility/src/Rupee50DispenseChain.java b/Behavioural Patterns/Chain of Responsibility/src/Rupee50DispenseChain.java index 8c656b7..1cb0b6f 100644 --- a/Behavioural Patterns/Chain of Responsibility/src/Rupee50DispenseChain.java +++ b/Behavioural Patterns/Chain of Responsibility/src/Rupee50DispenseChain.java @@ -7,7 +7,7 @@ public void dispense(int amount) { System.out.println("Number of 50 Rs notes are " + String.valueOf(notes)); this.nextChain.dispense(remainingAmount); } else if (amount > 0) { - this.nextChain.dispense(amount); + nextChain(amount); } } }