【Spring Security OAuth2】- App认证框架- 重构短信登录

作者简介：大家好，我是smart哥，前中兴通讯、美团架构师，现某互联网公司CTO

联系qq：184480602，加我进群，大家一起学习，一起进步，一起对抗互联网寒冬

学习必须往深处挖，挖的越深，基础越扎实！

阶段1、深入多线程

阶段2、深入多线程设计模式

阶段3、深入juc源码解析

阶段4、深入jdk其余源码解析

阶段5、深入jvm源码解析

码哥源码部分

码哥讲源码-原理源码篇【2024年最新大厂关于线程池使用的场景题】

码哥讲源码【炸雷啦！炸雷啦！黄光头他终于跑路啦！】

码哥讲源码-【jvm课程前置知识及c/c++调试环境搭建】

​​​​​​码哥讲源码-原理源码篇【揭秘join方法的唤醒本质上决定于jvm的底层析构函数】

码哥源码-原理源码篇【Doug Lea为什么要将成员变量赋值给局部变量后再操作？】

码哥讲源码【你水不是你的错,但是你胡说八道就是你不对了！】

码哥讲源码【谁再说Spring不支持多线程事务，你给我抽他！】

终结B站没人能讲清楚红黑树的历史，不服等你来踢馆！

打脸系列【020-3小时讲解MESI协议和volatile之间的关系，那些将x86下的验证结果当作最终结果的水货们请闭嘴】

重构短信登录

现有问题如上图：

1. 浏览器中使用session存储验证码
2. app中午cookie概念（无session）

解决方案：

1. app发送和验证 验证码必须携带一个deviceId （设备id）
2. 浏览器按之前的逻辑走

也就是说，这里只是验证码的存储发生了变化，那么抽出来一个存储接口，浏览器和app做不同的适配即可

    package cn.mrcode.imooc.springsecurity.securitycore.validate.code;
    
    import org.springframework.web.context.request.ServletWebRequest;
    
    /**
     * <pre>
     * 验证码存储仓库接口
     * </pre>
     * @author zhuqiang
     * @version 1.0.0
     * @date 2023/8/8 13:58
     * @since 1.0.0
     */
    public interface ValidateCodeRepository {
        /**
         * 保存验证码
         * @param request
         * @param code
         * @param validateCodeType
         */
        void save(ServletWebRequest request, ValidateCode code, ValidateCodeType validateCodeType);
    
        /**
         * 获取验证码
         * @param request
         * @param validateCodeType
         * @return
         */
        ValidateCode get(ServletWebRequest request, ValidateCodeType validateCodeType);
    
        /**
         * 移除验证码
         * @param request
         * @param validateCodeType
         */
        void remove(ServletWebRequest request, ValidateCodeType validateCodeType);
    }

浏览器实现;（因为开始只有session的实现，重构线把session的抽取出来）；
把session相关操作全部抽到这里了

    package cn.mrcode.imooc.springsecurity.securitybrowser.validate.code.impl;
    
    import cn.mrcode.imooc.springsecurity.securitycore.validate.code.ValidateCode;
    import cn.mrcode.imooc.springsecurity.securitycore.validate.code.ValidateCodeRepository;
    import cn.mrcode.imooc.springsecurity.securitycore.validate.code.ValidateCodeType;
    import cn.mrcode.imooc.springsecurity.securitycore.validate.code.impl.AbstractValidateCodeProcessor;
    import org.springframework.social.connect.web.HttpSessionSessionStrategy;
    import org.springframework.social.connect.web.SessionStrategy;
    import org.springframework.stereotype.Component;
    import org.springframework.web.context.request.ServletWebRequest;
    
    /**
     * ${desc}
     * @author zhuqiang
     * @version 1.0.1 2023/8/8 14:06
     * @date 2023/8/8 14:06
     * @since 1.0
     */
    @Component
    public class SessionValidateCodeRepository implements ValidateCodeRepository {
        /** 操作session的工具类 */
        private SessionStrategy sessionStrategy = new HttpSessionSessionStrategy();
        /** 验证码放入session的时候前缀 */
        public final static String SESSION_KEY_PREFIX = "SESSION_KEY_FOR_CODE";
    
        @Override
        public void save(ServletWebRequest request, ValidateCode code, ValidateCodeType validateCodeType) {
            sessionStrategy.setAttribute(request, getSessionKey(validateCodeType), code);
        }
    
        @Override
        public ValidateCode get(ServletWebRequest request, ValidateCodeType validateCodeType) {
            String sessionKey = getSessionKey(validateCodeType);
            // 拿到创建 create() 存储到session的code验证码对象
            return (ValidateCode) sessionStrategy.getAttribute(request, sessionKey);
        }
    
        @Override
        public void remove(ServletWebRequest request, ValidateCodeType validateCodeType) {
            sessionStrategy.removeAttribute(request, getSessionKey(validateCodeType));
        }
    
        /**
         * 构建验证码放入session时的key; 在保存的时候也使用该key
         * {@link AbstractValidateCodeProcessor#save(org.springframework.web.context.request.ServletWebRequest, cn.mrcode.imooc.springsecurity.securitycore.validate.code.ValidateCode)}
         * @param validateCodeType
         * @return
         */
        private String getSessionKey(ValidateCodeType validateCodeType) {
            return SESSION_KEY_PREFIX + validateCodeType.toString().toUpperCase();
        }
    }

