aljaz/ditto
1
0
Fork 0
mirror of https://github.com/aljazceru/ditto.git synced 2026-01-07 07:34:25 +01:00
Code Issues Packages Projects Releases Wiki Activity

stats: fix race conditions (on Postgres)

Browse Source
This commit is contained in:
Alex Gleason
2024-05-24 20:11:22 -05:00
parent 250998405a
commit 04018015c5
1 changed files with 19 additions and 2 deletions
Download Patch File Download Diff File Expand all files Collapse all files

21
src/utils/stats.ts
View File

@@ -6,6 +6,7 @@ import { SetRequired } from 'type-fest';
import { DittoTables } from '@/db/DittoTables.ts';
import { getTagSet } from '@/utils/tags.ts';
import { Conf } from '@/config.ts';
interface UpdateStatsOpts {
kysely: Kysely<DittoTables>;
@@ -153,8 +154,16 @@ export async function updateAuthorStats(
notes_count: 0,
};
const prev = await getAuthorStats(kysely, pubkey);
let query = kysely
.selectFrom('author_stats')
.selectAll()
.where('pubkey', '=', pubkey);
if (Conf.db.dialect === 'postgres') {
query = query.forUpdate();
}
const prev = await query.executeTakeFirst();
const stats = fn(prev ?? empty);
if (prev) {
@@ -195,8 +204,16 @@ export async function updateEventStats(
reactions: '{}',
};
const prev = await getEventStats(kysely, eventId);
let query = kysely
.selectFrom('event_stats')
.selectAll()
.where('event_id', '=', eventId);
if (Conf.db.dialect === 'postgres') {
query = query.forUpdate();
}
const prev = await query.executeTakeFirst();
const stats = fn(prev ?? empty);
if (prev) {