pg-interleave 0.1.0 → 0.3.0

package/dist/index.d.ts

@@ -2,6 +2,7 @@ import type { Client } from "pg";
 export interface InterceptOptions {
     client?: Client;
     clientPid?: number;
+    sessionTag?: string;
     onQueryMatch: string | RegExp;
     action?: "PAUSE_BEFORE_QUERY" | "PAUSE_AFTER_RESPONSE" | "pause_before_query" | "pause_after_response";
     times?: number;
@@ -16,7 +17,16 @@ export declare class BarrierHandle {
 export declare class PgInterleaveClient {
     private readonly controlUrl;
     constructor(controlUrl?: string);
+    /**
+     * Appends session routing comment to an SQL query for connection pool compatibility
+     */
+    static tag(sql: string, sessionTag: string): string;
     reset(): Promise<void>;
+    resumeNext(): Promise<{
+        status: string;
+        barrier_id?: string;
+    }>;
+    barriers(): Promise<string[]>;
     hasTimeouts(): Promise<boolean>;
     getPid(client: Client): Promise<number>;
     intercept(options: InterceptOptions): Promise<BarrierHandle>;

package/dist/index.js

@@ -42,12 +42,34 @@ class PgInterleaveClient {
     constructor(controlUrl = "http://127.0.0.1:9090") {
         this.controlUrl = controlUrl;
     }
+    /**
+     * Appends session routing comment to an SQL query for connection pool compatibility
+     */
+    static tag(sql, sessionTag) {
+        return `/* interleave: session=${sessionTag} */ ${sql}`;
+    }
     async reset() {
         const res = await fetch(`${this.controlUrl}/reset`, { method: "POST" });
         if (!res.ok) {
             throw new Error(`Failed to reset proxy: HTTP ${res.status}`);
         }
     }
+    async resumeNext() {
+        const res = await fetch(`${this.controlUrl}/resume-next`, {
+            method: "POST",
+        });
+        if (!res.ok) {
+            throw new Error(`Failed to resume next barrier: HTTP ${res.status}`);
+        }
+        return (await res.json());
+    }
+    async barriers() {
+        const res = await fetch(`${this.controlUrl}/barriers`);
+        if (!res.ok) {
+            throw new Error(`Failed to fetch active barriers: HTTP ${res.status}`);
+        }
+        return (await res.json());
+    }
     async hasTimeouts() {
         const res = await fetch(`${this.controlUrl}/events`);
         if (!res.ok) {
@@ -76,6 +98,7 @@ class PgInterleaveClient {
             headers: { "Content-Type": "application/json" },
             body: JSON.stringify({
                 target_pid: pid ?? null,
+                session_tag: options.sessionTag ?? null,
                 query_pattern: pattern,
                 action,
                 times: options.times ?? 1,

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "pg-interleave",
-  "version": "0.1.0",
+  "version": "0.3.0",
   "description": "Deterministic PostgreSQL wire-protocol proxy client for concurrency and chaos testing",
   "main": "dist/index.js",
   "types": "dist/index.d.ts",