add option field

Signed-off-by: Frank Lou <mloufra@amazon.com>
opensearch-project · Dec 6, 2022 · 1ae4284 · 1ae4284
1 parent 6de8201
commit 1ae4284
Show file tree

Hide file tree

Showing 3 changed files with 21 additions and 14 deletions.
diff --git a/server/src/main/java/org/opensearch/extensions/DiscoveryExtension.java b/server/src/main/java/org/opensearch/extensions/DiscoveryExtension.java
@@ -32,7 +32,7 @@
 public class DiscoveryExtension extends DiscoveryNode implements Writeable, ToXContentFragment {
 
     private final PluginInfo pluginInfo;
-    private List<ExtensionDenpendency> dependencies;
+    private List<ExtensionDependency> dependencies;
 
     public DiscoveryExtension(
         String name,
@@ -44,7 +44,7 @@ public DiscoveryExtension(
         Map<String, String> attributes,
         Version version,
         PluginInfo pluginInfo,
-        List<ExtensionDenpendency> dependencies
+        List<ExtensionDependency> dependencies
     ) {
         super(name, id, ephemeralId, hostName, hostAddress, address, attributes, DiscoveryNodeRole.BUILT_IN_ROLES, version);
         this.pluginInfo = pluginInfo;
@@ -55,7 +55,10 @@ public DiscoveryExtension(
     public void writeTo(StreamOutput out) throws IOException {
         super.writeTo(out);
         pluginInfo.writeTo(out);
-        ((DiscoveryNode) dependencies).writeTo(out);
+        out.writeVInt(dependencies.size());
+        for(ExtensionDependency result : dependencies){
+            result.writeTo(out);
+        }
     }
 
     /**
@@ -67,7 +70,11 @@ public void writeTo(StreamOutput out) throws IOException {
     public DiscoveryExtension(final StreamInput in) throws IOException {
         super(in);
         this.pluginInfo = new PluginInfo(in);
-        this.dependencies = new ArrayList<ExtensionDenpendency>();
+        int size = in.readVInt();
+        dependencies = new ArrayList<>(size);
+        for(int i = 0; i < size; i++){
+            dependencies.add(new ExtensionDependency(in));
+        }
     }
 
     @Override

diff --git a/...arch/extensions/ExtensionDenpendency.java → ...earch/extensions/ExtensionDependency.java b/...arch/extensions/ExtensionDenpendency.java → ...earch/extensions/ExtensionDependency.java
@@ -1,8 +1,6 @@
 package org.opensearch.extensions;
 
 import java.io.IOException;
-import java.util.Collections;
-import java.util.List;
 import java.util.Objects;
 
 import org.opensearch.Version;
@@ -11,31 +9,30 @@
 import org.opensearch.common.io.stream.Writeable;
 
 public class ExtensionDenpendency implements Writeable {
-    public String uniqueId;
-    public Version version;
+    private String uniqueId;
+    private String version;
 
-    public ExtensionDenpendency(String uniqueId, Version version){
+    public ExtensionDenpendency(String uniqueId, String version){
         this.uniqueId = uniqueId;
         this.version = version;
-        List<ExtensionDenpendency> dependencies = Collections.<ExtensionDenpendency>emptyList();
     }
 
     public ExtensionDenpendency(StreamInput in) throws IOException {
         uniqueId = in.readString();
-        version = Version.readVersion(in);
+        version = in.readString();
     }
 
     @Override
     public void writeTo(StreamOutput out) throws IOException {
         out.writeString(uniqueId);
-        Version.writeVersion(version, out);
+        Version.fromString(getVersion());
     }
 
     public String getUniqueId(){
         return uniqueId;
     }
 
-    public Version getVersion(){
+    public String getVersion(){
         return version;
     }
 

diff --git a/server/src/main/java/org/opensearch/extensions/ExtensionsOrchestrator.java b/server/src/main/java/org/opensearch/extensions/ExtensionsOrchestrator.java
@@ -15,6 +15,7 @@
 import java.nio.file.Files;
 import java.nio.file.Path;
 import java.util.ArrayList;
+import java.util.Collections;
 import java.util.HashMap;
 import java.util.List;
 import java.util.Map;
@@ -314,6 +315,8 @@ private void extensionsDiscovery() throws IOException {
                     logger.info("Duplicate uniqueId " + extension.getUniqueId() + ". Did not load extension: " + extension);
                 } else {
                     try {
+                        List<ExtensionDependency> dependenices = Collections.<ExtensionDependency>emptyList();
+                        dependenices.add(new ExtensionDependency(extension.getUniqueId(), extension.getVersion()));
                         DiscoveryExtension discoveryExtension = new DiscoveryExtension(
                             extension.getName(),
                             extension.getUniqueId(),
@@ -334,7 +337,7 @@ private void extensionsDiscovery() throws IOException {
                                 new ArrayList<String>(),
                                 Boolean.parseBoolean(extension.hasNativeController())
                             ),
-                            new ArrayList<ExtensionDenpendency>()
+                            dependenices
                         );
                         extensionIdMap.put(extension.getUniqueId(), discoveryExtension);
                         logger.info("Loaded extension with uniqueId " + extension.getUniqueId() + ": " + extension);