還在業務中用if else，策略模式瞭解一下

前段時間，我將公司系統中的批量審單的功能進行了重構，用到了java的併發編程進行異步化處理，數據庫的樂觀鎖機制處理多線程併發更新數據。其中批量審單的業務處理涉及到多種任務類型，對應不一樣的業務方法進行處理，好比轉倉，轉快遞，添加贈品，刪除贈品，拆分訂單，批量駁回，批量做廢等等，其中就用到了策略模式。

if else模式

if ("BATCH_CHANGE_WAREHOUSE".equals(taskType)) {
            //批量轉倉邏輯
        } else if ("BATCH_CHANGE_SHIPPING".equals(taskType)) {
            //批量轉快遞邏輯
        } else if ("BATCH_REPLACE_ORDER_GOODS".equals(taskType)) {
            //批量替換訂單商品邏輯
        } else if ("BATCH_DELETE_ORDER_GOODS".equals(taskType)) {
            //批量刪除訂單商品邏輯
        } else if ("BATCH_ADD_MEMO".equals(taskType)) {
            //批量添加備註邏輯
        } else {
            //任務類型未知
            System.out.println("任務類型沒法處理");
        }
複製代碼

看起來，思路清晰，if,else分支也很清楚，但不以爲代碼很臃腫，維護起來麻煩嗎，尤爲是其餘人來接鍋的時候，連看下去的慾望都沒有了。這時候你須要用策略模式消除其中的if else，進行一下簡單的重構！

策略模式

一、首先抽象業務處理器

public abstract class InspectionSolver {

    public abstract void solve(Long orderId, Long userId);

    public abstract String[] supports();
}
複製代碼

二、將業務處理器和其支持處理的類型放到一個容器中，java裏Map就是最經常使用的容器之一

@Component
public class InspectionSolverChooser implements ApplicationContextAware{


    private Map<String, InspectionSolver> chooseMap = new HashMap<>();

    public InspectionSolver choose(String type) {
        return chooseMap.get(type);
    }

    @PostConstruct
    public void register() {
        Map<String, InspectionSolver> solverMap = context.getBeansOfType(InspectionSolver.class);
        for (InspectionSolver solver : solverMap.values()) {
            for (String support : solver.supports()) {
                chooseMap.put(support,solver);
            }
        }
    }

    private ApplicationContext context;

    @Override
    public void setApplicationContext(ApplicationContext applicationContext) throws BeansException {
        this.context=applicationContext;
    }
}
複製代碼

這裏是在應用啓動的時候，加載spring容器中全部InspectionSolver類型的處理器，放到InspectionSolverChooser的map容器中。注意是InspectionSolver類型，因此定義的處理器都得繼承InspectionSolver，其次是spring容器中的才能加載，因此定義的處理器都得放到spring容器中（@Component註解不能少）

三、定義不一樣的處理器

@Component
public class ChangeWarehouseSolver extends InspectionSolver {

    @Override
    public void solve(Long orderId, Long userId) {
        System.out.println("訂單"+orderId+"開始進行批量轉倉了。。");
    }

    @Override
    public String[] supports() {
        return new String[] {InspectionConstant.INSPECTION_TASK_TYPE_BATCH_CHANGE_WAREHOUSE};
    }
}

@Component
public class ChangeShippingSolver extends InspectionSolver{

    @Override
    public void solve(Long orderId, Long userId) {
        System.out.println("訂單"+orderId+"開始進行轉快遞了。。");
    }

    @Override
    public String[] supports() {
        return new String[] {InspectionConstant.INSPECTION_TASK_TYPE_BATCH_CHANGE_SHIPPING};
    }
}

@Component
public class ReplaceOrderGoodsSolver extends InspectionSolver{

    @Override
    public void solve(Long orderId, Long userId) {
        System.out.println("訂單"+orderId+"開始進行替換商品了");
    }

    @Override
    public String[] supports() {
        return new String[]{InspectionConstant.INSPECTION_TASK_TYPE_BATCH_REPLACE_ORDER_GOODS};
    }
}
複製代碼

四、測試類

@RunWith(SpringJUnit4ClassRunner.class)
@SpringBootTest(classes=Application.class)// 指定spring-boot的啓動類
public class InspectionTest {

    @Autowired
    private InspectionSolverChooser chooser;

    @Test
    public void test() throws Exception{
        //準備數據
        String taskType = InspectionConstant.INSPECTION_TASK_TYPE_BATCH_CHANGE_WAREHOUSE;
        Long orderId = 12345L;
        Long userId = 123L;
        //獲取任務類型對應的solver
        InspectionSolver solver = chooser.choose(taskType);
        if (solver == null) {
            throw new RuntimeException("任務類型暫時沒法處理!");
        }
        //調用不一樣solver的方法進行處理
        solver.solve(orderId,userId);
    }
}
複製代碼

在測試類中我消除了可能一長段的if else，從選擇器InspectionSolverChooser中根據type的不一樣取出不一樣的任務處理器InspectionSolver，而後調用其solve()方法進行任務處理，不一樣處理器調用的固然就是不一樣的solve()方法了，目的達到。