archive-gh-me/node-postgres
1
0
Fork 0
mirror of https://github.com/brianc/node-postgres.git synced 2025-12-08 20:16:25 +00:00
Code Issues Projects Releases Wiki Activity

Slightly better types

Browse Source
This commit is contained in:
Brian Carlson 2025-07-18 11:22:04 -05:00
parent 64a9aa2eb5
commit 0815f05d15
1 changed files with 5 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

7
packages/pg-transaction/src/index.ts
View File

@ -8,7 +8,10 @@ function isPool(clientOrPool: Client | Pool): clientOrPool is Pool {
return 'idleCount' in clientOrPool
}
async function transaction<T>(clientOrPool: Client | Pool, cb: (client: Client) => Promise<T>): Promise<T> {
async function transaction<T>(
clientOrPool: Client | Pool,
cb: (client: Client | PoolClient) => Promise<T>
): Promise<T> {
let client: Client | PoolClient
if (isPool(clientOrPool)) {
// It's a Pool
@ -19,7 +22,7 @@ async function transaction<T>(clientOrPool: Client | Pool, cb: (client: Client)
}
await client.query('BEGIN')
try {
const result = await cb(client as Client)
const result = await cb(client)
await client.query('COMMIT')
return result
} catch (error) {