From e9ba46c84ea4159ef6d721a03eb1497234e58e3b Mon Sep 17 00:00:00 2001
From: syuilo <4439005+syuilo@users.noreply.github.com>
Date: Fri, 20 Sep 2024 09:36:08 +0900
Subject: [PATCH] Update ReactionsBufferingService.ts

---
 .../src/core/ReactionsBufferingService.ts     | 23 +++++++++++++++----
 1 file changed, 19 insertions(+), 4 deletions(-)

diff --git a/packages/backend/src/core/ReactionsBufferingService.ts b/packages/backend/src/core/ReactionsBufferingService.ts
index d179abc2ef..f5beca993f 100644
--- a/packages/backend/src/core/ReactionsBufferingService.ts
+++ b/packages/backend/src/core/ReactionsBufferingService.ts
@@ -9,6 +9,8 @@ import { DI } from '@/di-symbols.js';
 import type { MiNote } from '@/models/Note.js';
 import { bindThis } from '@/decorators.js';
 
+const REDIS_PREFIX = 'reactionsBuffer';
+
 @Injectable()
 export class ReactionsBufferingService {
 	constructor(
@@ -19,17 +21,17 @@ export class ReactionsBufferingService {
 
 	@bindThis
 	public async create(note: MiNote, reaction: string) {
-		this.redisClient.hincrby(`reactionsBuffer:${note.id}`, reaction, 1);
+		this.redisClient.hincrby(`${REDIS_PREFIX}:${note.id}`, reaction, 1);
 	}
 
 	@bindThis
 	public async delete(note: MiNote, reaction: string) {
-		this.redisClient.hincrby(`reactionsBuffer:${note.id}`, reaction, -1);
+		this.redisClient.hincrby(`${REDIS_PREFIX}:${note.id}`, reaction, -1);
 	}
 
 	@bindThis
 	public async get(note: MiNote) {
-		const result = await this.redisClient.hgetall(`reactionsBuffer:${note.id}`);
+		const result = await this.redisClient.hgetall(`${REDIS_PREFIX}:${note.id}`);
 		const delta = {};
 		for (const [name, count] of Object.entries(result)) {
 			delta[name] = parseInt(count);
@@ -43,7 +45,7 @@ export class ReactionsBufferingService {
 
 		const pipeline = this.redisClient.pipeline();
 		for (const note of notes) {
-			pipeline.hgetall(`reactionsBuffer:${note.id}`);
+			pipeline.hgetall(`${REDIS_PREFIX}:${note.id}`);
 		}
 		const results = await pipeline.exec();
 
@@ -59,4 +61,17 @@ export class ReactionsBufferingService {
 
 		return deltas;
 	}
+
+	@bindThis
+	public async bake() {
+		const bufferedNoteIds = [];
+		let cursor = '0';
+		do {
+			const result = await this.redisClient.scan(cursor, 'MATCH', `${REDIS_PREFIX}:*`, 'COUNT', '1000');
+			cursor = result[0];
+			bufferedNoteIds.push(...result[1]);
+		} while (cursor !== '0');
+
+		console.log(bufferedNoteIds);
+	}
 }