fix(graphql) Fix null pointer exception when fetching entity aspects …

…via graphql (#5857)
datahub-project · Sep 7, 2022 · 494f38e · 494f38e
1 parent 42a847c
commit 494f38e
Showing 1 changed file with 7 additions and 6 deletions.
diff --git a/...b-graphql-core/src/main/java/com/linkedin/datahub/graphql/WeaklyTypedAspectsResolver.java b/...b-graphql-core/src/main/java/com/linkedin/datahub/graphql/WeaklyTypedAspectsResolver.java
@@ -74,12 +74,13 @@ public CompletableFuture<List<RawAspect>> get(DataFetchingEnvironment environmen
 
                     DataMap renderSpec = aspectSpec.getRenderSpec();
 
-                    AspectRenderSpec resultRenderSpec = new AspectRenderSpec();
-
-                    resultRenderSpec.setDisplayType(renderSpec.getString("displayType"));
-                    resultRenderSpec.setDisplayName(renderSpec.getString("displayName"));
-                    resultRenderSpec.setKey(renderSpec.getString("key"));
-                    result.setRenderSpec(resultRenderSpec);
+                    if (renderSpec != null) {
+                        AspectRenderSpec resultRenderSpec = new AspectRenderSpec();
+                        resultRenderSpec.setDisplayType(renderSpec.getString("displayType"));
+                        resultRenderSpec.setDisplayName(renderSpec.getString("displayName"));
+                        resultRenderSpec.setKey(renderSpec.getString("key"));
+                        result.setRenderSpec(resultRenderSpec);
+                    }
 
                     results.add(result);
                 } catch (IOException | RemoteInvocationException | URISyntaxException e) {