FeignApi通过header头传递tenantId

yuyu888 于 2023-05-08 发布

前言

承接上一章： Mybatis多租户解决方案； 我们的数据有时候需要通过feignapi调用其他服务， 同时要把tenantId传递过去，又不能改变feignapi的调用参数

数据发送

这里是通过RequestInterceptor拦截器实现的

@Slf4j
// @Configuration 可以通过 @FeignClient 中的 configuration指定，而不需要全局生效
public class FeignConfiguration {

	@Autowired
	private TenantContext tenantContext;
	
	@Bean
    public RequestInterceptor headerInterceptor() {
        return new RequestInterceptor() {
            @Override
            public void apply(RequestTemplate template) {
            	String tenantId = tenantContext.getTenantId(); // 这里是获取的 ThreadLocal 的设定好的数据
            	log.info("Feign RequestInterceptor::tenantId:" + tenantId);
                template.header("feign_api_tenant_id", tenantId);
            }
        };
    }
}

接收参数

通过注解+切面的方式，先获取到tenantId 再放到 ThreadLocal 里需要的时候取 这样就不用改feignapi的具体实现了, 具体实现略

从http header 中获取租户id

		HttpServletRequest request = ((ServletRequestAttributes) RequestContextHolder.getRequestAttributes()).getRequest();
		String tenantId = request.getHeader("feign_api_tenant_id");

这里是获取client_service_name 的方法以供参考


import lombok.extern.slf4j.Slf4j;
import org.aspectj.lang.JoinPoint;
import org.aspectj.lang.annotation.Aspect;
import org.aspectj.lang.annotation.Before;
import org.aspectj.lang.annotation.Pointcut;
import org.aspectj.lang.reflect.MethodSignature;
import org.springframework.stereotype.Component;
import org.springframework.web.bind.annotation.GetMapping;
import org.springframework.web.bind.annotation.PostMapping;
import org.springframework.web.bind.annotation.RequestMapping;

import javax.servlet.http.HttpServletRequest;
import java.lang.reflect.Method;

@Slf4j
@Aspect
@Component
public class FeignClientInfoAspect {
    private final HttpServletRequest request;

    public FeignClientInfoAspect(HttpServletRequest request) {
        this.request = request;
    }


    @Pointcut("within(@org.springframework.web.bind.annotation.RestController *) && @within(requestMapping)")
    public void restControllerMethods(RequestMapping requestMapping) {}

    @Before("restControllerMethods(requestMapping)")
    public void before(JoinPoint joinPoint, RequestMapping requestMapping) {
        // Check if the RequestMapping path starts with "/serviceapi/"
        String[] paths = requestMapping.value();
        boolean matches = false;
        for (String path : paths) {
            if (path.startsWith("/serviceapi/")) {
                matches = true;
                break;
            }
        }
        if (matches) {
            MethodSignature methodSignature = (MethodSignature) joinPoint.getSignature();
            Method method = methodSignature.getMethod();

            // Check if the method is annotated with @GetMapping or @PostMapping
            if (method.isAnnotationPresent(GetMapping.class) || method.isAnnotationPresent(PostMapping.class)) {
                String clientServiceName = request.getHeader("client_service_name");
                log.info("本次请求的来源是："+clientServiceName+"； 请求地址为："+String.valueOf(request.getRequestURL()));
            }
        }
    }
}