Spring Cloud Feign請(qǐng)求添加headers的實(shí)現(xiàn)方式

更新時(shí)間：2023年04月26日 17:05:26 作者：clonechen2021

這篇文章主要介紹了Spring Cloud Feign請(qǐng)求添加headers的實(shí)現(xiàn)方式，具有很好的參考價(jià)值，希望對(duì)大家有所幫助。如有錯(cuò)誤或未考慮完全的地方，望不吝賜教

背景

部門技術(shù)升級(jí)，HttpClient需要升級(jí)Feign調(diào)用，重構(gòu)某一個(gè)資方時(shí)遇到請(qǐng)求需要添加上自定義簽名headers，踩坑后記錄了下來(lái)

方案一

方法上的@RequestMapping注解添加headers信息

@RequestMapping注解的屬性中包含一個(gè)headers數(shù)組，所以嘗試使用，在指定的方法上@RequestMapping注解中添加需要的headers，可以是寫死的，也可以讀取配置，測(cè)試是有效的

同理@RequestMapping一組的@PostMapping，@GetMapping注解等均適用

@FeignClient(name = "server",url = "127.0.0.1:8080")
public interface FeignTest {
? ? @RequestMapping(value = "/test",headers = {"app=test-app","token=${test-app.token}"})
? ? String test();
}

方案二

接口上的@RequestMapping注解添加headers信息

針對(duì)單個(gè)方法可以在方法上的@RequestMapping注解中添加headers，如果同一個(gè)接口中所有的方法都需要同樣的headers時(shí)在方法上加就比較繁瑣了，可以在接口上的@RequestMapping注解中添加headers，使整個(gè)接口的方法均被添加同樣的headers

@FeignClient(name = "server",url = "127.0.0.1:8080")
@RequestMapping(value = "/",headers = {"app=test-app","token=${test-app.token}"})
public interface FeignTest {
? ? @RequestMapping(value = "/test")
? ? String test();
}

方案三

使用@Headers注解添加headers信息

@FeignClient(name = "server",url = "127.0.0.1:8080")
@Headers({"app: test-app","token: ${test-app.token}"})
public interface FeignTest {
? ? @RequestMapping(value = "/test")
? ? String test();
}

查看openfeign官方文檔發(fā)現(xiàn)其使用的是@Headers來(lái)添加headers，測(cè)試發(fā)現(xiàn)并沒(méi)有生效，spring cloud使用了自己的SpringMvcContract來(lái)解析注解，所以需要自己實(shí)現(xiàn)一個(gè)Contract來(lái)實(shí)現(xiàn)對(duì)@Headers注解的支持，具體實(shí)現(xiàn)參照http://www.dbjr.com.cn/article/282530.htm

方案四

自定義RequestInterceptor添加headers信息

feign提供了一個(gè)攔截器接口RequestInterceptor，實(shí)現(xiàn)RequestInterceptor接口就可以實(shí)現(xiàn)對(duì)feign請(qǐng)求的攔截，接口提供了一個(gè)方法apply()，實(shí)現(xiàn)apply()方法

@Component
public class FeignRequestInterceptor implements RequestInterceptor {

? ? @Value("${test-app.token}")
? ? private String token;
? ??
? ? @Override
? ? public void apply(RequestTemplate requestTemplate) {
? ? ? ? requestTemplate.header("app","test-app");//靜態(tài)
? ? ? ? requestTemplate.header("token",token);//讀配置
? ? }
}

實(shí)現(xiàn)apply()方法直接添加header會(huì)攔截所有的請(qǐng)求都加上headers，如果不是所有的feign請(qǐng)求都需要用到不建議此方法

方案五

自定義RequestInterceptor實(shí)現(xiàn)添加動(dòng)態(tài)數(shù)據(jù)到header

以上方案都不適合把動(dòng)態(tài)的數(shù)據(jù)放入headers中，而通常場(chǎng)景下可能經(jīng)常需要把計(jì)算的簽名，用戶id等動(dòng)態(tài)信息設(shè)置到headers，所以還需要一個(gè)更加完善的方案。

方案1/2/3均不能設(shè)置動(dòng)態(tài)的值，方案4可以設(shè)置動(dòng)態(tài)值，但是沒(méi)做到請(qǐng)求的區(qū)分，所以在方案4的基礎(chǔ)上進(jìn)行改進(jìn)得到了較為完善的方案5。

具體實(shí)現(xiàn)如下：

在請(qǐng)求調(diào)用代碼中，獲取到HttpServletRequest對(duì)象，將需要添加到headers中的值封裝成一個(gè)map后放入HttpServletRequest的attribute域中

ServletRequestAttributes attributes = (ServletRequestAttributes) RequestContextHolder.getRequestAttributes();
HttpServletRequest request = Objects.requireNonNull(attributes).getRequest();
String signedMsg = getSignedMsg(reqJson); // 計(jì)算簽名字符串
Map<String, String> reqMap = new HashMap<>();
reqMap.put("content-type", "application/json");//常量字段
reqMap.put("accessKey", accessKey);//常量字段
reqMap.put("signedMsg", signedMsg);//動(dòng)態(tài)計(jì)算/獲取字段
request.setAttribute("customizedRequestHeader", reqMap);

在自定義RequestInterceptor中獲取到HttpServletRequest對(duì)象的attribute域中指定的key，將key對(duì)應(yīng)map中的所有參數(shù)加入到headers。

@Component
public class FeignRequestInterceptor implements RequestInterceptor {

? ? @Override
? ? public void apply(RequestTemplate requestTemplate) {
? ? ? ? ServletRequestAttributes attributes = (ServletRequestAttributes) RequestContextHolder.getRequestAttributes();
? ? ? ? HttpServletRequest request = attributes.getRequest();

? ? ? ? // 設(shè)置自定義header
? ? ? ? // 設(shè)置request中的attribute到header以便轉(zhuǎn)發(fā)到Feign調(diào)用的服務(wù)
? ? ? ? Enumeration<String> reqAttrbuteNames = request.getAttributeNames();
? ? ? ? if (reqAttrbuteNames != null) {
? ? ? ? ? ? while (reqAttrbuteNames.hasMoreElements()) {
? ? ? ? ? ? ? ? String attrName = reqAttrbuteNames.nextElement();
? ? ? ? ? ? ? ? if (!"customizedRequestHeader".equalsIgnoreCase(attrName)) {
? ? ? ? ? ? ? ? ? ? continue;
? ? ? ? ? ? ? ? }
? ? ? ? ? ? ? ? Map<String,String> requestHeaderMap = (Map)request.getAttribute(attrName);
? ? ? ? ? ? ? ? for (Map.Entry<String, String> entry : requestHeaderMap.entrySet()) {
? ? ? ? ? ? ? ? ? ? requestTemplate.header(entry.getKey(), entry.getValue());
? ? ? ? ? ? ? ? }
? ? ? ? ? ? ? ? break;
? ? ? ? ? ? }
? ? ? ? }
? ? }
}