学习随笔01----一个简单的RPC

简单的手写RPC

基本流程

实现代码

1. 创建Rpc Client

public class RpcClient {
	public static void main(String[] args) {
		User user = (User)getProxy(User.class);
		Object result=user.findbyId(19L);
		System.out.println(result);
	}
	static class RpcHandler implements InvocationHandler{
		String interfaceInfoName;
		
		public RpcHandler(String interfaceInfoName) {
			this.interfaceInfoName = interfaceInfoName;
		}
		
		public Object invoke(Object proxy, Method method, Object[] args) throws Throwable {
			//链接rpc server
			Socket socket=new Socket("127.0.0.1",2020);
			OutputStream outputStream=socket.getOutputStream();
			ObjectOutputStream objectOutputStream=new ObjectOutputStream(outputStream);
			InputStream inputStream=socket.getInputStream();
			ObjectInputStream objectInputStream=new ObjectInputStream(inputStream);
			//将接口名,方法名,方法参数类型,参数发送给服务器
			objectOutputStream.writeUTF(interfaceInfoName);
			objectOutputStream.writeUTF(method.getName());
			objectOutputStream.writeObject(method.getParameterTypes());
			objectOutputStream.writeObject(args);
			//接受结果
			Object result=objectInputStream.readObject();
			return result;
		}
	}
	public static Object getProxy(Class interfaceInfo){
		return Proxy.newProxyInstance(
				interfaceInfo.getClassLoader(), 
				new Class[]{interfaceInfo},
				new RpcHandler(interfaceInfo.getName()));
	}
}

2. 创建Rpc Server

public class RpcServer {
	public static void main(String[] args) throws Exception {
		//接受Client发送的数据
		ServerSocket serverSocket=new ServerSocket(2020);
		System.out.println("start");
		while (true) {
			Socket socket= serverSocket.accept();
			ObjectInputStream objectInputStream=new ObjectInputStream(socket.getInputStream());
			ObjectOutputStream objectOutputStream=new ObjectOutputStream(socket.getOutputStream());
			String interfaceName=objectInputStream.readUTF();
			String MethodName=objectInputStream.readUTF();
			Class[] ParameterType=(Class[])objectInputStream.readObject();
			Object[] Args=(Object[])objectInputStream.readObject();
			
			//根据接口名找到是实现类
			System.out.println(interfaceName);
			//查找当前接口的实现类(因为是简单所以直接找的实现类)
			Class impelClass=UserImpl.class;
			Object object=impelClass.newInstance();
			
			Method method = impelClass.getMethod(MethodName, ParameterType);
			Object result=method.invoke(object, Args);
			objectOutputStream.writeObject(result);
			//关闭
			objectInputStream.close();
			objectOutputStream.close();
		}
	}

}

最后结果

RPC Client控制台

RPC Server控制台