RPC框架的最简单实现

一个最简单的RPC框架实现包含暴露服务、引用服务、Socket通讯三部分。

RPC框架简易实现

public class RpcFramework { public static void exportService(final Object service, int port) throws Exception { ServerSocket serverSocket = new ServerSocket(port); while (true) { final Socket socket = serverSocket.accept(); new Thread(new Runnable() { @Override public void run() { ObjectInputStream reader = null; ObjectOutputStream writer = null; try { reader = new ObjectInputStream(socket.getInputStream()); String methodName = reader.readUTF(); Class[] argumentsType = (Class[]) reader.readObject(); Object[] arguments = (Object[]) reader.readObject(); Method method = service.getClass().getMethod(methodName, argumentsType); Object result = method.invoke(service, arguments); writer = new ObjectOutputStream(socket.getOutputStream()); writer.writeObject(result); } catch (Exception e) { if (null != writer) { try { writer.writeObject(e); } catch (IOException e1) { e1.printStackTrace(); } } } finally { if (null != writer) { try { writer.close(); } catch (IOException e) { e.printStackTrace(); } } if (null != reader) { try { reader.close(); } catch (IOException e) { e.printStackTrace(); } } } } }).start(); } } public static <T> T referenceService(Class<T> interfaceClass, final String host, final int port) { return (T) Proxy.newProxyInstance(interfaceClass.getClassLoader(), new Class[]{interfaceClass}, new InvocationHandler() { @Override public Object invoke(Object proxy, Method method, Object[] args) throws Throwable { Socket socket = new Socket(host, port); ObjectOutputStream writer = null; ObjectInputStream reader = null; try { writer = new ObjectOutputStream(socket.getOutputStream()); writer.writeUTF(method.getName()); writer.writeObject(method.getParameterTypes()); writer.writeObject(args); reader = new ObjectInputStream(socket.getInputStream()); return reader.readObject(); } finally { if (null != writer) { writer.close(); } if (null != reader) { reader.close(); } } } }); } }

接口

public interface HelloService { String sayHello(String name); } public interface ByeService { String sayBye(String name); }

接口实现

public class HelloServiceImpl implements HelloService { @Override public String sayHello(String name) { return name + " say hello"; } } public class ByeServiceImpl implements ByeService { @Override public String sayBye(String name) { return name +" bye bye"; } }

服务提供方

public class Provider { public static void main(String[] args) throws Exception { final HelloService helloService = new HelloServiceImpl(); new Thread(new Runnable() { @Override public void run() { try { RpcFramework.exportService(helloService, 20880); } catch (Exception e) { e.printStackTrace(); } } }).start(); final ByeService byeService = new ByeServiceImpl(); new Thread(new Runnable() { @Override public void run() { try { RpcFramework.exportService(byeService, 20881); } catch (Exception e) { e.printStackTrace(); } } }).start(); } }

服务消费方

public class Consumer { public static void main(String[] args) { HelloService helloService = RpcFramework.referenceService(HelloService.class, "127.0.0.1", 20880); System.out.println(helloService.sayHello("tom")); ByeService byeService = RpcFramework.referenceService(ByeService.class, "127.0.0.1", 20881); System.out.println(byeService.sayBye("jack")); } }

输出

tom say hello jack bye bye