Dubbo之 RpcContext
RpcContext
RpcContext 是一个临时状态记录器,通过上下文存放当前调用过程中所需的环境信息。
通过RpcContext可以获取 Request、Response、Url等。
注意
当接收到RPC请求,或发起RPC请求时,RpcContext的状态都会变化。
比如:A调B,B再调C,则B机器上,在B调C之前,RpcContext记录的是A调B的信息,在B调C之后,RpcContext记录的是B调C的信息。
ThreadLocal
RpcContext类内部有一个ThreadLocal 对象LOCAL, 用来设置/获取变量。
ThreadLocal 是线程副本变量。每个线程都有自己的副本变量,不会互相干扰,保证线程安全。
源码如下:
private static final ThreadLocal<RpcContext> LOCAL = new ThreadLocal<RpcContext>() {
@Override
protected RpcContext initialValue() {
return new RpcContext();
}
};
getContext() :
注意,RpcContext.getContext() 方法,在新版本中,已被替换为 RpcContext.getServiceContext()。
RpcContext.getServiceContext()
RpcContext类通过 ThreadLocal 对象LOCAL 来获取 RpcContext。
源码如下:
/**
* get context.
*
* @return context
*/
public static RpcContext getContext() {
return LOCAL.get();
}
/**
* remove context.
*
*/
public static void removeContext() {
LOCAL.remove();
}
RpcContext的应用:
- RpcContext 获取 Request:
HttpServletRequest request= (HttpServletRequest) RpcContext.getContext().getRequest();
拿到了 request ,很多参数,也都能通过 request 取到了。
- RpcContext 获取最后一次调用的提供方IP地址:
String serverIP = RpcContext.getContext().getRemoteHost();
- RpcContext 获取 url:
URL url = RpcContext.getContext().getUrl();
- RpcContext 获取 localHost和端口:
String localHost = RpcContext.getContext().getLocalHost();
int port = RpcContext.getContext().getLocalPort();
RpcContext 传递参数:
- RpcContext 传递参数:
在服务提供者,可以set传递参数。
RpcContext.getServiceContext().setAttachment(参数名称,具体值)
- RpcContext 获取参数:
在服务提供者,可以get获取参数。
客户端,如果在请求的 header 设置了参数,也可以通过这种方式获取。可以用 postman 测试一下。
RpcContext.getServiceContext().getAttachment(参数名称)
参考资料:
https://dubbo.apache.org/zh/docs/v2.7/user/examples/context/