update service cache when receiving update service event

sinkedship · Dec 10, 2021 · 30f3b2c · 30f3b2c
1 parent b4659b1
commit 30f3b2c
Showing 1 changed file with 14 additions and 1 deletion.
diff --git a/...rus-registry/src/main/java/com/sinkedship/cerberus/registry/k8s/K8sServiceDiscoverer.java b/...rus-registry/src/main/java/com/sinkedship/cerberus/registry/k8s/K8sServiceDiscoverer.java
@@ -138,7 +138,20 @@ public void onAdd(V1Service service) {
             }
 
             @Override
-            public void onUpdate(V1Service oldObj, V1Service newObj) {
+            public void onUpdate(V1Service oldSvc, V1Service newSvc) {
+                if (newSvc.getMetadata() != null && newSvc.getSpec() != null &&
+                        newSvc.getSpec().getPorts() != null) {
+                    String svcName = newSvc.getMetadata().getName();
+                    String ip = newSvc.getSpec().getClusterIP();
+                    for (V1ServicePort port : newSvc.getSpec().getPorts()) {
+                        String portName = port.getName();
+                        K8sServiceMetaData metaData = new K8sServiceMetaData(svcName, portName);
+                        cache.put(metaData, new CerberusService.Builder(Object.class)
+                                .metaData(metaData).host(ip).port(port.getPort())
+                                .build());
+                        LOGGER.debug("update svc, meta-data:{} with ip:{}, port:{}", metaData, ip, port.getPort());
+                    }
+                }
             }
 
             @Override