@@ -370,7 +370,9 @@ private JsonNode searchEntity(JsonNode inputJson, ISearchService service, String
370370 ViewTemplate viewTemplate = viewTemplateManager .getViewTemplate (inputJson );
371371 if (viewTemplate != null ) {
372372 ViewTransformer vTransformer = new ViewTransformer ();
373- resultNode .set (ENTITY_LIST , vTransformer .transform (viewTemplate , resultNode .get (ENTITY_LIST )));
373+ String entityName = resultNode .fields ().next ().getKey ();
374+ ObjectNode transformerInput = JsonNodeFactory .instance .objectNode ().set (entityName , resultNode .get (entityName ).get (ENTITY_LIST ));
375+ ((ObjectNode ) resultNode .get (entityName )).set (ENTITY_LIST , vTransformer .transform (viewTemplate , transformerInput ).get (entityName ));
374376 }
375377 // Search is tricky to support LD. Needs a revisit here.
376378 logger .debug ("searchEntity ends" );
@@ -754,7 +756,9 @@ public JsonNode getAuditLog(JsonNode inputJson, String userId) throws Exception
754756 ViewTemplate viewTemplate = viewTemplateManager .getViewTemplate (inputJson );
755757 if (viewTemplate != null ) {
756758 ViewTransformer vTransformer = new ViewTransformer ();
757- resultNode .set (ENTITY_LIST , vTransformer .transform (viewTemplate , resultNode .get (ENTITY_LIST )));
759+ String entityName = resultNode .fields ().next ().getKey ();
760+ ObjectNode transformerInput = JsonNodeFactory .instance .objectNode ().set (entityName , resultNode .get (entityName ).get (ENTITY_LIST ));
761+ ((ObjectNode ) resultNode .get (entityName )).set (ENTITY_LIST , vTransformer .transform (viewTemplate , transformerInput ).get (entityName ));
758762 }
759763 logger .debug ("get audit log ends" );
760764
0 commit comments