diff --git a/src/server/ReactDOMNodeStreamRenderer.js b/src/server/ReactDOMNodeStreamRenderer.js
index 2fa1606..5234ba3 100644
--- a/src/server/ReactDOMNodeStreamRenderer.js
+++ b/src/server/ReactDOMNodeStreamRenderer.js
@@ -99,12 +99,7 @@ function originalRenderToNodeStream(element, cache, streamingStart, memLife=0) {
return new ReactMarkupReadableStream(element, false, cache, streamingStart, memLife);
}
-export function renderToNodeStream(element, cache, res) {
-
- const htmlStart =
- 'Page';
-
- const htmlEnd = '
';
+export function renderToNodeStream(element, cache, res, htmlStart, htmlEnd, memLife) {
const streamingStart = {
sliceStartCount: htmlStart.length,
@@ -114,7 +109,7 @@ export function renderToNodeStream(element, cache, res) {
cacheStream.pipe(res);
cacheStream.write(htmlStart);
- const stream = originalRenderToNodeStream(element, cache, streamingStart);
+ const stream = originalRenderToNodeStream(element, cache, streamingStart, memLife);
stream.pipe(cacheStream, { end: false });
stream.on("end", () => {
cacheStream.end(htmlEnd);
@@ -131,11 +126,7 @@ function originalRenderToStaticNodeStream(element, cache, streamingStart, memLif
return new ReactMarkupReadableStream(element, true, cache, streamingStart, memLife);
}
-export function renderToStaticNodeStream(element, cache, res) {
- const htmlStart =
- 'Page';
-
- const htmlEnd = '
';
+export function renderToStaticNodeStream(element, cache, res, htmlStart, htmlEnd, memLife) {
const streamingStart = {
sliceStartCount: htmlStart.length,
@@ -145,7 +136,7 @@ export function renderToStaticNodeStream(element, cache, res) {
cacheStream.pipe(res);
cacheStream.write(htmlStart);
- const stream = originalRenderToStaticNodeStream(element, cache, streamingStart);
+ const stream = originalRenderToStaticNodeStream(element, cache, streamingStart, memLife);
stream.pipe(cacheStream, { end: false });
stream.on("end", () => {
cacheStream.end(htmlEnd);