Update server.js

rumkin · web-flow · commit 1cefd6c89721 · 2016-11-04T02:39:34.000+03:00
Make header prepending more elegant and simple.
diff --git a/server.js b/server.js
@@ -59,21 +59,22 @@ app.get('*', (req, res) => {
   var s = Date.now()
   const context = { url: req.url }
   const renderStream = renderer.renderToStream(context)
-  let firstChunk = true
-
-  renderStream.on('data', chunk => {
-    if (firstChunk) {
-      res.write(html.head)
-      // embed initial store state
-      if (context.initialState) {
-        res.write(
-          `<script>window.__INITIAL_STATE__=${
-            serialize(context.initialState, { isJSON: true })
-          }</script>`
-        )
-      }
-      firstChunk = false
+  
+  
+  // Write head once
+  renderStream.once('data', () => {
+    res.write(html.head)
+    // embed initial store state
+    if (context.initialState) {
+      res.write(
+        `<script>window.__INITIAL_STATE__=${
+          serialize(context.initialState, { isJSON: true })
+        }</script>`
+      )
     }
+  })
+  
+  renderStream.on('data', chunk => {
     res.write(chunk)
   })
 
