fix(BA-596): Add project_node field to GQL UserNode type

changes/3529.feature.md

@@ -0,0 +1 @@
+Add `project_node` field of type GroupConnection to GQL `UserNode` type

docs/manager/graphql-reference/schema.graphql

@@ -732,6 +732,9 @@
   totp_activated: Boolean
   totp_activated_at: DateTime
   sudo_session_enabled: Boolean
+
+  """Added in 25.2.0."""
+  project_nodes(filter: String, order: String, offset: Int, before: String, after: String, first: Int, last: Int): GroupConnection
 }
 
 """Added in 24.03.0"""

docs/manager/rest-reference/openapi.json

@@ -3,7 +3,7 @@
   "info": {
     "title": "Backend.AI Manager API",
     "description": "Backend.AI Manager REST API specification",
-    "version": "24.12.1",
+    "version": "25.1.1",
     "contact": {
       "name": "Lablup Inc.",
       "url": "https://docs.backend.ai",

src/ai/backend/manager/models/gql_models/group.py

@@ -32,28 +32,6 @@
     from ..gql import GraphQueryContext
     from ..scaling_group import ScalingGroup
 
-_queryfilter_fieldspec: Mapping[str, FieldSpecItem] = {
-    "id": ("id", None),
-    "row_id": ("id", None),
-    "name": ("name", None),
-    "is_active": ("is_active", None),
-    "created_at": ("created_at", dtparse),
-    "modified_at": ("modified_at", dtparse),
-    "domain_name": ("domain_name", None),
-    "resource_policy": ("resource_policy", None),
-}
-
-_queryorder_colmap: Mapping[str, OrderSpecItem] = {
-    "id": ("id", None),
-    "row_id": ("id", None),
-    "name": ("name", None),
-    "is_active": ("is_active", None),
-    "created_at": ("created_at", None),
-    "modified_at": ("modified_at", None),
-    "domain_name": ("domain_name", None),
-    "resource_policy": ("resource_policy", None),
-}
-
 
 class GroupInput(graphene.InputObjectType):
     type = graphene.String(
@@ -116,6 +94,28 @@ class Meta:
         UserConnection,
     )
 
+    queryfilter_fieldspec: Mapping[str, FieldSpecItem] = {
+        "id": ("id", None),
+        "row_id": ("id", None),
+        "name": ("name", None),
+        "is_active": ("is_active", None),
+        "created_at": ("created_at", dtparse),
+        "modified_at": ("modified_at", dtparse),
+        "domain_name": ("domain_name", None),
+        "resource_policy": ("resource_policy", None),
+    }
+
+    queryorder_colmap: Mapping[str, OrderSpecItem] = {
+        "id": ("id", None),
+        "row_id": ("id", None),
+        "name": ("name", None),
+        "is_active": ("is_active", None),
+        "created_at": ("created_at", None),
+        "modified_at": ("modified_at", None),
+        "domain_name": ("domain_name", None),
+        "resource_policy": ("resource_policy", None),
+    }
+
     @classmethod
     def from_row(
         cls,
@@ -227,12 +227,12 @@ async def get_connection(
     ) -> ConnectionResolverResult[Self]:
         graph_ctx: GraphQueryContext = info.context
         _filter_arg = (
-            FilterExprArg(filter_expr, QueryFilterParser(_queryfilter_fieldspec))
+            FilterExprArg(filter_expr, QueryFilterParser(cls.queryfilter_fieldspec))
             if filter_expr is not None
             else None
         )
         _order_expr = (
-            OrderExprArg(order_expr, QueryOrderParser(_queryorder_colmap))
+            OrderExprArg(order_expr, QueryOrderParser(cls.queryorder_colmap))
             if order_expr is not None
             else None
         )

src/ai/backend/manager/models/gql_models/user.py

@@ -3,7 +3,9 @@
 from typing import (
     TYPE_CHECKING,
     Mapping,
+    Optional,
     Self,
+    cast,
 )
 
 import graphene
@@ -14,6 +16,7 @@
 from ..base import (
     FilterExprArg,
     OrderExprArg,
+    PaginatedConnectionField,
     generate_sql_info_for_gql_connection,
 )
 from ..gql_relay import AsyncNode, Connection, ConnectionResolverResult
@@ -23,6 +26,7 @@
 
 if TYPE_CHECKING:
     from ..gql import GraphQueryContext
+    from .group import GroupNode
 
 
 class UserNode(graphene.ObjectType):
@@ -53,6 +57,11 @@ class Meta:
     totp_activated_at = GQLDateTime()
     sudo_session_enabled = graphene.Boolean()
 
+    project_nodes = PaginatedConnectionField(
+        "ai.backend.manager.models.gql_models.group.GroupConnection",
+        description="Added in 25.2.0.",
+    )
+
     @classmethod
     def from_row(cls, ctx: GraphQueryContext, row: UserRow) -> Self:
         return cls(
@@ -176,6 +185,61 @@ async def get_connection(
             total_cnt = await db_session.scalar(cnt_query)
             return ConnectionResolverResult(result, cursor, pagination_order, page_size, total_cnt)
 
+    async def resolve_project_nodes(
+        self,
+        info: graphene.ResolveInfo,
+        filter: Optional[str] = None,
+        order: Optional[str] = None,
+        offset: Optional[int] = None,
+        after: Optional[str] = None,
+        first: Optional[int] = None,
+        before: Optional[str] = None,
+        last: Optional[int] = None,
+    ) -> ConnectionResolverResult[GroupNode]:
+        from ..group import AssocGroupUserRow, GroupRow
+        from .group import GroupNode
+
+        graph_ctx: GraphQueryContext = info.context
+        _filter_arg = (
+            FilterExprArg(filter, QueryFilterParser(GroupNode.queryfilter_fieldspec))
+            if filter is not None
+            else None
+        )
+        _order_expr = (
+            OrderExprArg(order, QueryOrderParser(GroupNode.queryorder_colmap))
+            if order is not None
+            else None
+        )
+        (
+            query,
+            cnt_query,
+            _,
+            cursor,
+            pagination_order,
+            page_size,
+        ) = generate_sql_info_for_gql_connection(
+            info,
+            GroupRow,
+            GroupRow.id,
+            _filter_arg,
+            _order_expr,
+            offset,
+            after=after,
+            first=first,
+            before=before,
+            last=last,
+        )
+        j = sa.join(GroupRow, AssocGroupUserRow)
+        prj_query = query.select_from(j).where(AssocGroupUserRow.user_id == self.id)
+        cnt_query = cnt_query.select_from(j).where(AssocGroupUserRow.user_id == self.id)
+        result: list[GroupNode] = []
+        async with graph_ctx.db.begin_readonly_session() as db_session:
+            total_cnt = await db_session.scalar(cnt_query)
+            async for row in await db_session.stream_scalars(prj_query):
+                prj_row = cast(GroupRow, row)
+                result.append(GroupNode.from_row(graph_ctx, prj_row))
+            return ConnectionResolverResult(result, cursor, pagination_order, page_size, total_cnt)
+
 
 class UserConnection(Connection):
     class Meta: