（工厂+策略）实现登录功能

原始代码

业务层UserService

@Service public class UserService {      public LoginResp login(LoginReq loginReq){          if(loginReq.getType().equals("account")){             System.out.println("用户名密码登录");              //执行用户密码登录逻辑              return new LoginResp();          }else if(loginReq.getType().equals("sms")){             System.out.println("手机号验证码登录");              //执行手机号验证码登录逻辑              return new LoginResp();         }else if (loginReq.getType().equals("we_chat")){             System.out.println("微信登录");              //执行用户微信登录逻辑              return new LoginResp();         }         LoginResp loginResp = new LoginResp();         loginResp.setSuccess(false);         System.out.println("登录失败");         return loginResp;     } } 

注意：我们重点讲的是设计模式，并不是登录的逻辑，所以以上代码并没有真正的实现登录功能

问题分析

• 业务层代码大量使用到了if...else，在后期阅读代码的时候会非常不友好，大量使用if...else性能也不高
• 如果业务发生变更，比如现在新增了QQ登录方式，这个时候需要修改业务层代码，违反了开闭原则

解决：

使用工厂方法设计模式+策略模式解决

改进代码

要将上述代码改为使用工厂模式和策略模式，首先我们需要定义几个关键元素：策略接口、具体策略类、策略工厂以及修改UserService来使用这些策略。下面是改造后的代码示例：

1. 定义策略接口

public interface LoginStrategy {     LoginResp login(LoginReq loginReq); } 

2. 创建具体策略类

AccountLoginStrategy

@Service public class AccountLoginStrategy implements LoginStrategy {     @Override     public LoginResp login(LoginReq loginReq) {         System.out.println("用户名密码登录");         // 执行用户密码登录逻辑         return new LoginResp();     } } 

SmsLoginStrategy

@Service public class SmsLoginStrategy implements LoginStrategy {     @Override     public LoginResp login(LoginReq loginReq) {         System.out.println("手机号验证码登录");         // 执行手机号验证码登录逻辑         return new LoginResp();     } } 

WeChatLoginStrategy

@Service public class WeChatLoginStrategy implements LoginStrategy {     @Override     public LoginResp login(LoginReq loginReq) {         System.out.println("微信登录");         // 执行用户微信登录逻辑         return new LoginResp();     } } 

3. 实现策略工厂

@Service public class LoginStrategyFactory {          @Autowired     private AccountLoginStrategy accountLoginStrategy;          @Autowired     private SmsLoginStrategy smsLoginStrategy;          @Autowired     private WeChatLoginStrategy weChatLoginStrategy;          public LoginStrategy getLoginStrategy(String type) {         switch (type) {             case "account":                 return accountLoginStrategy;             case "sms":                 return smsLoginStrategy;             case "we_chat":                 return weChatLoginStrategy;             default:                 throw new IllegalArgumentException("Invalid login type");         }     } } 

4. 修改UserService使用策略

@Service public class UserService {      @Autowired     private LoginStrategyFactory loginStrategyFactory;      public LoginResp login(LoginReq loginReq){         LoginStrategy strategy = loginStrategyFactory.getLoginStrategy(loginReq.getType());         if(strategy == null){             LoginResp loginResp = new LoginResp();             loginResp.setSuccess(false);             System.out.println("登录失败: 无效的登录类型");             return loginResp;         }         return strategy.login(loginReq);     } } 

通过这种方式，UserService不再直接包含各种登录逻辑，而是通过策略工厂获取对应的策略对象来处理登录请求，这使得代码更加灵活和可扩展。如果需要添加新的登录方式，只需增加一个新的策略类并修改工厂类即可，无需修改UserService的逻辑。