教你用Springboot實現(xiàn)攔截器獲取header內容

更新時間：2021年05月21日 16:34:12 作者：鹽城吊霸天

項目中遇到一個需求,對接上游系統(tǒng)是涉及到需要增加請求頭,請求頭的信息是動態(tài)獲取的,需要動態(tài)從下游拿到之后轉給上游,文中非常詳細的介紹了該需求的實現(xiàn),需要的朋友可以參考下

分析

既然需要動態(tài)獲取那么只有兩種方式：要么每次下游請求過來時從請求頭中獲取，要么定義統(tǒng)一的攔截器自動獲取。

實現(xiàn)

那么我們就先來實現(xiàn)一下吧。

第一種比較簡單，直接使用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，而且如果有幾十個接口那么都要這么寫，過于累贅

還是使用攔截器的方式吧

具體如下：

首先，定義一個攔截器，需要實現(xiàn)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中 后續(xù)直接從中獲取
 
           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/*");
 
    }
 
}

執(zhí)行效果如下：

請求頭里設置app-user信息

可以看到Controller中已經(jīng)拿到請求頭里的信息

獲取head的代碼：

@RequestMapping(name = "獲取用戶詳情信息",value = "/getUserDetail",method=GET)
 
public String getUserDetail(@RequestParam(name = "search") String search){
 
    log.info("-----> heard info here: "+ThreadLocalUtils.get("app-user"));
 
    return "SUCCESS";
}