feat: add maxAge for ssr

QingWei-Li · QingWei-Li · commit 0144e6e51f9d · 2017-12-29T16:47:45.000+08:00
diff --git a/lib/commands/start.js b/lib/commands/start.js
@@ -6,6 +6,7 @@ const Renderer = require('docsify-server-renderer')
 const fs = require('fs')
 const util = require('../util/index')
 const chalk = require('chalk')
+const LRU = require('lru-cache')
 
 var defaultConfig = {
   template: `<!DOCTYPE html>
@@ -37,7 +38,7 @@ function loadConfig (config) {
 module.exports = function (path, configFile, port) {
   let config = defaultConfig
   const pkg = util.pkg()
-  let ctx = util.cwd('.')
+  const ctx = util.cwd('.')
 
   path = path || './'
 
@@ -50,37 +51,51 @@ module.exports = function (path, configFile, port) {
     const tpl = pkg.docsify.template
 
     config = pkg.docsify
-    config.template = (tpl && util.exists(util.resolve(ctx, tpl)))
-      ? util.read(tpl)
-      : defaultConfig.template
+    config.template =
+      tpl && util.exists(util.resolve(ctx, tpl))
+        ? util.read(tpl)
+        : defaultConfig.template
   }
 
   var renderer = new Renderer(Object.assign(defaultConfig, config))
   var server = connect()
+  const cached = new LRU(config.maxAge || 0)
 
   server.use(serveStatic(path))
-  server.use(function(req, res) {
+  server.use(function (req, res) {
     serveStatic(path)(req, res, function () {
-      if (/\.(jpg|jpeg|gif|png|svg|ico|mp4|webm|ogg|ogv|js|css|md)(?:\?v=[0-9.]+)?$/.test(req.url)) {
+      if (
+        /\.(jpg|jpeg|gif|png|svg|ico|mp4|webm|ogg|ogv|js|css|md)(?:\?v=[0-9.]+)?$/.test(
+          req.url
+        )
+      ) {
         res.writeHead(404)
         res.end()
       }
-      renderer.renderToString(req.url)
+
+      Promise.resolve(cached.get(req.url) || renderer.renderToString(req.url))
         .then(function (html) {
+          cached.set(req.url)
           res.end(html)
         })
         .catch(function (err) {
+          console.error(err)
           res.writeHead(404)
           res.end()
         })
     })
   })
   server.listen(port || 4000)
 
-  const msg = '\n'
-    + chalk.blue('[SSR]')
-    + ' Serving ' + chalk.green(`${path}`) + ' now.\n'
-    + 'Listening at ' + chalk.green(`http://localhost:${port}`) + '\n'
+  const msg =
+    '\n' +
+    chalk.blue('[SSR]') +
+    ' Serving ' +
+    chalk.green(`${path}`) +
+    ' now.\n' +
+    'Listening at ' +
+    chalk.green(`http://localhost:${port}`) +
+    '\n'
 
   console.log(msg)
 }
diff --git a/package.json b/package.json
@@ -32,9 +32,10 @@
     "connect-livereload": "^0.6.0",
     "cp-file": "^4.1.1",
     "docsify": ">=3",
-    "docsify-server-renderer": "^4.1.9",
+    "docsify-server-renderer": "^4.5.7",
     "fs-extra": "^2.1.2",
     "livereload": "^0.6.2",
+    "lru-cache": "^4.1.1",
     "open": "^0.0.5",
     "serve-static": "^1.12.1",
     "update-notifier": "^2.1.0",
