feat: migrate throttler config for new throttler version

toeverything · Mar 5, 2024 · 3f16f57 · 3f16f57
1 parent d8287d7
commit 3f16f57
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
@@ -55,7 +55,7 @@ export class AuthResolver {
  @Throttle({
  default: {
  limit: 10,
- ttl: 60,
+ ttl: 60 * 1000,
  },
  })
  @Public()
@@ -71,7 +71,7 @@ export class AuthResolver {
  @Throttle({
  default: {
  limit: 20,
- ttl: 60,
+ ttl: 60 * 1000,
  },
  })
  @ResolveField(() => ClientTokenType, {
@@ -103,7 +103,7 @@ export class AuthResolver {
  @Throttle({
  default: {
  limit: 10,
- ttl: 60,
+ ttl: 60 * 1000,
  },
  })
  @Mutation(() => UserType)
@@ -123,7 +123,7 @@ export class AuthResolver {
  @Throttle({
  default: {
  limit: 10,
- ttl: 60,
+ ttl: 60 * 1000,
  },
  })
  @Mutation(() => UserType)
@@ -141,7 +141,7 @@ export class AuthResolver {
  @Throttle({
  default: {
  limit: 5,
- ttl: 60,
+ ttl: 60 * 1000,
  },
  })
  @Mutation(() => UserType)
@@ -170,7 +170,7 @@ export class AuthResolver {
  @Throttle({
  default: {
  limit: 5,
- ttl: 60,
+ ttl: 60 * 1000,
  },
  })
  @Mutation(() => UserType)
@@ -199,7 +199,7 @@ export class AuthResolver {
  @Throttle({
  default: {
  limit: 5,
- ttl: 60,
+ ttl: 60 * 1000,
  },
  })
  @Mutation(() => Boolean)
@@ -232,7 +232,7 @@ export class AuthResolver {
  @Throttle({
  default: {
  limit: 5,
- ttl: 60,
+ ttl: 60 * 1000,
  },
  })
  @Mutation(() => Boolean)
@@ -267,7 +267,7 @@ export class AuthResolver {
  @Throttle({
  default: {
  limit: 5,
- ttl: 60,
+ ttl: 60 * 1000,
  },
  })
  @Mutation(() => Boolean)
@@ -293,7 +293,7 @@ export class AuthResolver {
  @Throttle({
  default: {
  limit: 5,
- ttl: 60,
+ ttl: 60 * 1000,
  },
  })
  @Mutation(() => Boolean)

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

diff --git a/packages/backend/server/src/core/users/resolver.ts b/packages/backend/server/src/core/users/resolver.ts
@@ -53,7 +53,7 @@ export class UserResolver {
  @Throttle({
  default: {
  limit: 10,
- ttl: 60,
+ ttl: 60 * 1000,
  },
  })
  @Query(() => UserOrLimitedUser, {
@@ -86,15 +86,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',
@@ -108,7 +118,7 @@ export class UserResolver {
  @Throttle({
  default: {
  limit: 10,
- ttl: 60,
+ ttl: 60 * 1000,
  },
  })
  @Mutation(() => UserType, {
@@ -143,7 +153,7 @@ export class UserResolver {
  @Throttle({
  default: {
  limit: 10,
- ttl: 60,
+ ttl: 60 * 1000,
  },
  })
  @Mutation(() => UserType, {
@@ -191,7 +201,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
@@ -27,7 +27,7 @@ export class WorkspaceManagementResolver {
  @Throttle({
  default: {
  limit: 10,
- ttl: 60,
+ ttl: 60 * 1000,
  },
  })
  @Mutation(() => Int)
@@ -46,7 +46,7 @@ export class WorkspaceManagementResolver {
  @Throttle({
  default: {
  limit: 10,
- ttl: 60,
+ ttl: 60 * 1000,
  },
  })
  @Mutation(() => Int)
@@ -65,7 +65,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
@@ -191,7 +191,7 @@ export class WorkspaceResolver {
  @Throttle({
  default: {
  limit: 10,
- ttl: 30,
+ ttl: 30 * 1000,
  },
  })
  @Public()
@@ -224,6 +224,12 @@ export class WorkspaceResolver {
  return workspace;
  }
 
+ @Throttle({
+ default: {
+ limit: 10,
+ ttl: 120 * 1000,
+ },
+ })
  @Mutation(() => WorkspaceType, {
  description: 'Create a new workspace',
  })
@@ -300,6 +306,12 @@ export class WorkspaceResolver {
  });
  }
 
+ @Throttle({
+ default: {
+ limit: 10,
+ ttl: 120 * 1000,
+ },
+ })
  @Mutation(() => Boolean)
  async deleteWorkspace(
  @CurrentUser() user: CurrentUser,
@@ -318,6 +330,12 @@ export class WorkspaceResolver {
  return true;
  }
 
+ @Throttle({
+ default: {
+ limit: 10,
+ ttl: 120 * 1000,
+ },
+ })
  @Mutation(() => String)
  async invite(
  @CurrentUser() user: CurrentUser,
@@ -407,7 +425,7 @@ export class WorkspaceResolver {
  @Throttle({
  default: {
  limit: 10,
- ttl: 30,
+ ttl: 30 * 1000,
  },
  })
  @Public()
@@ -468,6 +486,12 @@ export class WorkspaceResolver {
  };
  }
 
+ @Throttle({
+ default: {
+ limit: 10,
+ ttl: 120 * 1000,
+ },
+ })
  @Mutation(() => Boolean)
  async revoke(
  @CurrentUser() user: CurrentUser,
@@ -512,6 +536,12 @@ export class WorkspaceResolver {
  return this.permissions.acceptWorkspaceInvitation(inviteId, workspaceId);
  }
 
+ @Throttle({
+ default: {
+ limit: 10,
+ ttl: 120 * 1000,
+ },
+ })
  @Mutation(() => Boolean)
  async leaveWorkspace(
  @CurrentUser() user: CurrentUser,