策略模式：【利于扩展，消除if的模式】

携手创作，共同成长！这是我参与「掘金日新计划 · 8 月更文挑战」的第2天，点击查看活动详情

1.前言

对于大多数多种类型的逻辑代码，我们一般的做法一般是通过if来进行判断，然后再{}里进行逻辑代码的编写，如下述

package org.example.designpattern.strategy;

public class LoginNoDesign {

    /**
     * 登录方法, 不使用策略模式时, 对于流程的扩展如新增一种登录方式只能添加if
     *
     * @param username 用户名
     * @param password 密码
     * @param type     登录方式
     */
    public void login(String username, String password, String type) {
        if ("mobile".equals(type)) {
            // 手机号登录
            System.out.println("手机号登录成功");
        } else {
            // 用户名密码登录
            System.out.println("用户名密码登录成功");
        }
    }
}

复制代码

但这种代码给整个方法增添了一种臃肿，往往一个方法需要大几十甚至上百行代码。

二.策略模式介绍

策略模式是行为模式的一种，也是消除if的重要手段，通过对于同类逻辑的不同策略来实现替代，如我们有多种登陆方式，那么可以先定义某种type对应某种策略，在使用时通过type来获取对应执行策略。

三.代码实现

package org.example.designpattern.strategy;

public interface ILogin {

    /**
     * 登录方法
     *
     * @param username 用户名
     * @param password 密码
     * @param type     登录方式
     */
    void login(String username, String password, String type);
}

复制代码

package org.example.designpattern.strategy;

public class MobileLogin implements ILogin{

    @Override
    public void login(String username, String password, String type) {
        System.out.println("手机号登录成功");
    }
}

复制代码

package org.example.designpattern.strategy;

public class UsernameLogin implements ILogin{

    @Override
    public void login(String username, String password, String type) {
        System.out.println("用户名密码登录成功");
    }
}

复制代码

测试

package org.example.designpattern.strategy;

import org.junit.Test;

public class StrategyTests {

    @Test
    public void test() {
        LoginNoDesign loginNoDesign = new LoginNoDesign();
        loginNoDesign.login("root", "root", "username");

        // 策略模式
        ILogin login = new MobileLogin();
        login.login("13903291525", "", "mobile");
    }
}

复制代码

对比

通过类型将各自实现分开，这样将原来的耦合的代码分开，在某种类型逻辑需要修改维护时也不需要测试其他已经跑通的逻辑。

扩展

结合工厂模式，预定义对应类型所执行的策略

package org.example.designpattern.strategy;

import java.util.Map;
import java.util.concurrent.ConcurrentHashMap;

public class LoginServiceFactory {

    private static Map<String, ILogin> loginMap = new ConcurrentHashMap<>();

    static {
        loginMap.put("mobile", new MobileLogin());
        loginMap.put("username", new UsernameLogin());
    }

    public static ILogin getLoginService(String type) {
        return loginMap.get(type);
    }
}

复制代码

测试

package org.example.designpattern.strategy;

import org.junit.Test;

public class StrategyTests {

    @Test
    public void test() {
        LoginNoDesign loginNoDesign = new LoginNoDesign();
        loginNoDesign.login("root", "root", "username");

        // 策略模式
        ILogin login = new MobileLogin();
        login.login("13903291525", "");

        // 结合工厂
        ILogin login1 = LoginServiceFactory.getLoginService("mobile");
        login1.login("13903291525", "");
    }
}
复制代码

四.总结

策略模式并不复杂，通过策略模式可以优化掉代码中的大量if，可以极大的提高系统的扩展性