Refactor incremental rendering to use async handling for update chunks

Author: AbanoubGhadban

packages/react-on-rails-pro-node-renderer/src/worker.ts

@@ -315,15 +315,15 @@ export default function run(config: Partial<Config>) {
           }
         },
 
-        onUpdateReceived: (obj: unknown) => {
+        onUpdateReceived: async (obj: unknown) => {
           if (!incrementalSink) {
             log.error({ msg: 'Unexpected update chunk received after rendering was aborted', obj });
             return;
           }
 
           try {
             log.info(`Received a new update chunk ${JSON.stringify(obj)}`);
-            incrementalSink.add(obj);
+            await incrementalSink.add(obj);
           } catch (err) {
             // Log error but don't stop processing
             log.error({ err, msg: 'Error processing update chunk' });

packages/react-on-rails-pro-node-renderer/src/worker/handleIncrementalRenderRequest.ts

@@ -5,7 +5,7 @@ import { getRequestBundleFilePath } from '../shared/utils';
 
 export type IncrementalRenderSink = {
   /** Called for every subsequent NDJSON object after the first one */
-  add: (chunk: unknown) => void;
+  add: (chunk: unknown) => Promise<void>;
   handleRequestClosed: () => void;
 };
 
@@ -93,11 +93,11 @@ export async function handleIncrementalRenderRequest(
     return {
       response,
       sink: {
-        add: (chunk: unknown) => {
+        add: async (chunk: unknown) => {
           try {
             assertIsUpdateChunk(chunk);
             const bundlePath = getRequestBundleFilePath(chunk.bundleTimestamp);
-            executionContext.runInVM(chunk.updateChunk, bundlePath).catch((err: unknown) => {
+            await executionContext.runInVM(chunk.updateChunk, bundlePath).catch((err: unknown) => {
               log.error({ msg: 'Error running incremental render chunk', err, chunk });
             });
           } catch (err) {

packages/react-on-rails-pro-node-renderer/src/worker/vm.ts

@@ -365,7 +365,7 @@ export async function buildExecutionContext(
         const objectResult = await result;
         result = JSON.stringify(objectResult);
       }
-      if (log.level === 'debug') {
+      if (log.level === 'debug' && result) {
         log.debug(`result from JS:
   ${smartTrim(result)}`);
         const debugOutputPathResult = path.join(serverBundleCachePath, 'result.json');