Java设计模式在实际项目中的运用-策略模式

项目背景

以前写项目时，登录功能都是按照需求写不同业务逻辑去实现登录，需要大量的if…else判断登录场景。但这种写法拓展性不高，万一哪天要新增支持的登录方式(例如weibo、qq)，还需要去改源代码。这时设计模式中的策略模式就可以排上用场了。我在实习的项目中刚好需要实现登录功能，就想到用策略模式实现登录功能，顺便加强自己的印象。先给大家理解一下策略模式的概念。

策略模式

意图：定义一系列的算法,把它们一个个封装起来, 并且使它们可相互替换。
主要解决：在有多种算法相似的情况下，使用 if…else 所带来的复杂和难以维护。
我的个人理解就是，把功能抽象出来，根据不同的场景使用不用的解决方法(实现类)。

实战

首先定义一个接口(登录策略)，所有的登录实现(账号密码、微信登录)

    /**
    * @Description: 会员登录(使用策略模式方便后期拓展手机登录或者第三方登录)
    * @Author: 黎勇炫
    * @Date: 2022/2/28 14:43
    */
    Member getMember(MemberVO memberVO);

    /**
    * @Description: 返回当前策略支持的 登录 类型
    * @Author: 黎勇炫
    * @Date: 2022/2/28 15:17
    */
    String getSupportedType();
}

编写具体的实现策略(例如手机号登录)

@Service
public class MobileStrategy implements LoginStrategy{
    private static final Logger log = LoggerFactory.getLogger(LoginStrategy.class);

    @Override
    public Member getMember(MemberVO memberVO) {
        // 这里写具体的登录实现，返回会员信息
    }

    @Override
    public String getSupportedType() {
        // 常量-》登陆类型(mobile)
        return LoginConstant.LOGIN_MOBILE;
    }
}

编写一个工厂类，把所有的实现类加载进工厂(利用构造器注入+动态参数，参数为接口，spring会自动帮我们把所有的实现类加载进策略工厂)

@Component
public class LoginStrategyFactory {

     /**
       * 存放不同的策略实现
       */
    private List<LoginStrategy> strategies = new ArrayList<>();

     /**
       * 通过构造器动态传参，自动把所有的策略实现加载进工厂
       */
    @Autowired
    public LoginStrategyFactory(LoginStrategy...args ) {
        Arrays.stream(args).forEach(strategys->{
            strategies.add(strategys);
        });
    }

     /**
       * 根据登录类型返回登录策略
       */
    public  LoginStrategy getStrategy(String type) {
        for (LoginStrategy strategy : strategies) {
            if(strategy.getSupportedType().equals(type)){
                return strategy;
            }
        }
        return null;
    }
}

最后，我们在service层根据前台传来的登录类型到工厂里面拿到实现类，根据登录类型实现登录功能。

    @Autowired
    private LoginStrategyFactory loginStrategyFactory;

    public String doLogin(MemberVO member) {

        if(member == null){
            throw new BizException(BizCodeEnum.USERNAME_OR_PASSWORD_NULL);
        }

        // 获取登录策略
        LoginStrategy strategy = loginStrategyFactory.getStrategy(member.getLoginType());
        // 获取会员信息
        Member loginMember = strategy.getMember(member);
        //登录成功后的业务逻辑
    }