versia-pub/server
2
0
Fork 0
mirror of https://github.com/versia-pub/server.git synced 2026-03-13 05:49:16 +01:00
Code Issues Projects Releases Packages Wiki Activity Actions

feat: ✨ Add CLI command to rebuild index

Browse source
This commit is contained in:
Jesse Wierzbinski 2024-06-28 23:58:43 -10:00
parent 19c15f7e96
commit 93b8609411
No known key found for this signature in database
3 changed files with 89 additions and 6 deletions
Download patch file Download diff file Expand all files Collapse all files

22
classes/search/search-manager.ts
View file

@ -184,16 +184,18 @@ export class SonicSearchManager {
* Rebuild search indexes
* @param indexes Indexes to rebuild
* @param batchSize Size of each batch
* @param progressCallback Callback for progress updates
*/
async rebuildSearchIndexes(
indexes: SonicIndexType[],
batchSize = 100,
progressCallback?: (progress: number) => void,
): Promise<void> {
for (const index of indexes) {
if (index === SonicIndexType.Accounts) {
await this.rebuildAccountsIndex(batchSize);
await this.rebuildAccountsIndex(batchSize, progressCallback);
} else if (index === SonicIndexType.Statuses) {
await this.rebuildStatusesIndex(batchSize);
await this.rebuildStatusesIndex(batchSize, progressCallback);
}
}
}
@ -201,8 +203,12 @@ export class SonicSearchManager {
/**
* Rebuild accounts index
* @param batchSize Size of each batch
* @param progressCallback Callback for progress updates
*/
private async rebuildAccountsIndex(batchSize: number): Promise<void> {
private async rebuildAccountsIndex(
batchSize: number,
progressCallback?: (progress: number) => void,
): Promise<void> {
const accountCount = await User.getCount();
const batchCount = Math.ceil(accountCount / batchSize);
@ -221,15 +227,19 @@ export class SonicSearchManager {
),
),
);
this.logger.info`Indexed accounts batch ${i + 1}/${batchCount}`;
progressCallback?.((i + 1) / batchCount);
}
}
/**
* Rebuild statuses index
* @param batchSize Size of each batch
* @param progressCallback Callback for progress updates
*/
private async rebuildStatusesIndex(batchSize: number): Promise<void> {
private async rebuildStatusesIndex(
batchSize: number,
progressCallback?: (progress: number) => void,
): Promise<void> {
const statusCount = await Note.getCount();
const batchCount = Math.ceil(statusCount / batchSize);
@ -245,7 +255,7 @@ export class SonicSearchManager {
),
),
);
this.logger.info`Indexed statuses batch ${i + 1}/${batchCount}`;
progressCallback?.((i + 1) / batchCount);
}
}