Fix gRPC streams

Author: geropl

components/server/src/api/server.ts

@@ -255,22 +255,25 @@ export class API {
                         });
                     }
 
-                    let subjectId: SubjectId | undefined = undefined;
+                    // Because we can't await before returning that generator, we await inside the generator, and create child contexts with that SubjectId
                     return wrapAsyncGenerator(
                         (async function* () {
                             try {
-                                subjectId = await auth();
-
-                                const generator = await apply<AsyncGenerator<any>>();
-                                for await (const item of generator) {
-                                    yield item;
+                                const subjectId = await auth();
+                                const generator = await runWithSubjectId(subjectId, () => apply<AsyncGenerator<any>>());
+                                while (true) {
+                                    const { value, done } = await runWithSubjectId(subjectId, () => generator.next());
+                                    if (done) {
+                                        break;
+                                    }
+                                    yield value;
                                 }
                                 done();
                             } catch (e) {
                                 handleError(e);
                             }
                         })(),
-                        (f) => runWithSubjectId(subjectId, f),
+                        (f) => withRequestContext(f),
                     );
                 };
             },