support d2 symlink node in indis flow

linkedin · Sep 18, 2023 · a24e6ab · a24e6ab
1 parent 1fce408
commit a24e6ab
Show file tree

Hide file tree

Showing 6 changed files with 229 additions and 22 deletions.
diff --git a/CHANGELOG.md b/CHANGELOG.md
@@ -14,6 +14,9 @@ and what APIs have changed, if applicable.
 
 ## [Unreleased]
 
+## [29.45.2] - 2023-09-18
+- support d2 symlink in indis flow
+
 ## [29.45.1] - 2023-09-05
 - add @Nullable annotations to pegasus java getters and setters with mode
 
@@ -5527,7 +5530,8 @@ patch operations can re-use these classes for generating patch messages.
 
 ## [0.14.1]
 
-[Unreleased]: https://github.com/linkedin/rest.li/compare/v29.45.1...master
+[Unreleased]: https://github.com/linkedin/rest.li/compare/v29.45.2...master
+[29.45.2]: https://github.com/linkedin/rest.li/compare/v29.45.1...v29.45.2
 [29.45.1]: https://github.com/linkedin/rest.li/compare/v29.45.0...v29.45.1
 [29.45.0]: https://github.com/linkedin/rest.li/compare/v30.0.0...v29.45.0
 [30.0.0]: https://github.com/linkedin/rest.li/compare/v29.44.0...v30.0.0

diff --git a/d2/src/main/java/com/linkedin/d2/xds/XdsClient.java b/d2/src/main/java/com/linkedin/d2/xds/XdsClient.java
@@ -24,6 +24,7 @@
 public abstract class XdsClient
 {
   private static final String D2_NODE_TYPE_URL = "type.googleapis.com/indis.D2Node";
+  private static final String D2_SYMLINK_NODE_TYPE_URL = "type.googleapis.com/indis.D2SymlinkNode";
   private static final String D2_NODE_MAP_TYPE_URL = "type.googleapis.com/indis.D2NodeMap";
 
   interface ResourceWatcher
@@ -44,6 +45,11 @@ interface D2NodeResourceWatcher extends ResourceWatcher
     void onChanged(D2NodeUpdate update);
   }
 
+  interface D2SymlinkNodeResourceWatcher extends ResourceWatcher
+  {
+    void onChanged(String resourceName, D2SymlinkNodeUpdate update);
+  }
+
   interface D2NodeMapResourceWatcher extends ResourceWatcher
   {
     void onChanged(D2NodeMapUpdate update);
@@ -76,6 +82,28 @@ public String getVersion()
     }
   }
 
