Infrastructure/astro
0
Fork 0
mirror of https://github.com/withastro/astro.git synced 2024-12-30 22:03:56 -05:00
Code Issues Activity

feat: transaction fn error stub

Browse source
This commit is contained in:
bholmesdev 2024-03-18 17:29:55 -04:00
parent ccbb19ab51
commit 580d95a2bc
1 changed files with 16 additions and 1 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

17
packages/db/src/runtime/db-client.ts
View file

@ -2,16 +2,30 @@ import type { InStatement } from '@libsql/client';
import { createClient } from '@libsql/client'; import { createClient } from '@libsql/client';
import type { LibSQLDatabase } from 'drizzle-orm/libsql'; import type { LibSQLDatabase } from 'drizzle-orm/libsql';
import { drizzle as drizzleLibsql } from 'drizzle-orm/libsql'; import { drizzle as drizzleLibsql } from 'drizzle-orm/libsql';
import { drizzle as drizzleProxy } from 'drizzle-orm/sqlite-proxy'; import { drizzle as drizzleProxy, type SqliteRemoteDatabase } from 'drizzle-orm/sqlite-proxy';
import { z } from 'zod'; import { z } from 'zod';
import { safeFetch } from './utils.js'; import { safeFetch } from './utils.js';
import { AstroDbError } from '../core/errors.js';
const isWebContainer = !!process.versions?.webcontainer; const isWebContainer = !!process.versions?.webcontainer;
function addTransactionErrorStub(db: SqliteRemoteDatabase) {
Object.assign(db, {
transaction: async () => {
throw new AstroDbError({
message:
'The `transaction()` function is not supported. If you need to execute multiple queries with rollbacks, try `db.batch()`.',
hint: 'Form more on `db.batch()`, see https://docs.astro.build/en/guides/astro-db/#batch-transactions',
});
},
});
}
export function createLocalDatabaseClient({ dbUrl }: { dbUrl: string }): LibSQLDatabase { export function createLocalDatabaseClient({ dbUrl }: { dbUrl: string }): LibSQLDatabase {
const url = isWebContainer ? 'file:content.db' : dbUrl; const url = isWebContainer ? 'file:content.db' : dbUrl;
const client = createClient({ url }); const client = createClient({ url });
const db = drizzleLibsql(client); const db = drizzleLibsql(client);
addTransactionErrorStub(db);
return db; return db;
} }
@ -131,5 +145,6 @@ export function createRemoteDatabaseClient(appToken: string, remoteDbURL: string
return results; return results;
} }
); );
addTransactionErrorStub(db);
return db; return db;
} }