chore: websocket app 의존성 추가

Co-authored-by: Summer Min <[email protected]>
Co-authored-by: ez <[email protected]>

Author: 3 people

apps/websocket/src/red-lock/red-lock.module.ts

@@ -0,0 +1,27 @@
+import { Module, forwardRef } from '@nestjs/common';
+import Redis from 'ioredis';
+import Redlock from 'redlock';
+import { RedisModule } from '../redis/redis.module';
+const RED_LOCK_TOKEN = 'RED_LOCK';
+const REDIS_CLIENT_TOKEN = 'REDIS_CLIENT';
+
+@Module({
+  imports: [forwardRef(() => RedisModule)],
+  providers: [
+    {
+      provide: RED_LOCK_TOKEN,
+      useFactory: (redisClient: Redis) => {
+        return new Redlock([redisClient], {
+          driftFactor: 0.01,
+          retryCount: 10,
+          retryDelay: 200,
+          retryJitter: 200,
+          automaticExtensionThreshold: 500,
+        });
+      },
+      inject: [REDIS_CLIENT_TOKEN],
+    },
+  ],
+  exports: [RED_LOCK_TOKEN],
+})
+export class RedLockModule {}

apps/websocket/src/redis/redis.module.ts

@@ -0,0 +1,27 @@
+import { Module, forwardRef } from '@nestjs/common';
+import { ConfigModule, ConfigService } from '@nestjs/config';
+import { RedisService } from './redis.service';
+import Redis from 'ioredis';
+import { RedLockModule } from '../red-lock/red-lock.module';
+
+// 의존성 주입할 때 redis client를 식별할 토큰
+const REDIS_CLIENT_TOKEN = 'REDIS_CLIENT';
+
+@Module({
+  imports: [ConfigModule, forwardRef(() => RedLockModule)], // ConfigModule 추가
+  providers: [
+    RedisService,
+    {
+      provide: REDIS_CLIENT_TOKEN,
+      inject: [ConfigService], // ConfigService 주입
+      useFactory: (configService: ConfigService) => {
+        return new Redis({
+          host: configService.get<string>('REDIS_HOST'),
+          port: configService.get<number>('REDIS_PORT'),
+        });
+      },
+    },
+  ],
+  exports: [RedisService, REDIS_CLIENT_TOKEN],
+})
+export class RedisModule {}

apps/websocket/src/redis/redis.service.ts

@@ -0,0 +1,64 @@
+import { Injectable } from '@nestjs/common';
+import { Inject } from '@nestjs/common';
+import Redis from 'ioredis';
+import Redlock from 'redlock';
+const REDIS_CLIENT_TOKEN = 'REDIS_CLIENT';
+const RED_LOCK_TOKEN = 'RED_LOCK';
+
+type RedisPage = {
+  title?: string;
+  content?: string;
+};
+
+@Injectable()
+export class RedisService {
+  constructor(
+    @Inject(REDIS_CLIENT_TOKEN) private readonly redisClient: Redis,
+    @Inject(RED_LOCK_TOKEN) private readonly redisLock: Redlock,
+  ) {}
+
+  async getAllKeys(pattern) {
+    return await this.redisClient.keys(pattern);
+  }
+
+  createStream() {
+    return this.redisClient.scanStream();
+  }
+
+  async get(key: string) {
+    const data = await this.redisClient.hgetall(key);
+    return Object.fromEntries(
+      Object.entries(data).map(([field, value]) => [field, value]),
+    ) as RedisPage;
+  }
+
+  async set(key: string, value: object) {
+    // 락을 획득할 때까지 기다린다.
+    const lock = await this.redisLock.acquire([`user:${key}`], 1000);
+    try {
+      await this.redisClient.hset(key, Object.entries(value));
+    } finally {
+      lock.release();
+    }
+  }
+
+  async setField(key: string, field: string, value: string) {
+    // 락을 획득할 때까지 기다린다.
+    const lock = await this.redisLock.acquire([`user:${key}`], 1000);
+    try {
+      return await this.redisClient.hset(key, field, value);
+    } finally {
+      lock.release();
+    }
+  }
+
+  async delete(key: string) {
+    // 락을 획득할 때까지 기다린다.
+    const lock = await this.redisLock.acquire([`user:${key}`], 1000);
+    try {
+      return await this.redisClient.del(key);
+    } finally {
+      lock.release();
+    }
+  }
+}