server.js 4.1 KB

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