RPC客户端如何实现-KRPC源码解析

最新推荐文章于 2021-08-02 20:20:49 发布

weixin_33873846

最新推荐文章于 2021-08-02 20:20:49 发布

阅读量321

点赞数

文章标签：网络 netty java

原文链接：https://juejin.im/post/5b1359f35188257d4849cbae

版权

1.前言

这篇文章主要结合KRPC（我自己开源的一个RPC框）代码详细分析一下RPC客户端具体实现。在一篇文章了解RPC框架原理文中，我们主要讲述了一次调用RPC调用中各流程，这篇文章就结合KRPC的代码仔细讲解一下

开始前，我先说一下KRPC的网络传输中的内容： 1.服务实现名字。server端需要你服务实现的名字，才能知道你调用的是哪个实现的方法，跟web项目中的controller写的路径一样。

2.方法名字。知道了调用的是哪个类，接下来就需要调用的哪个方法了，无需多言。

3.方法参数名字。KRPC会获取每个方法参数的class的全路径（原因在下面展开讲）

4.方法中传入的值。

2.源码分析

2.1 客户端端初始化

KRPC在调用时，必须要先执行

KRPC.init("/opt/krpc/client/client.xml");
复制代码

该方法内部解析该配置文件，把配置文件中的服务及其地址解析后并放入内存缓存中，供后面TCP请求时，快速获取到服务地址。

2.2 动态代理

可以看到，我们调用的都是接口，并且我们没有引入接口的实现包，调用后怎么能获取到server端的数据呢？

这就引入了动态代理，由代理类替我们处理接口调用的动作。这里使用的是jdk的动态代理实现方式。

调用者接口代理获取方法

UserService service = ProxyFactory.create(UserService.class, "user", "userService");
复制代码

ProxyFactory#create方法

	public static <T> T  create(Class<?> type, String serviceName,String serviceImpleName) {
	ProxyHandler handler = new ProxyHandler(serviceName,serviceImpleName);
	return (T) handler.bind(new Class<?>[]{type});
复制代码

ProxyHandler.java

@Override
	public Object invoke(Object proxy, Method method, Object[] args) throws Throwable {

		// 构造请求request
		Request request = new Request();
		request.setMethodName(method.getName());
		request.setServiceImplName(serviceImplName);
		request.setParamsValues(Arrays.asList(args));
		Class[] sourceTypes = method.getParameterTypes();
		List<String> paramsTypeName = new ArrayList<String>();
		
		for (int i = 0; i < args.length; i++) {
			paramsTypeName.add(sourceTypes[i].getName());
		}
		request.setParamsTypesName(paramsTypeName);

		Class returnClass = method.getReturnType();

		return RequestHandler.request(serviceName, request, returnClass);
	}

复制代码

上面放了3段代码，是动态代理相关的全部代理，是的，就这么简单。通过ProxyFactory将接口绑定，获取接口代理，这样调用接口中的方法，直接交给ProxyHandler#invoke方法来执行。

invoke中主要是构建网络请求的参数Request类，然后调用请求控制类request。

我们在前言中有讲到Request的内容。这里说明一下对于参数Class为什么传递类的全路径名字，而不是Class类。

1.因为服务端的类是通过URLClassLoader动态加载进来的，客户端直接传递客户端这边的Class会报ClassNotFound异常

2.传递起类路径数据长度更小

服务端只需要根据类名，通过Class.forname加载进来就ok了。

2.3. 序列化

序列化是RPC框架中重要的一个环结，KRPC采用了Hessian序列化方式，common中的序列化工具

因为在服务端的各service的类都是动态加载进来的

public static Object deserialize(byte[] by, ClassLoader classLoader) throws IOException {
		if (by == null)
			throw new NullPointerException();

		ByteArrayInputStream is = new ByteArrayInputStream(by);
		ClassLoader old = null;
		if (classLoader != null) {
			old = Thread.currentThread().getContextClassLoader();
			// 切换当前线程classloader，保证动态加载的类不会报CNF
			Thread.currentThread().setContextClassLoader(classLoader);
		}
		HessianInput hi = new HessianInput(is);
		Object obj = hi.readObject();

		if (classLoader != null) {
			Thread.currentThread().setContextClassLoader(old);
		}
		return obj;
	}

复制代码

所以在凡序列中，server端需要传入动态加载的classLoader。 Hessian在凡序列化时，会获取当前线程的ClassLoader，所以我们在外面修改了当前线程的classloader（这也是迭代的目标，这样做有些不稳妥）。

同时我也引入了压缩功能，这样让传输的字节更少。

2.4.TCP请求

传输的数据准备好了，也从配置文件中获取到服务的地址，接下来就要进行TCP请求了。

public static byte[] send(byte[] sendData,String host,int port,int timeout) throws UnknownHostException, IOException {
		Socket socket = new Socket(host,port);
		socket.setSoTimeout(timeout);
		
		OutputStream os = socket.getOutputStream();
		InputStream is = socket.getInputStream();
		byte resultArray[] = null;
		try {
			os.write(sendData);
			os.flush();
			socket.shutdownOutput();
			resultArray = IOUtils.toByteArray(is);
		} catch (Exception e) {
			e.printStackTrace();
		} finally {
			os.close();
			is.close();
			socket.close();
		}
		return resultArray;
	}

复制代码