feat: auto rm blocked account after update

cooderl · Mar 25, 2024 · fe10dba · fe10dba
1 parent e74f6c6
commit fe10dba
Show file tree

Hide file tree

Showing 2 changed files with 17 additions and 1 deletion.
diff --git a/apps/server/src/trpc/trpc.router.ts b/apps/server/src/trpc/trpc.router.ts
@@ -98,6 +98,7 @@ export class TrpcRouter {
           update: data,
           create: input,
         });
+        this.trpcService.removeBlockedAccount(id);
 
         return account;
       }),
@@ -118,12 +119,15 @@ export class TrpcRouter {
           where: { id },
           data,
         });
+        this.trpcService.removeBlockedAccount(id);
         return account;
       }),
     delete: this.trpcService.protectedProcedure
       .input(z.string())
       .mutation(async ({ input: id }) => {
         await this.prismaService.account.delete({ where: { id } });
+        this.trpcService.removeBlockedAccount(id);
+
         return id;
       }),
   });

diff --git a/apps/server/src/trpc/trpc.service.ts b/apps/server/src/trpc/trpc.service.ts
@@ -75,7 +75,9 @@ export class TrpcService {
           const blockedAccounts = blockedAccountsMap.get(today);
 
           if (Array.isArray(blockedAccounts)) {
-            blockedAccounts.push(id);
+            if (id) {
+              blockedAccounts.push(id);
+            }
             blockedAccountsMap.set(today, blockedAccounts);
           } else {
             blockedAccountsMap.set(today, [id]);
@@ -87,6 +89,16 @@ export class TrpcService {
     );
   }
 
+  removeBlockedAccount = (vid: string) => {
+    const today = this.getTodayDate();
+
+    const blockedAccounts = blockedAccountsMap.get(today);
+    if (Array.isArray(blockedAccounts)) {
+      const newBlockedAccounts = blockedAccounts.filter((id) => id !== vid);
+      blockedAccountsMap.set(today, newBlockedAccounts);
+    }
+  };
+
   private getTodayDate() {
     return dayjs.tz(new Date(), 'Asia/Shanghai').format('YYYY-MM-DD');
   }