优雅实现策略器模式
- 场景
- 废话少说,源码地址。
- 讲解
- 关于本文
- 创建登陆策略
- 创建环境(注册器)
- 简化代码 创建父类对象,也方便后期扩展
- 具体策略实现类
- 用户名密码登陆
- 手机号登陆
- 触发点
- 调试
- 简言
场景
登陆场景使用(登陆之后返回用户信息和token所做操作基本一致,杜绝多个判断)
- 用户名密码登陆,
- 指纹登录
- 手机号登陆
楼主之前写过一次通过注解实现的,那个看起来没有注册器实现优雅。
策略器可以使用的场景很多,主要为了杜绝多if判断,把if判断需实现的信息放在自己的实现类中,提升代码可阅读性,提高执行效率,降低出错概率。
八股文:可扩展性高,灵活性高,代码复用性高,单一职责原则,可替换性高。
废话少说,源码地址。
代码地址
git clone https://github.com/gwy572294624/strategy-demo-new
讲解
chatGpt关于策略模式的八股文
策略模式(Strategy Pattern)是一种行为型设计模式,用于将不同的算法封装成独立的策略类,并使这些策略类可以相互替换,以实现在运行时动态地选择不同的算法。
在策略模式中,有三个主要的角色:
-
环境(Context):环境类是策略模式的核心,其内部持有一个策略对象的引用。环境类将具体的任务委派给策略对象进行处理,而不直接实现具体的算法逻辑。
-
策略(Strategy):策略类是一个接口或抽象类,它定义了具体算法的方法。不同的策略类实现了不同的算法逻辑。策略类之间可以相互替换,提供了算法的灵活配置。
-
具体策略(Concrete Strategies):具体策略类是策略模式的实现类,实现了策略接口或抽象类中定义的具体算法逻辑。
策略模式的工作流程如下:
-
客户端创建一个环境对象,并通过构造函数或设置方法将具体的策略对象传递给环境对象。
-
客户端根据需求选择合适的策略对象。
-
环境对象在执行任务时,会将具体的算法委派给当前持有的策略对象进行处理。
-
当需要切换算法时,客户端可以动态地替换环境对象的策略对象,实现不同的算法逻辑。
策略模式的优点包括灵活性、可替换性、可扩展性和代码复用性。它可以避免使用大量的条件语句或开关语句,将不同的算法逻辑封装在独立的策略类中,使系统更加灵活、易于扩展和维护。
关于本文
策略模式组成是通过环境,策略,具体实现三部分组成,本文将环境(context)通过注册器实现,实现自动化注入容器中,维护起来更方便。主要用于优化系统中多if判断场景,大一点的系统应该都能碰到,所以希望可以帮助到大家。
创建登陆策略
public interface LoginStrategy {/*** 检查参数* @return*/default Result checkParam(LoginPatamDTO loginPatamDTO){return Result.err();}/*** 登陆* @param loginPatamDTO* @return*/default Result<LoginUserInfoVO> login(LoginPatamDTO loginPatamDTO){return Result.err();}
}
创建环境(注册器)
将bean对象放在指定的map容器中。
@Component
public class LoginCommonStrategyRegistry {@Autowiredprivate BeanFactory beanFactory;private Map<LoginTypeEnum, LoginStrategy> loginrCommonStrategyMap = new ConcurrentHashMap<>();public Map<LoginTypeEnum, LoginStrategy> mapGet() {return this.loginrCommonStrategyMap;}/*** 初始化策略*/public void registry(LoginTypeEnum loginTypeEnum, Class glass) {LoginStrategy funderCommonStrategy = (LoginStrategy)beanFactory.getBean(glass);this.loginrCommonStrategyMap.put(loginTypeEnum, funderCommonStrategy);}
}
简化代码 创建父类对象,也方便后期扩展
@PostConstruct 是 Java 中的一个注解,用于标记一个方法,在对象创建之后(通过构造函数创建并注入依赖后),在依赖注入完成之后立即执行。它的作用是在对象初始化阶段执行一些必要的操作,例如初始化资源、建立连接、加载数据等。
@Service
public abstract class AbstractLoginService implements LoginStrategy {private final LoginCommonStrategyRegistry loginCommonStrategyRegistry;public AbstractLoginService(LoginCommonStrategyRegistry loginCommonStrategyRegistry){this.loginCommonStrategyRegistry = loginCommonStrategyRegistry;}/*** 注册当前类到策略map中*/@PostConstructprotected abstract void initRegistry();
}
具体策略实现类
用户名密码登陆
@Service
@Slf4j
public class UserNameLoginService extends AbstractLoginService {public UserNameLoginService(LoginCommonStrategyRegistry loginCommonStrategyRegistry) {super(loginCommonStrategyRegistry);}@Overrideprotected void initRegistry() {super.loginCommonStrategyRegistry.registry(LoginTypeEnum.USERNAME_TYPE, this.getClass());}@Overridepublic Result checkParam(LoginPatamDTO loginPatamDTO) {if (ObjectUtils.isEmpty(loginPatamDTO.getUserName())) {return Result.err("用户名不能为空");}if (ObjectUtils.isEmpty(loginPatamDTO.getPassword())) {return Result.err("密码不能为空");}return Result.suc();}@Overridepublic Result<LoginUserInfoVO> login(LoginPatamDTO loginPatamDTO) {log.info("这里是用户名和密码登陆");if (loginPatamDTO.getUserName().equals("admin") && loginPatamDTO.getPassword().equals("admin")) {// 创建默认对象 模拟登陆返回LoginUserInfoVO loginUserInfoVO = LoginUserInfoVO.loginUserInfoVOCreate(loginPatamDTO.getLoginTypeEnum());return Result.suc(loginUserInfoVO);} else {return Result.err("用户名或者密码不正确");}}
}
手机号登陆
@Service
@Slf4j
public class PhoneLoginService extends AbstractLoginService {public PhoneLoginService(LoginCommonStrategyRegistry loginCommonStrategyRegistry) {super(loginCommonStrategyRegistry);}@Overrideprotected void initRegistry() {super.loginCommonStrategyRegistry.registry(LoginTypeEnum.PHONE_TYPE, this.getClass());}@Overridepublic Result checkParam(LoginPatamDTO loginPatamDTO) {if (ObjectUtils.isEmpty(loginPatamDTO.getPhone())) {return Result.err("手机号不能为空");}if (ObjectUtils.isEmpty(loginPatamDTO.getSmsCode())) {return Result.err("验证码不能为空");}return Result.suc();}@Overridepublic Result<LoginUserInfoVO> login(LoginPatamDTO loginPatamDTO) {log.info("这里是手机号登陆");if (loginPatamDTO.getPhone().equals("17777777777") && loginPatamDTO.getSmsCode().equals("7777")) {// 创建默认对象 模拟登陆返回LoginUserInfoVO loginUserInfoVO = LoginUserInfoVO.loginUserInfoVOCreate(loginPatamDTO.getLoginTypeEnum());return Result.suc(loginUserInfoVO);} else {return Result.err("验证码不正确");}}
}
触发点
如果没有写策略模式 那代码应该是这样
if(type.equals("用户名登陆")){// 用户名密码登陆验证
}else if(type.equals("手机号登陆")){// 手机号登陆
}else if(.......){....
}
使用策略模式触发点确实简化了不少代码,让代码优雅起来了。后续扩展啥的都很方便。
@RestController
public class LoginController {/*** 模拟redis*/public static Map<String, String> userInforedis = new ConcurrentHashMap<>();private final LoginCommonStrategyRegistry loginCommonStrategyRegistry;public LoginController(LoginCommonStrategyRegistry loginCommonStrategyRegistry) {this.loginCommonStrategyRegistry = loginCommonStrategyRegistry;}@RequestMapping(value = "/a/login")public Result<LoginUserInfoVO> login(LoginPatamDTO loginPatamDTO) {if (ObjectUtils.isEmpty(loginPatamDTO.getLoginTypeEnum())) {return Result.err("无效的登陆方式");}LoginStrategy loginStrategy = loginCommonStrategyRegistry.mapGet().get(loginPatamDTO.getLoginTypeEnum());if (loginStrategy == null) {return Result.err("无效的登陆方式");}Result result = loginStrategy.checkParam(loginPatamDTO);if (result.getCode() == 0) {return result;}Result<LoginUserInfoVO> login = loginStrategy.login(loginPatamDTO);if (login.getCode() == 1) {userInforedis.put(login.getDate().getToken(), JSON.toJSONString(login.getDate()));}return login;}}
调试
http://127.0.0.1:8080/a/login?loginTypeEnum=USERNAME_TYPE&userName=admin&password=admin{"code":1,"msg":"成功","date":{"token":"57a44abe-e731-40f1-a9f6-1069710ac6fa","userInfoBO":{"userId":"这个是id","userRealName":"这个是名字","userSex":"这个是性别","userDeptIds":"这个是部门","loginTypeEnum":"USERNAME_TYPE"}}}
简言
在接入策略模式前一定要想好调用交互逻辑,如果系统有现成的最好使用现成的。当然也要避免循环依赖注入。
希望本文可以帮到大家。