+  static final class D2SymlinkNodeUpdate implements ResourceUpdate
+  {
+    String _version;
+    XdsD2.D2SymlinkNode _nodeData;
+
+    D2SymlinkNodeUpdate(String version, XdsD2.D2SymlinkNode nodeData)
+    {
+      _version = version;
+      _nodeData = nodeData;
+    }
+
+    XdsD2.D2SymlinkNode getNodeData()
+    {
+      return _nodeData;
+    }
+
+    public String getVersion()
+    {
+      return _version;
+    }
+  }
+
   static final class D2NodeMapUpdate implements ResourceUpdate
   {
     String _version;
@@ -100,14 +128,19 @@ public String getVersion()
 
   enum ResourceType
   {
-    UNKNOWN, D2_NODE, D2_NODE_MAP;
+    // TODO: add D2_SYMLINK_NODE type
+    UNKNOWN, D2_NODE, D2_SYMLINK_NODE, D2_NODE_MAP;
 
     static ResourceType fromTypeUrl(String typeUrl)
     {
       if (typeUrl.equals(D2_NODE_TYPE_URL))
       {
         return D2_NODE;
       }
+      if (typeUrl.equals(D2_SYMLINK_NODE_TYPE_URL))
+      {
+        return D2_SYMLINK_NODE;
+      }
       if (typeUrl.equals(D2_NODE_MAP_TYPE_URL))
       {
         return D2_NODE_MAP;
@@ -121,6 +154,8 @@ String typeUrl()
       {
         case D2_NODE:
           return D2_NODE_TYPE_URL;
+        case D2_SYMLINK_NODE:
+          return D2_SYMLINK_NODE_TYPE_URL;
         case D2_NODE_MAP:
           return D2_NODE_MAP_TYPE_URL;
         case UNKNOWN:

diff --git a/d2/src/main/java/com/linkedin/d2/xds/XdsClientImpl.java b/d2/src/main/java/com/linkedin/d2/xds/XdsClientImpl.java
@@ -51,6 +51,7 @@ public class XdsClientImpl extends XdsClient
   private static final Logger _log = LoggerFactory.getLogger(XdsClientImpl.class);
 
   private final Map<String, ResourceSubscriber> _d2NodeSubscribers = new HashMap<>();
+  private final Map<String, ResourceSubscriber> _d2SymlinkNodeSubscribers = new HashMap<>();
   private final Map<String, ResourceSubscriber> _d2NodeMapSubscribers = new HashMap<>();
 
   private final Node _node;
@@ -143,6 +144,28 @@ private void handleD2NodeResponse(DiscoveryResponseData data)
     handleResourceUpdate(updates, data.getResourceType(), data.getNonce(), errors);
   }
 
+  private void handleD2SymlinkNodeResponse(DiscoveryResponseData data)
+  {
+    Map<String, D2SymlinkNodeUpdate> updates = new HashMap<>();
+    List<String> errors = new ArrayList<>();
+
+    for (Resource resource: data.getResourcesList())
+    {
+      String resourceName = resource.getName();
+      try
+      {
+        XdsD2.D2SymlinkNode symlinkNode = resource.getResource().unpack(XdsD2.D2SymlinkNode.class);
+        updates.put(resourceName, new D2SymlinkNodeUpdate(resource.getVersion(), symlinkNode));
+      } catch (InvalidProtocolBufferException e)
+      {
+        _log.warn("Failed to unpack D2SymlinkNode response", e);
+        errors.add("Failed to unpack D2SymlinkNode response");
+      }
+    }
+
+    handleResourceUpdate(updates, data.getResourceType(), data.getNonce(), errors);
+  }
+
   private void handleD2NodeMapResponse(DiscoveryResponseData data)
   {
     Map<String, D2NodeMapUpdate> updates = new HashMap<>();
@@ -214,6 +237,8 @@ private Map<String, ResourceSubscriber> getResourceSubscriberMap(ResourceType ty
     {
       case D2_NODE:
         return _d2NodeSubscribers;
+      case D2_SYMLINK_NODE:
+        return _d2SymlinkNodeSubscribers;
       case D2_NODE_MAP:
         return _d2NodeMapSubscribers;
       case UNKNOWN:
@@ -257,6 +282,9 @@ private void notifyWatcher(ResourceWatcher watcher, ResourceUpdate update)
         case D2_NODE:
           ((D2NodeResourceWatcher) watcher).onChanged((D2NodeUpdate) update);
           break;
+        case D2_SYMLINK_NODE:
+          ((D2SymlinkNodeResourceWatcher) watcher).onChanged(_resource, (D2SymlinkNodeUpdate) update);
+          break;
         case D2_NODE_MAP:
           ((D2NodeMapResourceWatcher) watcher).onChanged((D2NodeMapUpdate) update);
           break;
@@ -519,11 +547,15 @@ private void handleResponse(DiscoveryResponseData response)
       _responseReceived = true;
       String respNonce = response.getNonce();
       ResourceType resourceType = response.getResourceType();
+      // TODO: handle D2_SYMLINK_NODE type
       switch (resourceType)
       {
         case D2_NODE:
           handleD2NodeResponse(response);
           break;
+        case D2_SYMLINK_NODE:
+          handleD2SymlinkNodeResponse(response);
+          break;
         case D2_NODE_MAP:
           handleD2NodeMapResponse(response);
           break;