基于session实现共享登录

基于session实现登录

1.发送短信验证码

 

@Override
public Result sendCode(String phone, HttpSession session) {
    //1.校验手机号是否合规
    if (RegexUtils.isPhoneInvalid(phone)) {
        //2.不合规直接返回 错误信息
        return Result.fail("手机号错误");
    }
    //3.如果合规生成验证码
    String code = RandomUtil.randomString(4);
    //4.将验证码保存到session中
    session.setAttribute("code",code);
    //5.发送验证码
    log.info("code: {}",code);
    //返回ok
    return Result.ok();
}

ServletContext：上下文对象，在服务器启动时被创建，关闭时被注销，被所有Servlet共享，可在web.xml中进行配置，存放一些初始化数据，拥有最长的生命周期。

HttpSession：会话对象，浏览器请求服务器时被创建，关闭浏览器窗口或页面不刷新过期时被销毁，拥有较长的生命周期。

ServletRequest：请求对象，浏览器每次发送请求时被创建，响应结束之后被销毁，用于存放来自页面的参数和浏览器信息，生命周期最短。

2.短信验证、登录、注册功能

 

@Override
public Result login(LoginFormDTO loginForm, HttpSession session) {
    //1.校验手机号的格式
    String phone = loginForm.getPhone();
    if (RegexUtils.isPhoneInvalid(phone)) {
        //2.不一致直接报错
        return Result.fail("手机号错误");
    }
    //3.比较验证码
    Object cacheCode = session.getAttribute("code");
    String code = loginForm.getCode();
    if(session==null || !cacheCode.toString().equals(code)){
        //4.不一致直接报错
        return Result.fail("错误信息");
    }
    //5.根据手机号查询用户
    LambdaQueryWrapper<User> query = new LambdaQueryWrapper<>();
    query.eq(User::getPhone,loginForm.getPhone());
    User user = this.getOne(query);
    if(user==null){
        //6.不存在直接创建新用户保存到数据库中
        user=createUserWithPhone(loginForm.getPhone());
    }
    //7.最终将用户信息保存到session中
    session.setAttribute("user",user);
    return Result.ok();
}
​
private User createUserWithPhone(String phone) {
    User user = new User();
    user.setPhone(phone);
    user.setNickName(USER_NICK_NAME_PREFIX+RandomUtil.randomString(6));
    this.save(user);
    return user;
​
}

3.登录校验功能

 

创建拦截器

public class LoginInterceptor implements HandlerInterceptor {
    @Override
    public boolean preHandle(HttpServletRequest request, HttpServletResponse response, Object handler) throws Exception {
        //1.获取session
        HttpSession session = request.getSession();
        //2.从session中拿到用户信息
        Object user = session.getAttribute("user");
        if(user==null){
            //3.如果不存在直接拦截  返回401状态
            response.setStatus(401);
            return false;
        }
        //4.存在  保存到ThreadLocal中实现共享
        UserHolder.saveUser((User) user);
        //5.放行
        return true;
    }
​
    @Override
    public void afterCompletion(HttpServletRequest request, HttpServletResponse response, Object handler, Exception ex) throws Exception {
        //避免造成内存泄露
        UserHolder.removeUser();
    }
}

注册拦截器

//注册拦截器  及其相关配置
@Configuration
public class MvcConfig implements WebMvcConfigurer {
    //添加拦截器
​
    @Override
    public void addInterceptors(InterceptorRegistry registry) {
        registry.addInterceptor(new LoginInterceptor()).excludePathPatterns(
                "/shop/**",
                "/voucher/**",
                "/shop-type/**",
                "/upload/**",
                "/blog/hot",
                "/user/code",
                "/user/login"
        );
    }
}

public class UserHolder {
    private static final ThreadLocal<User> tl = new ThreadLocal<>();
​
    public static void saveUser(User user){
        tl.set(user);
    }
​
    public static User getUser(){
        return tl.get();
    }
​
    public static void removeUser(){
        tl.remove();
    }
}

ThreadLocal叫做线程变量，意思是ThreadLocal中*填充的变量*属于*当前线程*，该变量对其他线程而言是隔离的，也就是说该变量是当前线程独有的变量。ThreadLocal为变量在每个线程中都创建了一个副本，那么每个线程可以访问自己内部的副本变量。