rpc调用 java_手撕Java之Rpc调用

参照《分布式服务框架原理与实践-李林锋》的书，

代码调出功能来了。

向更高的服务架构技能进发！

颤抖吧，老IT！

package echorpc;

public interface EchoService {

String echo(String ping);

}

package echorpc;

public class EchoServiceImpl implements EchoService{

@Override

public String echo(String ping) {

return ping != null ? ping + " --> I am ok." : " I am ok.";

}

}

package echorpc;

import java.io.IOException;

import java.io.ObjectInputStream;

import java.io.ObjectOutputStream;

import java.lang.reflect.Method;

import java.net.InetSocketAddress;

import java.net.ServerSocket;

import java.net.Socket;

import java.util.concurrent.Executor;

import java.util.concurrent.Executors;

public class RpcExporter {

static Executor executor =

Executors.newFixedThreadPool(Runtime.getRuntime().availableProcessors());

public static void exporter(String hostName, int port) throws Exception {

ServerSocket server = new ServerSocket();

server.bind(new InetSocketAddress(hostName, port));

try {

while(true) {

executor.execute(new ExporterTask(server.accept()));

}

} finally {

server.close();

}

}

private static class ExporterTask implements Runnable{

Socket client = null;

public ExporterTask(Socket client) {

this.client = client;

}

@Override

public void run() {

ObjectInputStream input = null;

ObjectOutputStream output = null;

try {

input = new ObjectInputStream(client.getInputStream());

String interfaceName = input.readUTF();

Class> service = Class.forName(interfaceName);

String methodName = input.readUTF();

Class>[] parameterTypes = (Class>[])input.readObject();

Object[] arguments = (Object[])input.readObject();

Method method = service.getMethod(methodName, parameterTypes);

Object result = method.invoke(service.newInstance(),arguments);

output = new ObjectOutputStream(client.getOutputStream());

output.writeObject(result);

} catch(Exception e) {

e.printStackTrace();

} finally {

if (output != null)

try {

output.close();

} catch(IOException e) {

e.printStackTrace();

}

if (input != null)

try {

input.close();

} catch(IOException e) {

e.printStackTrace();

}

if (client != null)

try {

client.close();

} catch(IOException e) {

e.printStackTrace();

}

}

}

}

}

package echorpc;

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.InetSocketAddress;

import java.net.Socket;

public class RpcImporter {

@SuppressWarnings("unchecked")

public S importer(final Class> serviceClass, final InetSocketAddress addr) {

return (S) Proxy.newProxyInstance(serviceClass.getClassLoader(),

new Class>[] {serviceClass.getInterfaces()[0]},

new InvocationHandler() {

@Override

public Object invoke(Object proxy, Method method, Object[] args) throws Throwable {

Socket socket = null;

ObjectOutputStream output = null;

ObjectInputStream input = null;

try {

socket = new Socket();

socket.connect(addr);

output = new ObjectOutputStream(socket.getOutputStream());

output.writeUTF(serviceClass.getName());

output.writeUTF(method.getName());

output.writeObject(method.getParameterTypes());

output.writeObject(args);

input = new ObjectInputStream(socket.getInputStream());

return input.readObject();

} finally {

if (socket != null)

socket.close();

if (input != null)

input.close();

if (output != null)

output.close();

}

}

});

}

}

package echorpc;

import java.net.InetSocketAddress;

public class echorpc {

public static void main(String[] args) throws Exception {

// TODO Auto-generated method stub

new Thread(new Runnable() {

@Override

public void run() {

try {

RpcExporter.exporter("localhost", 8088);

} catch (Exception e) {

e.printStackTrace();

}

}

}

).start();

RpcImporter importer = new RpcImporter();

EchoService echo = importer.importer(EchoServiceImpl.class,

new InetSocketAddress("localhost", 8088));

System.out.println(echo.echo("Are you all ok?"));

}

}