Change: fix mongo atomic operation

ambrosus · Jan 21, 2025 · 4280510 · 4280510
1 parent 0b09a5e
commit 4280510
Showing 1 changed file with 20 additions and 13 deletions.
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 },
         );
@@ -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 },
       );