Merge pull request #9 from kandarppatel28/master

Added hit rate feature #5
bmusuko · Oct 3, 2020 · b64a175 · b64a175 · vercel · Oct 3, 2020
2 parents ba7d694 + 26a691c
commit b64a175
Show file tree

Hide file tree

Showing 2 changed files with 9 additions and 0 deletions.
diff --git a/models/Shortener.ts b/models/Shortener.ts
@@ -18,6 +18,9 @@ const ShortenerSchema = new mongoose.Schema({
   password: {
     type: String,
   },
+  visitCount: {
+    type: Number
+  },
 });
 
 const Shortener =

diff --git a/pages/api/short/index.ts b/pages/api/short/index.ts
@@ -38,6 +38,12 @@ export default async (req: NextApiRequest, res: NextApiResponse) => {
       if (!realLink) {
         return responseGenerator(res, 404, "link not found");
       }
+      //Increment visitCount
+      await Shortener.findOneAndUpdate(
+        { generated_link: link },
+        { visitCount: ( realLink.visitCount || 0 ) + 1 }
+      );
+
       if (realLink["is_password"]) {
         return responseGenerator(res, 200, "get link", {
           is_password: true,