fix: 스케줄러 undefined 이슈 해결

Author: ezcolin2

apps/backend/src/redis/redis.service.ts

@@ -4,8 +4,8 @@ import Redis from 'ioredis';
 const REDIS_CLIENT_TOKEN = 'REDIS_CLIENT';
 
 type RedisPage = {
-  title: string;
-  content: string;
+  title?: string;
+  content?: string
 };
 @Injectable()
 export class RedisService {
@@ -15,8 +15,12 @@ export class RedisService {
     @Inject(REDIS_CLIENT_TOKEN) private readonly redisClient: Redis,
   ) {}
 
-  async getAllKeys() {
-    return await this.redisClient.keys('*');
+  async getAllKeys(pattern) {
+    return await this.redisClient.keys(pattern);
+  }
+
+  createStream() {
+    return this.redisClient.scanStream();
   }
 
   async get(key: string) {

apps/backend/src/tasks/tasks.service.ts

@@ -2,7 +2,6 @@ import { Injectable, Logger } from '@nestjs/common';
 import { Cron, CronExpression } from '@nestjs/schedule';
 import { RedisService } from '../redis/redis.service';
 import { PageService } from '../page/page.service';
-
 @Injectable()
 export class TasksService {
   private readonly logger = new Logger(TasksService.name);
@@ -11,38 +10,84 @@ export class TasksService {
     private readonly pageService: PageService,
   ) {}
 
-  @Cron(CronExpression.EVERY_30_SECONDS)
+  @Cron(CronExpression.EVERY_10_SECONDS)
   async handleCron() {
     this.logger.log('스케줄러 시작');
     // 시작 시간
     const startTime = performance.now();
-
-    // redis의 모든 값을 가져와서 database에 저장
-    const keys = await this.redisService.getAllKeys();
-    const pages = [];
-    for await (const key of keys) {
-      const { title, content } = await this.redisService.get(key);
-      const jsonContent = JSON.parse(content);
-      Object.assign({
-        id: parseInt(key),
+    const keys = await this.redisService.getAllKeys('page:*');
+    Promise.all(
+      keys.map(async (key) => {
+        const { title, content } = await this.redisService.get(key);
+        const pageId = parseInt(key.split(':')[1]);
+        if (title === null) {
+          const jsonContent = JSON.parse(content);
+          await this.pageService.updatePage(pageId, {
+            content: jsonContent,
+          });
+        } else if (content === null) {
+          await this.pageService.updatePage(pageId, {
+            title,
+          });
+        } else {
+          const jsonContent = JSON.parse(content);
+          await this.pageService.updatePage(pageId, {
+            title,
+            content: jsonContent,
+          });
+          await this.redisService.delete(key);
+        }
+      }),
+    )
+      .then(() => {
+        const endTime = performance.now();
+        this.logger.log(`갱신 개수 : ${keys.length}개`);
+        this.logger.log(`실행 시간 : ${(endTime - startTime) / 1000}초`);
+      })
+      .catch((err) => {
+        this.logger.error(err);
       });
-      pages.push({
-        id: parseInt(key),
-        title,
-        content: jsonContent,
-        version: 1,
-      });
-      // this.pageService.updatePage(parseInt(key), {
-      //   title,
-      //   content: jsonContent,
-      // });
-      // this.logger.log('데이터베이스 갱신');
-    }
-    await this.pageService.updateBulkPage(pages);
-
-    // 끝 시간
-    const endTime = performance.now();
-    this.logger.log(`갱신 개수 : ${pages.length}개`);
-    this.logger.log(`실행 시간 : ${(endTime - startTime) / 1000}초`);
+    // scan stream을 가져온다.
+    // const stream = this.redisService.createStream();
+    // let totalCount = 0;
+    // stream.on('data', (keys) => {
+    //   console.log(keys);
+    //   totalCount += keys.length;
+    //   stream.pause();
+    //   Promise.all(
+    //     keys.map(async (key) => {
+    //       const { title, content } = await this.redisService.get(key);
+    //       if (title === null) {
+    //         const jsonContent = JSON.parse(content);
+    //         await this.pageService.updatePage(parseInt(key), {
+    //           content: jsonContent,
+    //         });
+    //       } else if (content === null) {
+    //         await this.pageService.updatePage(parseInt(key), {
+    //           title,
+    //         });
+    //       } else {
+    //         const jsonContent = JSON.parse(content);
+    //         await this.pageService.updatePage(parseInt(key), {
+    //           title,
+    //           content: jsonContent,
+    //         });
+    //         await this.redisService.delete(key);
+    //       }
+    //     }),
+    //   )
+    //     .then(() => {
+    //       stream.resume();
+    //     })
+    //     .catch((err) => {
+    //       this.logger.error(err);
+    //       stream.resume();
+    //     });
+    // });
+    // stream.on('end', () => {
+    //   const endTime = performance.now();
+    //   this.logger.log(`갱신 개수 : ${totalCount}개`);
+    //   this.logger.log(`실행 시간 : ${(endTime - startTime) / 1000}초`);
+    // });
   }
 }