diff --git a/src/main/java/millions/calculators/PurchaseCalculatorFactory.java b/src/main/java/millions/calculators/PurchaseCalculatorFactory.java deleted file mode 100644 index 7f496cf..0000000 --- a/src/main/java/millions/calculators/PurchaseCalculatorFactory.java +++ /dev/null @@ -1,18 +0,0 @@ -package millions.calculators; - -import millions.Share; - -public class PurchaseCalculatorFactory extends TransactionCalculatorFactory { - Share share; - - @Override - public void setShare(Share share) { - this.share = share; - } - - @Override - public TransactionCalculator createCalculator() { - return new PurchaseCalculator(share); - } - -} diff --git a/src/main/java/millions/calculators/SaleCalculatorFactory.java b/src/main/java/millions/calculators/SaleCalculatorFactory.java deleted file mode 100644 index fd16818..0000000 --- a/src/main/java/millions/calculators/SaleCalculatorFactory.java +++ /dev/null @@ -1,16 +0,0 @@ -package millions.calculators; - -import millions.Share; - -public class SaleCalculatorFactory extends TransactionCalculatorFactory { - Share share; - - @Override - public void setShare(Share share) { - this.share = share; - } - @Override - public TransactionCalculator createCalculator() { - return new SaleCalculator(share); - } -} diff --git a/src/main/java/millions/calculators/TransactionCalculatorFactory.java b/src/main/java/millions/calculators/TransactionCalculatorFactory.java index 106ca73..dd79b39 100644 --- a/src/main/java/millions/calculators/TransactionCalculatorFactory.java +++ b/src/main/java/millions/calculators/TransactionCalculatorFactory.java @@ -2,7 +2,15 @@ import millions.Share; -public abstract class TransactionCalculatorFactory { - public abstract void setShare(Share share); - public abstract TransactionCalculator createCalculator(); +public class TransactionCalculatorFactory { + + private TransactionCalculatorFactory() {} + + public TransactionCalculator createPurchaseCalculator(Share share) { + return new PurchaseCalculator(share); + } + + public TransactionCalculator createSaleCalculator(Share share) { + return new SaleCalculator(share); + } }