来至于阿里liangf:如有冒犯,请原谅
RPCFrameWork:
package com.sunchao.demo; import java.io.IOException; import java.io.ObjectInputStream; import java.io.ObjectOutputStream; import java.lang.reflect.InvocationHandler; import java.lang.reflect.Method; import java.lang.reflect.Proxy; import java.net.ServerSocket; import java.net.Socket; /** * RPCFramework * * copy from the alibaba liangf * * @author Administrator * */ public class RPCFramework { /**暴露 服务 * * @param service * @param port * @throws IOException */ public static void export(final Object service, int port) throws IOException{ if(service == null) throw new IllegalArgumentException("service instance == null"); if(port <= 0 || port > 65535) throw new IllegalArgumentException("invard port" + port); System.out.println("Export service " + service.getClass().getName() + "on port" + port); @SuppressWarnings("resource") ServerSocket server = new ServerSocket(port); for(;;){ try{ final Socket socket = server.accept(); new Thread(new Runnable(){ @Override public void run() { try{ try{ ObjectInputStream input = new ObjectInputStream( socket.getInputStream()); try{ String methodName = input.readUTF(); //method parameter type Class<?>[] parameterTypes = (Class<?>[]) input.readObject(); //method parameter Object[] arguments = (Object[])input.readObject(); ObjectOutputStream output = new ObjectOutputStream (socket.getOutputStream()); try{ Method method = service.getClass().getMethod (methodName, parameterTypes); Object result = method.invoke(service, arguments); output.writeObject(result); }catch(Throwable t){ output.writeObject(t); }finally{ output.close(); } }finally{ input.close(); } }finally{ socket.close(); } }catch(Exception e){ e.printStackTrace(); } } }).start(); } catch(Exception e1){ e1.printStackTrace(); } } } /** * 引用服务 * * @param interfaceClass 接口类型 * @param host 主机名 * @param port 端口号 * @return 远程服务 */ @SuppressWarnings("unchecked") public static <T> T refer(final Class<T> interfaceClass, final String host, final int port){ if(interfaceClass == null) throw new IllegalArgumentException("Interface class == null"); if(! interfaceClass.isInterface()) throw new IllegalArgumentException ("The " + interfaceClass.getName() + " must be interface class "); if(host == null || host.length() == 0) throw new IllegalArgumentException("host == null"); if(port <= 0 || port > 65535) throw new IllegalArgumentException("Invalid port " + port); System.out.println("Get remote service " + interfaceClass.getName() + " from service " + host + ":" + 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); try{ ObjectOutputStream output = new ObjectOutputStream(socket.getOutputStream()); try{ output.writeUTF(method.getName()); output.writeObject(method.getParameterTypes()); output.writeObject(args); ObjectInputStream input = new ObjectInputStream(socket.getInputStream()); try{ Object result = input.readObject(); if(result instanceof Throwable) throw (Throwable)result; return result; }finally{ input.close(); } }finally{ output.close(); } }finally{ socket.close(); } } }); } }
服务接口:
package com.sunchao.demo; /** * service interface * * @author Administrator * */ public interface HelloService { String hello(String name); }
服务实现:
package com.sunchao.demo; public class HelloServiceImpl implements HelloService { @Override public String hello(String name) { return "hello " + name; } }
服务器端服务启动:
package com.sunchao.demo; import java.io.IOException; /** * 暴漏服务 * * @author Administrator * */ public class RPCProvider { public static void main(String args[]) throws IOException{ HelloService service = new HelloServiceImpl(); RPCFramework.export(service, 1234); } }
客户端服务代理:
package com.sunchao.demo; /** * customer * * @author Administrator * */ public class RPCConsumer { public static void main(String args[]) throws InterruptedException{ HelloService service = RPCFramework.refer (HelloService.class, "10.1.58.54", 1234); for(int i = 0; i < Integer.MAX_VALUE; i++){ String hello = service.hello("world" + i); System.out.println(hello); Thread.sleep(1000); } } }