监听nacos

kxs-gateway/src/main/java/com/kxs/gateway/api/config/ServiceChangeNotifier.java

@@ -0,0 +1,65 @@
+package com.kxs.gateway.api.config;
+
+import com.alibaba.nacos.api.common.Constants;
+import com.alibaba.nacos.client.naming.event.InstancesChangeEvent;
+import com.alibaba.nacos.common.notify.NotifyCenter;
+import com.alibaba.nacos.common.notify.listener.Subscriber;
+import com.alibaba.nacos.common.notify.Event;
+import jakarta.annotation.PostConstruct;
+import jakarta.annotation.Resource;
+import lombok.extern.slf4j.Slf4j;
+import org.springframework.stereotype.Component;
+import org.springframework.cache.Cache;
+import org.springframework.cache.CacheManager;
+import org.springframework.cloud.loadbalancer.core.CachingServiceInstanceListSupplier;
+
+/**
+ * 服务变更监听
+ *
+ * @author butterfly
+ * @date 2023-09-24
+ */
+@Slf4j
+@Component
+public class ServiceChangeNotifier extends Subscriber<InstancesChangeEvent> {
+    
+    /**
+     * 由于会有多个类型的 CacheManager bean, 这里的 defaultLoadBalancerCacheManager 名称不可修改
+     */
+    @Resource
+	private CacheManager defaultLoadBalancerCacheManager;
+
+	@PostConstruct
+	public void init() {
+        // 注册当前自定义的订阅者以获取通知
+		NotifyCenter.registerSubscriber(this);
+	}
+
+	@Override
+	public void onEvent(InstancesChangeEvent event) {
+		String serviceName = event.getServiceName();
+        // 使用 dubbo 时包含 rpc 服务类会注册以 providers: 或者 consumers: 开头的服务
+        // 由于不是正式的服务, 这里需要进行排除, 如果未使用 dubbo 则不需要该处理
+		if (serviceName.contains(":")) {
+			return;
+		}
+        // serviceName 格式为 groupName@@name
+        String split = Constants.SERVICE_INFO_SPLITER;
+		if (serviceName.contains(split)) {
+			serviceName = serviceName.substring(serviceName.indexOf(split) + split.length());
+		}
+		log.info("服务上下线: {}", serviceName);
+		// 手动更新服务列表
+        Cache cache = defaultLoadBalancerCacheManager.getCache(
+            CachingServiceInstanceListSupplier.SERVICE_INSTANCE_CACHE_NAME);
+		if (cache != null) {
+			cache.evictIfPresent(serviceName);
+		}
+	}
+
+	@Override
+	public Class<? extends Event> subscribeType() {
+		return InstancesChangeEvent.class;
+	}
+
+}