Java实现一个简单的RPC框架(六) 注册机制

最新推荐文章于 2024-08-11 22:39:05 发布

v_xchen_v

最新推荐文章于 2024-08-11 22:39:05 发布

阅读量1.7k

点赞数

分类专栏： C++入门到放弃设计模式

本文链接：https://blog.csdn.net/v_xchen_v/article/details/53634264

版权

本文探讨了在Java中实现简单RPC框架时如何引入服务注册机制，以避免每次新增RPC方法都需要修改核心处理函数的问题。通过创建一个包含方法映射的字典，可以根据请求的command字段找到并调用相应的函数。

摘要由CSDN通过智能技术生成

就是Server要声明支持那些RPC方法，比如我们之前Server支持‘sayHello’和‘add’方法，其它的方法就无法识别。

我们是之前把注册功能写在了procRequest函数中，但是这样写的话，我么每次新增一个rpc方法的时候就需要修改这个函数，如何避免？引入服务注册机制。

String procRequest(String reqStr)
	{
		
		//return "Hello World";
		mRequest req =decode(reqStr);
		mResponse rsp = new mResponse();
		rsp.id = req.id;
		if(req.command.equals("sayHello"))
		{
			rsp.result = "Hello";
		}else if(req.command.equals("add"))
		{
			rsp.result = Integer.toString(req.arg1+req.arg2);
		}
		else
		{
			System.out.println("req command"+req.command);
			rsp.result = "unknown command";
		}		
		return encode(rsp);
	}

服务注册机制，就是将command与具体的function object绑定起来，生成第一dict。

｛‘sayHello’:sayHello,'add':add｝

有了dict，每次收到req，只要提取出command字段，就可以从dict中找出对应的function，调用该function即可。

Java代码实现：

Client.java不变

import java.io.BufferedReader;
import java.io.BufferedWriter;
import java.io.IOException;
import java.io.InputStreamReader;
import java.io.PrintWriter;
import java.net.Socket;
import java.net.UnknownHostException;

public class Client {
	public static void main(String[] args) {
		mClient client = new mClient();
		client.setReqId(1);
		client.setReqCommand("sayHello");
		String reqStr = client.req.getRequest();
		try {
			Socket socket = new Socket("localhost",12345);
			PrintWriter out = new PrintWriter(socket.getOutputStream());
			//将reqStr传给server
			out.println(reqStr);
			out.flush();