app实现

    package cn.mrcode.imooc.springsecurity.securityapp.validate.code.impl;
    
    import cn.mrcode.imooc.springsecurity.securitycore.validate.code.ValidateCode;
    import cn.mrcode.imooc.springsecurity.securitycore.validate.code.ValidateCodeException;
    import cn.mrcode.imooc.springsecurity.securitycore.validate.code.ValidateCodeRepository;
    import cn.mrcode.imooc.springsecurity.securitycore.validate.code.ValidateCodeType;
    import cn.mrcode.imooc.springsecurity.securitycore.validate.code.impl.AbstractValidateCodeProcessor;
    import org.apache.commons.lang3.StringUtils;
    import org.springframework.beans.factory.annotation.Autowired;
    import org.springframework.boot.autoconfigure.data.redis.RedisAutoConfiguration;
    import org.springframework.data.redis.core.RedisTemplate;
    import org.springframework.stereotype.Component;
    import org.springframework.web.context.request.ServletWebRequest;
    
    import java.util.concurrent.TimeUnit;
    
    /**
     * ${desc}
     * @author zhuqiang
     * @version 1.0.1 2023/8/8 14:06
     * @date 2023/8/8 14:06
     * @since 1.0
     */
    @Component
    public class RedisValidateCodeRepository implements ValidateCodeRepository {
        /**
         * @see RedisAutoConfiguration#redisTemplate(org.springframework.data.redis.connection.RedisConnectionFactory)
         */
        @Autowired
        private RedisTemplate<Object, Object> redisTemplate;
        /** 验证码放入redis规则模式：CODE_{TYPE}_{DEVICEId} */
        private final static String CODE_KEY_PATTERN = "CODE_%s_%s";
    
        @Override
        public void save(ServletWebRequest request, ValidateCode code, ValidateCodeType validateCodeType) {
            redisTemplate.opsForValue().set(buildKey(request, validateCodeType), code, 180, TimeUnit.MINUTES);
        }
    
        @Override
        public ValidateCode get(ServletWebRequest request, ValidateCodeType validateCodeType) {
            String key = buildKey(request, validateCodeType);
            // 拿到创建 create() 存储到session的code验证码对象
            return (ValidateCode) redisTemplate.opsForValue().get(key);
        }
    
        @Override
        public void remove(ServletWebRequest request, ValidateCodeType validateCodeType) {
            String key = buildKey(request, validateCodeType);
            redisTemplate.delete(key);
        }
    
        /**
         * 构建验证码放入redis时的key; 在保存的时候也使用该key
         * {@link AbstractValidateCodeProcessor#save(ServletWebRequest, ValidateCode)}
         * @param validateCodeType
         * @return
         */
        private String buildKey(ServletWebRequest request, ValidateCodeType validateCodeType) {
            String deviceId = request.getHeader("deviceId");
            if (StringUtils.isBlank(deviceId)) {
                throw new ValidateCodeException("请在请求头中携带deviceId参数");
            }
            return String.format(CODE_KEY_PATTERN, validateCodeType, deviceId);
        }
    }

测试

打开之前屏蔽掉的图形验证码过滤器配置
发送验证码的是不用登陆的

    GET /code/sms?mobile=13012345678 HTTP/1.1
    Host: localhost:8080
    deviceId: 1

登陆,记得携带client信息

    POST /authentication/mobile HTTP/1.1
    Host: localhost:8080
    Authorization: Basic bXlpZDpteWlk
    deviceId: 1
    Content-Type: application/x-www-form-urlencoded
    
    smsCode=270058&mobile=13012345678

因为图形验证码和短信验证码存储只有存储这一块变更了。所以只要短信验证码的可以使用，图形验证码的也应该可以使用