Change: fix mongo atomic operation

ambrosus · Jan 21, 2025 · 8c83633 · 8c83633
1 parent 0b09a5e
commit 8c83633
Show file tree

Hide file tree

Showing 2 changed files with 27 additions and 16 deletions.
diff --git a/src/documentation.ts b/src/documentation.ts
@@ -116,7 +116,11 @@ paths:
     get:
       summary: Get user by address
       parameters:
-        - $ref: '#/components/parameters/GlobalAddressHeader'
+      - name: address
+        in: query
+        required: true
+        schema:
+          type: string
       responses:
         "200":
           description: User data
@@ -469,7 +473,7 @@ paths:
       description: Retrieve the list of followed.
       parameters:
         - name: address
-          in: query 
+          in: query
           required: true
           description: The address of the user whose followers are being retrieved.
           schema:

diff --git a/src/server.ts b/src/server.ts
@@ -68,7 +68,6 @@ const startServer = async () => {
         };
         address = address.toLowerCase();
         const secret = mapSecret.get(address);
-
         if (!signature || !address || !secret) {
           return reply.status(400).send({
             error: "Missing address or signature in headers",
@@ -132,9 +131,11 @@ async function checkTokenExist(
       await request.server.mongo.db?.collection("token").updateOne(
         { tokenAddress },
         {
-          tokenAddress,
-          like: 0,
-          timestamp: new Date(),
+          $set: {
+            tokenAddress,
+            like: 0,
+            timestamp: new Date(),
+          },
         },
         { upsert: true },
       );
@@ -202,10 +203,12 @@ async function addOrUpdateUser(request: FastifyRequest, reply: FastifyReply) {
     await request.server.mongo.db?.collection("user").updateOne(
       { address },
       {
-        address,
-        userName,
-        image,
-        timestamp: new Date(),
+        $set: {
+          address,
+          userName,
+          image,
+          timestamp: new Date(),
+        },
       },
       { upsert: true },
     );
@@ -244,9 +247,11 @@ async function addOrDeleteLike(request: FastifyRequest, reply: FastifyReply) {
         .updateOne(
           { address, tokenAddress },
           {
-            address,
-            tokenAddress,
-            timestamp: new Date(),
+            $set: {
+              address,
+              tokenAddress,
+              timestamp: new Date(),
+            },
           },
           { upsert: true },
         );
@@ -274,7 +279,7 @@ async function addOrDeleteLike(request: FastifyRequest, reply: FastifyReply) {
 
 async function getUser(request: FastifyRequest, reply: FastifyReply) {
   try {
-    let { address } = request.headers as { address?: string };
+    let { address } = request.query as { address?: string };
     address = address.toLowerCase();
     const user = await request.server.mongo.db
       ?.collection("user")
@@ -458,8 +463,10 @@ async function addfollow(request: FastifyRequest, reply: FastifyReply) {
       await request.server.mongo.db?.collection("followers").updateOne(
         { address, userAddress },
         {
-          address,
-          userAddress,
+          $set: {
+            address,
+            userAddress,
+          },
         },
         { upsert: true },
       );