在网站的应用场景下,对于用户的信息展示与账号设置的都是一个绕不过的问题。解决这个问题,最重要的是,怎么拦截未登录用户进入这个功能内?这就意味着必须在每次动态请求时,校验登录状态以及让请求持有用户(在服务器端可查询到对应用户)。当然,我们必须考虑多线程并发的情况,简单地将用户信息放在一个公共的类或者一个容器内,可能会导致数据冲突。现在流行的做法就是利用拦截器去保证动态请求的合法性,而不是在重复造轮子式一次次检查,并且利用 ThreadLocal 进行数据线程隔离保护。本文就将介绍在拦截器的作用下,用户的信息展示与账户设置该如何正确地处理。当然,接下来,也会有更多更有技术力的文章等候着你,欢迎大家关注本人的博客。

文章目录

    • 实现思路
      • 拦截动态请求思路:
      • 登录信息展示思路:
      • 账户设置之修改用户头像思路:
      • 账户设置之修改用户密码思路:
    • 代码实现
      • 拦截动态请求配置类 WebConfig 设置
      • 用户信息工具类 hostHolder
      • 查询 cookie 中信息工具类 CookieUtil
      • 请求持有用户拦截器 LoginTicketInterceptor
      • 自定义注解配置 @LoginRequired
      • 检查登录状态拦截器 LoginRequiredInterceptor
      • 修改用户头像核心类 uploadHeader
      • 修改密码 updatePassword
    • 相关Bug
      • java.io.FileNotFoundException: xxxxx (系统找不到指定的路径。)

实现思路

关于注册部分的相关知识可看,博主的另一篇文章:一文教你学会实现以邮件激活的注册账户代码,而关于登录部分的相关知识可看博主这一篇文章:基于验证码检验的登录就该这么实现,欢迎关注博主本人专栏:后端开发

拦截动态请求思路:

  1. 利用 ThreadLocal 实现保存用户信息工具类 hostHolder

  2. 利用实现 HandlerInterceptor 来进行拦截用户动态请求

    • 请求开始前,查询并检查登录凭证是否有效,有效则返回根据凭证字符查询登录凭证,并利用其中的 userId 查询用户,再调用 hostHolder 存储用户信息,到该线程里。
    • 请求模板引擎返回前,将线程保存的需要进行展示的用户信息存入模板中
    • 请求结束时,清理线程存储的用户信息,以免占用太多内存空间
  3. 设置拦截目标路径

    • 第一种方法:通过实现 WebMvcConfigurer 来直接完成拦截器配置目标路径。
    • 第二种方法:通过自定义注解,并放置在需要拦截的方法之上,配合 WebMvcConfigurer 实现拦截指定方法
  4. 检查登录状态:

    • 在实现了 WebMvcConfigurer 的配置类中配置注解拦截器拦截所有请求。
    • 通过自定义注解 @LoginRequired 实现拦截方法,并检查登录状态
    • 利用拦截器判断请求的方法是否存在 @LoginRequired 注解,存在的话,利用 hostHolder 检查登录状态
    • 如果为未登录状态,并且请求了登录后才能使用的方法,必须将请求重定向至登录页面
    • 如果为已登录状态,则不管请求方法。

当然,可能会有这样的疑惑,为什么要使用第一种方法实现请求持有用户信息,为什么要额外使用第二种方法实现检查登录状态?因为请求持有用户信息是需要绝大多数请求都要实现的,毕竟前端头部的用户信息展示都在每个页面中存在,并且,未登录状态的用户也可以访问首页信息,如果也进行拦截,会较为麻烦,需要在 config 方法进行配置,解耦较为困难。而检查登录状态,只需要拦截特定方法即可。

如果存在多个拦截器,并且都是拦截所有请求的开始之前,那它们的拦截顺序又是怎么样的呢?拦截顺序按的是实现了 WebMvcConfigurer 的配置类 WebConfig 的注册顺序决定的。

登录信息展示思路:

在拦截器拦截请求方式完成后,已经让需要展示的登录信息存入模板引擎中,直接在模板引擎中调用即可。

