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

原始代码

业务层UserService

1. @Service
2. public class UserService {
4.     public LoginResp login(LoginReq loginReq){
6.         if(loginReq.getType().equals("account")){
7.             System.out.println("用户名密码登录");
9.             //执行用户密码登录逻辑
11.             return new LoginResp();
13.         }else if(loginReq.getType().equals("sms")){
14.             System.out.println("手机号验证码登录");
16.             //执行手机号验证码登录逻辑
18.             return new LoginResp();
19.         }else if (loginReq.getType().equals("we_chat")){
20.             System.out.println("微信登录");
22.             //执行用户微信登录逻辑
24.             return new LoginResp();
25.         }
26.         LoginResp loginResp = new LoginResp();
27.         loginResp.setSuccess(false);
28.         System.out.println("登录失败");
29.         return loginResp;
30.     }
31. }

复制代码

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

问题分析

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

解决：

使用工厂方法计划模式+策略模式解决
改进代码

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

1. public interface LoginStrategy {
2.     LoginResp login(LoginReq loginReq);
3. }

复制代码

2. 创建详细策略类

AccountLoginStrategy

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

复制代码

SmsLoginStrategy

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

复制代码

WeChatLoginStrategy

1. @Service
2. public class WeChatLoginStrategy implements LoginStrategy {
3.     @Override
4.     public LoginResp login(LoginReq loginReq) {
5.         System.out.println("微信登录");
6.         // 执行用户微信登录逻辑
7.         return new LoginResp();
8.     }
9. }

复制代码

3. 实现策略工厂

1. @Service
2. public class LoginStrategyFactory {
3.    
4.     @Autowired
5.     private AccountLoginStrategy accountLoginStrategy;
6.    
7.     @Autowired
8.     private SmsLoginStrategy smsLoginStrategy;
9.    
10.     @Autowired
11.     private WeChatLoginStrategy weChatLoginStrategy;
12.    
13.     public LoginStrategy getLoginStrategy(String type) {
14.         switch (type) {
15.             case "account":
16.                 return accountLoginStrategy;
17.             case "sms":
18.                 return smsLoginStrategy;
19.             case "we_chat":
20.                 return weChatLoginStrategy;
21.             default:
22.                 throw new IllegalArgumentException("Invalid login type");
23.         }
24.     }
25. }

复制代码

4. 修改UserService使用策略

1. @Service
2. public class UserService {
4.     @Autowired
5.     private LoginStrategyFactory loginStrategyFactory;
7.     public LoginResp login(LoginReq loginReq){
8.         LoginStrategy strategy = loginStrategyFactory.getLoginStrategy(loginReq.getType());
9.         if(strategy == null){
10.             LoginResp loginResp = new LoginResp();
11.             loginResp.setSuccess(false);
12.             System.out.println("登录失败: 无效的登录类型");
13.             return loginResp;
14.         }
15.         return strategy.login(loginReq);
16.     }
17. }

复制代码

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

免责声明：如果侵犯了您的权益，请联系站长，我们会及时删除侵权内容，谢谢合作！更多信息从访问主页：qidao123.com:ToB企服之家，中国第一个企服评测及商务社交产业平台。