9c88af4b20
This replaces usage of our `chainMethods` helper to dynamically chain queries with [drizzle dynamic query building](https://orm.drizzle.team/docs/dynamic-query-building). This is more type-safe, more readable and requires less code
44 lines
1.3 KiB
TypeScript
44 lines
1.3 KiB
TypeScript
import type { SQLiteSelect } from 'drizzle-orm/sqlite-core'
|
|
|
|
import { count, sql } from 'drizzle-orm'
|
|
|
|
import type { CountDistinct, SQLiteAdapter } from './types.js'
|
|
|
|
export const countDistinct: CountDistinct = async function countDistinct(
|
|
this: SQLiteAdapter,
|
|
{ db, joins, tableName, where },
|
|
) {
|
|
// When we don't have any joins - use a simple COUNT(*) query.
|
|
if (joins.length === 0) {
|
|
const countResult = await db
|
|
.select({
|
|
count: count(),
|
|
})
|
|
.from(this.tables[tableName])
|
|
.where(where)
|
|
return Number(countResult[0]?.count)
|
|
}
|
|
|
|
let query: SQLiteSelect = db
|
|
.select({
|
|
count: sql`COUNT(1) OVER()`,
|
|
})
|
|
.from(this.tables[tableName])
|
|
.where(where)
|
|
.groupBy(this.tables[tableName].id)
|
|
.limit(1)
|
|
.$dynamic()
|
|
|
|
joins.forEach(({ condition, table }) => {
|
|
query = query.leftJoin(table, condition)
|
|
})
|
|
|
|
// When we have any joins, we need to count each individual ID only once.
|
|
// COUNT(*) doesn't work for this well in this case, as it also counts joined tables.
|
|
// SELECT (COUNT DISTINCT id) has a very slow performance on large tables.
|
|
// Instead, COUNT (GROUP BY id) can be used which is still slower than COUNT(*) but acceptable.
|
|
const countResult = await query
|
|
|
|
return Number(countResult[0]?.count)
|
|
}
|