/* * Licensed to the Apache Software Foundation (ASF) under one or more * contributor license agreements. See the NOTICE file distributed with * this work for additional information regarding copyright ownership. * The ASF licenses this file to You under the Apache License, Version 2.0 * (the "License"); you may not use this file except in compliance with * the License. You may obtain a copy of the License at * * http://www.apache.org/licenses/LICENSE-2.0 * * Unless required by applicable law or agreed to in writing, software * distributed under the License is distributed on an "AS IS" BASIS, * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. * See the License for the specific language governing permissions and * limitations under the License. */ package com.alibaba.dubbo.registry.nacos; import org.apache.dubbo.common.URL; import org.apache.dubbo.common.function.ThrowableFunction; import org.apache.dubbo.registry.client.ServiceDiscovery; import org.apache.dubbo.registry.client.ServiceInstance; import org.apache.dubbo.registry.client.event.listener.ServiceInstancesChangedListener; import com.alibaba.dubbo.registry.nacos.util.NacosNamingServiceUtils; import com.alibaba.nacos.api.naming.NamingService; import com.alibaba.nacos.api.naming.listener.NamingEvent; import com.alibaba.nacos.api.naming.pojo.Instance; import com.alibaba.nacos.api.naming.pojo.ListView; import java.util.Collection; import java.util.LinkedHashSet; import java.util.List; import java.util.Set; import java.util.stream.Collectors; import static com.alibaba.dubbo.registry.nacos.util.NacosNamingServiceUtils.createNamingService; import static com.alibaba.dubbo.registry.nacos.util.NacosNamingServiceUtils.getGroup; import static com.alibaba.dubbo.registry.nacos.util.NacosNamingServiceUtils.toInstance; import static org.apache.dubbo.common.function.ThrowableConsumer.execute; /** * Nacos {@link ServiceDiscovery} implementation * * @see ServiceDiscovery * @since 2.7.5 */ public class NacosServiceDiscovery implements ServiceDiscovery { private String group; private NamingService namingService; @Override public void initialize(URL registryURL) throws Exception { this.namingService = createNamingService(registryURL); this.group = getGroup(registryURL); } @Override public void destroy() { this.namingService = null; } @Override public void register(ServiceInstance serviceInstance) throws RuntimeException { execute(namingService, service -> { Instance instance = toInstance(serviceInstance); service.registerInstance(instance.getServiceName(), group, instance); }); } @Override public void update(ServiceInstance serviceInstance) throws RuntimeException { // TODO: Nacos should support unregister(serviceInstance); register(serviceInstance); } @Override public void unregister(ServiceInstance serviceInstance) throws RuntimeException { execute(namingService, service -> { Instance instance = toInstance(serviceInstance); service.deregisterInstance(instance.getServiceName(), group, instance); }); } @Override public Set<String> getServices() { return ThrowableFunction.execute(namingService, service -> { ListView<String> view = service.getServicesOfServer(0, Integer.MAX_VALUE, group); return new LinkedHashSet<>(view.getData()); }); } @Override public List<ServiceInstance> getInstances(String serviceName) throws NullPointerException { return ThrowableFunction.execute(namingService, service -> service.selectInstances(serviceName, true) .stream().map(NacosNamingServiceUtils::toServiceInstance) .collect(Collectors.toList()) ); } @Override public void addServiceInstancesChangedListener(ServiceInstancesChangedListener listener) throws NullPointerException, IllegalArgumentException { execute(namingService, service -> { service.subscribe(listener.getServiceName(), e -> { // Register Nacos EventListener if (e instanceof NamingEvent) { NamingEvent event = (NamingEvent) e; handleEvent(event, listener); } }); }); } private void handleEvent(NamingEvent event, ServiceInstancesChangedListener listener) { String serviceName = event.getServiceName(); Collection<ServiceInstance> serviceInstances = event.getInstances() .stream() .map(NacosNamingServiceUtils::toServiceInstance) .collect(Collectors.toList()); dispatchServiceInstancesChangedEvent(serviceName, serviceInstances); } }