ChrisSamo632 commented on code in PR #6544:
URL: https://github.com/apache/nifi/pull/6544#discussion_r1014601052


##########
nifi-nar-bundles/nifi-elasticsearch-bundle/nifi-elasticsearch-restapi-processors/src/main/java/org/apache/nifi/processors/elasticsearch/ElasticsearchRestProcessor.java:
##########
@@ -117,13 +123,66 @@ default String getQuery(final FlowFile input, final 
ProcessContext context, fina
     }
 
     default Map<String, String> getUrlQueryParameters(final ProcessContext 
context, final FlowFile flowFile) {
+        return getUrlQueryParameters(context, flowFile != null ? 
flowFile.getAttributes() : null);
+    }
+
+    default Map<String, String> getUrlQueryParameters(final ProcessContext 
context, final Map<String, String> attributes) {
         return context.getProperties().entrySet().stream()
                 // filter non-null dynamic properties
                 .filter(e -> e.getKey().isDynamic() && e.getValue() != null)
                 // convert to Map of URL parameter keys and values
                 .collect(Collectors.toMap(
-                    e -> e.getKey().getName(),
-                    e -> 
context.getProperty(e.getKey()).evaluateAttributeExpressions(flowFile).getValue()
+                        e -> e.getKey().getName(),
+                        e -> 
context.getProperty(e.getKey()).evaluateAttributeExpressions(attributes).getValue()
                 ));
     }
+
+    String VERIFICATION_STEP_INDEX_EXISTS = "Elasticsearch Index Exists";
+
+    @Override
+    default List<ConfigVerificationResult> verify(final ProcessContext 
context, final ComponentLog verificationLogger, final Map<String, String> 
attributes) {
+        final List<ConfigVerificationResult> results = new ArrayList<>();
+        final ConfigVerificationResult.Builder indexExistsResult = new 
ConfigVerificationResult.Builder()
+                .verificationStepName(VERIFICATION_STEP_INDEX_EXISTS);
+
+        ElasticSearchClientService verifyClientService = null;
+        String index = null;
+        boolean indexExists = false;
+        if (context.getProperty(CLIENT_SERVICE).isSet()) {
+            verifyClientService = 
context.getProperty(CLIENT_SERVICE).asControllerService(ElasticSearchClientService.class);
+            if (context.getProperty(INDEX).isSet()) {
+                index = 
context.getProperty(INDEX).evaluateAttributeExpressions(attributes).getValue();
+                try {
+                    if (verifyClientService.exists(index, 
getUrlQueryParameters(context, attributes))) {
+                        
indexExistsResult.outcome(ConfigVerificationResult.Outcome.SUCCESSFUL)
+                                .explanation(String.format("Index [%s] 
exists", index));
+                        indexExists = true;
+                    } else {
+                        
indexExistsResult.outcome(ConfigVerificationResult.Outcome.SUCCESSFUL)

Review Comment:
   Correct, I've tried to describe this further in 
https://github.com/apache/nifi/pull/6544#discussion_r1014601005



-- 
This is an automated message from the Apache Git Service.
To respond to the message, please log on to GitHub and use the
URL above to go to the specific comment.

To unsubscribe, e-mail: issues-unsubscr...@nifi.apache.org

For queries about this service, please contact Infrastructure at:
us...@infra.apache.org

Reply via email to