feat: migrate throttler config for new throttler version

toeverything · Mar 29, 2024 · 1a96495 · 1a96495
1 parent db49ad9
commit 1a96495
Show file tree

Hide file tree

Showing 5 changed files with 64 additions and 24 deletions.
diff --git a/packages/backend/server/src/core/auth/resolver.ts b/packages/backend/server/src/core/auth/resolver.ts
@@ -59,7 +59,7 @@ export class AuthResolver {
   @Throttle({
     default: {
       limit: 10,
-      ttl: 60,
+      ttl: 60 * 1000,
     },
   })
   @Public()
@@ -75,7 +75,7 @@ export class AuthResolver {
   @Throttle({
     default: {
       limit: 20,
-      ttl: 60,
+      ttl: 60 * 1000,
     },
   })
   @ResolveField(() => ClientTokenType, {
@@ -107,7 +107,7 @@ export class AuthResolver {
   @Throttle({
     default: {
       limit: 10,
-      ttl: 60,
+      ttl: 60 * 1000,
     },
   })
   @Mutation(() => UserType)
@@ -137,7 +137,7 @@ export class AuthResolver {
   @Throttle({
     default: {
       limit: 10,
-      ttl: 60,
+      ttl: 60 * 1000,
     },
   })
   @Mutation(() => UserType)
@@ -165,7 +165,7 @@ export class AuthResolver {
   @Throttle({
     default: {
       limit: 5,
-      ttl: 60,
+      ttl: 60 * 1000,
     },
   })
   @Mutation(() => UserType)
@@ -196,7 +196,7 @@ export class AuthResolver {
   @Throttle({
     default: {
       limit: 5,
-      ttl: 60,
+      ttl: 60 * 1000,
     },
   })
   @Mutation(() => UserType)
@@ -226,7 +226,7 @@ export class AuthResolver {
   @Throttle({
     default: {
       limit: 5,
-      ttl: 60,
+      ttl: 60 * 1000,
     },
   })
   @Mutation(() => Boolean)
@@ -259,7 +259,7 @@ export class AuthResolver {
   @Throttle({
     default: {
       limit: 5,
-      ttl: 60,
+      ttl: 60 * 1000,
     },
   })
   @Mutation(() => Boolean)
@@ -297,7 +297,7 @@ export class AuthResolver {
   @Throttle({
     default: {
       limit: 5,
-      ttl: 60,
+      ttl: 60 * 1000,
     },
   })
   @Mutation(() => Boolean)
@@ -323,7 +323,7 @@ export class AuthResolver {
   @Throttle({
     default: {
       limit: 5,
-      ttl: 60,
+      ttl: 60 * 1000,
     },
   })
   @Mutation(() => Boolean)

diff --git a/packages/backend/server/src/core/user/management.ts b/packages/backend/server/src/core/user/management.ts
@@ -27,7 +27,7 @@ export class UserManagementResolver {
   @Throttle({
     default: {
       limit: 10,
-      ttl: 60,
+      ttl: 60 * 1000,
     },
   })
   @Mutation(() => Int)
@@ -52,7 +52,7 @@ export class UserManagementResolver {
   @Throttle({
     default: {
       limit: 10,
-      ttl: 60,
+      ttl: 60 * 1000,
     },
   })
   @Mutation(() => Int)
@@ -73,7 +73,7 @@ export class UserManagementResolver {
   @Throttle({
     default: {
       limit: 10,
-      ttl: 60,
+      ttl: 60 * 1000,
     },
   })
   @Query(() => [UserType])

diff --git a/packages/backend/server/src/core/user/resolver.ts b/packages/backend/server/src/core/user/resolver.ts
@@ -54,7 +54,7 @@ export class UserResolver {
   @Throttle({
     default: {
       limit: 10,
-      ttl: 60,
+      ttl: 60 * 1000,
     },
   })
   @Query(() => UserOrLimitedUser, {
@@ -90,15 +90,25 @@ export class UserResolver {
     };
   }
 
-  @Throttle({ default: { limit: 10, ttl: 60 } })
+  @Throttle({
+    default: {
+      limit: 10,
+      ttl: 60 * 1000,
+    },
+  })
   @ResolveField(() => UserQuotaType, { name: 'quota', nullable: true })
   async getQuota(@CurrentUser() me: User) {
     const quota = await this.quota.getUserQuota(me.id);
 
     return quota.feature;
   }
 
-  @Throttle({ default: { limit: 10, ttl: 60 } })
+  @Throttle({
+    default: {
+      limit: 10,
+      ttl: 60 * 1000,
+    },
+  })
   @ResolveField(() => Int, {
     name: 'invoiceCount',
     description: 'Get user invoice count',
@@ -121,7 +131,7 @@ export class UserResolver {
   @Throttle({
     default: {
       limit: 10,
-      ttl: 60,
+      ttl: 60 * 1000,
     },
   })
   @Mutation(() => UserType, {
@@ -156,7 +166,7 @@ export class UserResolver {
   @Throttle({
     default: {
       limit: 10,
-      ttl: 60,
+      ttl: 60 * 1000,
     },
   })
   @Mutation(() => UserType, {
@@ -204,7 +214,7 @@ export class UserResolver {
   @Throttle({
     default: {
       limit: 10,
-      ttl: 60,
+      ttl: 60 * 1000,
     },
   })
   @Mutation(() => DeleteAccount)

diff --git a/packages/backend/server/src/core/workspaces/management.ts b/packages/backend/server/src/core/workspaces/management.ts
@@ -26,7 +26,7 @@ export class WorkspaceManagementResolver {
   @Throttle({
     default: {
       limit: 10,
-      ttl: 60,
+      ttl: 60 * 1000,
     },
   })
   @Mutation(() => Int)
@@ -45,7 +45,7 @@ export class WorkspaceManagementResolver {
   @Throttle({
     default: {
       limit: 10,
-      ttl: 60,
+      ttl: 60 * 1000,
     },
   })
   @Mutation(() => Int)
@@ -64,7 +64,7 @@ export class WorkspaceManagementResolver {
   @Throttle({
     default: {
       limit: 10,
-      ttl: 60,
+      ttl: 60 * 1000,
     },
   })
   @Query(() => [WorkspaceType])

diff --git a/packages/backend/server/src/core/workspaces/resolvers/workspace.ts b/packages/backend/server/src/core/workspaces/resolvers/workspace.ts
@@ -194,7 +194,7 @@ export class WorkspaceResolver {
   @Throttle({
     default: {
       limit: 10,
-      ttl: 30,
+      ttl: 30 * 1000,
     },
   })
   @Public()
@@ -227,6 +227,12 @@ export class WorkspaceResolver {
     return workspace;
   }
 
+  @Throttle({
+    default: {
+      limit: 10,
+      ttl: 120 * 1000,
+    },
+  })
   @Mutation(() => WorkspaceType, {
     description: 'Create a new workspace',
   })
@@ -303,6 +309,12 @@ export class WorkspaceResolver {
     });
   }
 
+  @Throttle({
+    default: {
+      limit: 10,
+      ttl: 120 * 1000,
+    },
+  })
   @Mutation(() => Boolean)
   async deleteWorkspace(
     @CurrentUser() user: CurrentUser,
@@ -321,6 +333,12 @@ export class WorkspaceResolver {
     return true;
   }
 
+  @Throttle({
+    default: {
+      limit: 10,
+      ttl: 120 * 1000,
+    },
+  })
   @Mutation(() => String)
   async invite(
     @CurrentUser() user: CurrentUser,
@@ -425,7 +443,7 @@ export class WorkspaceResolver {
   @Throttle({
     default: {
       limit: 10,
-      ttl: 30,
+      ttl: 30 * 1000,
     },
   })
   @Public()
@@ -486,6 +504,12 @@ export class WorkspaceResolver {
     };
   }
 
+  @Throttle({
+    default: {
+      limit: 10,
+      ttl: 120 * 1000,
+    },
+  })
   @Mutation(() => Boolean)
   async revoke(
     @CurrentUser() user: CurrentUser,
@@ -530,6 +554,12 @@ export class WorkspaceResolver {
     return this.permissions.acceptWorkspaceInvitation(inviteId, workspaceId);
   }
 
+  @Throttle({
+    default: {
+      limit: 10,
+      ttl: 120 * 1000,
+    },
+  })
   @Mutation(() => Boolean)
   async leaveWorkspace(
     @CurrentUser() user: CurrentUser,