原来使用 Spring 实现策略模式可以这么简单!

武培轩

共 5788字,需浏览 12分钟

 ·

2021-02-02 03:42








策略模式作为一种软件设计模式,指对象有某个行为,但是在不同的场景中,该行为有不同的实现算法,可以替代代码中大量的 if-else。

比如我们生活中的场景:买东西结账可以使用微信支付、支付宝支付或者银行卡支付,这些交易方式就是不同的策略。

那么在什么时候使用策略模式呢?

在《阿里巴巴Java开发手册》中有提到当超过 3 层的 if-else 的逻辑判断代码可以使用策略模式来实现。

在 Spring 中实现策略模式的方式有很多种,下面通过一个案例来演示下,比如有个需求需要实现支持第三方登录,目前需要支持以下三种登录方式:

  • 微信登录
  • QQ 登录
  • 微博登录

下面将通过策略模式来实现这个需求,其中策略模式结构如下图所示:

策略模式结构如下图所示:

策略模式结构

主要包括一个登录接口类和几种登录方式的实现方式,并利用简单工厂来获取对应的处理器。

定义策略接口

首先定义一个登录的策略接口 LoginHandler,其中包括两个方法:

  1. 获取策略类型的方法
  2. 处理策略逻辑的方法
public interface LoginHandler<T extends Serializable{

    /**
     * 获取登录类型
     *
     * @return
     */

    LoginType getLoginType();

    /**
     * 登录
     *
     * @param request
     * @return
     */

    LoginResponse handleLogin(LoginRequest request);
}

其中,LoginHandlergetLoginType 方法用来获取登录的类型(即策略类型),用于根据客户端传递的参数直接获取到对应的策略实现。

客户端传递的相关参数都被封装为 LoginRequest,传递给 handleLogin 进行处理。

@Data
public class LoginRequest {

    private LoginType loginType;

    private Long userId;
}

其中,根据需求定义登录类型枚举如下:

public enum LoginType {
    QQ,
    WE_CHAT,
    WEI_BO;
}

实现策略接口

在定义好策略接口后,我们就需要根据各种第三方登录来实现对应的处理逻辑就可以了。

微信登录

@Component
public class WeChatLoginHandler implements LoginHandler<String{

    private final Logger logger = LoggerFactory.getLogger(this.getClass());

    /**
     * 获取登录类型
     *
     * @return
     */

    @Override
    public LoginType getLoginType() {
        return LoginType.WE_CHAT;
    }

    /**
     * 登录
     *
     * @param request
     * @return
     */

    @Override
    public LoginResponse handleLogin(LoginRequest request) {
        logger.info("微信登录:userId:{}", request.getUserId());
        String weChatName = getWeChatName(request);
        return LoginResponse.success("微信登录成功", weChatName);
    }

    private String getWeChatName(LoginRequest request) {
        return "wupx";
    }
}

QQ 登录

@Component
public class QQLoginHandler implements LoginHandler<Serializable{

    private final Logger logger = LoggerFactory.getLogger(this.getClass());

    /**
     * 获取登录类型
     *
     * @return
     */

    @Override
    public LoginType getLoginType() {
        return LoginType.QQ;
    }

    /**
     * 登录
     *
     * @param request
     * @return
     */

    @Override
    public LoginResponse handleLogin(LoginRequest request) {
        logger.info("QQ登录:userId:{}", request.getUserId());
        return LoginResponse.success("QQ登录成功"null);
    }
}

微博登录

@Component
public class WeiBoLoginHandler implements LoginHandler<Serializable{

    private final Logger logger = LoggerFactory.getLogger(this.getClass());

    /**
     * 获取登录类型
     *
     * @return
     */

    @Override
    public LoginType getLoginType() {
        return LoginType.WEI_BO;
    }

    /**
     * 登录
     *
     * @param request
     * @return
     */

    @Override
    public LoginResponse handleLogin(LoginRequest request) {
        logger.info("微博登录:userId:{}", request.getUserId());
        return LoginResponse.success("微博登录成功"null);
    }
}

创建策略的简单工厂

@Component
public class LoginHandlerFactory implements InitializingBeanApplicationContextAware {
    private static final Map> LOGIN_HANDLER_MAP = new EnumMap<>(LoginType.class);
    private ApplicationContext appContext;

    /**
     * 根据登录类型获取对应的处理器
     *
     * @param loginType 登录类型
     * @return 登录类型对应的处理器
     */

    public LoginHandler getHandler(LoginType loginType) {
        return LOGIN_HANDLER_MAP.get(loginType);
    }

    @Override
    public void afterPropertiesSet() throws Exception {
        // 将 Spring 容器中所有的 LoginHandler 注册到 LOGIN_HANDLER_MAP
        appContext.getBeansOfType(LoginHandler.class)
                .values()
                .forEach(handler -> LOGIN_HANDLER_MAP.put(handler.getLoginType(), handler))
;
    }

    @Override
    public void setApplicationContext(ApplicationContext applicationContext) throws BeansException {
        appContext = applicationContext;
    }
}

我们让 LoginHandlerFactory实现 InitializingBean 接口,在 afterPropertiesSet 方法中,基于 Spring 容器将所有 LoginHandler 自动注册到 LOGIN_HANDLER_MAP,从而 Spring 容器启动完成后, getHandler 方法可以直接通过 loginType 来获取对应的登录处理器。

创建登录服务

在登录服务中,我们通过 LoginHandlerFactory 来获取对应的登录处理器,从而处理不同类型的第三方登录:

@Service
public class LoginServiceImpl implements LoginService {
    @Autowired
    private LoginHandlerFactory loginHandlerFactory;

    @Override
    public LoginResponse login(LoginRequest request) {
        LoginType loginType = request.getLoginType();
        // 根据 loginType 找到对应的登录处理器
        LoginHandler loginHandler =
                loginHandlerFactory.getHandler(loginType);
        // 处理登录
        return loginHandler.handleLogin(request);
    }
}

Factory 只负责获取 Handler,Handler 只负责处理具体的登录,Service 只负责逻辑编排,从而达到功能上的低耦合高内聚。

测试

写一个 Controller:

@RestController
public class LoginController {

    @Autowired
    private LoginService loginService;

    /**
     * 登录
     */

    @PostMapping("/login")
    public LoginResponse login(@RequestParam LoginType loginType, @RequestParam Long userId) {
        LoginRequest loginRequest = new LoginRequest();
        loginRequest.setLoginType(loginType);
        loginRequest.setUserId(userId);
        return loginService.login(loginRequest);
    }
}

然后用 Postman 测下下:

微信登录
QQ登录

是不是很简单呢?如果需求又要加需求,需要支持 GitHub 第三方登录。

此时我们只需要添加一个新的策略实现,然后在登录枚举中加入对应的类型即可:

@Component
public class GitHubLoginHandler implements LoginHandler {

    private final Logger logger = LoggerFactory.getLogger(this.getClass());

    /**
     * 获取登录类型
     *
     * @return
     */
    @Override
    public LoginType getLoginType() {
        return LoginType.GIT_HUB;
    }

    /**
     * 登录
     *
     * @param request
     * @return
     */
    @Override
    public LoginResponse handleLogin(LoginRequest request) {
        logger.info("GitHub登录:userId:{}", request.getUserId());
        return LoginResponse.success("GitHub登录成功", null);
    }
}

此时不需要修改任何代码 ,因为 Spring 容器重启时会自动将 GitHubLoginHandler 注册到 LoginHandlerFactory 中,使用 Spring 实现策略模式就是这么简单,还不快学起来!


       
   
说完观察者和发布订阅模式的区别,面试官不留我吃饭了
模板方法模式——看看 JDK 和 Spring 是如何优雅复用代码的
从原型模式到浅拷贝和深拷贝

觉得不错,点个在看~

浏览 34
点赞
评论
收藏
分享

手机扫一扫分享

分享
举报
评论
图片
表情
推荐
点赞
评论
收藏
分享

手机扫一扫分享

分享
举报