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

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("139****1525", "", "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("139****1525", ""); // 结合工厂 ILogin login1 = LoginServiceFactory.getLoginService("mobile");
        login1.login("139****1525", "");
    }
}