UNPKG

@melleb/pgtx

Version:

Simple node-postgres wrapper that abstracts transactions and savepoints

142 lines (97 loc) 5.34 kB
# pgtx Keep your PostgreSQL transactions safe and simple with pgtx. --- `pgtx` is a lightweight TypeScript library that simplifies the use of nested transactions with savepoints in PostgreSQL. With pgtx, you can effortlessly manage savepoints in your database transactions without having to explicitly manage them yourself. The main motivation to write this libary was to be able to execute tests in transactions without affecting the code-under-test, which might also use transactions. Using `pgtx` you can wrap each test in its own transaction, leveraging eitehr PostgreSQL transactions or savepoints, whichever is appropriate. ## Features - **Nested Transactions**: Automatically handles savepoints, allowing you to create nested transactions. - **Commit and Rollback**: Easily commit or rollback transactions, including nested ones. - **Seamless Integration**: Works as a drop-in wrapper around the existing `pg` client. ## Installation This package has a peer dependency on `pg` so in order to install `pgtx` via npm execute: ```bash $ npm install pg pgtx ``` ... or use your package manager du jour. Other than the peer dependency it has no direct dependencies. ## Usage To use pgtx, wrap your PostgreSQL client with `pgtx`. You can then perform transactions as usual, and pgtx will manage savepoints automatically. `pgtx` has two modes: 1. **Automated mode** runs inside nested functions and commits after the function ends. To roll back call `tx.rollback()` or throw an error. 2. **Explicit mode** requires the user to call `tx.commit()` or `tx.rollback()` explicitly. ### Usage -- Automated mode The caller provides an async function to the transaction method and committing happens after completion of this method. ```typescript import pgtx, { PgtxClient } from 'pgtx'; import { Client } from 'pg'; const client = new Client({ /* connection config */ }); const db: PgtxClient = pgtx(client); await db.transaction(async (tx) => { // Your transactional code here, e.g. await db.query('INSERT INTO ...', [value]); await tx.transaction(async (nestedTx) => { // Nested transaction handled with savepoint // Savepoint is automatically released here }); await tx.transaction(async (nestedTx) => { // Or call rollback in order to rollback to the previous savepoint await nestedTx.rollback(); }); // If an exception is caught it is rolled back and propagated so all // parent transactions can handle the exception as well try { await tx.transation(async (nestedTx) => { throw new Error('Some issue occured'); }); } catch (e) { // Handle recovery scenario } // Commit is applied automatically }); ``` ### Usage -- Explicit mode The caller is responsible for calling either `commit()` or `rollback`. Be sure to handle failure scenario's using `try`/`catch`. ```typescript const client = new Client({ /* connection config */ }); const db: PgtxClient = pgtx(client); const tx = await db.transaction(); // Other code here, e.g.: await tx.query('INSERT INTO ...', [value]); const txNested = await tx.transaction(); // note the use of `tx` here // Other code here, e.g.: await txNested.query('INSERT INTO ...', [value]); await txNested.rollback(); await tx.commit(); ``` ## API **`PgtxClient`** extends `pg.Client` and adds: - `transaction(txFn?: (tx: PgtxClient) => void): Promise<PgtxClient>`: Begins a new transaction or savepoint if already in a transaction. If `txFn` is not defined it sets the client to the child context. - `rollback(): Promise<PgtxClient>`: Rolls back the current transaction or savepoint. Returns the client in the parent context. - `commit(): Promise<PgtxClient>`: Commits the current transaction or releases the savepoint -- although typically you will not use this method. Returns the client in the parent context. ## Changelog This library is using a semver versioning scheme. Changes are recorded in the [changelog](./CHANGELOG.md) using [Changesets](https://www.npmjs.com/package/@changesets/cli). ## Contributions Contributions are welcome! Please submit issues or pull requests to the GitHub repository. ### Development The tests of this library require access to a postgres database. Note that the tests will modify data in this database, so make sure to run them in a separate database. A connection to the database is established based on a `DATABASE_URL` environment variable. This variable needs to be present in either the environment itself or a `.env` file in the root of the solution. The connection string should be formatted as follows: `postgresql://user:password@localhost:5432/dbname?sslmode=disable`. To get started: 1. Start database 2. Either set the `DATABASE_URL` or `cp .env.dist .env` and edit `DATABASE_URL` variable 3. Run tests with `npm test` -- everything should pass 4. Make changes to codebase in a separate branch 5. Run `npx changset` to describe the changes you made 6. Create a Pull Request A code owner can review the PR and publish a new version on NPM with the following commands: 1. `npx changeset version`, this captures the change description in the PR and appends it to the changelog 2. `npx changeset publish`, which publishes the changes on [NPM](https://www.npmjs.com/package/@melleb/pgtx). ## License This library is licensed under the [MIT license](./LICENSE.md).