java 单机接口限流处理方案
3035
2022-10-23
教你用Springboot实现拦截器获取header内容
分析
既然需要动态获取那么只有两种方式:要么每次下游请求过来时从请求头中获取,要么定义统一的拦截器自动获取。
实现
那么我们就先来实现一下吧。
第一种比较简单,直接使用springboot获取请求头的方式,从controller方法入口处使用: @RequestHeader(value = "xxxx",required = false) String appUser的方式获取请求头
代码如下:
@RequestMapping(name = "获取用户详情信息",value = "/getUserDetail",method=GET)
public String getUserDetail(@RequestHeader(value = "app-user",required = false) String appUser, @RequestParam(name = "search") String search){
log.info("------>拿到的请求头是:"+appUser);
return "SUCCESS";
}
此种方法虽然可以获取,但是太过low b,而且如果有几十个接口那么都要这么写,过于累赘
还是使用拦截器的方式吧
具体如下:
首先,定义一个拦截器,需要实现HandlerInterceptor,其中包含preHandler以及afterCompletion分别对应调用前以及调用完成后
package com.xxxx.cc.admin.interceptor;
import com.alibaba.fastjson.JSONObject;
import com.ddmc.cc.admin.support.ThreadLocalUtils;
import lombok.extern.slf4j.Slf4j;
import org.apache.commons.lang.StringUtils;
import org.springframework.stereotype.Component;
import org.springframework.web.servlet.HandlerInterceptor;
import javax.servlet.http.HttpServletRequest;
import javax.servlet.http.HttpServletResponse;
/**
* Copyright (C), 2017-2021
* Author: sjugg
* Date: 2021/5/19 10:39
* FileName: AuthInterceptor
* Description: 交易前置拦截器
*/
@Component
@Slf4j
public class AuthInfoInterceptor implements HandlerInterceptor {
@Override
public boolean preHandle(HttpServletRequest request, HttpServletResponse response, Object handler) throws Exception {
String userInfo = request.getHeader("app-user");
log.info(">>>>>>>拦截到api相关请求头<<<<<<<<"+userInfo);
if(StringUtils.isNotEmpty(userInfo)){
//直接搂下来,放到ThreadLocal中 后续直接从中获取
ThreadLocalUtils.set(MaiCaiApiConstants.MAI_CAI_APP_USER,userInfo);
}
return true;//注意 这里必须是true否则请求将就此终止。
}
@Override
public void afterCompletion(HttpServletRequest request, HttpServletResponse response, Object handler, Exception ex) throws Exception {
//移除app-user
ThreadLocalUtils.remove("app-user");
log.info("移除请求头中的app-user:"+ThreadLocalUtils.get("app-user"));
}
}
后面只要从ThreadLocal中按key去拿就可以了。
另外需要做的是
添加web过滤器
1、添加web 过滤器,管理过滤的类,以及过滤的请求路径
2、继承WebMvcConfigurationSupport 类 和上面一样Generate,复写父类的方法 addInterceptors
3、通过registry.addInterceptor拦截类,addPathPatterns拦截路径,excludePathPatterns添加允许路径
这里我拦截的是以xxxx开头的请求。
package com.ddmc.cc.admin.config;
import com.ddmc.cc.admin.interceptor.AuthInfoInterceptor;
import org.springframework.beans.factory.annotation.Autowired;
import org.springframework.web.servlet.config.annotation.InterceptorRegistry;
import org.springframework.web.servlet.config.annotation.WebMvcConfigurer;
/**
*
*
* @author sjugg
* @since 2021/5/13 4:43 下午
*/
@Configuration
public class WebConfig implements WebMvcConfigurer {
@Autowired
private AuthInfoInterceptor authInfoInterceptor;
@Override
public void addInterceptors(InterceptorRegistry registry) {
registry.addInterceptor(authInfoInterceptor).addPathPatterns("/xxxx/*");
}
}
执行效果如下:
请求头里设置app-user信息
可以看到Controller中已经拿到请求头里的信息
获取head的代码:
@RequestMapping(name = "获取用户详情信息",value = "/getUserDetail",method=GET)
public String getUhttp://serDetail(@RequestParam(name = "search") String search){
log.info("-----> heard info here: "+ThreadLocalUtils.get("app-user"));
return "SUCCESS";
}
方法执行完之后对应的threadlocal变量也被清除
版权声明:本文内容由网络用户投稿,版权归原作者所有,本站不拥有其著作权,亦不承担相应法律责任。如果您发现本站中有涉嫌抄袭或描述失实的内容,请联系我们jiasou666@gmail.com 处理,核实后本网站将在24小时内删除侵权内容。
发表评论
暂时没有评论,来抢沙发吧~