[WIP] Subscriptions

example/api/src/model/post_query.py

@@ -1,3 +1,4 @@
+from cpl.dependency.event_bus import EventBusABC
 from cpl.graphql.query_context import QueryContext
 from cpl.graphql.schema.db_model_graph_type import DbModelGraphType
 from cpl.graphql.schema.filter.db_model_filter import DbModelFilter
@@ -5,6 +6,7 @@ from cpl.graphql.schema.input import Input
 from cpl.graphql.schema.mutation import Mutation
 from cpl.graphql.schema.sort.sort import Sort
 from cpl.graphql.schema.sort.sort_order import SortOrder
+from cpl.graphql.schema.subscription import Subscription
 from model.author_dao import AuthorDao
 from model.author_query import AuthorGraphType, AuthorFilter
 from model.post import Post
@@ -19,6 +21,7 @@ class PostFilter(DbModelFilter[Post]):
         self.string_field("title")
         self.string_field("content")
 
+
 class PostSort(Sort[Post]):
     def __init__(self):
         Sort.__init__(self)
@@ -26,6 +29,7 @@ class PostSort(Sort[Post]):
         self.field("title", SortOrder)
         self.field("content", SortOrder)
 
+
 class PostGraphType(DbModelGraphType[Post]):
 
     def __init__(self, authors: AuthorDao):
@@ -42,7 +46,7 @@ class PostGraphType(DbModelGraphType[Post]):
         def r_name(ctx: QueryContext):
             return ctx.user.username == "admin"
 
-        self.object_field("author", AuthorGraphType, resolver=_a).with_public(True)# .with_require_any([], [r_name]))
+        self.object_field("author", AuthorGraphType, resolver=_a).with_public(True)  # .with_require_any([], [r_name]))
         self.string_field(
             "title",
             resolver=lambda root: root.title,
@@ -64,6 +68,7 @@ class PostCreateInput(Input[Post]):
         self.string_field("content").with_required()
         self.int_field("author_id").with_required()
 
+
 class PostUpdateInput(Input[Post]):
     title: str
     content: str
@@ -75,13 +80,31 @@ class PostUpdateInput(Input[Post]):
         self.string_field("title").with_required(False)
         self.string_field("content").with_required(False)
 
+
+class PostSubscription(Subscription):
+    def __init__(self, bus: EventBusABC):
+        Subscription.__init__(self)
+        self._bus = bus
+
+        async def post_changed():
+            async for event in await self._bus.subscribe("postChange"):
+                print("Event:", event, type(event))
+                yield event
+
+        def selector(event: Post, info) -> bool:
+            return True
+
+        self.subscription_field("postChange", PostGraphType, post_changed, selector)
+
+
 class PostMutation(Mutation):
 
-    def __init__(self, posts: PostDao, authors: AuthorDao):
+    def __init__(self, posts: PostDao, authors: AuthorDao, bus: EventBusABC):
         Mutation.__init__(self)
 
         self._posts = posts
         self._authors = authors
+        self._bus = bus
 
         self.field("create", int, resolver=self.create_post).with_public().with_required().with_argument(
             "input",
@@ -112,6 +135,7 @@ class PostMutation(Mutation):
         post.content = input.content if input.content is not None else post.content
 
         await self._posts.update(post)
+        await self._bus.publish("postChange", post)
         return True
 
     async def delete_post(self, id: int) -> bool:
@@ -127,4 +151,3 @@ class PostMutation(Mutation):
             return False
         await self._posts.restore(post)
         return True
-