一起写个Dubbo——2. 注册多个服务

最新推荐文章于 2024-09-06 09:10:13 发布

原创

最新推荐文章于 2024-09-06 09:10:13 发布 · 1.3w 阅读

24 ·

CC 4.0 BY-SA版权

文章标签：

#java #rpc

本文原载于我的博客，地址：https://blog.guoziyang.top/archives/63/，项目地址：https://github.com/CN-GuoZiyang/My-RPC-Framework

本文对应的commit为8467b19，完整的项目目录

上一节中，我们使用 JDK 序列化和 Socket 实现了一个最基本的 RPC 框架，服务端测试时是这样的：

public class TestServer {
   
   
    public static void main(String[] args) {
   
   
        HelloService helloService = new HelloServiceImpl();
        RpcServer rpcServer = new RpcServer();
        rpcServer.register(helloService, 9000);
    }
}

在注册完 helloService 后，服务器就自行启动了，也就是说，一个服务器只能注册一个服务，这个设计非常不好（毕竟是简易实现）。这一节，我们就将服务的注册和服务器启动分离，使得服务端可以提供多个服务。

服务注册表

我们需要一个容器，这个容器很简单，就是保存一些本地服务的信息，并且在获得一个服务名字的时候能够返回这个服务的信息。创建一个 ServiceRegistry 接口：

package top.guoziyang.rpc.registry;

public interface ServiceRegistry {
   
   
    <T> void register(T service);
    Object getService(String serviceName);
}

一目了然，一个register注册服务信息，一个getService获取服务信息。

我们新建一个默认的注册表类 DefaultServiceRegistry 来实现这个接口，提供服务注册服务，如下：

public class DefaultServiceRegistry implements ServiceRegistry {
   
   

    private static final Logger logger = LoggerFactory.getLogger(DefaultServiceRegistry.class);

    private final Map<String, Object> serviceMap = new ConcurrentHashMap<>();
    private final Set<String> registeredService = ConcurrentHashMap.newKeySet();

    @Override
    public synchronized <T> void register(T service) {
   
   
        String serviceName = service.getClass().getCanonicalName();
        if(registeredService.contains(serviceName)) return;
        registeredService