From 23404f9c35899dd4c9170b400fcc41edd4decd9a Mon Sep 17 00:00:00 2001
From: Aditya Radhakrishnan <aditya@acryl.io>
Date: Mon, 14 Mar 2022 11:09:05 -0700
Subject: [PATCH] fix(containers) - configure domain resolver for containers
 (#4404)

---
 .../java/com/linkedin/datahub/graphql/GmsGraphQLEngine.java   | 4 ++++
 1 file changed, 4 insertions(+)

diff --git a/datahub-graphql-core/src/main/java/com/linkedin/datahub/graphql/GmsGraphQLEngine.java b/datahub-graphql-core/src/main/java/com/linkedin/datahub/graphql/GmsGraphQLEngine.java
index 4e381201f9..a079be8327 100644
--- a/datahub-graphql-core/src/main/java/com/linkedin/datahub/graphql/GmsGraphQLEngine.java
+++ b/datahub-graphql-core/src/main/java/com/linkedin/datahub/graphql/GmsGraphQLEngine.java
@@ -514,6 +514,10 @@ public class GmsGraphQLEngine {
             .type("Container", typeWiring -> typeWiring
                 .dataFetcher("relationships", new EntityRelationshipsResultResolver(graphClient))
                 .dataFetcher("entities", new ContainerEntitiesResolver(entityClient))
+                .dataFetcher("domain", new LoadableTypeResolver<>(domainType, (env) -> {
+                    final Container container = env.getSource();
+                    return container.getDomain() != null ? container.getDomain().getUrn() : null;
+                }))
                 .dataFetcher("platform",
                     new LoadableTypeResolver<>(dataPlatformType,
                         (env) -> ((Container) env.getSource()).getPlatform().getUrn()))