自定义实现一个RPC框架（七）

忘是亡心i 2023-02-13 11:52 2阅读 0赞

### 文章目录 ###

*  自定义实现一个RPC框架（七）：
 *   *   *  地址：https://gitee.com/liyunfengfengfeng/rpc
         *  框架API

# 自定义实现一个RPC框架（七）： #

### 地址：https://gitee.com/liyunfengfengfeng/rpc ###

### 框架API ###

package com.luban.framework;
    
    import com.luban.register.RemoteMapRegister;
    
    import java.lang.reflect.InvocationHandler;
    import java.lang.reflect.Method;
    import java.lang.reflect.Proxy;
    import java.util.List;
    
    public class ProxyFactory<T> { 
    
        @SuppressWarnings("unchecked")
        public static <T> T getProxy(final Class interfaceClass) { 
            return (T)Proxy.newProxyInstance(interfaceClass.getClassLoader(), new Class[]{ interfaceClass}, new InvocationHandler() { 
                @Override
                public Object invoke(Object proxy, Method method, Object[] args) throws Throwable { 
                    String mock = System.getProperty("mock");
                    if (mock != null && mock.startsWith("return:")) { 
                        String result = mock.replace("return:", "");
                        return result;
                    }
    
                    Invocation invocation = new Invocation(interfaceClass.getName(), method.getName(), args, method.getParameterTypes());
                    List<URL> urlList = RemoteMapRegister.get(interfaceClass.getName());
                    URL url = LoadBalance.random(urlList);
                    Protocol protocol = ProtocolFactory.getProtocol();
                    String result = protocol.send(url, invocation);
                    return result;
                }
            });
        }
    
    }

package com.luban.framework;
    
    import java.io.Serializable;
    
    public class URL implements Serializable { 
        private String hostname;
        private Integer port;
    
        public URL(String hostname, Integer port) { 
            this.hostname = hostname;
            this.port = port;
        }
    
        public String getHostname() { 
            return hostname;
        }
    
        public void setHostname(String hostname) { 
            this.hostname = hostname;
        }
    
        public Integer getPort() { 
            return port;
        }
    
        public void setPort(Integer port) { 
            this.port = port;
        }
    }

package com.luban.framework;
    
    import com.luban.protocol.dubbo.DubboProtocol;
    import com.luban.protocol.http.HttpProtocol;
    
    public class ProtocolFactory { 
    
        public static Protocol getProtocol() { 
    
            String name = System.getProperty("protocolName");
            if (name == null || name.equals("")) name = "http";
            switch (name) { 
                case "http":
                    return new HttpProtocol();
                case "dubbo":
                    return new DubboProtocol();
                default:
                    break;
            }
            return new HttpProtocol();
        }
    }

package com.luban.framework;
    
    /** * 在扩展类的 jar 包内 [1]，放置扩展点配置文件 META-INF/dubbo/接口全 * 限定名，内容为：配置名=扩展实现类全限定名，多个实现类用换行符分隔。 * resources * META-INF * services * com.luban.framework.Protocol */
    public interface Protocol { 
    
        void start(URL url);
    
        String send(URL url, Invocation invocation);
    }

package com.luban.framework;
    
    import java.util.Arrays;
    import java.util.List;
    import java.util.Random;
    
    /** * *************书山有路勤为径*************** * 鲁班学院 * 往期资料加木兰老师 QQ: 2746251334 * VIP课程加安其拉老师 QQ: 3164703201 * 讲师：周瑜老师 * *************学海无涯苦作舟*************** */
    public class LoadBalance { 
    
        public static URL random(List<URL> list) { 
            Random random =new Random();
            int n = random.nextInt(list.size());
            return list.get(n);
        }
    
    
    
    }

package com.luban.framework;
    
    import java.io.Serializable;
    
    
    public class Invocation implements Serializable { 
    
        private String interfaceName;
        private String methodName;
        private Object[] params;
        private Class[] paramType;
    
        public Invocation(String interfaceName, String methodName, Object[] params, Class[] paramType) { 
            this.interfaceName = interfaceName;
            this.methodName = methodName;
            this.params = params;
            this.paramType = paramType;
        }
    
        public String getInterfaceName() { 
            return interfaceName;
        }
    
        public void setInterfaceName(String interfaceName) { 
            this.interfaceName = interfaceName;
        }
    
        public String getMethodName() { 
            return methodName;
        }
    
        public void setMethodName(String methodName) { 
            this.methodName = methodName;
        }
    
        public Object[] getParams() { 
            return params;
        }
    
        public void setParams(Object[] params) { 
            this.params = params;
        }
    
        public Class[] getParamType() { 
            return paramType;
        }
    
        public void setParamType(Class[] paramType) { 
            this.paramType = paramType;
        }
    }

package com.luban.comsumer;
    
    import com.luban.framework.Invocation;
    import com.luban.framework.ProxyFactory;
    import com.luban.protocol.http.HttpClient;
    import com.luban.provider.api.HelloService;
    
    public class Consumer { 
    
        public static void main(String[] args) { 
    
            HelloService helloService = ProxyFactory.getProxy(HelloService.class);
            String xxx = helloService.sayHello("xxx");
            System.out.println(xxx);
    
    
    
        }
    }
    
      public static void main(String[] args) { 
    
            HelloService helloService = ProxyFactory.getProxy(HelloService.class);
            String xxx = helloService.sayHello("xxx");
            System.out.println(xxx);
    
    
    
        }
    }