✨ feat: 채팅 필터링 결과를 별도의 이벤트로 전송받을 수 있게 분리

boostcampwm-2024 · Nov 26, 2024 · cad0b3d · cad0b3d
1 parent 29027c6
commit cad0b3d
Show file tree

Hide file tree

Showing 3 changed files with 50 additions and 9 deletions.
diff --git a/Backend/apps/api/src/chats/chats.controller.ts b/Backend/apps/api/src/chats/chats.controller.ts
@@ -11,6 +11,6 @@ export class ChatsController {
   @Post()
   @UseGuards(JwtAuthGuard)
   async sendChat(@Body(ValidationPipe) sendChatDto: SendChatDto, @Req() req: Request & { user: UserEntity }) {
-    this.chatsService.publishChat({ ...sendChatDto, userId: req.user.id, nickname: req.user.nickname });
+    this.chatsService.ingestChat({ ...sendChatDto, userId: req.user.id, nickname: req.user.nickname });
   }
 }
diff --git a/Backend/apps/api/src/chats/chats.service.ts b/Backend/apps/api/src/chats/chats.service.ts
@@ -8,13 +8,14 @@ import { firstValueFrom } from 'rxjs';
 
 @Injectable()
 export class ChatsService {
+  private isFlushing = false;
   constructor(
     @InjectRedis() private readonly redisClient: Redis,
     private readonly httpService: HttpService,
     private readonly configService: ConfigService,
   ) {}
 
-  async publishChat({
+  async ingestChat({
     channelId,
     message,
     userId,
@@ -25,9 +26,11 @@ export class ChatsService {
     userId: number;
     nickname: string;
   }) {
-    const filteringResult = await this.clovaFiltering(message);
-    const chat = JSON.stringify({ content: message, userId, nickname, timestamp: new Date(), filteringResult });
-    await this.redisClient.multi().publish(`${channelId}:chat`, chat).rpush(`${channelId}:chats`, chat).exec();
+    const chatId = crypto.randomUUID();
+    const chat = { content: message, userId, nickname, timestamp: new Date(), channelId, chatId };
+    const chatString = JSON.stringify(chat);
+    await this.redisClient.multi().publish(`${channelId}:chat`, chatString).lpush('chatQueue', chatId).exec();
+    this.clovaFiltering(chat);
   }
 
   async readViewers(channelId: UUID) {
@@ -38,7 +41,7 @@ export class ChatsService {
     this.redisClient.del(`${channelId}:chats`);
   }
 
-  async clovaFiltering(message: string) {
+  async clovaFiltering(chat) {
     const postData = {
       messages: [
         {
@@ -47,7 +50,7 @@ export class ChatsService {
         },
         {
           role: 'user',
-          content: message,
+          content: chat.content,
         },
       ],
     };
@@ -61,6 +64,32 @@ export class ChatsService {
       }),
     );
 
-    return data?.result?.message?.content?.includes('true');
+    chat.filteringResult = data?.result?.message?.content?.includes('false');
+    await this.redisClient
+      .multi()
+      .publish(
+        `${chat.channelId}:filter`,
+        JSON.stringify({ chatId: chat.chatId, filteringResult: chat.filteringResult }),
+      )
+      .hset('chatCache', chat.chatId, JSON.stringify(chat))
+      .exec();
+    this.flushChat();
+  }
+
+  async flushChat() {
+    if (!this.isFlushing) {
+      this.isFlushing = true;
+      while (true) {
+        const frontChatId = await this.redisClient.lindex('chatQueue', 0);
+        const chatString = await this.redisClient.hget('chatCache', frontChatId);
+        if (!chatString) {
+          break;
+        } else {
+          const chat = JSON.parse(chatString);
+          await this.redisClient.multi().rpush(`${chat.channelId}:chats`, chatString).lpop('chatQueue').exec();
+        }
+      }
+      this.isFlushing = false;
+    }
   }
 }
diff --git a/Backend/apps/chats/src/chats.gateway.ts b/Backend/apps/chats/src/chats.gateway.ts
@@ -25,13 +25,21 @@ export class ChatsGateway implements OnGatewayDisconnect, OnGatewayConnection {
   constructor(@InjectRedis() private redisClient: Redis) {
     const subscriber = this.redisClient.duplicate();
     subscriber.psubscribe('*:chat');
+    subscriber.psubscribe('*:filter');
     subscriber.on('pmessage', async (pattern, channel, message) => {
       if (pattern === '*:chat') {
-        const channelId = channel.slice(0, -5) as UUID;
+        const channelId = channel.split(':')[0];
         const chat = plainToInstance(ChatDto, message);
         this.emitChat({ channelId, chat });
       }
     });
+
+    subscriber.on('pmessage', async (pattern, channel, message) => {
+      if (pattern === '*:filter') {
+        const channelId = channel.split(':')[0];
+        this.emitFilter({ channelId, filteringResult: JSON.parse(message) });
+      }
+    });
   }
 
   async handleConnection(socket: Socket) {
@@ -67,4 +75,8 @@ export class ChatsGateway implements OnGatewayDisconnect, OnGatewayConnection {
   async emitChat({ channelId, chat }) {
     this.server.to(channelId).emit('chat', [chat]);
   }
+
+  async emitFilter({ channelId, filteringResult }) {
+    this.server.to(channelId).emit('filter', [filteringResult]);
+  }
 }