Improve code readability using Stream

Signed-off-by: David Osorno <[email protected]>
opensearch-project · Mar 13, 2024 · 0f1792b · 0f1792b
1 parent 46c296f
commit 0f1792b
Show file tree

Hide file tree

Showing 2 changed files with 13 additions and 22 deletions.
diff --git a/src/main/java/org/opensearch/security/dlic/rest/api/MultiTenancyConfigApiAction.java b/src/main/java/org/opensearch/security/dlic/rest/api/MultiTenancyConfigApiAction.java
@@ -12,13 +12,13 @@
 package org.opensearch.security.dlic.rest.api;
 
 import java.io.IOException;
-import java.util.ArrayList;
 import java.util.List;
 import java.util.Map;
 import java.util.Objects;
 import java.util.Optional;
 import java.util.Set;
 import java.util.stream.Collectors;
+import java.util.stream.IntStream;
 
 import com.google.common.collect.ImmutableList;
 import com.google.common.collect.ImmutableMap;
@@ -216,28 +216,19 @@ private void updateAndValidatesValues(final ConfigV7 config, final JsonNode json
     }
 
     private List<DashboardSignInOption> getNewSignInOptions(JsonNode newOptions, Authc authc) {
-        List<DashboardSignInOption> options = new ArrayList<>();
-        for (int i = 0; i < newOptions.size(); i++) {
-            try {
-                String option = newOptions.get(i).asText();
-                if (isOptionConfiguredAtBackEnd(authc, option)) {
-                    options.add(DashboardSignInOption.valueOf(option));
-                }
-            } catch (Exception e) {
-                throw new IllegalArgumentException("Invalid sign-in option: " + e.getMessage());
-            }
-        }
-        return options;
-    }
 
-    private boolean isOptionConfiguredAtBackEnd(Authc authc, String option) {
-        for (String key : authc.getDomains().keySet()) {
-            if (key.contains(option.toLowerCase())) {
+        Set<String> domains = authc.getDomains().keySet();
+
+        return IntStream.range(0, newOptions.size()).mapToObj(newOptions::get).map(JsonNode::asText).filter(option -> {
+            // Checking if the new sign-in options are set in backend.
+            if (option.equals(DashboardSignInOption.ANONYMOUS.toString())
+                || domains.stream().anyMatch(domain -> domain.contains(option.toLowerCase()))) {
                 return true;
+            } else {
+                throw new IllegalArgumentException(
+                    "Validation failure: " + option.toUpperCase() + " authentication provider is not available for this cluster."
+                );
             }
-        }
-        throw new IllegalArgumentException(
-            "Validation failure: " + option.toUpperCase() + " authentication provider is not available for this cluster."
-        );
+        }).map(DashboardSignInOption::valueOf).collect(Collectors.toList());
     }
 }
diff --git a/src/test/java/org/opensearch/security/dlic/rest/api/MultiTenancyConfigApiTest.java b/src/test/java/org/opensearch/security/dlic/rest/api/MultiTenancyConfigApiTest.java
@@ -207,7 +207,7 @@ private void verifyTenantUpdateFailed(final Header... header) throws Exception {
         assertThat(
             invalidSignInOption.getBody(),
             invalidSignInOption.findValueInJson("error.reason"),
-            containsString("Invalid sign-in option")
+            containsString("authentication provider is not available for this cluster")
         );
     }