fix: add temp cache

Author: velopert

src/server/CacheManager.ts

@@ -18,9 +18,10 @@ export default class CacheManager {
       return null;
     }
   }
-  set(url: string, html: string) {
+  async set(url: string, html: string) {
     const rule = checkCacheRule(url);
     if (!rule) return;
-    return redis.set(createCacheKey(url), html);
+    await redis.set(createCacheKey(url), html);
+    await redis.expire(createCacheKey(url), 300);
   }
 }

src/server/rateLimitMiddleware.ts

@@ -11,6 +11,13 @@ const parseNumber = (value: string | null) => {
 const rateLimitMiddleware: Middleware = async (ctx, next) => {
   const ip = ctx.request.ips.slice(-1)[0] || ctx.request.ip;
 
+  const isBlockedUrl = await redis.get(`${ctx.url}:blocked`);
+  if (isBlockedUrl === '1') {
+    ctx.status = 429;
+    ctx.body = { msg: 'Too many request' };
+    return;
+  }
+
   const key = `${ip}:pageview`;
   const blockedKey = `${ip}:blocked`;
 

src/server/ssrMiddleware.ts

@@ -1,5 +1,8 @@
 import { Middleware } from 'koa';
 import serverRender from './serverRender';
+import CacheManager from './CacheManager';
+
+const cache = new CacheManager();
 
 const ssrMiddleware: Middleware = async (ctx, next) => {
   const ip = ctx.request.ips.slice(-1)[0] || ctx.request.ip;
@@ -9,25 +12,36 @@ const ssrMiddleware: Middleware = async (ctx, next) => {
     return;
   }
 
-  console.log(ip + ' >> ' + ctx.url);
+  const isLoggedIn = !!(
+    ctx.cookies.get('refresh_token') || ctx.cookies.get('access_token')
+  );
+
+  if (!isLoggedIn) {
+    const exists = await cache.get(ctx.url);
+    if (exists !== null) {
+      return exists;
+    }
+  }
 
   try {
     const result = await serverRender({
       url: ctx.url,
       cookie: ctx.headers.cookie,
-      loggedIn: !!(
-        ctx.cookies.get('refresh_token') || ctx.cookies.get('access_token')
-      ),
+      loggedIn: isLoggedIn,
     });
     if (!result) {
       return next();
     }
     ctx.body = result.html;
     ctx.status = result.statusCode;
+
+    if (!isLoggedIn) {
+      await cache.set(ctx.url, result.html ?? '');
+    }
   } catch (e) {
     ctx.throw(500, e as any);
   }
-  console.log(ip + '<< ' + ctx.url);
+  console.log(ip + ' << ' + ctx.url);
 };
 
 export default ssrMiddleware;