File tree Expand file tree Collapse file tree 1 file changed +20
-0
lines changed Expand file tree Collapse file tree 1 file changed +20
-0
lines changed Original file line number Diff line number Diff line change @@ -120,6 +120,15 @@ def __next__(self) -> ConvexValue:
120
120
raise ConvexError (result ["message" ], result ["data" ])
121
121
return result ["value" ]
122
122
123
+ def __aiter__ (self ) -> QuerySubscription :
124
+ return self
125
+
126
+ async def __anext__ (self ) -> ConvexValue :
127
+ result = await self .safe_inner_sub ().anext ()
128
+ if result ["type" ] == "convexerror" :
129
+ raise ConvexError (result ["message" ], result ["data" ])
130
+ return result ["value" ]
131
+
123
132
def unsubscribe (self ) -> None :
124
133
"""Unsubscribe from the query and drop this subscription from the active query set.
125
134
@@ -162,6 +171,17 @@ def __next__(self) -> Optional[Dict[SubscriberId, ConvexValue]]:
162
171
result [k ] = result [k ]
163
172
return result
164
173
174
+ def __aiter__ (self ) -> QuerySetSubscription :
175
+ return self
176
+
177
+ async def __anext__ (self ) -> Optional [Dict [SubscriberId , ConvexValue ]]:
178
+ result = await self .safe_inner_sub ().anext ()
179
+ if not result :
180
+ return result
181
+ for k in result :
182
+ result [k ] = result [k ]
183
+ return result
184
+
165
185
166
186
class ConvexClient :
167
187
"""WebSocket-based Convex Client.
You can’t perform that action at this time.
0 commit comments