09ad6e4280
- Abstract shared sql code to a new drizzle package - Adds sqlite package, not ready to publish until drizzle patches some issues - Add `transactionOptions` to allow customizing or disabling db transactions - Adds "experimental" label to the `schemaName` property until drizzle patches an issue
33 lines
1.1 KiB
TypeScript
33 lines
1.1 KiB
TypeScript
import type { DrizzleAdapter } from '@payloadcms/drizzle/types'
|
|
import type { Payload } from 'payload'
|
|
|
|
import { isMongoose } from './isMongoose.js'
|
|
|
|
export async function resetDB(_payload: Payload, collectionSlugs: string[]) {
|
|
if (isMongoose(_payload) && 'collections' in _payload.db && collectionSlugs.length > 0) {
|
|
await _payload.db.collections[collectionSlugs[0]].db.dropDatabase()
|
|
} else if ('drizzle' in _payload.db) {
|
|
const db = _payload.db as unknown as DrizzleAdapter
|
|
|
|
// Alternative to: await db.drizzle.execute(sql`drop schema public cascade; create schema public;`)
|
|
|
|
// Deleting the schema causes issues when restoring the database from a snapshot later on. That's why we only delete the table data here,
|
|
// To avoid having to re-create any table schemas / indexes / whatever
|
|
const schema = db.drizzle._.schema
|
|
if (!schema) {
|
|
return
|
|
}
|
|
|
|
const queries = Object.values(schema)
|
|
.map((table: any) => {
|
|
return `DELETE FROM ${db.schemaName ? db.schemaName + '.' : ''}${table.dbName};`
|
|
})
|
|
.join('')
|
|
|
|
await db.execute({
|
|
drizzle: db.drizzle,
|
|
raw: queries,
|
|
})
|
|
}
|
|
}
|