SpringBoot之HandlerInterceptor拦截器的使用

5.1 用户上下文类

---

package com.nobody.context;

import com.nobody.enums.AuthenticationTypeEnum;

import lombok.Getter;

import lombok.Setter;

import lombok.ToString;

/**

• @Description 用户上下文

• @Author Mr.nobody

• @Date 2020/10/25

• @Version 1.0

*/

@Getter

@Setter

@ToString

public class UserContext {

// 用户名称

private String name;

// 用户ID

private String userId;

// 用户类型

private AuthenticationTypeEnum type;

}

5.2 校验访问权限注解

---

package com.nobody.annotation;

import com.nobody.enums.AuthenticationTypeEnum;

import java.lang.annotation.*;

/**

• @Description 校验访问权限注解

• @Author Mr.nobody

• @Date 2020/10/25

• @Version 1.0

*/

@Target({ElementType.TYPE, ElementType.METHOD})

@Retention(RetentionPolicy.RUNTIME)

@Inherited

@Documented

public @interface UserAuthenticate {

/**

• 是否需要校验访问权限 默认不校验

• @return

*/

boolean permission() default false;

/**

• 验证类型，默认游客

• @return

*/

AuthenticationTypeEnum type() default AuthenticationTypeEnum.VISITOR;

}

5.3 用户上下文操作类

---

package com.nobody.context;

import com.nobody.enums.AuthenticationTypeEnum;

import com.nobody.exception.RestAPIError;

import com.nobody.exception.RestException;

import org.springframework.beans.factory.annotation.Autowired;

import org.springframework.stereotype.Component;

import org.springframework.util.StringUtils;

import javax.servlet.http.Cookie;

import javax.servlet.http.HttpServletRequest;

import javax.servlet.http.HttpServletResponse;

import java.util.Objects;

import java.util.UUID;

/**

• @Description 用户上下文操作类

• @Author Mr.nobody

• @Date 2020/10/25

• @Version 1.0

*/

@Component

public class UserContextManager {

private static final String COOKIE_KEY = “__userToken”;

// @Autowired

// private RedisService redisService;

/**

• 获取用户上下文信息

• @param request

• @return

*/

public UserContext getUserContext(HttpServletRequest request) {

String userToken = getUserToken(request, COOKIE_KEY);

if (!StringUtils.isEmpty(userToken)) {

// 从缓存或者第三方获取用户信息

// String userContextStr = redisService.getString(userToken);

// if (!StringUtils.isEmpty(userContextStr)) {

// return JSON.parseObject(userContextStr, UserContext.class);

// }

// 因为演示，没集成Redis，故简单new对象

UserContext userContext = new UserContext();

userContext.setName(“Mr.nobody”);

userContext.setUserId(“0000001”);

userContext.setType(AuthenticationTypeEnum.ADMIN);

return userContext;

}

throw new RestException(RestAPIError.AUTH_ERROR);

}

public String getUserToken(HttpServletRequest request, String cookieKey) {

Cookie[] cookies = request.getCookies();

if (null != cookies) {

for (Cookie cookie : cookies) {

if (Objects.equals(cookie.getName(), cookieKey)) {

return cookie.getValue();

}

}

}

return null;

}

/**

• 保存用户上下文信息

• @param response

• @param userContextStr

*/

public void saveUserContext(HttpServletResponse response, String userContextStr) {

// 用户token实际根据自己业务进行生成，此处简单用UUID

String userToken = UUID.randomUUID().toString();

// 设置cookie

Cookie cookie = new Cookie(COOKIE_KEY, userToken);

cookie.setPath(“/”);

response.addCookie(cookie);

// redis缓存

// redisService.setString(userToken, userContextStr, 3600);

}

}

5.4 方法参数解析器类

---

package com.nobody.context;

import lombok.extern.slf4j.Slf4j;

import org.springframework.beans.factory.annotation.Autowired;

import org.springframework.core.MethodParameter;

import org.springframework.stereotype.Component;

import org.springframework.web.bind.support.WebDataBinderFactory;

import org.springframework.web.context.request.NativeWebRequest;

import org.springframework.web.method.support.HandlerMethodArgumentResolver;

import org.springframework.web.method.support.ModelAndViewContainer;

import javax.servlet.http.HttpServletRequest;

/**

• @Description 对有UserContext参数的接口，进行拦截注入用户信息

• @Author Mr.nobody

• @Date 2020/10/25

• @Version 1.0

*/

@Component

@Slf4j

