feat(ssr): context.onRenderComplete

Guillaume Chau · Guillaume Chau · commit fa286b051f54 · 2018-11-07T01:48:43.000+01:00
diff --git a/src/server/create-renderer.js b/src/server/create-renderer.js
@@ -79,6 +79,9 @@ export function createRenderer ({
       }, cb)
       try {
         render(component, write, context, err => {
+          if (context && context.onRenderComplete) {
+            context.onRenderComplete(context)
+          }
           if (template) {
             result = templateRenderer.renderSync(result, context)
           }
@@ -106,13 +109,23 @@ export function createRenderer ({
         render(component, write, context, done)
       })
       if (!template) {
+        if (context && context.onRenderComplete) {
+          renderStream.once('beforeEnd', () => {
+            context.onRenderComplete(context)
+          })
+        }
         return renderStream
       } else {
         const templateStream = templateRenderer.createStream(context)
         renderStream.on('error', err => {
           templateStream.emit('error', err)
         })
         renderStream.pipe(templateStream)
+        if (context && context.onRenderComplete) {
+          renderStream.once('beforeEnd', () => {
+            context.onRenderComplete(context)
+          })
+        }
         return templateStream
       }
     }
diff --git a/src/server/render-stream.js b/src/server/render-stream.js
@@ -42,6 +42,7 @@ export default class RenderStream extends stream.Readable {
     })
 
     this.end = () => {
+      this.emit('beforeEnd')
       // the rendering is finished; we should push out the last of the buffer.
       this.done = true
       this.push(this.buffer)
