fix: csr add reactToString

moligit · Mar 10, 2020 · 579c08b · 579c08b
1 parent 79aa0e6
commit 579c08b
Show file tree

Hide file tree

Showing 2 changed files with 7 additions and 3 deletions.
diff --git a/packages/ykfe-utils/package.json b/packages/ykfe-utils/package.json
@@ -1,6 +1,6 @@
 {
   "name": "ykfe-utils",
-  "version": "3.5.3",
+  "version": "3.5.4",
   "description": "utils for ykfe",
   "main": "lib/index.js",
   "module": "es/index.js",

diff --git a/packages/ykfe-utils/src/utils/index.ts b/packages/ykfe-utils/src/utils/index.ts
@@ -1,7 +1,7 @@
 import { resolve } from 'path'
 import { Readable, PassThrough } from 'stream'
 import React from 'react'
-import { renderToNodeStream } from 'react-dom/server'
+import { renderToNodeStream, renderToString as reactRenderToString } from 'react-dom/server'
 import { Config } from '../interface/config'
 
 const { renderToString } = require('rax-server-renderer')
@@ -12,7 +12,11 @@ const logGreen = (text: string) => {
 }
 
 const reactToStream = (Component: React.FunctionComponent, props: object, config: Config) => {
-  return config.isRax ? renderToString(React.createElement(Component, props)) : renderToNodeStream(React.createElement(Component, props))
+  if (config.useReactToString) {
+    return reactRenderToString(React.createElement(Component, props))
+  } else {
+    return config.isRax ? renderToString(React.createElement(Component, props)) : renderToNodeStream(React.createElement(Component, props))
+  }
 }
 
 const getVersion = (str: string) => {