Don't fail create workflow on already-consumed OpenSearch REST params

Signed-off-by: Daniel Widdis <[email protected]>
opensearch-project · Feb 27, 2024 · f146098 · f146098
1 parent 2c6dfd5
commit f146098
Showing 1 changed file with 5 additions and 4 deletions.
diff --git a/src/main/java/org/opensearch/flowframework/rest/RestCreateWorkflowAction.java b/src/main/java/org/opensearch/flowframework/rest/RestCreateWorkflowAction.java
@@ -78,18 +78,17 @@ protected RestChannelConsumer prepareRequest(RestRequest request, NodeClient cli
         String workflowId = request.param(WORKFLOW_ID);
         String[] validation = request.paramAsStringArray(VALIDATION, new String[] { "all" });
         boolean provision = request.paramAsBoolean(PROVISION_WORKFLOW, false);
-        final List<String> validCreateParams = List.of(WORKFLOW_ID, VALIDATION, PROVISION_WORKFLOW);
         // If provisioning, consume all other params and pass to provision transport action
         Map<String, String> params = provision
             ? request.params()
                 .keySet()
                 .stream()
-                .filter(k -> !validCreateParams.contains(k))
+                .filter(k -> !request.consumedParams().contains(k))
                 .collect(Collectors.toMap(Function.identity(), request::param))
             : request.params()
                 .entrySet()
                 .stream()
-                .filter(e -> !validCreateParams.contains(e.getKey()))
+                .filter(e -> !request.consumedParams().contains(e.getKey()))
                 .collect(Collectors.toMap(Map.Entry::getKey, Map.Entry::getValue));
         if (!flowFrameworkSettings.isFlowFrameworkEnabled()) {
             FlowFrameworkException ffe = new FlowFrameworkException(
@@ -105,7 +104,9 @@ protected RestChannelConsumer prepareRequest(RestRequest request, NodeClient cli
             params.keySet().stream().forEach(request::param);
             request.content();
             FlowFrameworkException ffe = new FlowFrameworkException(
-                "Only the parameters " + validCreateParams + " are permitted unless the provision parameter is set to true.",
+                "Only the parameters "
+                    + List.of(WORKFLOW_ID, VALIDATION, PROVISION_WORKFLOW)
+                    + " are permitted unless the provision parameter is set to true.",
                 RestStatus.BAD_REQUEST
             );
             return channel -> channel.sendResponse(