Add high performance batch imports

packages/backend/src/db/meilisearch.ts

@@ -97,8 +97,14 @@ export default hasConfig ? {
 			filter: constructedFilters
 		});
 	},
-	ingestNote: (note : Note) => {
+	ingestNote: (note: Note | Note[]) => {
-		logger.info("Indexing note in MeiliSearch: " + note.id);
+		if (note instanceof Note) {
+			note = [note];
+		}
+
+		let indexingBatch: MeilisearchNote[] = [];
+
+		note.forEach(note => {
 
 			let attachmentType = "";
 			if (note.attachedFileTypes.length > 0) {
@@ -115,18 +121,24 @@ export default hasConfig ? {
 				}
 			}
 
-		return posts.addDocuments([
+			indexingBatch.push({
-			{
 					id: note.id.toString(),
-				text: note.text,
+					text: note.text ? note.text : "",
 					userId: note.userId,
-				userHost: note.userHost,
+					userHost: note.userHost ? note.userHost : "",
-				channelId: note.channelId,
+					channelId: note.channelId ? note.channelId : "",
 					mediaAttachment: attachmentType,
-				userName: note.user?.username,
+					userName: note.user?.username ? note.user.username : "",
 					createdAt: note.createdAt.getTime() / 1000 // division by 1000 is necessary because Node returns in ms-accuracy
 				}
-		]);
+			)
+		});
+
+		let indexingIDs = indexingBatch.map(note => note.id);
+
+		logger.info("Indexing notes in MeiliSearch: " + indexingIDs.join(","));
+
+		return posts.addDocuments(indexingBatch);
 	},
 	serverStats: async () => {
 		let health : Health = await client.health();

packages/backend/src/queue/processors/background/index-all-notes.ts

@@ -5,6 +5,7 @@ import { Notes } from "@/models/index.js";
 import { MoreThan } from "typeorm";
 import { index } from "@/services/note/create.js";
 import {Note} from "@/models/entities/note.js";
+import meilisearch from "../../../db/meilisearch.js";
 
 const logger = queueLogger.createSubLogger("index-all-notes");
 
@@ -58,7 +59,12 @@ export default async function indexAllNotes(
 
 		for (let i = 0; i < notes.length; i += batch) {
 			const chunk = notes.slice(i, i + batch);
-			await Promise.all(chunk.map((note) => index(note)));
+
+			if (meilisearch) {
+				await meilisearch.ingestNote(chunk)
+			}
+
+			await Promise.all(chunk.map((note) => index(note, true)));
 
 			indexedCount += chunk.length;
 			const pct = (indexedCount / total) * 100;

packages/backend/src/services/note/create.ts

@@ -749,7 +749,7 @@ async function insertNote(
 	}
 }
 
-export async function index(note: Note): Promise<void> {
+export async function index(note: Note, reindexing: boolean): Promise<void> {
 	if (!note.text) return;
 
 	if (config.elasticsearch && es) {
@@ -778,7 +778,7 @@ export async function index(note: Note): Promise<void> {
 		);
 	}
 
-	if (meilisearch) {
+	if (meilisearch && !reindexing) {
 		await meilisearch.ingestNote(note);
 	}
 }