server.js 4.7 KB

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