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