ToB企服应用市场:ToB评测及商务社交产业平台

标题: （工厂+策略）实现登录功能 [打印本页]

作者: 王海鱼 时间: 2024-6-1 18:01
标题: （工厂+策略）实现登录功能
原始代码

业务层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的逻辑。

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

欢迎光临 ToB企服应用市场:ToB评测及商务社交产业平台 (https://dis.qidao123.com/)