首先阐述一下作者遇到的使用场景:一次项目开发中,rpc调用使用的是dubbo,因为在本地联调时没有用户登录,所以开发的过程中忽略了用户登录信息的获取,当所有接口都开发完成后,发现不少接口是需要获取用户登录信息的,当时我们有两个微服务,在这我称为 A 服务和 B 服务,A 调用 B 服务。因为框架的原因,用户信息只能在 A 获取,在 B 服务直接获取不到,如果这个时候在 A 服务获取到用户信息后,通过传参的方式把用户信息传递到 B 服务,这样做也可以,只是接口多的话工作量也大很多,用户信息对原有接口侵入性也比较强,耦合性也比较高。所以查了资料后,作者使用了以下代码解决了dubbo服务直接调用的变量传递。
以下代码是项目中二次封装好的原代码:
/**
* @Description
* @Author: jinglonglong
* @Date:2021-6-9
**/
public class RequestUtils {
public static HttpServletRequest request() {
ServletRequestAttributes attributes = (ServletRequestAttributes) RequestContextHolder.getRequestAttributes();
return attributes.getRequest();
}
//获取用户信息
public static EmployeePO getUser(){
//从RpcContext中获取
String user = RpcContext.getContext().getAttachment("user");
if(StringUtils.isNotBlank(user)){
return JSON.parseObject(user,EmployeePO.class);
}
return null;
}
//set用户信息
public static void setUser(){
//设置到RpcContext中
EmployeePO employeePO = SessionUtils.getEmployeePO(request());
RpcContext.getContext().setAttachment("user",JSON.toJSONString(employeePO));
}
}
真正使用到的是代码是:
//上游服务存信息
RpcContext.getContext().setAttachment("user",object);
//下游服务取信息
RpcContext.getContext().getAttachment("user");
原理其实是使用到了 ThreadLocal 。