Working Flow Framework APIs except Search

Signed-off-by: Daniel Widdis <[email protected]>
opensearch-project · Apr 1, 2024 · b7384f2 · b7384f2
1 parent f493cfe
commit b7384f2
Show file tree

Hide file tree

Showing 21 changed files with 686 additions and 289 deletions.
diff --git a/src/main/java/org/opensearch/flowframework/FlowFrameworkPlugin.java b/src/main/java/org/opensearch/flowframework/FlowFrameworkPlugin.java
@@ -26,6 +26,7 @@
 import org.opensearch.env.Environment;
 import org.opensearch.env.NodeEnvironment;
 import org.opensearch.flowframework.common.FlowFrameworkSettings;
+import org.opensearch.flowframework.indices.DynamoDbUtil.DDBClient;
 import org.opensearch.flowframework.indices.FlowFrameworkIndicesHandler;
 import org.opensearch.flowframework.rest.RestCreateWorkflowAction;
 import org.opensearch.flowframework.rest.RestDeleteWorkflowAction;
@@ -112,9 +113,11 @@ public Collection<Object> createComponents(
         Settings settings = environment.settings();
         flowFrameworkSettings = new FlowFrameworkSettings(clusterService, settings);
         MachineLearningNodeClient mlClient = new MachineLearningNodeClient(client);
-        EncryptorUtils encryptorUtils = new EncryptorUtils(clusterService, client);
+        DDBClient ddbClient = new DDBClient(client);
+        EncryptorUtils encryptorUtils = new EncryptorUtils(clusterService, client, ddbClient);
         FlowFrameworkIndicesHandler flowFrameworkIndicesHandler = new FlowFrameworkIndicesHandler(
             client,
+            ddbClient,
             clusterService,
             encryptorUtils,
             xContentRegistry
@@ -134,7 +137,14 @@ public Collection<Object> createComponents(
             flowFrameworkSettings
         );
 
-        return List.of(workflowStepFactory, workflowProcessSorter, encryptorUtils, flowFrameworkIndicesHandler, flowFrameworkSettings);
+        return List.of(
+            workflowStepFactory,
+            workflowProcessSorter,
+            encryptorUtils,
+            flowFrameworkIndicesHandler,
+            flowFrameworkSettings,
+            ddbClient
+        );
     }
 
     @Override

diff --git a/src/main/java/org/opensearch/flowframework/indices/DynamoDBDemo.java b/src/main/java/org/opensearch/flowframework/indices/DynamoDBDemo.java