JDK动态代理在RPC框架中的应用

RPC框架中一般都有3个角色：服务提供者、服务消费者和注册中心。服务提供者将服务注册到注册中心，服务消费者从注册中心拉取服务的地址，并根据服务地址向服务提供者发起RPC调用。动态代理在这个RPC调用的过程中有什么作用？对于服务消费者，一般只会依赖服务接口，而服务的具体实现是在服务提供者这一端的，服务消费者和服务提供者分别部署在不同的机器上，服务消费者调用接口中的方法时怎么能够得到结果呢？JDK的动态代理就派上用场了。服务消费者使用JDK的动态代理技术，可以创建接口的代理对象，并在回调函数中将自己要调用的接口名称、方法签名信息通过http或者tcp的方式发送给服务提供者，服务提供者再通过反射的方式调用本地的服务，最后将结果通过http或tcp的方式返回给消费者。

 

下面是一个简单的演示程序：

import java.lang.reflect.InvocationHandler;
import java.lang.reflect.Method;
import java.lang.reflect.Proxy;
import java.util.HashMap;
import java.util.Map;

/**
 * 动态代理在RPC中的使用
 *
 * @author syj
 */
public class JDKProxyTest {

    // ---------------------------  模拟RPC客户端  ---------------------------

    // 客户端依赖服务端的接口(实现类在服务端)
    private static IUserService userService;

    public static void main(String[] args) {
        // 根据接口创建代理对象
        userService = (IUserService) Proxy.newProxyInstance(
                JDKProxyTest.class.getClassLoader(),
                new Class[]{IUserService.class},
                new InvocationHandler() {
                    // 在回调方法模拟进行RPC调用(通过http或者tcp与服务端通信)
                    @Override
                    public Object invoke(Object proxy, Method method, Object[] params) throws Throwable {
                        return rpcInvoke(IUserService.class.getSimpleName(), method.getName(), method.getParameterTypes(), params);
                    }
                }
        );
        // 本地调用
        String result = userService.sayHello("hello");
        System.out.println(">>>> result =" + result);
    }


    // ---------------------------  模拟RPC服务端  ---------------------------

    /**
     * 反射调用
     *
     * @param methodName     方法名称
     * @param parameterTypes 方法参数类型
     * @param parameters     方法参数
     * @return
     */
    private static Object rpcInvoke(String interfaceName, String methodName, Class<?>[] parameterTypes, Object[] parameters) {
        Object result = null;
        try {
            // 根据接口名称从Bean容器中获取Bean实例
            Object serviceBean = beanMap.get(interfaceName);
            // 反射调用
            Class<?> serviceClass = serviceBean.getClass();
            Method method = serviceClass.getMethod(methodName, parameterTypes);
            method.setAccessible(true);
            // 得到调用结果
            result = method.invoke(serviceBean, parameters);
        } catch (Exception e) {
            e.printStackTrace();
        }
        return result;
    }

    // 模拟Bean容器, key为接口名称, value为bean实例
    private static Map<String, Object> beanMap = new HashMap<String, Object>() {{
        put("IUserService", new UserServiceImpl());
    }};
}

接口(服务提供者和服务的消费者都会依赖该接口)：

public interface IUserService {
    String sayHello(String content);
}

服务提供者实现类：

public class UserServiceImpl implements IUserService {
    @Override
    public String sayHello(String content) {
        return content + "::" + System.currentTimeMillis();
    }
}

 

其实，JDK的动态代理技术，不仅可以应用在RPC框架中，也可以应用在所有基于客户端和服务端通信的架构中，比如微服务架构中的注册中心、配置中心、消息中心等。