public class UserContextResolver implements HandlerMethodArgumentResolver {

@Autowired

private UserContextManager userContextManager;

@Override

public Object resolveArgument(MethodParameter parameter, ModelAndViewContainer mavContainer,

NativeWebRequest webRequest, WebDataBinderFactory binderFactory) {

log.info(“>>> resolveArgument – begin…”);

HttpServletRequest request = webRequest.getNativeRequest(HttpServletRequest.class);

// 从缓存获取用户信息赋值到接口参数中

return userContextManager.getUserContext(request);

}

/**

• 只对UserContext参数进行拦截赋值

• @param methodParameter

• @return

*/

@Override

public boolean supportsParameter(MethodParameter methodParameter) {

if (methodParameter.getParameterType().equals(UserContext.class)) {

return true;

}

return false;

}

}

六、测试验证

=================================================================

package com.nobody.controller;

import com.alibaba.fastjson.JSON;

import com.nobody.annotation.UserAuthenticate;

import com.nobody.context.UserContext;

import com.nobody.context.UserContextManager;

import com.nobody.enums.AuthenticationTypeEnum;

import com.nobody.pojo.model.GeneralResult;

import org.springframework.beans.factory.annotation.Autowired;

import org.springframework.web.bind.annotation.GetMapping;

import org.springframework.web.bind.annotation.RequestMapping;

import org.springframework.web.bind.annotation.RestController;

import javax.servlet.http.HttpServletResponse;

/**

• @Description

• @Author Mr.nobody

• @Date 2020/10/25

• @Version 1.0

*/

@RestController

自我介绍一下，小编13年上海交大毕业，曾经在小公司待过，也去过华为、OPPO等大厂，18年进入阿里一直到现在。

深知大多数Java工程师，想要提升技能，往往是自己摸索成长或者是报班学习，但对于培训机构动则几千的学费，着实压力不小。自己不成体系的自学效果低效又漫长，而且极易碰到天花板技术停滞不前！

因此收集整理了一份《2024年Java开发全套学习资料》，初衷也很简单，就是希望能够帮助到想自学提升又不知道该从何学起的朋友，同时减轻大家的负担。

既有适合小白学习的零基础资料，也有适合3年以上经验的小伙伴深入学习提升的进阶课程，基本涵盖了95%以上Java开发知识点，真正体系化！

由于文件比较大，这里只是将部分目录大纲截图出来，每个节点里面都包含大厂面经、学习笔记、源码讲义、实战项目、讲解视频，并且后续会持续更新

如果你觉得这些内容对你有帮助，可以添加V获取：vip1024b （备注Java）

最后

针对最近很多人都在面试，我这边也整理了相当多的面试专题资料，也有其他大厂的面经。希望可以帮助到大家。

上述的面试题答案都整理成文档笔记。 也还整理了一些面试资料&最新2021收集的一些大厂的面试真题（都整理成文档，小部分截图）

以上就是本文的全部内容，希望对大家的学习有所帮助，也希望大家多多支持。

一个人可以走的很快，但一群人才能走的更远。如果你从事以下工作或对以下感兴趣，欢迎戳这里加入程序员的圈子，让我们一起学习成长！

AI人工智能、Android移动开发、AIGC大模型、C C#、Go语言、Java、Linux运维、云计算、MySQL、PMP、网络安全、Python爬虫、UE5、UI设计、Unity3D、Web前端开发、产品经理、车载开发、大数据、鸿蒙、计算机网络、嵌入式物联网、软件测试、数据结构与算法、音视频开发、Flutter、IOS开发、PHP开发、.NET、安卓逆向、云计算

存中…(img-0MrhDoju-1712455290039)]

上述的面试题答案都整理成文档笔记。 也还整理了一些面试资料&最新2021收集的一些大厂的面试真题（都整理成文档，小部分截图）

[外链图片转存中…(img-RcpqyRAH-1712455290039)]

以上就是本文的全部内容，希望对大家的学习有所帮助，也希望大家多多支持。

一个人可以走的很快，但一群人才能走的更远。如果你从事以下工作或对以下感兴趣，欢迎戳这里加入程序员的圈子，让我们一起学习成长！

AI人工智能、Android移动开发、AIGC大模型、C C#、Go语言、Java、Linux运维、云计算、MySQL、PMP、网络安全、Python爬虫、UE5、UI设计、Unity3D、Web前端开发、产品经理、车载开发、大数据、鸿蒙、计算机网络、嵌入式物联网、软件测试、数据结构与算法、音视频开发、Flutter、IOS开发、PHP开发、.NET、安卓逆向、云计算