账户设置之修改用户头像思路:

  1. 利用 MultipartFile 来获取上传的文件,如果是多个文件,即为数组,单个文件即为个体。
  2. 先判断上传的用户头像是否为空,为空返回错误信息,否则,继续
  3. 利用 MultipartFile 的 getOriginalFilename() 实现获取文件原始名称
  4. 通过判断文件名称最后一个 “.” 的位置,截取字符串,获取文件后缀名
  5. 如果后缀名为空,返回错误信息,否则,继续
  6. 为了防止用户上传文件名与已有用户上传的文件名重复,故采取数字加字母结合的一段随机字符串并加上后缀名组成文件名
  7. 在 application.yml 配置上传路径,并注入到修改方法所属的类中
  8. 利用 java.io.File 对象存储文件,并配合 MultipartFile.transferTo()方法实现上传文件到对应目录。注意捕获异常,记录日志。
  9. 更新数据库中的用户头像路径为 web 路径:域名/user/header/***.suffix
  10. 实现一个获取头像方法,实现读取服务器指定路径(物理路径)下的用户头像文件,并将文件输出到 域名/user/header/***.suffix 中。

账户设置之修改用户密码思路:

  1. 判断新密码是否为空,两次输入的新密码是否一致,新密码长度是否符合要求等等基本格式检查
  2. 比较旧密码是否与数据库中的密码一致,注意采用了加盐 md5 加密
  3. 如果旧密码校验成功,就可以生成新的盐值,更新用户数据库,并与新密码组成新密码字符,再经过 md5 加密,更新用户数据库即可。


项目代码已上传:项目代码

代码实现

拦截动态请求配置类 WebConfig 设置

  • loginTicketInterceptor:请求持有用户拦截器
  • loginRequiredInterceptor:检查登录状态拦截器
  • excludePathPatterns:拦截器不拦截的静态请求配置
@Configurationpublic class WebConfig implements WebMvcConfigurer {    @Autowired    private LoginTicketInterceptor loginTicketInterceptor;    @Autowired    private LoginRequiredInterceptor loginRequiredInterceptor;    @Override    public void addInterceptors(InterceptorRegistry registry) {        registry.addInterceptor(loginTicketInterceptor)        .excludePathPatterns("/**/*.css", "/**/*.js", "/**/*.png", "/**/*.jpg","/**/*.jpeg","/**/*.webp");        registry.addInterceptor(loginRequiredInterceptor)                .excludePathPatterns("/**/*.css", "/**/*.js", "/**/*.png", "/**/*.jpg","/**/*.jpeg","/**/*.webp");    }}

用户信息工具类 hostHolder

  • ThreadLocal 底层方法:都是获取线程,并统一在 ThreadLocalMap 统一存储对应线程的 map 内容,不同线程拥有不同的 ThreadLocalMap 空间。

    set 方法:

    public void set(T value) {    Thread t = Thread.currentThread();    ThreadLocalMap map = getMap(t);    if (map != null) {        map.set(this, value);    } else {        createMap(t, value);    }}

    get 方法:

    public T get() {    Thread t = Thread.currentThread();    ThreadLocalMap map = getMap(t);    if (map != null) {        ThreadLocalMap.Entry e = map.getEntry(this);        if (e != null) {            @SuppressWarnings("unchecked")            T result = (T)e.value;            return result;        }    }    return setInitialValue();}

    remove方法:

    public void remove() {    ThreadLocalMap m = getMap(Thread.currentThread());    if (m != null) {        m.remove(this);    }}
  • 实现方案就是在 hostHolder 封装多个关于 user 的方法,底层实现是利用 ThreadLocal 实现线程隔离

  • set():存入用户,get():获取用户,remove():移除用户

@Componentpublic class HostHolder {    private ThreadLocal<User> users = new ThreadLocal<>();    public void setUser(User user){        users.set(user);    }    public User getUser(){        return users.get();    }    public void clear(){        users.remove();    }}

查询 cookie 中信息工具类 CookieUtil

  • request.getCookies():获取所有 cookie 内容
  • 遍历所有 cookie 内容,找到对应信息并返回 cookie.getValue() 对应的值
// 查询 cookie 中指定的信息public class CookieUtil {    public static String getValue(HttpServletRequest request, String cookieName){        if (request == null || cookieName == null){            throw new IllegalArgumentException("参数为空!");        }        Cookie[] cookies = request.getCookies();        if (cookies != null){            for (Cookie cookie : cookies){                if (cookie.getName().equals(cookieName)){                    return cookie.getValue();                }            }        }        return null;    }}

