server.js 5.6 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110
  1. import fs from 'fs'
  2. import path from 'path'
  3. import { fileURLToPath } from 'url'
  4. import express from 'express'
  5. import serverStatic from 'serve-static'
  6. import { createServer as createViteServer } from 'vite'
  7. import ssrManifest from './dist/client/.vite/ssr-manifest.json' assert { type: "json" }
  8. import * as PRODserver from './dist/server/entry-server.js'
  9. const isProd = process.env.NODE_ENV === 'production'
  10. const __dirname = path.dirname(fileURLToPath(import.meta.url))
  11. async function createServer() {
  12. const app = express()
  13. // 以中间件模式创建 Vite 应用,并将 appType 配置为 'custom'
  14. // 这将禁用 Vite 自身的 HTML 服务逻辑
  15. // 并让上级服务器接管控制
  16. const vite = await createViteServer({
  17. server: { middlewareMode: 'ssr' },
  18. appType: 'custom'
  19. })
  20. if(!isProd){
  21. // 使用 vite 的 Connect 实例作为中间件
  22. // 如果你使用了自己的 express 路由(express.Router()),你应该使用 router.use
  23. // 当服务器重启(例如用户修改了 vite.config.js 后),
  24. // `vite.middlewares` 仍将保持相同的引用
  25. // (带有 Vite 和插件注入的新的内部中间件堆栈)。
  26. // 即使在重新启动后,以下内容仍然有效。
  27. app.use(vite.middlewares)
  28. }else{
  29. app.use(serverStatic(path.resolve(__dirname,'dist/client'),{index:false}))
  30. }
  31. // 参考https://cn.vitejs.dev/guide/ssr.html
  32. app.use('*', async(req, res) => {
  33. const url = req.originalUrl
  34. let template
  35. let render
  36. let html
  37. try {
  38. if(!isProd){
  39. console.log("开启开发服务端渲染")
  40. // 1. 读取 index.html
  41. template = fs.readFileSync(
  42. path.resolve(__dirname, 'index.html'),
  43. 'utf-8',
  44. )
  45. // 2. 应用 Vite HTML 转换。这将会注入 Vite HMR 客户端,
  46. // 同时也会从 Vite 插件应用 HTML 转换。
  47. // 例如:@vitejs/plugin-react 中的 global preambles
  48. template = await vite.transformIndexHtml(url, template)
  49. // 3a. 加载服务器入口。vite.ssrLoadModule 将自动转换
  50. // 你的 ESM 源码使之可以在 Node.js 中运行!无需打包
  51. // 并提供类似 HMR 的根据情况随时失效。
  52. render = (await vite.ssrLoadModule('/src/entry-server.ts')).render
  53. // 3b. 从 Vite 5.1 版本开始,你可以试用实验性的 createViteRuntime
  54. // API。
  55. // 这个 API 完全支持热更新(HMR),其工作原理与 ssrLoadModule 相似
  56. // 如果你想尝试更高级的用法,可以考虑在另一个线程,甚至是在另一台机器上,
  57. // 使用 ViteRuntime 类来创建运行环境。
  58. // const runtime = await vite.createViteRuntime(server)
  59. // const { render } = await runtime.executeEntrypoint('/src/entry-server.ts')
  60. }else{
  61. console.log("开启生产服务端渲染")
  62. template = fs.readFileSync(
  63. path.resolve(__dirname, 'dist/client/index.html'),
  64. 'utf-8',
  65. )
  66. render = PRODserver.render
  67. }
  68. // 4. 渲染应用的 HTML。这假设 entry-server.js 导出的 `render`
  69. // 函数调用了适当的 SSR 框架 API。
  70. // 例如 ReactDOMServer.renderToString()
  71. if(!isProd){
  72. const { appHtml, state } = await render(url)
  73. const { title,keywords, description } = state.meta
  74. // 5. 注入渲染后的应用程序 HTML 到模板中。
  75. html = template
  76. .replace(`<!--ssr-outlet-->`, appHtml)
  77. .replace('\'<!--vuex-state-->\'', JSON.stringify(state))
  78. .replace('<title>',`<title>${title}`)
  79. .replace('<meta name="keywords" content="">',`<meta name="keywords" content="${keywords}">`)
  80. .replace('<meta name="description" content="">',`<meta name="description" content="${description}">`)
  81. }else{
  82. const manifest = ssrManifest
  83. const { appHtml, state,preloadLinks } = await render(url,manifest)
  84. const { title,keywords, description } = state.meta
  85. html = template
  86. .replace(`<!--preload-links-->`, preloadLinks)
  87. .replace(`<!--ssr-outlet-->`, appHtml)
  88. .replace('\'<!--vuex-state-->\'', JSON.stringify(state))
  89. .replace('<title>',`<title>${title}`)
  90. .replace('<meta name="keywords" content="">',`<meta name="keywords" content="${keywords}">`)
  91. .replace('<meta name="description" content="">',`<meta name="description" content="${description}">`)
  92. }
  93. // 6. 返回渲染后的 HTML。
  94. res.status(200).set({ 'Content-Type': 'text/html' }).end(html)
  95. } catch (e) {
  96. // 如果捕获到了一个错误,让 Vite 来修复该堆栈,这样它就可以映射回
  97. // 你的实际源码中。
  98. // vite.ssrFixStacktrace(e)
  99. // console.log(e)
  100. // next(e)
  101. }
  102. })
  103. app.listen(5173, () => {
  104. console.log("node server.js",isProd ? `运行 生产环境` : `运行 开发环境`)
  105. })
  106. }
  107. createServer()