@hexaijs/postgres 0.1.0 → 0.2.1

package/README.md

@@ -274,26 +274,92 @@ await tableManager.dropAllTables();
 await ensureConnection(client);  // Safe to call multiple times
 ```
 
-### Test Fixtures
+### PostgresUnitOfWorkForTesting
 
-The package exports test utilities from `@hexaijs/postgres/test`:
+A test-specific `UnitOfWork` implementation that runs inside an external transaction. This allows tests to rollback all changes after each test, keeping the database clean without truncating tables.
 
 ```typescript
-import { useDatabase, useClient, useTableManager } from "@hexaijs/postgres/test";
+import { PostgresUnitOfWorkForTesting } from "@hexaijs/postgres/test";
+import { Client } from "pg";
+
+describe("OrderService", () => {
+    let client: Client;
+    let uow: PostgresUnitOfWorkForTesting;
+
+    beforeEach(async () => {
+        client = new Client({ connectionString: "postgres://..." });
+        await client.connect();
+        await client.query("BEGIN");  // Start external transaction
+        uow = new PostgresUnitOfWorkForTesting(client);
+    });
 
-describe("OrderRepository", () => {
-    // Creates database before tests, drops after
-    const dbUrl = useDatabase("order_test_db");
+    afterEach(async () => {
+        await client.query("ROLLBACK");  // Rollback all changes
+        await client.end();
+    });
 
-    // Provides connected client
-    const client = useClient("order_test_db");
+    it("should create order", async () => {
+        await uow.wrap(async (c) => {
+            await c.query("INSERT INTO orders (id) VALUES ($1)", ["order-1"]);
+        });
 
-    it("should persist orders", async () => {
-        // Use client for assertions
+        // Verify within the same transaction
         const result = await client.query("SELECT * FROM orders");
         expect(result.rows).toHaveLength(1);
     });
+
+    // After this test, ROLLBACK cleans up - no data persists
+});
+```
+
+**How it works:**
+
+| Operation | Production (`PostgresUnitOfWork`) | Testing (`PostgresUnitOfWorkForTesting`) |
+|-----------|-----------------------------------|------------------------------------------|
+| Start | `BEGIN` | `SAVEPOINT` |
+| Commit | `COMMIT` | `RELEASE SAVEPOINT` |
+| Rollback | `ROLLBACK` | `ROLLBACK TO SAVEPOINT` |
+
+**Key behaviors:**
+
+- **abortError propagation**: When a nested `EXISTING` operation throws (even if caught), the entire transaction is marked as aborted and will rollback - matching production behavior.
+- **NESTED savepoints**: `Propagation.NESTED` creates independent savepoints that can rollback without affecting the parent.
+- **Propagation.NEW**: Logs a warning and creates a new savepoint instead (true separate transactions are not possible within the external transaction).
+- **Single client**: Does not support concurrent `Promise.all` wrap calls (PostgreSQL limitation with single connection).
+
+```typescript
+// abortError behavior - matches production
+await uow.wrap(async (c) => {
+    await insertOrder(c, 1);
+
+    try {
+        await uow.wrap(async () => {
+            throw new Error("fails");
+        });
+    } catch {
+        // Caught, but transaction is already marked as aborted
+    }
+
+    await insertOrder(c, 2);  // Executes, but will be rolled back
+});
+// Result: Both orders rolled back (abortError propagation)
+
+// NESTED savepoint - independent rollback
+await uow.wrap(async (c) => {
+    await insertOrder(c, 1);
+
+    try {
+        await uow.wrap(async (sp) => {
+            await insertOrder(sp, 2);
+            throw new Error("fails");
+        }, { propagation: Propagation.NESTED });
+    } catch {
+        // Only savepoint rolled back
+    }
+
+    await insertOrder(c, 3);
 });
+// Result: Orders 1 and 3 committed, order 2 rolled back
 ```
 
 ## API Highlights
@@ -301,6 +367,7 @@ describe("OrderRepository", () => {
 | Export | Description |
 |--------|-------------|
 | `PostgresUnitOfWork` | Transaction management with `AsyncLocalStorage` context |
+| `PostgresUnitOfWorkForTesting` | Test-specific UnitOfWork that runs inside external transaction |
 | `PostgresEventStore` | Event store implementation with batch insert support |
 | `PostgresConfig` | Immutable configuration with builder pattern |
 | `postgresConfig` | Config spec for `defineConfig` integration |

package/dist/config/postgres-config-spec.d.ts

@@ -1,4 +1,4 @@
-import type { ConfigSpec } from "@hexaijs/utils/config";
+import type { ConfigSpec } from "ezcfg";
 import { PostgresConfig, type FromEnvOptions } from "./postgres-config";
 export declare class PostgresConfigSpec implements ConfigSpec<PostgresConfig> {
     private readonly prefix;

package/dist/config/postgres-config-spec.d.ts.map

@@ -1 +1 @@
-{"version":3,"file":"postgres-config-spec.d.ts","sourceRoot":"","sources":["../../src/config/postgres-config-spec.ts"],"names":[],"mappings":"AAAA,OAAO,KAAK,EAAE,UAAU,EAAE,MAAM,uBAAuB,CAAC;AACxD,OAAO,EAAE,cAAc,EAAE,KAAK,cAAc,EAAE,MAAM,mBAAmB,CAAC;AAExE,qBAAa,kBAAmB,YAAW,UAAU,CAAC,cAAc,CAAC;IAI7D,OAAO,CAAC,QAAQ,CAAC,MAAM;IACvB,OAAO,CAAC,QAAQ,CAAC,IAAI;IAJzB,QAAQ,CAAC,KAAK,cAAc;gBAGP,MAAM,EAAE,MAAM,EACd,IAAI,GAAE,cAAc,CAAC,MAAM,CAAS;IAGzD,OAAO,CAAC,MAAM,EAAE,MAAM,EAAE,GAAG,cAAc,GAAG,SAAS;CAQxD;AAED;;;;;;;;;;;;;;;;;;;;GAoBG;AACH,wBAAgB,cAAc,CAC1B,MAAM,EAAE,MAAM,EACd,IAAI,GAAE,cAAc,CAAC,MAAM,CAAS,GACrC,kBAAkB,CAEpB"}
+{"version":3,"file":"postgres-config-spec.d.ts","sourceRoot":"","sources":["../../src/config/postgres-config-spec.ts"],"names":[],"mappings":"AAAA,OAAO,KAAK,EAAE,UAAU,EAAE,MAAM,OAAO,CAAC;AACxC,OAAO,EAAE,cAAc,EAAE,KAAK,cAAc,EAAE,MAAM,mBAAmB,CAAC;AAExE,qBAAa,kBAAmB,YAAW,UAAU,CAAC,cAAc,CAAC;IAI7D,OAAO,CAAC,QAAQ,CAAC,MAAM;IACvB,OAAO,CAAC,QAAQ,CAAC,IAAI;IAJzB,QAAQ,CAAC,KAAK,cAAc;gBAGP,MAAM,EAAE,MAAM,EACd,IAAI,GAAE,cAAc,CAAC,MAAM,CAAS;IAGzD,OAAO,CAAC,MAAM,EAAE,MAAM,EAAE,GAAG,cAAc,GAAG,SAAS;CAQxD;AAED;;;;;;;;;;;;;;;;;;;;GAoBG;AACH,wBAAgB,cAAc,CAC1B,MAAM,EAAE,MAAM,EACd,IAAI,GAAE,cAAc,CAAC,MAAM,CAAS,GACrC,kBAAkB,CAEpB"}

package/dist/config/postgres-config.d.ts

@@ -1,4 +1,4 @@
-import type { DatabaseConfig } from "@hexaijs/utils/config";
+import type { DatabaseConfig } from "ezcfg";
 export interface PoolOptions {
     size?: number;
     connectionTimeout?: number;

package/dist/config/postgres-config.d.ts.map

@@ -1 +1 @@
-{"version":3,"file":"postgres-config.d.ts","sourceRoot":"","sources":["../../src/config/postgres-config.ts"],"names":[],"mappings":"AAAA,OAAO,KAAK,EAAE,cAAc,EAAE,MAAM,uBAAuB,CAAC;AAE5D,MAAM,WAAW,WAAW;IACxB,IAAI,CAAC,EAAE,MAAM,CAAC;IACd,iBAAiB,CAAC,EAAE,MAAM,CAAC;IAC3B,WAAW,CAAC,EAAE,MAAM,CAAC;CACxB;AAED,MAAM,WAAW,cAAc;IAC3B;;;;OAIG;IACH,IAAI,CAAC,EAAE,KAAK,GAAG,QAAQ,CAAC;CAC3B;AAED,qBAAa,cAAe,YAAW,cAAc;IACjD,SAAgB,IAAI,EAAE,MAAM,CAAC;IAC7B,SAAgB,QAAQ,EAAE,MAAM,CAAC;IACjC,SAAgB,IAAI,EAAE,MAAM,CAAC;IAC7B,SAAgB,IAAI,EAAE,MAAM,CAAC;IAC7B,SAAgB,QAAQ,CAAC,EAAE,MAAM,CAAC;IAClC,SAAgB,IAAI,CAAC,EAAE,WAAW,CAAC;gBAEvB,MAAM,EAAE;QAChB,QAAQ,EAAE,MAAM,CAAC;QACjB,IAAI,CAAC,EAAE,MAAM,CAAC;QACd,IAAI,CAAC,EAAE,MAAM,CAAC;QACd,IAAI,CAAC,EAAE,MAAM,CAAC;QACd,QAAQ,CAAC,EAAE,MAAM,CAAC;QAClB,IAAI,CAAC,EAAE,WAAW,CAAC;KACtB;WASa,OAAO,CAAC,KAAK,EAAE,MAAM,GAAG,cAAc;IAIpD;;;;;;;;;;;;;;;OAeG;WACW,OAAO,CACjB,MAAM,EAAE,MAAM,EACd,OAAO,CAAC,EAAE,cAAc,GACzB,cAAc;IAiCjB,OAAO,CAAC,MAAM,CAAC,QAAQ;IAqBhB,YAAY,CAAC,QAAQ,EAAE,MAAM,GAAG,cAAc;IAW9C,QAAQ,CAAC,IAAI,EAAE,MAAM,GAAG,cAAc;IAWtC,YAAY,CAAC,QAAQ,EAAE,MAAM,GAAG,cAAc;IAW9C,QAAQ,CAAC,IAAI,EAAE,MAAM,GAAG,cAAc;IAWtC,QAAQ,CAAC,IAAI,EAAE,MAAM,GAAG,cAAc;IAWtC,YAAY,CAAC,IAAI,EAAE,MAAM,GAAG,cAAc;IAW1C,qBAAqB,CAAC,iBAAiB,EAAE,MAAM,GAAG,cAAc;IAWhE,eAAe,CAAC,WAAW,EAAE,MAAM,GAAG,cAAc;IAWpD,QAAQ,IAAI,MAAM;CA2B5B"}
+{"version":3,"file":"postgres-config.d.ts","sourceRoot":"","sources":["../../src/config/postgres-config.ts"],"names":[],"mappings":"AAAA,OAAO,KAAK,EAAE,cAAc,EAAE,MAAM,OAAO,CAAC;AAE5C,MAAM,WAAW,WAAW;IACxB,IAAI,CAAC,EAAE,MAAM,CAAC;IACd,iBAAiB,CAAC,EAAE,MAAM,CAAC;IAC3B,WAAW,CAAC,EAAE,MAAM,CAAC;CACxB;AAED,MAAM,WAAW,cAAc;IAC3B;;;;OAIG;IACH,IAAI,CAAC,EAAE,KAAK,GAAG,QAAQ,CAAC;CAC3B;AAED,qBAAa,cAAe,YAAW,cAAc;IACjD,SAAgB,IAAI,EAAE,MAAM,CAAC;IAC7B,SAAgB,QAAQ,EAAE,MAAM,CAAC;IACjC,SAAgB,IAAI,EAAE,MAAM,CAAC;IAC7B,SAAgB,IAAI,EAAE,MAAM,CAAC;IAC7B,SAAgB,QAAQ,CAAC,EAAE,MAAM,CAAC;IAClC,SAAgB,IAAI,CAAC,EAAE,WAAW,CAAC;gBAEvB,MAAM,EAAE;QAChB,QAAQ,EAAE,MAAM,CAAC;QACjB,IAAI,CAAC,EAAE,MAAM,CAAC;QACd,IAAI,CAAC,EAAE,MAAM,CAAC;QACd,IAAI,CAAC,EAAE,MAAM,CAAC;QACd,QAAQ,CAAC,EAAE,MAAM,CAAC;QAClB,IAAI,CAAC,EAAE,WAAW,CAAC;KACtB;WASa,OAAO,CAAC,KAAK,EAAE,MAAM,GAAG,cAAc;IAIpD;;;;;;;;;;;;;;;OAeG;WACW,OAAO,CACjB,MAAM,EAAE,MAAM,EACd,OAAO,CAAC,EAAE,cAAc,GACzB,cAAc;IAiCjB,OAAO,CAAC,MAAM,CAAC,QAAQ;IAqBhB,YAAY,CAAC,QAAQ,EAAE,MAAM,GAAG,cAAc;IAW9C,QAAQ,CAAC,IAAI,EAAE,MAAM,GAAG,cAAc;IAWtC,YAAY,CAAC,QAAQ,EAAE,MAAM,GAAG,cAAc;IAW9C,QAAQ,CAAC,IAAI,EAAE,MAAM,GAAG,cAAc;IAWtC,QAAQ,CAAC,IAAI,EAAE,MAAM,GAAG,cAAc;IAWtC,YAAY,CAAC,IAAI,EAAE,MAAM,GAAG,cAAc;IAW1C,qBAAqB,CAAC,iBAAiB,EAAE,MAAM,GAAG,cAAc;IAWhE,eAAe,CAAC,WAAW,EAAE,MAAM,GAAG,cAAc;IAWpD,QAAQ,IAAI,MAAM;CA2B5B"}

package/dist/test-fixtures/config.d.ts

@@ -1,4 +1,4 @@
-export declare const getTestConfig: () => import("@hexaijs/utils/config").InferConfigType<{
+export declare const getTestConfig: () => import("ezcfg").InferConfigType<{
     db: import("../config").PostgresConfigSpec;
     migrationsDir: string;
 }>;

package/dist/test-fixtures/config.js

@@ -6,8 +6,8 @@ Object.defineProperty(exports, "__esModule", { value: true });
 exports.getTestConfig = void 0;
 const node_path_1 = __importDefault(require("node:path"));
 const config_1 = require("../config");
-const config_2 = require("@hexaijs/utils/config");
-exports.getTestConfig = (0, config_2.defineConfig)({
+const ezcfg_1 = require("ezcfg");
+exports.getTestConfig = (0, ezcfg_1.defineConfig)({
     db: (0, config_1.postgresConfig)("HEXAI_DB"),
     migrationsDir: node_path_1.default.join(__dirname + "/../migrations"),
 });

package/dist/test-fixtures/config.js.map

@@ -1 +1 @@
-{"version":3,"file":"config.js","sourceRoot":"","sources":["../../src/test-fixtures/config.ts"],"names":[],"mappings":";;;;;;AAAA,0DAA6B;AAE7B,qCAA0C;AAC1C,kDAAqD;AAExC,QAAA,aAAa,GAAG,IAAA,qBAAY,EAAC;IACtC,EAAE,EAAE,IAAA,uBAAc,EAAC,UAAU,CAAC;IAC9B,aAAa,EAAE,mBAAI,CAAC,IAAI,CAAC,SAAS,GAAG,gBAAgB,CAAC;CACzD,CAAC,CAAC"}
+{"version":3,"file":"config.js","sourceRoot":"","sources":["../../src/test-fixtures/config.ts"],"names":[],"mappings":";;;;;;AAAA,0DAA6B;AAE7B,qCAA0C;AAC1C,iCAAqC;AAExB,QAAA,aAAa,GAAG,IAAA,oBAAY,EAAC;IACtC,EAAE,EAAE,IAAA,uBAAc,EAAC,UAAU,CAAC;IAC9B,aAAa,EAAE,mBAAI,CAAC,IAAI,CAAC,SAAS,GAAG,gBAAgB,CAAC;CACzD,CAAC,CAAC"}

package/dist/test.d.ts

@@ -1,6 +1,8 @@
 import { Client } from "pg";
+import { UnitOfWork } from "@hexaijs/core";
 import { TableManager } from "./helpers";
 import { PostgresConfig } from "./config";
+import { PostgresTransactionOptions } from "./types";
 export declare function createTestContext(dbUrl: string | PostgresConfig): {
     client: Client;
     newClient: () => Client;
@@ -8,4 +10,16 @@ export declare function createTestContext(dbUrl: string | PostgresConfig): {
     setup: () => Promise<void>;
     teardown: () => Promise<void>;
 };
+export declare class PostgresUnitOfWorkForTesting implements UnitOfWork<Client, PostgresTransactionOptions> {
+    private client;
+    private executorStorage;
+    constructor(client: Client);
+    getClient(): Client;
+    wrap<T = unknown>(fn: (client: Client) => Promise<T>, options?: Partial<PostgresTransactionOptions>): Promise<T>;
+    private getCurrentExecutor;
+    private resolveOptions;
+    private resolveExecutor;
+    private createExecutor;
+    private executeInContext;
+}
 //# sourceMappingURL=test.d.ts.map

package/dist/test.d.ts.map

@@ -1 +1 @@
-{"version":3,"file":"test.d.ts","sourceRoot":"","sources":["../src/test.ts"],"names":[],"mappings":"AAAA,OAAO,EAAE,MAAM,EAAE,MAAM,IAAI,CAAC;AAE5B,OAAO,EAAoC,YAAY,EAAE,MAAM,WAAW,CAAC;AAC3E,OAAO,EAAE,cAAc,EAAE,MAAM,UAAU,CAAC;AAG1C,wBAAgB,iBAAiB,CAAC,KAAK,EAAE,MAAM,GAAG,cAAc;;;;iBAUpC,OAAO,CAAC,IAAI,CAAC;oBAeV,OAAO,CAAC,IAAI,CAAC;EAa3C"}
+{"version":3,"file":"test.d.ts","sourceRoot":"","sources":["../src/test.ts"],"names":[],"mappings":"AACA,OAAO,EAAE,MAAM,EAAE,MAAM,IAAI,CAAC;AAE5B,OAAO,EAAe,UAAU,EAAE,MAAM,eAAe,CAAC;AACxD,OAAO,EAAoC,YAAY,EAAE,MAAM,WAAW,CAAC;AAC3E,OAAO,EAAE,cAAc,EAAE,MAAM,UAAU,CAAC;AAE1C,OAAO,EAAE,0BAA0B,EAAE,MAAM,SAAS,CAAC;AAErD,wBAAgB,iBAAiB,CAAC,KAAK,EAAE,MAAM,GAAG,cAAc;;;;iBAUpC,OAAO,CAAC,IAAI,CAAC;oBAeV,OAAO,CAAC,IAAI,CAAC;EAa3C;AAED,qBAAa,4BACT,YAAW,UAAU,CAAC,MAAM,EAAE,0BAA0B,CAAC;IAI7C,OAAO,CAAC,MAAM;IAF1B,OAAO,CAAC,eAAe,CAAoD;gBAEvD,MAAM,EAAE,MAAM;IAE3B,SAAS,IAAI,MAAM;IAQpB,IAAI,CAAC,CAAC,GAAG,OAAO,EAClB,EAAE,EAAE,CAAC,MAAM,EAAE,MAAM,KAAK,OAAO,CAAC,CAAC,CAAC,EAClC,OAAO,GAAE,OAAO,CAAC,0BAA0B,CAAM,GAClD,OAAO,CAAC,CAAC,CAAC;IASb,OAAO,CAAC,kBAAkB;IAI1B,OAAO,CAAC,cAAc;IAStB,OAAO,CAAC,eAAe;IAYvB,OAAO,CAAC,cAAc;IAItB,OAAO,CAAC,gBAAgB;CAM3B"}

package/dist/test.js

@@ -1,7 +1,10 @@
 "use strict";
 Object.defineProperty(exports, "__esModule", { value: true });
+exports.PostgresUnitOfWorkForTesting = void 0;
 exports.createTestContext = createTestContext;
+const node_async_hooks_1 = require("node:async_hooks");
 const pg_1 = require("pg");
+const core_1 = require("@hexaijs/core");
 const helpers_1 = require("./helpers");
 const config_1 = require("./config");
 const run_hexai_migrations_1 = require("./run-hexai-migrations");
@@ -38,4 +41,216 @@ function createTestContext(dbUrl) {
         teardown,
     };
 }
+class PostgresUnitOfWorkForTesting {
+    client;
+    executorStorage = new node_async_hooks_1.AsyncLocalStorage();
+    constructor(client) {
+        this.client = client;
+    }
+    getClient() {
+        const executor = this.getCurrentExecutor();
+        if (!executor) {
+            throw new Error("Unit of work not started");
+        }
+        return this.client;
+    }
+    async wrap(fn, options = {}) {
+        const resolvedOptions = this.resolveOptions(options);
+        const executor = this.resolveExecutor(resolvedOptions);
+        return this.executeInContext(executor, (exec) => exec.execute(fn, resolvedOptions));
+    }
+    getCurrentExecutor() {
+        return this.executorStorage.getStore() ?? null;
+    }
+    resolveOptions(options) {
+        return {
+            propagation: core_1.Propagation.EXISTING,
+            ...options,
+        };
+    }
+    resolveExecutor(options) {
+        if (options.propagation === core_1.Propagation.NEW) {
+            console.warn("[PostgresUnitOfWorkForTesting] Propagation.NEW is not fully supported in testing mode. Using savepoint instead.");
+            return this.createExecutor();
+        }
+        return this.getCurrentExecutor() ?? this.createExecutor();
+    }
+    createExecutor() {
+        return new TestTransactionExecutor(this.client);
+    }
+    executeInContext(executor, callback) {
+        return this.executorStorage.run(executor, () => callback(executor));
+    }
+}
+exports.PostgresUnitOfWorkForTesting = PostgresUnitOfWorkForTesting;
+class TestTransactionExecutor {
+    client;
+    initialized = false;
+    closed = false;
+    abortError;
+    nestingDepth = 0;
+    savepointCounter = 0;
+    savepoints = [];
+    savepointName;
+    constructor(client) {
+        this.client = client;
+        this.savepointName = `test_sp_${Date.now()}_${Math.random().toString(36).slice(2, 8)}`;
+    }
+    async execute(fn, options) {
+        await this.ensureStarted();
+        const executor = this.resolveExecutor(options.propagation);
+        return executor === this
+            ? this.runWithLifecycle(fn)
+            : executor.execute(fn, options);
+    }
+    getClient() {
+        return this.client;
+    }
+    async ensureStarted() {
+        if (this.initialized) {
+            return;
+        }
+        this.initialized = true;
+        await this.client.query(`SAVEPOINT ${this.savepointName}`);
+    }
+    async runWithLifecycle(fn) {
+        try {
+            return await this.executeWithNesting(fn);
+        }
+        catch (e) {
+            this.markAsAborted(e);
+            throw e;
+        }
+        finally {
+            await this.finalizeIfRoot();
+        }
+    }
+    async executeWithNesting(fn) {
+        this.nestingDepth++;
+        try {
+            return await fn(this.client);
+        }
+        finally {
+            this.nestingDepth--;
+        }
+    }
+    markAsAborted(error) {
+        this.abortError = error;
+    }
+    async finalizeIfRoot() {
+        if (this.nestingDepth === 0) {
+            await (this.isAborted() ? this.rollback() : this.commit());
+        }
+    }
+    resolveExecutor(propagation) {
+        if (this.nestingDepth === 0) {
+            return this;
+        }
+        return propagation === core_1.Propagation.NESTED
+            ? this.createSavepoint()
+            : (this.findActiveSavepoint() ?? this);
+    }
+    createSavepoint() {
+        this.savepointCounter++;
+        const savepoint = new TestSavepoint(`${this.savepointName}_nested_${this.savepointCounter}`, this.client, () => this.removeSavepoint());
+        this.savepoints.push(savepoint);
+        return savepoint;
+    }
+    findActiveSavepoint() {
+        for (let i = this.savepoints.length - 1; i >= 0; i--) {
+            if (!this.savepoints[i].isClosed()) {
+                return this.savepoints[i];
+            }
+        }
+    }
+    removeSavepoint() {
+        this.savepoints.pop();
+    }
+    async commit() {
+        if (this.closed) {
+            return;
+        }
+        this.closed = true;
+        await this.client.query(`RELEASE SAVEPOINT ${this.savepointName}`);
+    }
+    async rollback() {
+        if (this.closed) {
+            return;
+        }
+        this.closed = true;
+        await this.client.query(`ROLLBACK TO SAVEPOINT ${this.savepointName}`);
+    }
+    isAborted() {
+        return this.abortError !== undefined && !this.closed;
+    }
+}
+class TestSavepoint {
+    name;
+    client;
+    onClose;
+    initialized = false;
+    closed = false;
+    abortError;
+    nestingDepth = 0;
+    constructor(name, client, onClose) {
+        this.name = name;
+        this.client = client;
+        this.onClose = onClose;
+    }
+    async execute(fn) {
+        await this.ensureStarted();
+        return this.runWithLifecycle(fn);
+    }
+    isClosed() {
+        return this.closed;
+    }
+    async ensureStarted() {
+        if (this.initialized) {
+            return;
+        }
+        this.initialized = true;
+        await this.client.query(`SAVEPOINT ${this.name}`);
+    }
+    async runWithLifecycle(fn) {
+        this.nestingDepth++;
+        try {
+            return await fn(this.client);
+        }
+        catch (e) {
+            this.markAsAborted(e);
+            throw e;
+        }
+        finally {
+            this.nestingDepth--;
+            await this.finalizeIfRoot();
+        }
+    }
+    markAsAborted(error) {
+        this.abortError = error;
+    }
+    async finalizeIfRoot() {
+        if (this.nestingDepth === 0) {
+            await (this.isAborted() ? this.rollback() : this.commit());
+        }
+    }
+    async commit() {
+        if (this.closed) {
+            return;
+        }
+        this.closed = true;
+        await this.client.query(`RELEASE SAVEPOINT ${this.name}`);
+        this.onClose();
+    }
+    async rollback() {
+        if (this.closed) {
+            return;
+        }
+        this.closed = true;
+        await this.client.query(`ROLLBACK TO SAVEPOINT ${this.name}`);
+        this.onClose();
+    }
+    isAborted() {
+        return this.abortError !== undefined && !this.closed;
+    }
+}
 //# sourceMappingURL=test.js.map

package/dist/test.js.map

@@ -1 +1 @@
-{"version":3,"file":"test.js","sourceRoot":"","sources":["../src/test.ts"],"names":[],"mappings":";;AAMA,8CAsCC;AA5CD,2BAA4B;AAE5B,uCAA2E;AAC3E,qCAA0C;AAC1C,iEAA4D;AAE5D,SAAgB,iBAAiB,CAAC,KAA8B;IAC5D,MAAM,MAAM,GACR,OAAO,KAAK,KAAK,QAAQ,CAAC,CAAC,CAAC,uBAAc,CAAC,OAAO,CAAC,KAAK,CAAC,CAAC,CAAC,CAAC,KAAK,CAAC;IAEtE,MAAM,MAAM,GAAG,MAAM,CAAC,QAAQ,CAAC;IAC/B,MAAM,eAAe,GAAG,IAAI,yBAAe,CACvC,MAAM,CAAC,YAAY,CAAC,UAAU,CAAC,CAClC,CAAC;IACF,MAAM,YAAY,GAAG,IAAI,sBAAY,CAAC,MAAM,CAAC,CAAC;IAE9C,KAAK,UAAU,KAAK;QAChB,IAAI,CAAC;YACD,MAAM,eAAe,CAAC,YAAY,CAAC,MAAM,CAAC,CAAC;QAC/C,CAAC;QAAC,OAAO,CAAC,EAAE,CAAC;YACT,IAAI,IAAA,yBAAe,EAAC,CAAC,CAAC,IAAI,CAAC,CAAC,IAAI,KAAK,OAAO,EAAE,CAAC;gBAC3C,SAAS;YACb,CAAC;iBAAM,CAAC;gBACJ,MAAM,CAAC,CAAC;YACZ,CAAC;QACL,CAAC;QAED,MAAM,eAAe,CAAC,cAAc,CAAC,MAAM,CAAC,CAAC;QAC7C,MAAM,IAAA,yCAAkB,EAAC,MAAM,CAAC,CAAC;IACrC,CAAC;IAED,KAAK,UAAU,QAAQ;QACnB,MAAM,YAAY,CAAC,KAAK,EAAE,CAAC;QAC3B,MAAM,eAAe,CAAC,YAAY,CAAC,MAAM,CAAC,CAAC;QAC3C,MAAM,eAAe,CAAC,KAAK,EAAE,CAAC;IAClC,CAAC;IAED,OAAO;QACH,MAAM,EAAE,YAAY,CAAC,SAAS,EAAE;QAChC,SAAS,EAAE,GAAG,EAAE,CAAC,IAAI,WAAM,CAAC,KAAK,CAAC;QAClC,YAAY;QACZ,KAAK;QACL,QAAQ;KACX,CAAC;AACN,CAAC"}
+{"version":3,"file":"test.js","sourceRoot":"","sources":["../src/test.ts"],"names":[],"mappings":";;;AASA,8CAsCC;AA/CD,uDAAqD;AACrD,2BAA4B;AAE5B,wCAAwD;AACxD,uCAA2E;AAC3E,qCAA0C;AAC1C,iEAA4D;AAG5D,SAAgB,iBAAiB,CAAC,KAA8B;IAC5D,MAAM,MAAM,GACR,OAAO,KAAK,KAAK,QAAQ,CAAC,CAAC,CAAC,uBAAc,CAAC,OAAO,CAAC,KAAK,CAAC,CAAC,CAAC,CAAC,KAAK,CAAC;IAEtE,MAAM,MAAM,GAAG,MAAM,CAAC,QAAQ,CAAC;IAC/B,MAAM,eAAe,GAAG,IAAI,yBAAe,CACvC,MAAM,CAAC,YAAY,CAAC,UAAU,CAAC,CAClC,CAAC;IACF,MAAM,YAAY,GAAG,IAAI,sBAAY,CAAC,MAAM,CAAC,CAAC;IAE9C,KAAK,UAAU,KAAK;QAChB,IAAI,CAAC;YACD,MAAM,eAAe,CAAC,YAAY,CAAC,MAAM,CAAC,CAAC;QAC/C,CAAC;QAAC,OAAO,CAAC,EAAE,CAAC;YACT,IAAI,IAAA,yBAAe,EAAC,CAAC,CAAC,IAAI,CAAC,CAAC,IAAI,KAAK,OAAO,EAAE,CAAC;gBAC3C,SAAS;YACb,CAAC;iBAAM,CAAC;gBACJ,MAAM,CAAC,CAAC;YACZ,CAAC;QACL,CAAC;QAED,MAAM,eAAe,CAAC,cAAc,CAAC,MAAM,CAAC,CAAC;QAC7C,MAAM,IAAA,yCAAkB,EAAC,MAAM,CAAC,CAAC;IACrC,CAAC;IAED,KAAK,UAAU,QAAQ;QACnB,MAAM,YAAY,CAAC,KAAK,EAAE,CAAC;QAC3B,MAAM,eAAe,CAAC,YAAY,CAAC,MAAM,CAAC,CAAC;QAC3C,MAAM,eAAe,CAAC,KAAK,EAAE,CAAC;IAClC,CAAC;IAED,OAAO;QACH,MAAM,EAAE,YAAY,CAAC,SAAS,EAAE;QAChC,SAAS,EAAE,GAAG,EAAE,CAAC,IAAI,WAAM,CAAC,KAAK,CAAC;QAClC,YAAY;QACZ,KAAK;QACL,QAAQ;KACX,CAAC;AACN,CAAC;AAED,MAAa,4BAA4B;IAKjB;IAFZ,eAAe,GAAG,IAAI,oCAAiB,EAA2B,CAAC;IAE3E,YAAoB,MAAc;QAAd,WAAM,GAAN,MAAM,CAAQ;IAAG,CAAC;IAE/B,SAAS;QACZ,MAAM,QAAQ,GAAG,IAAI,CAAC,kBAAkB,EAAE,CAAC;QAC3C,IAAI,CAAC,QAAQ,EAAE,CAAC;YACZ,MAAM,IAAI,KAAK,CAAC,0BAA0B,CAAC,CAAC;QAChD,CAAC;QACD,OAAO,IAAI,CAAC,MAAM,CAAC;IACvB,CAAC;IAED,KAAK,CAAC,IAAI,CACN,EAAkC,EAClC,UAA+C,EAAE;QAEjD,MAAM,eAAe,GAAG,IAAI,CAAC,cAAc,CAAC,OAAO,CAAC,CAAC;QACrD,MAAM,QAAQ,GAAG,IAAI,CAAC,eAAe,CAAC,eAAe,CAAC,CAAC;QAEvD,OAAO,IAAI,CAAC,gBAAgB,CAAC,QAAQ,EAAE,CAAC,IAAI,EAAE,EAAE,CAC5C,IAAI,CAAC,OAAO,CAAC,EAAE,EAAE,eAAe,CAAC,CACpC,CAAC;IACN,CAAC;IAEO,kBAAkB;QACtB,OAAO,IAAI,CAAC,eAAe,CAAC,QAAQ,EAAE,IAAI,IAAI,CAAC;IACnD,CAAC;IAEO,cAAc,CAClB,OAA4C;QAE5C,OAAO;YACH,WAAW,EAAE,kBAAW,CAAC,QAAQ;YACjC,GAAG,OAAO;SACb,CAAC;IACN,CAAC;IAEO,eAAe,CACnB,OAAmC;QAEnC,IAAI,OAAO,CAAC,WAAW,KAAK,kBAAW,CAAC,GAAG,EAAE,CAAC;YAC1C,OAAO,CAAC,IAAI,CACR,iHAAiH,CACpH,CAAC;YACF,OAAO,IAAI,CAAC,cAAc,EAAE,CAAC;QACjC,CAAC;QACD,OAAO,IAAI,CAAC,kBAAkB,EAAE,IAAI,IAAI,CAAC,cAAc,EAAE,CAAC;IAC9D,CAAC;IAEO,cAAc;QAClB,OAAO,IAAI,uBAAuB,CAAC,IAAI,CAAC,MAAM,CAAC,CAAC;IACpD,CAAC;IAEO,gBAAgB,CACpB,QAAiC,EACjC,QAA2D;QAE3D,OAAO,IAAI,CAAC,eAAe,CAAC,GAAG,CAAC,QAAQ,EAAE,GAAG,EAAE,CAAC,QAAQ,CAAC,QAAQ,CAAC,CAAC,CAAC;IACxE,CAAC;CACJ;AA9DD,oEA8DC;AAED,MAAM,uBAAuB;IAUI;IATrB,WAAW,GAAG,KAAK,CAAC;IACpB,MAAM,GAAG,KAAK,CAAC;IACf,UAAU,CAAS;IAEnB,YAAY,GAAG,CAAC,CAAC;IACjB,gBAAgB,GAAG,CAAC,CAAC;IACrB,UAAU,GAAoB,EAAE,CAAC;IACjC,aAAa,CAAS;IAE9B,YAA6B,MAAc;QAAd,WAAM,GAAN,MAAM,CAAQ;QACvC,IAAI,CAAC,aAAa,GAAG,WAAW,IAAI,CAAC,GAAG,EAAE,IAAI,IAAI,CAAC,MAAM,EAAE,CAAC,QAAQ,CAAC,EAAE,CAAC,CAAC,KAAK,CAAC,CAAC,EAAE,CAAC,CAAC,EAAE,CAAC;IAC3F,CAAC;IAEM,KAAK,CAAC,OAAO,CAChB,EAAkC,EAClC,OAAmC;QAEnC,MAAM,IAAI,CAAC,aAAa,EAAE,CAAC;QAE3B,MAAM,QAAQ,GAAG,IAAI,CAAC,eAAe,CAAC,OAAO,CAAC,WAAW,CAAC,CAAC;QAC3D,OAAO,QAAQ,KAAK,IAAI;YACpB,CAAC,CAAC,IAAI,CAAC,gBAAgB,CAAC,EAAE,CAAC;YAC3B,CAAC,CAAC,QAAQ,CAAC,OAAO,CAAC,EAAE,EAAE,OAAO,CAAC,CAAC;IACxC,CAAC;IAEM,SAAS;QACZ,OAAO,IAAI,CAAC,MAAM,CAAC;IACvB,CAAC;IAEO,KAAK,CAAC,aAAa;QACvB,IAAI,IAAI,CAAC,WAAW,EAAE,CAAC;YACnB,OAAO;QACX,CAAC;QAED,IAAI,CAAC,WAAW,GAAG,IAAI,CAAC;QACxB,MAAM,IAAI,CAAC,MAAM,CAAC,KAAK,CAAC,aAAa,IAAI,CAAC,aAAa,EAAE,CAAC,CAAC;IAC/D,CAAC;IAEO,KAAK,CAAC,gBAAgB,CAC1B,EAAkC;QAElC,IAAI,CAAC;YACD,OAAO,MAAM,IAAI,CAAC,kBAAkB,CAAC,EAAE,CAAC,CAAC;QAC7C,CAAC;QAAC,OAAO,CAAC,EAAE,CAAC;YACT,IAAI,CAAC,aAAa,CAAC,CAAU,CAAC,CAAC;YAC/B,MAAM,CAAC,CAAC;QACZ,CAAC;gBAAS,CAAC;YACP,MAAM,IAAI,CAAC,cAAc,EAAE,CAAC;QAChC,CAAC;IACL,CAAC;IAEO,KAAK,CAAC,kBAAkB,CAC5B,EAAkC;QAElC,IAAI,CAAC,YAAY,EAAE,CAAC;QACpB,IAAI,CAAC;YACD,OAAO,MAAM,EAAE,CAAC,IAAI,CAAC,MAAM,CAAC,CAAC;QACjC,CAAC;gBAAS,CAAC;YACP,IAAI,CAAC,YAAY,EAAE,CAAC;QACxB,CAAC;IACL,CAAC;IAEO,aAAa,CAAC,KAAY;QAC9B,IAAI,CAAC,UAAU,GAAG,KAAK,CAAC;IAC5B,CAAC;IAEO,KAAK,CAAC,cAAc;QACxB,IAAI,IAAI,CAAC,YAAY,KAAK,CAAC,EAAE,CAAC;YAC1B,MAAM,CAAC,IAAI,CAAC,SAAS,EAAE,CAAC,CAAC,CAAC,IAAI,CAAC,QAAQ,EAAE,CAAC,CAAC,CAAC,IAAI,CAAC,MAAM,EAAE,CAAC,CAAC;QAC/D,CAAC;IACL,CAAC;IAEO,eAAe,CACnB,WAAwB;QAExB,IAAI,IAAI,CAAC,YAAY,KAAK,CAAC,EAAE,CAAC;YAC1B,OAAO,IAAI,CAAC;QAChB,CAAC;QAED,OAAO,WAAW,KAAK,kBAAW,CAAC,MAAM;YACrC,CAAC,CAAC,IAAI,CAAC,eAAe,EAAE;YACxB,CAAC,CAAC,CAAC,IAAI,CAAC,mBAAmB,EAAE,IAAI,IAAI,CAAC,CAAC;IAC/C,CAAC;IAEO,eAAe;QACnB,IAAI,CAAC,gBAAgB,EAAE,CAAC;QACxB,MAAM,SAAS,GAAG,IAAI,aAAa,CAC/B,GAAG,IAAI,CAAC,aAAa,WAAW,IAAI,CAAC,gBAAgB,EAAE,EACvD,IAAI,CAAC,MAAM,EACX,GAAG,EAAE,CAAC,IAAI,CAAC,eAAe,EAAE,CAC/B,CAAC;QACF,IAAI,CAAC,UAAU,CAAC,IAAI,CAAC,SAAS,CAAC,CAAC;QAChC,OAAO,SAAS,CAAC;IACrB,CAAC;IAEO,mBAAmB;QACvB,KAAK,IAAI,CAAC,GAAG,IAAI,CAAC,UAAU,CAAC,MAAM,GAAG,CAAC,EAAE,CAAC,IAAI,CAAC,EAAE,CAAC,EAAE,EAAE,CAAC;YACnD,IAAI,CAAC,IAAI,CAAC,UAAU,CAAC,CAAC,CAAC,CAAC,QAAQ,EAAE,EAAE,CAAC;gBACjC,OAAO,IAAI,CAAC,UAAU,CAAC,CAAC,CAAC,CAAC;YAC9B,CAAC;QACL,CAAC;IACL,CAAC;IAEO,eAAe;QACnB,IAAI,CAAC,UAAU,CAAC,GAAG,EAAE,CAAC;IAC1B,CAAC;IAEO,KAAK,CAAC,MAAM;QAChB,IAAI,IAAI,CAAC,MAAM,EAAE,CAAC;YACd,OAAO;QACX,CAAC;QAED,IAAI,CAAC,MAAM,GAAG,IAAI,CAAC;QACnB,MAAM,IAAI,CAAC,MAAM,CAAC,KAAK,CAAC,qBAAqB,IAAI,CAAC,aAAa,EAAE,CAAC,CAAC;IACvE,CAAC;IAEO,KAAK,CAAC,QAAQ;QAClB,IAAI,IAAI,CAAC,MAAM,EAAE,CAAC;YACd,OAAO;QACX,CAAC;QAED,IAAI,CAAC,MAAM,GAAG,IAAI,CAAC;QACnB,MAAM,IAAI,CAAC,MAAM,CAAC,KAAK,CACnB,yBAAyB,IAAI,CAAC,aAAa,EAAE,CAChD,CAAC;IACN,CAAC;IAEO,SAAS;QACb,OAAO,IAAI,CAAC,UAAU,KAAK,SAAS,IAAI,CAAC,IAAI,CAAC,MAAM,CAAC;IACzD,CAAC;CACJ;AAED,MAAM,aAAa;IAQM;IACA;IACA;IATb,WAAW,GAAG,KAAK,CAAC;IACpB,MAAM,GAAG,KAAK,CAAC;IACf,UAAU,CAAS;IAEnB,YAAY,GAAG,CAAC,CAAC;IAEzB,YACqB,IAAY,EACZ,MAAc,EACd,OAAmB;QAFnB,SAAI,GAAJ,IAAI,CAAQ;QACZ,WAAM,GAAN,MAAM,CAAQ;QACd,YAAO,GAAP,OAAO,CAAY;IACrC,CAAC;IAEG,KAAK,CAAC,OAAO,CAAI,EAAkC;QACtD,MAAM,IAAI,CAAC,aAAa,EAAE,CAAC;QAC3B,OAAO,IAAI,CAAC,gBAAgB,CAAC,EAAE,CAAC,CAAC;IACrC,CAAC;IAEM,QAAQ;QACX,OAAO,IAAI,CAAC,MAAM,CAAC;IACvB,CAAC;IAEO,KAAK,CAAC,aAAa;QACvB,IAAI,IAAI,CAAC,WAAW,EAAE,CAAC;YACnB,OAAO;QACX,CAAC;QAED,IAAI,CAAC,WAAW,GAAG,IAAI,CAAC;QACxB,MAAM,IAAI,CAAC,MAAM,CAAC,KAAK,CAAC,aAAa,IAAI,CAAC,IAAI,EAAE,CAAC,CAAC;IACtD,CAAC;IAEO,KAAK,CAAC,gBAAgB,CAC1B,EAAkC;QAElC,IAAI,CAAC,YAAY,EAAE,CAAC;QACpB,IAAI,CAAC;YACD,OAAO,MAAM,EAAE,CAAC,IAAI,CAAC,MAAM,CAAC,CAAC;QACjC,CAAC;QAAC,OAAO,CAAC,EAAE,CAAC;YACT,IAAI,CAAC,aAAa,CAAC,CAAU,CAAC,CAAC;YAC/B,MAAM,CAAC,CAAC;QACZ,CAAC;gBAAS,CAAC;YACP,IAAI,CAAC,YAAY,EAAE,CAAC;YACpB,MAAM,IAAI,CAAC,cAAc,EAAE,CAAC;QAChC,CAAC;IACL,CAAC;IAEO,aAAa,CAAC,KAAY;QAC9B,IAAI,CAAC,UAAU,GAAG,KAAK,CAAC;IAC5B,CAAC;IAEO,KAAK,CAAC,cAAc;QACxB,IAAI,IAAI,CAAC,YAAY,KAAK,CAAC,EAAE,CAAC;YAC1B,MAAM,CAAC,IAAI,CAAC,SAAS,EAAE,CAAC,CAAC,CAAC,IAAI,CAAC,QAAQ,EAAE,CAAC,CAAC,CAAC,IAAI,CAAC,MAAM,EAAE,CAAC,CAAC;QAC/D,CAAC;IACL,CAAC;IAEO,KAAK,CAAC,MAAM;QAChB,IAAI,IAAI,CAAC,MAAM,EAAE,CAAC;YACd,OAAO;QACX,CAAC;QAED,IAAI,CAAC,MAAM,GAAG,IAAI,CAAC;QACnB,MAAM,IAAI,CAAC,MAAM,CAAC,KAAK,CAAC,qBAAqB,IAAI,CAAC,IAAI,EAAE,CAAC,CAAC;QAC1D,IAAI,CAAC,OAAO,EAAE,CAAC;IACnB,CAAC;IAEO,KAAK,CAAC,QAAQ;QAClB,IAAI,IAAI,CAAC,MAAM,EAAE,CAAC;YACd,OAAO;QACX,CAAC;QAED,IAAI,CAAC,MAAM,GAAG,IAAI,CAAC;QACnB,MAAM,IAAI,CAAC,MAAM,CAAC,KAAK,CAAC,yBAAyB,IAAI,CAAC,IAAI,EAAE,CAAC,CAAC;QAC9D,IAAI,CAAC,OAAO,EAAE,CAAC;IACnB,CAAC;IAEO,SAAS;QACb,OAAO,IAAI,CAAC,UAAU,KAAK,SAAS,IAAI,CAAC,IAAI,CAAC,MAAM,CAAC;IACzD,CAAC;CACJ"}

package/package.json

@@ -3,7 +3,7 @@
   "publishConfig": {
     "access": "public"
   },
-  "version": "0.1.0",
+  "version": "0.2.1",
   "description": "PostgreSQL support for hexai",
   "license": "MIT",
   "author": "Sangwoo Hyun <wkdny.hyun@gmail.com>",
@@ -52,9 +52,9 @@
     "node-pg-migrate": "^7.9.0"
   },
   "peerDependencies": {
+    "ezcfg": "^0.1.0",
     "pg": "^8.0.0",
-    "@hexaijs/core": "^0.2.0",
-    "@hexaijs/utils": "^0.1.0"
+    "@hexaijs/core": "^0.2.0"
   },
   "devDependencies": {
     "@types/pg": "^8.11.8",