Merge pull request #28 from Bricks666/feature/cache-policy

Feature/cache policy
Bricks666 · Mar 25, 2024 · 52d8901 · 52d8901
2 parents 5571018 + 50bfeca
commit 52d8901
Show file tree

Hide file tree

Showing 3 changed files with 26 additions and 6 deletions.
diff --git a/src/app.module.ts b/src/app.module.ts
@@ -26,7 +26,7 @@ const modules = [
 		store: redisStore as unknown as CacheStore,
 		isGlobal: true,
 		max: 50,
-		ttl: 5,
+		ttl: +process.env.CACHE_TTL,
 		url: process.env.REDIS_URL,
 	}),
 	ConfigModule.forRoot({

diff --git a/src/rooms/controllers/lib.ts b/src/rooms/controllers/lib.ts
@@ -0,0 +1,3 @@
+export const getUserRoomsCacheKey = (userId: number): string => {
+	return `rooms_${userId}`;
+};
diff --git a/src/rooms/controllers/rooms.controller.ts b/src/rooms/controllers/rooms.controller.ts
@@ -5,7 +5,8 @@ import {
 	Body,
 	Put,
 	Delete,
-	UseInterceptors
+	UseInterceptors,
+	Inject
 } from '@nestjs/common';
 import {
 	ApiBody,
@@ -16,7 +17,7 @@ import {
 	ApiParam,
 	ApiTags
 } from '@nestjs/swagger';
-import { CacheInterceptor } from '@nestjs/cache-manager';
+import { CACHE_MANAGER, Cache, CacheInterceptor } from '@nestjs/cache-manager';
 import { Auth, CurrentUser } from '@/auth/lib';
 import { SecurityUserDto } from '@/users/dto';
 import { IntParam } from '@/shared';
@@ -25,12 +26,16 @@ import { RoomsService } from '../services';
 import { CreateRoomDto, RoomDto, UpdateRoomDto } from '../dto';
 import { IsOwner } from '../lib';
 import { WithRights } from '../types';
+import { getUserRoomsCacheKey } from './lib';
 
 @ApiTags('Комнаты')
 @Controller('rooms')
 @Auth()
 export class RoomsController {
-	constructor(private readonly roomsService: RoomsService) {}
+	constructor(
+		private readonly roomsService: RoomsService,
+		@Inject(CACHE_MANAGER) private readonly cacheManager: Cache
+	) {}
 
 	@ApiOperation({
 		summary: 'Возврат всех комнат авторизованного пользователя',
@@ -41,12 +46,24 @@ export class RoomsController {
 		description: 'Все комнаты, в которых состоит пользователь',
 	})
 	@Get('/')
-	@UseInterceptors(CacheInterceptor)
 	async getAll(
 		@CurrentUser() user: SecurityUserDto
 	): Promise<WithRights<RoomDto>[]> {
 		const { id, } = user;
-		return this.roomsService.getAll({ userId: id, });
+
+		const cacheKey = getUserRoomsCacheKey(id);
+
+		let rooms = await this.cacheManager.get<WithRights<RoomDto>[]>(cacheKey);
+
+		if (rooms) {
+			return rooms;
+		}
+
+		rooms = await this.roomsService.getAll({ userId: id, });
+
+		await this.cacheManager.set(cacheKey, rooms, +process.env.CACHE_TTL);
+
+		return rooms;
 	}
 
 	@ApiOperation({