[设计模式]springboot优雅实现策略器模式(加入注册器实现)

场景

登陆场景使用(登陆之后返回用户信息和token所做操作基本一致,杜绝多个判断)

  1. 用户名密码登陆,
  2. 指纹登录
  3. 手机号登陆
    楼主之前写过一次通过注解实现的,那个看起来没有注册器实现优雅。
    策略器可以使用的场景很多,主要为了杜绝多if判断,把if判断需实现的信息放在自己的实现类中,提升代码可阅读性,提高执行效率,降低出错概率。
    八股文:可扩展性高,灵活性高,代码复用性高,单一职责原则,可替换性高。

废话少说,源码地址。

代码地址 
git clone https://github.com/gwy572294624/strategy-demo-new


讲解

chatGpt关于策略模式的八股文
策略模式(Strategy Pattern)是一种行为型设计模式,用于将不同的算法封装成独立的策略类,并使这些策略类可以相互替换,以实现在运行时动态地选择不同的算法。

在策略模式中,有三个主要的角色:

  1. 环境(Context):环境类是策略模式的核心,其内部持有一个策略对象的引用。环境类将具体的任务委派给策略对象进行处理,而不直接实现具体的算法逻辑。

  2. 策略(Strategy):策略类是一个接口或抽象类,它定义了具体算法的方法。不同的策略类实现了不同的算法逻辑。策略类之间可以相互替换,提供了算法的灵活配置。

  3. 具体策略(Concrete Strategies):具体策略类是策略模式的实现类,实现了策略接口或抽象类中定义的具体算法逻辑。

策略模式的工作流程如下:

  1. 客户端创建一个环境对象,并通过构造函数或设置方法将具体的策略对象传递给环境对象。

  2. 客户端根据需求选择合适的策略对象。

  3. 环境对象在执行任务时,会将具体的算法委派给当前持有的策略对象进行处理。

  4. 当需要切换算法时,客户端可以动态地替换环境对象的策略对象,实现不同的算法逻辑。

策略模式的优点包括灵活性、可替换性、可扩展性和代码复用性。它可以避免使用大量的条件语句或开关语句,将不同的算法逻辑封装在独立的策略类中,使系统更加灵活、易于扩展和维护。

关于本文

策略模式组成是通过环境,策略,具体实现三部分组成,本文将环境(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 {

    @Autowired
    private 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中
     */
    @PostConstruct
    protected abstract void initRegistry();
}



具体策略实现类

用户名密码登陆


@Service
@Slf4j
public class UserNameLoginService extends AbstractLoginService {


    public UserNameLoginService(LoginCommonStrategyRegistry loginCommonStrategyRegistry) {
        super(loginCommonStrategyRegistry);
    }

    @Override
    protected void initRegistry() {
        super.loginCommonStrategyRegistry.registry(LoginTypeEnum.USERNAME_TYPE, this.getClass());
    }

    @Override
    public Result checkParam(LoginPatamDTO loginPatamDTO) {
        if (ObjectUtils.isEmpty(loginPatamDTO.getUserName())) {
            return Result.err("用户名不能为空");
        }
        if (ObjectUtils.isEmpty(loginPatamDTO.getPassword())) {
            return Result.err("密码不能为空");
        }
        return Result.suc();
    }

    @Override
    public 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);
    }

    @Override
    protected void initRegistry() {
        super.loginCommonStrategyRegistry.registry(LoginTypeEnum.PHONE_TYPE, this.getClass());
    }

    @Override
    public Result checkParam(LoginPatamDTO loginPatamDTO) {
        if (ObjectUtils.isEmpty(loginPatamDTO.getPhone())) {
            return Result.err("手机号不能为空");
        }
        if (ObjectUtils.isEmpty(loginPatamDTO.getSmsCode())) {
            return Result.err("验证码不能为空");
        }
        return Result.suc();
    }

    @Override
    public 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"}}}

简言

在接入策略模式前一定要想好调用交互逻辑,如果系统有现成的最好使用现成的。当然也要避免循环依赖注入。

希望本文可以帮到大家。

  • 0
    点赞
  • 0
    收藏
    觉得还不错? 一键收藏
  • 1
    评论
好的,下面我给你讲一下如何在Spring Boot实现策略模式。 首先,我们需要定义一个接口或者抽象类,用于定义策略模式中的抽象行为。以一个简单的计算为例,我们可以定义一个`Calculator`接口,其中包含一个`calculate`方法: ```java public interface Calculator { int calculate(int a, int b); } ``` 然后,我们需要定义具体的策略类,实现`Calculator`接口。假设我们有两种计算方式:加法和乘法,我们可以分别定义两个实现类: ```java @Component public class AddCalculator implements Calculator { @Override public int calculate(int a, int b) { return a + b; } } @Component public class MultiplyCalculator implements Calculator { @Override public int calculate(int a, int b) { return a * b; } } ``` 在上面的代码中,我们使用了`@Component`注解将这两个类注册Spring Bean,以便在其他地方可以自动注入它们。 接下来,我们需要一个Context类,用于选择具体的策略类。在Spring Boot中,我们可以使用`@Autowired`注解自动注入所有实现了`Calculator`接口的Bean,并使用`@Qualifier`注解来指定具体的实现类。例如: ```java @Service public class CalculatorService { @Autowired private List<Calculator> calculators; public int calculate(String operator, int a, int b) { for (Calculator calculator : calculators) { if (calculator.getClass().getSimpleName().equalsIgnoreCase(operator)) { return calculator.calculate(a, b); } } throw new IllegalArgumentException("Invalid operator: " + operator); } } ``` 在上面的代码中,我们使用了一个`List<Calculator>`来注入所有实现了`Calculator`接口的Bean。然后,我们在`calculate`方法中根据传入的运算符选择具体的策略类,并调用它的`calculate`方法进行计算。 最后,我们可以在Controller中使用`CalculatorService`类来实现具体的业务逻辑。例如: ```java @RestController public class CalculatorController { @Autowired private CalculatorService calculatorService; @GetMapping("/calculate") public int calculate(@RequestParam String operator, @RequestParam int a, @RequestParam int b) { return calculatorService.calculate(operator, a, b); } } ``` 在上面的代码中,我们使用了`@RequestParam`注解来接收请求参数,然后调用`CalculatorService`的`calculate`方法进行计算,并返回结果。 这就是在Spring Boot实现策略模式的方法。希望对你有帮助!
评论 1
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值