请求持有用户拦截器 LoginTicketInterceptor

  • preHandle 拦截器返回的是布尔值,为 true 表示请求可以继续,为 false 表示请求不可继续
  • 调用 CookieUtil 从 cookie 中获取登录凭证字符,通过登录凭证字符查询数据库对应登录凭证对象,再根据对象存有的 userId 获取登录用户对象
  • hostHolder.setUser(user):存入线程空间中,位于服务器
// 让请求持有用户@Componentpublic class LoginTicketInterceptor implements HandlerInterceptor {    @Resource    @Qualifier("userServiceImpl")    private UserService userServiceImpl;    @Autowired    private HostHolder hostHolder;    // 拦截需要登录却未登录的接口    @Override    public boolean preHandle(HttpServletRequest request, HttpServletResponse response, Object handler) throws Exception {        // 查询凭证        String loginTicket = CookieUtil.getValue(request, "loginTicket");        if (loginTicket != null){            LoginTicket userTicket = userServiceImpl.getLoginTicket(loginTicket);            // 查询凭证是否有效            if (userTicket != null && userTicket.getStatus() ==1 && userTicket.getExpired().after(new Date())){                // 根据凭证查询用户                User user = userServiceImpl.findUserById(userTicket.getUserId());                // 多线程情况下,不能使用简单的公共方法或者容器可能会发生冲突,必须实现线程隔离                // 在本次请求中持有用户                hostHolder.setUser(user);            }        }        return true;    }    // 往模板里存入用户    @Override    public void postHandle(HttpServletRequest request, HttpServletResponse response, Object handler, ModelAndView modelAndView) throws Exception {        User user = hostHolder.getUser();        if (user != null && modelAndView != null){            modelAndView.addObject("loginUser", user);        }    }    // 请求结束,清理用户    @Override    public void afterCompletion(HttpServletRequest request, HttpServletResponse response, Object handler, Exception ex) throws Exception {        hostHolder.clear();    }}

自定义注解配置 @LoginRequired

  • @Target(ElementType.METHOD):指定注解放置位置为 Method
  • @Retention(RetentionPolicy.RUNTIME):指定注解生效时间,生命周期
