Address review comments - replace for loop with stream (JabRef#4539)

shubhamatlani · Dec 27, 2018 · 4a8b3c8 · 4a8b3c8
1 parent 2fa906b
commit 4a8b3c8
Showing 1 changed file with 4 additions and 6 deletions.
diff --git a/src/main/java/org/jabref/model/groups/WordKeywordGroup.java b/src/main/java/org/jabref/model/groups/WordKeywordGroup.java
@@ -1,14 +1,14 @@
 package org.jabref.model.groups;
 
 import java.util.ArrayList;
-import java.util.Arrays;
 import java.util.Collection;
 import java.util.Collections;
 import java.util.HashSet;
 import java.util.List;
 import java.util.Objects;
 import java.util.Optional;
 import java.util.Set;
+import java.util.stream.Collectors;
 
 import org.jabref.model.EntryTypes;
 import org.jabref.model.FieldChange;
@@ -119,11 +119,9 @@ public boolean contains(BibEntry entry) {
     private Set<String> getFieldContentAsWords(BibEntry entry) {
         if (onlySplitWordsAtSeparator) {
             if (BibEntry.TYPE_HEADER.equals(searchField)) {
-                for (String searchWord : searchWords) {
-                    Optional<EntryType> entryType = EntryTypes.getType(entry.getType(), BibDatabaseMode.BIBLATEX);
-                    if (entryType.isPresent() && entryType.get().getName().equals(searchWord)) {
-                        return new HashSet<>(Arrays.asList(searchWord));
-                    }
+                Optional<EntryType> entryType = EntryTypes.getType(entry.getType(), BibDatabaseMode.BIBLATEX);
+                if (entryType.isPresent()) {
+                    return searchWords.stream().filter(sw -> entryType.get().getName().equals(sw)).collect(Collectors.toSet());
                 }
             }
             return entry.getField(searchField)