PLT-191: enable mapVertexToAtlasEntityHeader2

atlanhq · Nov 15, 2023 · 3858e4f · 3858e4f
1 parent 169e5fb
commit 3858e4f
Show file tree

Hide file tree

Showing 2 changed files with 2 additions and 2 deletions.
diff --git a/repository/src/main/java/org/apache/atlas/discovery/EntityDiscoveryService.java b/repository/src/main/java/org/apache/atlas/discovery/EntityDiscoveryService.java
@@ -1132,7 +1132,7 @@ private void prepareSearchResult2(AtlasSearchResult ret, DirectIndexQueryResult
     }
 
     private AtlasEntityHeader enrichVertex(AtlasSearchResult ret, Set<String> resultAttributes, Result result, AtlasVertex vertex, boolean fetchCollapsedResults) throws AtlasBaseException {
-        AtlasEntityHeader header = entityRetriever.toAtlasEntityHeader(vertex, resultAttributes);
+        AtlasEntityHeader header = entityRetriever.toAtlasEntityHeader2(vertex, resultAttributes);
         if(RequestContext.get().includeClassifications()){
             header.setClassifications(entityRetriever.getAllClassifications(vertex));
         }

diff --git a/...sitory/src/main/java/org/apache/atlas/repository/store/graph/v2/EntityGraphRetriever.java b/...sitory/src/main/java/org/apache/atlas/repository/store/graph/v2/EntityGraphRetriever.java
@@ -1034,7 +1034,7 @@ private AtlasEntityHeader mapVertexToAtlasEntityHeader2(AtlasVertex entityVertex
         AtlasEntityType entityType = typeRegistry.getEntityTypeByName(typeName);
 
         if (entityType != null) {
-            entityType.getHeaderAttributes().values().forEach(headerAttribute->{
+            entityType.getHeaderAttributes().values().stream().parallel().forEach(headerAttribute->{
                 try {
                     Object attrValue = getVertexAttribute(entityVertex, headerAttribute);
                     if (attrValue != null) {