一简单的RPC实例(Java)

    来至于阿里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);
        }
        
    }
}

 

posted @ 2015-05-20 17:16  TomSun*star  阅读(1061)  评论(0编辑  收藏  举报