thread pool information to graphql layer

dagster-io · Jan 6, 2025 · 2050fcc · 2050fcc
1 parent 5ff2171
commit 2050fcc
Show file tree

Hide file tree

Showing 8 changed files with 1,017 additions and 391 deletions.
diff --git a/js_modules/dagster-ui/packages/ui-core/src/graphql/schema.graphql b/js_modules/dagster-ui/packages/ui-core/src/graphql/schema.graphql
diff --git a/js_modules/dagster-ui/packages/ui-core/src/graphql/types.ts b/js_modules/dagster-ui/packages/ui-core/src/graphql/types.ts
diff --git a/python_modules/dagster-graphql/dagster_graphql/schema/asset_graph.py b/python_modules/dagster-graphql/dagster_graphql/schema/asset_graph.py
@@ -240,6 +240,7 @@ class GrapheneAssetNode(graphene.ObjectType):
     backfillPolicy = graphene.Field(GrapheneBackfillPolicy)
     changedReasons = graphene.Field(non_null_list(GrapheneAssetChangedReason))
     computeKind = graphene.String()
+    pools = non_null_list(graphene.String)
     configField = graphene.Field(GrapheneConfigTypeField)
     dataVersion = graphene.Field(graphene.String(), partition=graphene.String())
     dataVersionByPartition = graphene.Field(
@@ -671,6 +672,9 @@ def _get_config_type(key: str):
             field_snap=node_def_snap.config_field_snap,
         )
 
+    def resolve_pools(self, _graphene_info: ResolveInfo) -> Sequence[str]:
+        return sorted([pool for pool in self._asset_node_snap.pools or set()])
+
     def resolve_computeKind(self, _graphene_info: ResolveInfo) -> Optional[str]:
         return self._asset_node_snap.compute_kind
 

diff --git a/python_modules/dagster-graphql/dagster_graphql/schema/solids.py b/python_modules/dagster-graphql/dagster_graphql/schema/solids.py
@@ -468,6 +468,7 @@ def resolve_assetNodes(self, graphene_info: ResolveInfo) -> Sequence["GrapheneAs
 class GrapheneSolidDefinition(graphene.ObjectType, ISolidDefinitionMixin):
     config_field = graphene.Field(GrapheneConfigTypeField)
     required_resources = non_null_list(GrapheneResourceRequirement)
+    pool = graphene.String()
 
     class Meta:
         interfaces = (GrapheneISolidDefinition,)
@@ -501,6 +502,9 @@ def resolve_required_resources(
             GrapheneResourceRequirement(key) for key in self._solid_def_snap.required_resource_keys
         ]
 
+    def resolve_pool(self, _graphene_info: ResolveInfo) -> Optional[str]:
+        return self._solid_def_snap.pool
+
 
 class GrapheneSolidStepStatsUnavailableError(graphene.ObjectType):
     class Meta: