別再滿屏的 if/else 了，試試策略模式，真香?。?/h1>
你還在寫滿屏的 if/ else/ switch 之類的判斷邏輯嗎?
棧長在開發(fā)人員的代碼中看過太多這樣的低級代碼了，真的太 low，極不好維護，本文棧長就教你如何用策略模式干掉 if/ else/ switch，讓你的代碼更優(yōu)雅。
什么是策略模式?
比如說對象的某個行為，在不同場景中有不同的實現(xiàn)方式，這樣就可以將這些實現(xiàn)方式定義成一組策略，每個實現(xiàn)類對應一個策略，在不同的場景就使用不同的實現(xiàn)類，并且可以自由切換策略。
策略模式結構圖如下：
策略模式需要一個策略接口，不同的策略實現(xiàn)不同的實現(xiàn)類，在具體業(yè)務環(huán)境中僅持有該策略接口，根據(jù)不同的場景使用不同的實現(xiàn)類即可。
面向接口編程，而不是面向?qū)崿F(xiàn)。
策略模式的優(yōu)點：
1、干掉繁瑣的 if、switch 判斷邏輯;
2、代碼優(yōu)雅、可復用、可讀性好;
3、符合開閉原則，擴展性好、便于維護;
策略模式的缺點：
1、策略如果很多的話，會造成策略類膨脹;
2、使用者必須清楚所有的策略類及其用途;
策略模式實戰(zhàn)
舉個實際的例子，XX 公司是做支付的，根據(jù)不同的客戶類型會有不同的支付方式和支付產(chǎn)品，比如：信用卡、本地支付，而本地支付在中國又有微信支付、支付寶、云閃付、等更多其他第三方支付公司，這時候策略模式就派上用場了。
傳統(tǒng)的 if/ else/ switch 等判斷寫法大家都會寫，這里就不貼代碼了，直接看策略模式怎么搞!
1、定義策略接口
定義一個策略接口，所有支付方式的接口。
策略接口：
- /**
- * 支付接口
- * @author: 棧長
- * @from: 公眾號Java技術棧
- */
- public interface IPayment {
- /**
- * 支付
- * @param order
- * @return
- */
- PayResult pay(Order order);
- }
訂單信息類：
- /**
- * 訂單信息
- * @author: 棧長
- * @from: 公眾號Java技術棧
- */
- @Data
- public class Order {
- /**
- * 金額
- */
- private int amount;
- /**
- * 支付類型
- */
- private String paymentType;
- }
返回結果類：
- /**
- * @author: 棧長
- * @from: 公眾號Java技術棧
- */
- @Data
- @AllArgsConstructor
- public class PayResult {
- /**
- * 支付結果
- */
- private String result;
- }
2、定義各種策略
定義各種支付策略，微信支付、支付寶、云閃付等支付實現(xiàn)類都實現(xiàn)這個接口。
微信支付實現(xiàn)：
- /**
- * 微信支付
- * @author: 棧長
- * @from: 公眾號Java技術棧
- */
- @Service("WechatPay")
- public class WechatPay implements IPayment {
- @Override
- public PayResult pay(Order order) {
- return new PayResult("微信支付成功");
- }
- }
支付寶實現(xiàn)：
- /**
- * 支付寶
- * @author: 棧長
- * @from: 公眾號Java技術棧
- */
- @Service("Alipay")
- public class Alipay implements IPayment {
- @Override
- public PayResult pay(Order order) {
- return new PayResult("支付寶支付成功");
- }
- }
云閃付實現(xiàn)：
- /**
- * 銀聯(lián)云閃付
- * @author: 棧長
- * @from: 公眾號Java技術棧
- */
- @Service("UnionPay")
- public class UnionPay implements IPayment {
- @Override
- public PayResult pay(Order order) {
- return new PayResult("云閃付支付成功");
- }
- }
這里我把所有支付方式類都用 @Service 注解生成 Bean 放入 Spring Bean 容器中了，在使用策略的時候就不用 new 支付對象了，可以直接使用 Bean，這樣更貼近業(yè)務。Spring 基礎教程就不介紹了，大家可以關注公眾號Java技術棧，回復：spring，歷史教程我都整理好了。
3、使用策略
有的文章使用了枚舉、HashMap 的方式來根據(jù)策略名稱映射策略實現(xiàn)類 ，這樣是沒有問題，但在使用了 Spring 框架的項目還是有點多此一舉，完全可以發(fā)揮 Spring 框架的優(yōu)勢，使用 Bean 名稱就能找到對應的策略實現(xiàn)類了。
參考示例代碼如下：
- /**
- * 支付服務
- * @author: 棧長
- * @from: 公眾號Java技術棧
- */
- @RestController
- public class PayService {
- @Autowired
- private ApplicationContext applicationContext;
- /**
- * 支付接口
- * @param amount
- * @param paymentType
- * @return
- */
- @RequestMapping("/pay")
- public PayResult pay(@RequestParam("amount") int amount,
- @RequestParam("paymentType") String paymentType) {
- Order order = new Order();
- order.setAmount(amount);
- order.setPaymentType(paymentType);
- // 根據(jù)支付類型獲取對應的策略 bean
- IPayment payment = applicationContext.getBean(order.getPaymentType(), IPayment.class);
- // 開始支付
- PayResult payResult = payment.pay(order);
- return payResult;
- }
- }
看示例代碼，我并沒有像策略模式結構圖中那樣新建一個 Context 類持有策略接口，那是標準的策略模式，其實道理是一樣的，關鍵是怎么施放策略。
測試一下：
http://localhost:8080/pay?amount=8800&paymentType=WechatPay
測試 OK，傳入不同的支付方式會調(diào)用不同的策略。
本節(jié)教程所有實戰(zhàn)源碼已上傳到這個倉庫：https://github.com/javastacks/javastack
策略模式在 JDK 中的應用
現(xiàn)在我們知道如何使用策略模式了，現(xiàn)在我們再看看 JDK 哪些地方運用了策略模式呢。
1、線程池中的拒絕策略
線程池的構造中有一個拒絕策略參數(shù)，默認是默認拒絕策略：
其實這就是一個策略接口：
下面有幾種拒絕策略的實現(xiàn)：
image-20210329161322406
在創(chuàng)建線程池的時候，就可以傳入不同的拒絕策略，這就是 JDK 中策略模式的經(jīng)典實現(xiàn)了。
2、比較器
JDK 中大量使用了 Comparator 這個策略接口：
策略接口有了，但策略需要開發(fā)人員自己定。
集合排序我們比較熟悉的了，不同的排序規(guī)則其實就是不同的策略：
這個策略模式使用了函數(shù)式編程接口，比較規(guī)則使用匿名內(nèi)部類或者 Lambda 表達式就搞定了，不需要每個規(guī)則定義一個實現(xiàn)類，這樣就大量省略策略類了。
這個策略模式可能藏的比較深，但也是 JDK 中經(jīng)典的策略模式的應用了。
不限于這兩個，其實還有更多，你還知道別的么?歡迎留言分享……
所以說，策略模式就在你身邊，你一直都在用，但可能沒有發(fā)覺。。
總結
使用策略模式，我們可以輕松干掉大量的 if/ else，代碼也更優(yōu)雅，還能很靈活的擴展。
像本文中支付的案例，后面我們想添加、刪除多少個支付方式都不用修改現(xiàn)有的代碼，所以就不會影響現(xiàn)有的業(yè)務，真正做到對擴展開放，對修改關閉。
當然，完全干掉 if/ else 是不可能的，不能過度設計，不能為了使用設計模式而使用設計模式，否則適得其反。但是，我們每個程序員都需要掌握策略模式，做到在系統(tǒng)中靈活駕馭，這樣才能寫出更優(yōu)雅、高質(zhì)量的代碼。
本節(jié)教程所有實戰(zhàn)源碼已上傳到這個倉庫：
https://github.com/javastacks/javastack
本文轉(zhuǎn)載自微信公眾號「Java技術?！?，可以通過以下二維碼關注。轉(zhuǎn)載本文請聯(lián)系Java技術棧公眾號。
