createCacheStream function in main exports

Author: adifiore

SSRtest/ModifiedReact.js

@@ -2730,6 +2730,53 @@ function renderToStaticNodeStream(element, cache, streamingStart, memLife=0) {
  return new ReactMarkupReadableStream(element, true, cache, streamingStart, memLife);
 }
 
+function createCacheStream(cache, streamingStart, memLife=0) {
+ const bufferedChunks = [];
+ return new Transform({
+ // transform() is called with each chunk of data
+ transform(data, enc, cb) {
+ // We store the chunk of data (which is a Buffer) in memory
+ bufferedChunks.push(data);
+ // Then pass the data unchanged onwards to the next stream
+ cb(null, data);
+ },
+
+ // flush() is called when everything is done
+ flush(cb) {
+ // We concatenate all the buffered chunks of HTML to get the full HTML, then cache it at "key"
+ let html = bufferedChunks.join("");
+ delete streamingStart.sliceStartCount; 
+
+ for (let component in streamingStart) {
+ let tagStack = [];
+ let tagStart;
+ let tagEnd;
+
+ do {
+ if (!tagStart) tagStart = streamingStart[component];
+ else tagStart = (html[tagEnd] === '<') ? tagEnd : html.indexOf('<', tagEnd);
+ tagEnd = html.indexOf('>', tagStart) + 1;
+ // Skip stack logic for void/self-closing elements and HTML comments 
+ if (html[tagEnd - 2] !== '/' && html[tagStart + 1] !== '!') {
+ // Push opening tags onto stack; pop closing tags off of stack
+ if (html[tagStart + 1] !== '/') tagStack.push(html.slice(tagStart, tagEnd));
+ else tagStack.pop();
+ }
+ } while (tagStack.length !== 0);
+ // cache component by slicing 'html'
+ if (memLife) {
+ cache.set(component, html.slice(streamingStart[component], tagEnd), memLife, (err) => {
+ if(err) console.log(err)
+ });
+ } else {
+ cache.set(component, html.slice(streamingStart[component], tagEnd));
+ }
+ }
+ cb();
+ }
+ });
+};
+
 class ComponentCache {
  constructor(config = {}) {
 
@@ -2765,6 +2812,7 @@ var ReactDOMServerNode = {
  renderToNodeStream: renderToNodeStream,
  renderToStaticNodeStream: renderToStaticNodeStream,
  ComponentCache: ComponentCache,
+ createCacheStream: createCacheStream,
  version: ReactVersion
 };
 

SSRtest/src/server/cacheStream.js

@@ -15,14 +15,7 @@ const createCacheStream = (cache, streamingStart, memLife=0) => {
 
  // flush() is called when everything is done
  flush(cb) {
- // We concatenate all the buffered chunks of HTML to get the full HTML
- // then cache it at "key"
-
- // console.log(cache);
- // console.log(bufferedChunks.join());
- // cache.set(key, Buffer.concat(bufferedChunks));
- // console.log("final", streamingStart.finalSliceStart);
- //joinedChunks = bufferedChunks.join(""); 
+ // We concatenate all the buffered chunks of HTML to get the full HTML, then cache it at "key"
  let html = bufferedChunks.join("");
  delete streamingStart.sliceStartCount; 
 
@@ -36,7 +29,6 @@ const createCacheStream = (cache, streamingStart, memLife=0) => {
  else tagStart = (html[tagEnd] === '<') ? tagEnd : html.indexOf('<', tagEnd);
  tagEnd = html.indexOf('>', tagStart) + 1;
  // Skip stack logic for void/self-closing elements and HTML comments 
- // Note: Does not account for tags inside HTML comments
  if (html[tagEnd - 2] !== '/' && html[tagStart + 1] !== '!') {
  // Push opening tags onto stack; pop closing tags off of stack
  if (html[tagStart + 1] !== '/') tagStack.push(html.slice(tagStart, tagEnd));

SSRtest/src/server/index.js

@@ -9,7 +9,7 @@ import App from '../shared/App';
 
 import createCacheStream from "./cacheStream";
 // can pass in max-size, otherwise defaults to 1 million
-// const cache = new ReactCC.ComponentCache();
+const cache = new ReactCC.ComponentCache();
 // import redis from 'redis';
 // const cache = redis.createClient();
 import memcached from 'memcached';