diff --git a/apps/server/src/modules/workspaces/controller.ts b/apps/server/src/modules/workspaces/controller.ts index bcc4e87b4d..58d7daf402 100644 --- a/apps/server/src/modules/workspaces/controller.ts +++ b/apps/server/src/modules/workspaces/controller.ts @@ -4,6 +4,7 @@ import { ForbiddenException, Get, Inject, + Logger, NotFoundException, Param, Res, @@ -20,6 +21,8 @@ import { PermissionService } from './permission'; @Controller('/api/workspaces') export class WorkspacesController { + private readonly logger = new Logger('WorkspacesController'); + constructor( @Inject(StorageProvide) private readonly storage: Storage, private readonly permission: PermissionService, @@ -75,6 +78,6 @@ export class WorkspacesController { res.setHeader('content-type', 'application/octet-stream'); res.send(update); - console.info('workspaces doc api: ', format(process.hrtime(start))); + this.logger.debug('workspaces doc api: ', format(process.hrtime(start))); } } diff --git a/apps/server/src/modules/workspaces/resolver.ts b/apps/server/src/modules/workspaces/resolver.ts index 155afd736c..03b7ec877d 100644 --- a/apps/server/src/modules/workspaces/resolver.ts +++ b/apps/server/src/modules/workspaces/resolver.ts @@ -2,6 +2,7 @@ import type { Storage } from '@affine/storage'; import { ForbiddenException, Inject, + Logger, NotFoundException, UseGuards, } from '@nestjs/common'; @@ -129,6 +130,8 @@ export class UpdateWorkspaceInput extends PickType( @Auth() @Resolver(() => WorkspaceType) export class WorkspaceResolver { + private readonly logger = new Logger('WorkspaceResolver'); + constructor( private readonly auth: AuthService, private readonly config: Config, @@ -404,7 +407,6 @@ export class WorkspaceResolver { @Args('workspaceId') workspaceId: string, @Args('email') email: string, @Args('permission', { type: () => Permission }) permission: Permission, - // TODO: add rate limit @Args('sendInviteMail', { nullable: true }) sendInviteMail: boolean ) { await this.permissionProvider.check(workspaceId, user.id, Permission.Admin); @@ -650,6 +652,7 @@ export class WorkspaceResolver { const { size } = await this.collectAllBlobSizes(user); if (size > quota) { + this.logger.log(`storage size limit exceeded: ${size} > ${quota}`); throw new ForbiddenException('storage size limit exceeded'); }