手撸一个RPC框架——傻瓜式教程（六）

最新推荐文章于 2024-07-09 23:47:49 发布

原创

最新推荐文章于 2024-07-09 23:47:49 发布 · 919 阅读

标签

#java #rpc

本文介绍了如何在自定义RPC框架中实现服务的自动注销功能，通过JVM的ShutdownHook在服务关闭时自动向Nacos注销服务，避免客户端调用失效服务。此外，还探讨了客户端的负载均衡策略，包括随机算法和轮询算法的实现，允许外部传入LoadBalancer进行选择。

RPC框架——傻瓜式教程（六）

今天的内容是自动注销服务和负载均衡策略

自动注销服务和负载均衡策略

自动注销服务

在（五）我们实现了服务的自动注册和发现，但是还是有问题的，如果你启动完成服务端后把服务端给关闭了，并不会自动地注销 Nacos 中对应的服务信息，这样就导致了当客户端再次向 Nacos 请求服务时，会获取到已经关闭的服务端信息，最终就有可能因为连接不到服务器而调用失败。

那么我们就需要一种办法，在服务端关闭之前自动向 Nacos 注销服务。但是有一个问题，我们不知道什么时候服务器会关闭，也就不知道这个方法调用的时机，就没有办法手工去调用。这时，我们就写一个在关闭系统后自动去调用的方法。

首先先写向 Nacos 注销所有服务的方法，这部分被放在了 NacosUtils 中作为一个静态方法，NacosUtils 是一个 Nacos 相关的工具类：
在这里插入图片描述

NacosUtils

   public class NacosUtil {

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

    private static final NamingService namingService;
    private static final Set<String> serviceNames = new HashSet<>();
    private static InetSocketAddress address;

    private static final String SERVER_ADDR = "127.0.0.1:8848";

    static {
        namingService = getNacosNamingService();
    }

    public static NamingService getNacosNamingService() {
        try {
            return NamingFactory.createNamingService(SERVER_ADDR);
        } catch (NacosException e) {
            logger.error("连接到Nacos时有错误发生: ", e);
            throw new RpcException(RpcError.FAILED_TO_CONNECT_TO_SERVICE_REGISTRY);
        }
    }

    public static void registerService(String serviceName, InetSocketAddress address) throws NacosException {
        namingService.registerInstance(serviceName, address.getHostName(), address.getPort());
        NacosUtil.address = address;
        serviceNames.add(serviceName);

    }

    public static List<Instance> getAllInstance

最低0.47元/天解锁文章