@Target(ElementType.METHOD)@Retention(RetentionPolicy.RUNTIME)public @interface LoginRequired {    // 检查登录状态注解}

检查登录状态拦截器 LoginRequiredInterceptor

  • handlerMethod.getMethod():获取请求方法

  • method.getAnnotation(LoginRequired.class):检查请求方法上是否存在 @LoginRequired 注解

  • loginRequired != null && hostHolder.getUser() == null:存在注解说明需要进行检查,检查登录状态为未登录,需要进行处理

  • response.sendRedirect(request.getContextPath() + “/login”):重定向到登录方法

// 检查登录状态请求@Componentpublic class LoginRequiredInterceptor implements HandlerInterceptor {    @Autowired    private HostHolder hostHolder;    @Override    public boolean preHandle(HttpServletRequest request, HttpServletResponse response, Object handler) throws Exception {        if (handler instanceof HandlerMethod){            HandlerMethod handlerMethod = (HandlerMethod) handler;            // 获取请求方法            Method method = handlerMethod.getMethod();            // 检查请求方法上是否存在 @LoginRequired 注解            LoginRequired loginRequired = method.getAnnotation(LoginRequired.class);            if (loginRequired != null && hostHolder.getUser() == null){                // 存在注解说明需要进行检查,检查登录状态为未登录,故重定向到登录请求。                response.sendRedirect(request.getContextPath() + "/login");                return false;            }        }        return true;    }}

修改用户头像核心类 uploadHeader

  • 添加了拦截注解的有 uploadHeader,而没有 getHeader 是因为未登录用户也能查看其他用户的头像,不可拦截
  • new FileInputStream(filenamePath):为自建输出流,spingboot 不会自动管理该 bean,需要自行关闭该输出流
  • upload:application.yml 配置的上传文件路径,尽可能为绝对路径,否则,请看相关 Bug 部分,在这里不多赘述
  • throw new RuntimeException():异常先抛出,后续再进行统一处理,本文暂不涉及,欢迎关注博主博客。
  • fileInputStream.read(buffer):利用缓存区传输数据较为快速
@LoginRequired@RequestMapping(path = "/setting",method = RequestMethod.GET)public String getSettingPage(){    return "/site/setting";}@LoginRequired@RequestMapping(path = "/upload",method = RequestMethod.POST)public String uploadHeader(MultipartFile headerImage, Model model){    if (headerImage == null){        model.addAttribute("imageError","您还没有上传图片!");        return "/site/setting";    }    // 获取上传的文件原始名称    String filename = headerImage.getOriginalFilename();    String suffix = filename.substring(filename.lastIndexOf("."));    if (StringUtils.isBlank(suffix)){        model.addAttribute("imageError","图片格式不正确!");        return "/site/setting";    }    // 生成随机名称,防止覆盖    filename = CommunityUtil.generateUUID() + suffix;    // 确定文件路径    File file = new File(upload+"/"+filename);    try {        headerImage.transferTo(file);    } catch (IOException e) {        logger.error("上传文件失败!"+e.getMessage());        throw new RuntimeException("上传文件失败,服务器异常!"+e.getMessage());    }    // 更新用户头像路径(web路径)    // 路径:域名/user/header/***.suffix    User user = hostHolder.getUser();    String headerUrl = domain + "/user/headerUrl/" + filename;    userServiceImpl.updateHeaderUrl(user.getId(),headerUrl);    return "redirect:/index";}// 获取头像@RequestMapping(path = "/headerUrl/{filename}",method = RequestMethod.GET)public void getHeader(@PathVariable("filename") String filename, HttpServletResponse response){    // 服务器路径    String filenamePath =  upload + "/" + filename;    // 文件后缀    String suffix = filenamePath.substring(filename.lastIndexOf("."));    // 响应图片    response.setContentType("image/" + suffix);    try(            OutputStream outputStream = response.getOutputStream();            FileInputStream fileInputStream = new FileInputStream(filenamePath);    ) {        byte[] buffer = new byte[1024];        int b = 0;        while ((b = fileInputStream.read(buffer)) != -1){            outputStream.write(buffer, 0, b);        }    } catch (IOException e) {        logger.error("读取头像失败:"+ e.getMessage());    }}

修改密码 updatePassword

// 修改密码@LoginRequired@RequestMapping(path = "/updatePassword", method = RequestMethod.POST)public String updatePassword(Model model, String newPassword, String oldPassword, String confirmPassword){    if (!newPassword.equals(confirmPassword)){        model.addAttribute("newPasswordError","两次填写的新密码不一致");        return "/site/setting";    }    if (newPassword.length() <= 8){        model.addAttribute("newPasswordError","新密码长度不能小于8位!");        return "/site/setting";    }    if (oldPassword.equals(newPassword)){        model.addAttribute("newPasswordError","新旧密码相同!请重新输入!");        return "/site/setting";    }    User user = hostHolder.getUser();    // 比较原密码是否正确    if (StringUtils.isBlank(oldPassword) || !user.getPassword().equals(CommunityUtil.md5(oldPassword + user.getSalt()))){        model.addAttribute("oldPasswordError","您输入的原密码错误!");        return "/site/setting";    }    // 更新密码    String salt = CommunityUtil.generateUUID().substring(0,5);    userServiceImpl.updatePassword(user.getId(),CommunityUtil.md5(newPassword+ salt));    userServiceImpl.updateSalt(user.getId(),salt);    model.addAttribute("msg","您的密码已修改完成,可用新密码登录!");    model.addAttribute("target","/index");    model.addAttribute("url",domain+"/index");    return "/site/operate-result";}


项目代码已上传:项目代码

相关Bug

java.io.FileNotFoundException: xxxxx (系统找不到指定的路径。)

出现原因:

  • 文件夹未建立

    自行创建路径,或者添加以下代码:

    File dir = new File("目录路径");if (!dir.exists()) {    dir.mkdirs(); // 创建目录}try {    String realPath = dir.getCanonicalPath(); // 获取真实路径    file.transferTo(new File(realPath + "/" + fileName)); // 转存} catch (IOException e) {    System.err.println(e);    throw new BadRequestException("上传失败");}
  • 路径书写为相对路径

当MultipartFile的 transferTo 的入参是相对路径时,会自动拼接成一个绝对路径,但这个绝对路径并不是真实存在的,所以转存时会报错。要么修改为绝对路径,要么使用第一个出现原因的解决代码。