实现要求：提供RPC上下文，客户端可以透传数据给服务端。
一个应用的基本设计包含几个重要的角色：
- 实体域
- 会话域
- 服务域
实体域就是应用中抽象出来的组件，如Spring中的Bean，Mybatis中的MappedStatement等。会话域代表着一次交互过程，如Mybatis中的SqlSession，而服务域就是组件的功能集，负责会话域和实体域的生命周期管理，如Spring的ApplicationContext，Shiro的SecurityManager等。
现在构造我们这个rpc demo中的会话域角色RpcContext。其主要职责就是负责在一次会话生命周期内，保存、传递相关参数数据。会话中的执行体就是线程了，那么理所应当的RpcContext就应该和当前执行线程绑定。很容易想到了ThreadLocal！
实现如下：
/**
* rpc上下文
*
* @author wqx
*
*/
public class RpcContext {
private static ThreadLocal<Map<String,Object>> context = new ThreadLocal<Map<String,Object> >(){
@Override
protected Map<String,Object> initialValue() {
Map<String,Object> m = new HashMap<String,Object>();
return m;
}
};
public static void addAttribute(String key, Object value){
context.get().put(key,value);
}
public static Object getAttribute(String key){
return context.get().get(key);
}
public static Map<String,Object> getAttributes(){
return Collections.unmodifiableMap(context.get());
}
}
RpcContext中的数据传输载体还是选择RpcRequest，将其包在请求体中即可。如下：
public class RpcRequest implements Serializable
{
/**
*
*/
private static final long serialVersionUID = -7102839100899303105L;
//方法名
private String methodName;
//参数类型
private Class<?>[] parameterTypes;
//参数列表
private Object[] args;
//参数
private Map<String,Object> context;
RpcBuilder中在发送请求前，需要从当前上下文中获取数据。传入RpcRequest对象。如下：
RpcRequest request = new RpcRequest(method.getName(), method.getParameterTypes(),args,RpcContext.getAttributes());
同理，在服务端接收到请求对象RpcRequest之后，需要将RpcRequest中传来的数据和当前上下文进行关联。Handler的run方法修改如下：
public void run() {
try{
ObjectInputStream in = null;
ObjectOutputStream out = null;
RpcResponse response = new RpcResponse();
try {
in = new ObjectInputStream(socket.getInputStream());
out = new ObjectOutputStream(socket.getOutputStream());
Object req = in.readObject();
if(req instanceof RpcRequest){
RpcRequest rpcRequest = (RpcRequest)req;
//关联客户端传来的上下文数据
RpcContext.context.set(rpcRequest.getContext());
Method method = service.getClass().getMethod(rpcRequest.getMethodName(), rpcRequest.getParameterTypes());
//。。。
}
测试：
业务接口增加测试方法：
public interface UserService {
/**
* 上下文测试，透明传输数据
*/
public Map<String,Object> rpcContextTest();
//。。。
}
public class UserServiceImpl implements UserService {
@Override
public Map<String,Object> rpcContextTest() {
Map<String,Object> map = new HashMap<String,Object>();
map.put("server","hahaha");
if(RpcContext.getAttributes() != null )
map.putAll(RpcContext.getAttributes());
return map;
}
//。。。
}
客户端测试代码：
@Test
public void testRpcContext(){
RpcContext.addAttribute("client","huhuhu");
Map<String, Object> res = userService.rpcContextTest();
Assert.assertEquals(res.get("server"),"hahaha");
Assert.assertEquals(res.get("client"),"huhuhu");
}
