mirror/ditto
mirror/ditto
1
0
Fork 0
mirror of https://gitlab.com/soapbox-pub/ditto.git synced 2025-12-06 11:29:46 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions

trends: make trends work in Postgres (probably)

Browse source
This commit is contained in:
Alex Gleason 2024-08-12 14:48:32 -05:00
parent 5c23cb02c2
commit 3a82c98c21
No known key found for this signature in database
GPG key ID: 7211D1F99744FBB7
1 changed files with 32 additions and 20 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

52
src/trends.ts
View file

@ -20,26 +20,38 @@ export async function getTrendingTagValues(
filter: NostrFilter,
): Promise<{ value: string; authors: number; uses: number }[]> {
if (dialect === 'postgres') {
const { rows } = await sql<{ value: string; authors: number; uses: number }>`
SELECT
LOWER(element.value) AS value,
COUNT(DISTINCT nostr_events.pubkey) AS authors,
COUNT(*) as "uses"
FROM
nostr_events,
jsonb_each_text(nostr_events.tags_index) kv,
jsonb_array_elements_text(kv.value::jsonb) element
WHERE
kv.key = ANY(${tagNames})
${filter.kinds ? sql`AND nostr_events.kind = ANY(${filter.kinds})` : sql``}
${typeof filter.since === 'number' ? sql`AND nostr_events.created_at >= ${filter.since}` : sql``}
${typeof filter.until === 'number' ? sql`AND nostr_events.created_at <= ${filter.until}` : sql``}
GROUP BY
LOWER(element.value)
ORDER BY
COUNT(DISTINCT nostr_events.pubkey) DESC
${typeof filter.limit === 'number' ? sql`LIMIT ${filter.limit}` : sql``};`
.execute(kysely);
let query = kysely
.selectFrom([
'nostr_events',
sql<{ key: string; value: string }>`jsonb_each_text(nostr_events.tags_index)`.as('kv'),
sql<{ key: string; value: string }>`jsonb_array_elements_text(kv.value::jsonb)`.as('element'),
])
.select(({ fn }) => [
fn<string>('lower', ['element.value']).as('value'),
fn.agg<number>('count', ['nostr_events.pubkey']).distinct().as('authors'),
fn.countAll<number>().as('uses'),
])
.where('kv.key', '=', (eb) => eb.fn.any(eb.val(tagNames)))
.groupBy((eb) => eb.fn<string>('lower', ['element.value']))
.orderBy((eb) => eb.fn.agg('count', ['nostr_events.pubkey']).distinct(), 'desc');
if (filter.kinds) {
query = query.where('nostr_events.kind', '=', ({ fn, val }) => fn.any(val(filter.kinds)));
}
if (filter.authors) {
query = query.where('nostr_events.pubkey', '=', ({ fn, val }) => fn.any(val(filter.authors)));
}
if (typeof filter.since === 'number') {
query = query.where('nostr_events.created_at', '>=', filter.since);
}
if (typeof filter.until === 'number') {
query = query.where('nostr_events.created_at', '<=', filter.until);
}
if (typeof filter.limit === 'number') {
query = query.limit(filter.limit);
}
const rows = await query.execute();
return rows.map((row) => ({
value: row.value,