Add rate limit to page view counter

Author: BlakeBr0

apps/api/src/plugins/cache.ts

@@ -38,6 +38,14 @@ class Cache {
   async hasUniqueLaunch(mod: string, version: string, ip: string) {
     return this.cache.has(`unique_launch:${mod}:${version}:${ip}`);
   }
+
+  async setPageView(mod: string, source: string, ip: string) {
+    return this.cache.set(`page_view:${mod}:${source}:${ip}`, 1, "10m");
+  }
+
+  async hasPageView(mod: string, source: string, ip: string) {
+    return this.cache.has(`page_view:${mod}:${source}:${ip}`);
+  }
 }
 
 export default async function (fastify: FastifyInstance) {

apps/api/src/routes/banners/routes.ts

@@ -5,6 +5,7 @@ import { z } from "zod";
 
 export const plugin: FastifyPluginAsyncZod = async fastify => {
   const db = fastify.mongo.db!;
+  const cache = fastify.cache;
 
   fastify.get(
     "/:mod_id",
@@ -41,15 +42,21 @@ export const plugin: FastifyPluginAsyncZod = async fastify => {
         return `Banner ${mod_id} not found.`;
       }
 
-      await db.collection<PageViews>(Collections.PageViews).updateOne(
-        { mod_id },
-        {
-          $inc: {
-            [`${source}.${dayjs().format("YYYY.M.D")}`]: 1
-          }
-        },
-        { upsert: true }
-      );
+      const hasPageView = await cache.hasPageView(mod_id, source, request.ip);
+
+      if (!hasPageView) {
+        await db.collection<PageViews>(Collections.PageViews).updateOne(
+          { mod_id },
+          {
+            $inc: {
+              [`${source}.${dayjs().format("YYYY.M.D")}`]: 1
+            }
+          },
+          { upsert: true }
+        );
+
+        await cache.setPageView(mod_id, source, request.ip);
+      }
 
       reply.redirect(
         `https://blakesmods.com/img/banner/${mod_id}_title.png`,