moon
/
ditto
1
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

Update stats before storing event

This commit is contained in:
Alex Gleason 2024-05-14 15:05:59 -05:00
parent a3597edb90
commit eef349f1e9
No known key found for this signature in database
GPG Key ID: 7211D1F99744FBB7
2 changed files with 12 additions and 12 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

6
src/pipeline.ts
View File

@ -107,10 +107,8 @@ async function storeEvent(event: DittoEvent, signal?: AbortSignal): Promise<void
if (deletion) { if (deletion) {
return Promise.reject(new RelayError('blocked', 'event was deleted')); return Promise.reject(new RelayError('blocked', 'event was deleted'));
} else { } else {
await Promise.all([ await updateStats(event).catch(debug);
Storages.db.event(event, { signal }).catch(debug), await Storages.db.event(event, { signal }).catch(debug);
updateStats(event).catch(debug),
]);
} }
} }

18
src/stats.ts
View File

@ -1,4 +1,4 @@
import { NostrEvent } from '@nostrify/nostrify'; import { NKinds, NostrEvent } from '@nostrify/nostrify';
import Debug from '@soapbox/stickynotes/debug'; import Debug from '@soapbox/stickynotes/debug';
import { InsertQueryBuilder } from 'kysely'; import { InsertQueryBuilder } from 'kysely';
@ -16,14 +16,14 @@ type StatDiff = AuthorStatDiff | EventStatDiff;
const debug = Debug('ditto:stats'); const debug = Debug('ditto:stats');
/** Store stats for the event in LMDB. */ /** Store stats for the event. */
async function updateStats(event: NostrEvent) { async function updateStats(event: NostrEvent) {
let prev: NostrEvent | undefined; let prev: NostrEvent | undefined;
const queries: InsertQueryBuilder<DittoTables, any, unknown>[] = []; const queries: InsertQueryBuilder<DittoTables, any, unknown>[] = [];
// Kind 3 is a special case - replace the count with the new list. // Kind 3 is a special case - replace the count with the new list.
if (event.kind === 3) { if (event.kind === 3) {
prev = await maybeGetPrev(event); prev = await getPrevEvent(event);
if (!prev || event.created_at >= prev.created_at) { if (!prev || event.created_at >= prev.created_at) {
queries.push(updateFollowingCountQuery(event)); queries.push(updateFollowingCountQuery(event));
} }
@ -153,12 +153,14 @@ function eventStatsQuery(diffs: EventStatDiff[]) {
} }
/** Get the last version of the event, if any. */ /** Get the last version of the event, if any. */
async function maybeGetPrev(event: NostrEvent): Promise<NostrEvent> { async function getPrevEvent(event: NostrEvent): Promise<NostrEvent | undefined> {
const [prev] = await Storages.db.query([ if (NKinds.replaceable(event.kind) || NKinds.parameterizedReplaceable(event.kind)) {
{ kinds: [event.kind], authors: [event.pubkey], limit: 1 }, const [prev] = await Storages.db.query([
]); { kinds: [event.kind], authors: [event.pubkey], limit: 1 },
]);
return prev; return prev;
}
} }
/** Set the following count to the total number of unique "p" tags in the follow list. */ /** Set the following count to the total number of unique "p" tags in the follow list. */