（六）SpringCloud系列——关于openfeign服务调用中上游数据的传递解决方案

前言

在微服务应用中，服务间的相互调用是必不可少的，openfeign是我们最常用到的服务调用组件，在实际的开发应用过程中，例如我们的微服务间相互调用会有一些鉴权的操作，从而保证我们服务的安全性，这样我们A服务调用B服务的时候也需要将A服务的一些授权信息传递给B服务，从而保证B服务的调用也可以通过授权，这样保证了整个调用链的安全。本小节我们简单介绍一下openfeign实现上游数据传递的解决方案。主要通过一个openfeign的拦截器RequestInterceptor实现。

正文

说明：通过RequestInterceptor 拦截器拦截我们的openfeign服务请求，将上游服务的请求头或者请求体中的数据封装到我们的openfeign调用的请求模板中，从而实现上游数据的传递。

package com.yundi.aiyundun.config;

import feign.RequestInterceptor;
import feign.RequestTemplate;
import lombok.extern.slf4j.Slf4j;
import org.springframework.context.annotation.Configuration;
import org.springframework.web.context.request.RequestContextHolder;
import org.springframework.web.context.request.ServletRequestAttributes;

import javax.servlet.http.HttpServletRequest;
import java.util.Enumeration;

/**
 * @Author: yanp
 * @Description: openfeign 拦截器:实现上游请求头和请求参数的传递
 * @Date: 2021/6/15 14:53
 * @Version: 1.0.0
 */
@Slf4j
@Configuration
public class OpenFeignRequestInterceptor implements RequestInterceptor {

    @Override
    public void apply(RequestTemplate requestTemplate) {
        ServletRequestAttributes attributes = (ServletRequestAttributes) RequestContextHolder.getRequestAttributes();
        HttpServletRequest request = attributes.getRequest();
        //1.获取上游请求头参数并封装到openfeign接口调用中
        Enumeration<String> headerNames = request.getHeaderNames();
        if (headerNames != null) {
            while (headerNames.hasMoreElements()) {
                String name = headerNames.nextElement();
                String value = request.getHeader(name);
                requestTemplate.header(name, value);
            }
        }
        //2.获取上游请求body参数并封装到openfeign接口调用中
        Enumeration<String> bodyNames = request.getParameterNames();
        StringBuffer body = new StringBuffer();
        if (bodyNames != null) {
            while (bodyNames.hasMoreElements()) {
                String name = bodyNames.nextElement();
                String value = request.getParameter(name);
                body.append(name).append("=").append(value).append("&");
            }
        }
        if (body.length() != 0) {
            body.deleteCharAt(body.length() - 1);
            requestTemplate.body(body.toString());
            log.info("openfeign interceptor body:{}", body.toString());
        }
    }
}

结语

关于openfeign服务调用中上游数据的传递解决方案就这里了，希望能对你有所帮助，我们下期见。。。