filter out empty value for entity features (#306)

opendistro-for-elasticsearch · Nov 4, 2020 · 75d564b · 75d564b
1 parent 2b78d10
commit 75d564b
Showing 1 changed file with 5 additions and 3 deletions.
diff --git a/src/main/java/com/amazon/opendistroforelasticsearch/ad/feature/SearchFeatureDao.java b/src/main/java/com/amazon/opendistroforelasticsearch/ad/feature/SearchFeatureDao.java
@@ -793,9 +793,11 @@ public void getFeaturesByEntities(
                     .stream()
                     .filter(agg -> AGG_NAME_TERM.equals(agg.getName()))
                     .flatMap(agg -> ((Terms) agg).getBuckets().stream())
-                    .collect(
-                        Collectors.toMap(Terms.Bucket::getKeyAsString, bucket -> parseBucket(bucket, detector.getEnabledFeatureIds()).get())
-                    );
+                    .collect(Collectors.toMap(Terms.Bucket::getKeyAsString, bucket -> parseBucket(bucket, detector.getEnabledFeatureIds())))
+                    .entrySet()
+                    .stream()
+                    .filter(entry -> entry.getValue().isPresent())
+                    .collect(Collectors.toMap(Entry::getKey, entry -> entry.getValue().get()));
 
                 listener.onResponse(results);
             }, listener::onFailure);