策略模式打折

src/main/java/com/github/hcsp/polymorphism/Discount95Strategy.java

@@ -1,3 +1,8 @@
 package com.github.hcsp.polymorphism;
 
-public class Discount95Strategy {}
+public class Discount95Strategy extends PriceCalculator {
+    @Override
+    public int discount(int price,User user){
+        return (int) (price*0.95);
+    }
+}

src/main/java/com/github/hcsp/polymorphism/DiscountStrategy.java

@@ -1,6 +1,6 @@
 package com.github.hcsp.polymorphism;
 
-public class DiscountStrategy {
+public class DiscountStrategy extends PriceCalculator {
     public int discount(int price, User user) {
         throw new UnsupportedOperationException();
     }

src/main/java/com/github/hcsp/polymorphism/NoDiscountStrategy.java

@@ -1,3 +1,9 @@
 package com.github.hcsp.polymorphism;
 
-public class NoDiscountStrategy {}
+public class NoDiscountStrategy extends PriceCalculator {
+    @Override
+    public int discount(int price, User user) {
+        return price;
+    }
+
+}

src/main/java/com/github/hcsp/polymorphism/OnlyVipDiscountStrategy.java

@@ -1,3 +1,13 @@
 package com.github.hcsp.polymorphism;
 
-public class OnlyVipDiscountStrategy {}
+public class OnlyVipDiscountStrategy extends PriceCalculator {
+    @Override
+    public int discount(int price,User user){
+        if(user.isVip()){
+            return (int) (price*0.95);
+        }else{
+            return price;
+        }
+    }
+
+}

src/main/java/com/github/hcsp/polymorphism/PriceCalculator.java

@@ -1,28 +1,15 @@
 package com.github.hcsp.polymorphism;
 
-public class PriceCalculator {
+public abstract class PriceCalculator {
     // 使用策略模式重构这个方法，实现三个策略：
     // NoDiscountStrategy 不打折
     // Discount95Strategy 全场95折
     // OnlyVipDiscountStrategy 只有VIP打95折，其他人保持原价
     // 重构后的方法签名：
     // public static int calculatePrice(DiscountStrategy strategy, int price, User user)
-    public static int calculatePrice(String discountStrategy, int price, User user) {
-        switch (discountStrategy) {
-            case "NoDiscount":
-                return price;
-            case "Discount95":
-                return (int) (price * 0.95);
-            case "OnlyVip":
-                {
-                    if (user.isVip()) {
-                        return (int) (price * 0.95);
-                    } else {
-                        return price;
-                    }
-                }
-            default:
-                throw new IllegalStateException("Should not be here!");
-        }
+    public  int calculatePrice(DiscountStrategy strategy, int price, User user) {
+        return strategy.discount(price,user);
+
     }
+
 }