misty.cafe/mistykey
1
0
Fork 0
Code Issues Pull requests 1 Projects Releases Packages Wiki Activity Actions

increment/decrement note counts through CollapsedQueueService

Browse source
This commit is contained in:
Hazelnoot 2025-06-25 13:24:03 -04:00
parent 28dff9aff9
commit 81fd01d5fa
3 changed files with 10 additions and 27 deletions
Download patch file Download diff file Expand all files Collapse all files

9
packages/backend/src/core/CollapsedQueueService.ts
View file

@ -24,7 +24,8 @@ export type UpdateInstanceJob = {
};
export type UpdateUserJob = {
updatedAt: Date,
updatedAt?: Date,
additionalNotes?: number,
};
@Injectable()
@ -74,8 +75,12 @@ export class CollapsedQueueService implements OnApplicationShutdown {
fiveMinuteInterval,
(oldJob, newJob) => ({
updatedAt: maxDate(oldJob.updatedAt, newJob.updatedAt),
additionalNotes: (oldJob.additionalNotes ?? 0) + (newJob.additionalNotes ?? 0),
}),
(id, job) => this.usersRepository.update({ id }, {
updatedAt: job.updatedAt,
notesCount: job.additionalNotes ? () => `"notesCount" + ${job.additionalNotes}` : undefined,
}),
(id, job) => this.usersRepository.update({ id }, { updatedAt: job.updatedAt }),
{
onError: this.onQueueError,
concurrency: 4,

13
packages/backend/src/core/NoteCreateService.ts
View file

@ -605,7 +605,7 @@ export class NoteCreateService implements OnApplicationShutdown {
if (!this.isRenote(note) || this.isQuote(note)) {
// Increment notes count (user)
await this.incNotesCountOfUser(user);
this.collapsedQueueService.updateUserQueue.enqueue(user.id, { additionalNotes: 1 });
}
this.collapsedQueueService.updateUserQueue.enqueue(user.id, { updatedAt: new Date() });
@ -890,17 +890,6 @@ export class NoteCreateService implements OnApplicationShutdown {
await this.searchService.indexNote(note);
}
@bindThis
private async incNotesCountOfUser(user: { id: MiUser['id']; }) {
await this.usersRepository.createQueryBuilder().update()
.set({
updatedAt: this.timeService.date,
notesCount: () => '"notesCount" + 1',
})
.where('id = :id', { id: user.id })
.execute();
}
@bindThis
private async extractMentionedUsers(user: { host: MiUser['host']; }, tokens: mfm.MfmNode[]): Promise<MiUser[]> {
if (tokens == null) return [];

15
packages/backend/src/core/NoteDeleteService.ts
View file

@ -141,14 +141,14 @@ export class NoteDeleteService {
if (!isPureRenote(note)) {
// Decrement notes count (user)
promises.push(this.decNotesCountOfUser(user));
this.collapsedQueueService.updateUserQueue.enqueue(user.id, { additionalNotes: -1 });
}
this.collapsedQueueService.updateUserQueue.enqueue(user.id, { updatedAt: new Date() });
for (const cascade of cascadingNotes) {
if (!isPureRenote(cascade)) {
promises.push(this.decNotesCountOfUser(cascade.user));
this.collapsedQueueService.updateUserQueue.enqueue(cascade.user.id, { additionalNotes: -1 });
}
// Don't mark cascaded user as updated (active)
}
@ -217,17 +217,6 @@ export class NoteDeleteService {
});
}
@bindThis
private async decNotesCountOfUser(user: { id: MiUser['id']; }) {
await this.usersRepository.createQueryBuilder().update()
.set({
updatedAt: this.timeService.date,
notesCount: () => '"notesCount" - 1',
})
.where('id = :id', { id: user.id })
.execute();
}
@bindThis
private async findCascadingNotes(note: MiNote): Promise<(MiNote & { user: MiUser })[]> {
const cascadingNotes: MiNote[] = [];