nacos 注册服务源码分析

Nacos 注册服务源码涉及的主要类有以下几个：

1. NamingService：命名服务接口，定义了注册、注销、查询服务的方法。

2. NamingServiceImpl：命名服务的实现类，实现了 NamingService 接口。

3. ServerListManager：服务列表管理器，负责管理服务列表。

4. Instance：服务实例对象，包含了服务实例的基本信息。

5. Service：服务对象，包含了服务的基本信息和服务实例列表。

6. ServiceInfo：服务信息对象，包含了所有服务和服务实例的信息。

7. ServerProxy：服务代理类，负责与服务注册中心通信。

下面对这些类进行详细的分析。

1. NamingService

命名服务接口，定义了注册、注销、查询服务的方法。

public interface NamingService {
    /**
     * 注册一个服务实例
     */
    void registerInstance(String serviceName, Instance instance) throws NacosException;

    /**
     * 注销一个服务实例
     */
    void deregisterInstance(String serviceName, Instance instance) throws NacosException;

    /**
     * 查询一个服务的所有实例
     */
    List<Instance> getAllInstances(String serviceName) throws NacosException;

    /**
     * 监听一个服务的实例变化
     */
    void subscribe(String serviceName, EventListener listener) throws NacosException;

    /**
     * 取消监听一个服务的实例变化
     */
    void unsubscribe(String serviceName, EventListener listener) throws NacosException;
}

2. NamingServiceImpl

命名服务的实现类，实现了 NamingService 接口。

public class NamingServiceImpl implements NamingService {
    private ServerProxy serverProxy;

    public NamingServiceImpl(ServerProxy serverProxy) {
        this.serverProxy = serverProxy;
    }

    @Override
    public void registerInstance(String serviceName, Instance instance) throws NacosException {
        serverProxy.registerService(serviceName, instance);
    }

    @Override
    public void deregisterInstance(String serviceName, Instance instance) throws NacosException {
        serverProxy.deregisterService(serviceName, instance);
    }

    @Override
    public List<Instance> getAllInstances(String serviceName) throws NacosException {
        return serverProxy.getAllInstances(serviceName);
    }

    @Override
    public void subscribe(String serviceName, EventListener listener) throws NacosException {
        serverProxy.subscribe(serviceName, listener);
    }

    @Override
    public void unsubscribe(String serviceName, EventListener listener) throws NacosException {
        serverProxy.unsubscribe(serviceName, listener);
    }
}

3. ServerListManager

服务列表管理器，负责管理服务列表。

public class ServerListManager {
    private DiscoveryConfig config;
    private List<ServerInfo> serverList = new ArrayList<>();
    private AtomicInteger index = new AtomicInteger(0);

    public ServerListManager(DiscoveryConfig config) {
        this.config = config;
        initServerList();
    }

    private void initServerList() {
        // 读取配置文件中的服务地址列表
        String serverListStr = config.getServerList();
        String[] serverArr = serverListStr.split(",");
        for (int i = 0; i < serverArr.length; i++) {
            String[] serverInfoArr = serverArr[i].split(":");
            String ip = serverInfoArr[0];
            int port = Integer.parseInt(serverInfoArr[1]);
            ServerInfo serverInfo = new ServerInfo(ip, port);
            serverList.add(serverInfo);
        }
    }

    public ServerInfo getNextServer() {
        // 轮询获取服务地址
        int i = index.getAndIncrement() % serverList.size();
        return serverList.get(i);
    }
}

4. Instance

服务实例对象，包含了服务实例的基本信息。

public class Instance {
    private String serviceName;
    private String ip;
    private int port;

    public Instance(String serviceName, String ip, int port) {
        this.serviceName = serviceName;
        this.ip = ip;
        this.port = port;
    }

    public String getServiceName() {
        return serviceName;
    }

    public void setServiceName(String serviceName) {
        this.serviceName = serviceName;
    }

    public String getIp() {
        return ip;
    }

    public void setIp(String ip) {
        this.ip = ip;
    }

    public int getPort() {
        return port;
    }

    public void setPort(int port) {
        this.port = port;
    }
}

5. Service

服务对象，包含了服务的基本信息和服务实例列表。

public class Service {
    private String name;
    private List<Instance> instances = new ArrayList<>();

    public Service(String name) {
        this.name = name;
    }

    public String getName() {
        return name;
    }

    public List<Instance> getInstances() {
        return instances;
    }

    public void addInstance(Instance instance) {
        instances.add(instance);
    }

    public void removeInstance(Instance instance) {
        instances.remove(instance);
    }
}

6. ServiceInfo

服务信息对象，包含了所有服务和服务实例的信息。

public class ServiceInfo {
    private Map<String, Service> serviceMap = new HashMap<>();

    public void addService(Service service) {
        serviceMap.put(service.getName(), service);
    }

    public void removeService(Service service) {
        serviceMap.remove(service.getName());
    }

    public List<Service> getServices() {
        return new ArrayList<>(serviceMap.values());
    }
}

7. ServerProxy

服务代理类，负责与服务注册中心通信。

public class ServerProxy {
    private ServerListManager serverListManager;

    public ServerProxy(ServerListManager serverListManager) {
        this.serverListManager = serverListManager;
    }

    public void registerService(String serviceName, Instance instance) throws NacosException {
        ServerInfo serverInfo = serverListManager.getNextServer();
        try (Socket socket = new Socket(serverInfo.getIp(), serverInfo.getPort())) {
            // 向注册中心发送注册请求
            // ...
        } catch (IOException e) {
            // 处理异常
        }
    }

    public void deregisterService(String serviceName, Instance instance) throws NacosException {
        // 向注册中心发送注销请求
        // ...
    }

    public List<Instance> getAllInstances(String serviceName) throws NacosException {
        // 从注册中心获取服务实例列表
        // ...
        return instances;
    }

    public void subscribe(String serviceName, EventListener listener) throws NacosException {
        // 向注册中心发送订阅请求
        // ...
    }

    public void unsubscribe(String serviceName, EventListener listener) throws NacosException {
        // 向注册中心发送取消订阅请求
        // ...
    }
}

以上就是 Nacos 注册服务源码分析的内容。