pg-interleave 0.3.2 → 0.5.0

package/dist/index.d.ts

@@ -4,6 +4,7 @@ export interface InterceptOptions {
     clientPid?: number;
     sessionTag?: string;
     onQueryMatch: string | RegExp;
+    withParams?: Array<string | number | boolean | null>;
     action?: "PAUSE_BEFORE_QUERY" | "PAUSE_AFTER_RESPONSE" | "pause_before_query" | "pause_after_response";
     times?: number;
 }
@@ -31,6 +32,7 @@ export declare class PgInterleaveClient {
     getPid(client: Client): Promise<number>;
     intercept(options: InterceptOptions): Promise<BarrierHandle>;
     injectTornCommit(client: Client, times?: number): Promise<void>;
+    flushTrace(): Promise<void>;
     injectDeadlock(options: {
         client?: Client;
         clientPid?: number;

package/dist/index.js

@@ -93,16 +93,24 @@ class PgInterleaveClient {
         const action = options.action
             ? options.action.toLowerCase()
             : "pause_before_query";
+        const serializedParams = options.withParams !== undefined
+            ? options.withParams.map((p) => (p === null || p === undefined ? null : String(p)))
+            : undefined;
+        const body = {
+            query_pattern: pattern,
+            action,
+            times: options.times ?? 1,
+        };
+        if (pid !== undefined)
+            body.target_pid = pid;
+        if (options.sessionTag !== undefined)
+            body.session_tag = options.sessionTag;
+        if (serializedParams !== undefined)
+            body.with_params = serializedParams;
         const res = await fetch(`${this.controlUrl}/intercept`, {
             method: "POST",
             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,
-            }),
+            body: JSON.stringify(body),
         });
         if (!res.ok) {
             const errBody = await res.text();
@@ -126,6 +134,14 @@ class PgInterleaveClient {
             throw new Error(`Failed to inject torn commit: HTTP ${res.status}`);
         }
     }
+    async flushTrace() {
+        const res = await fetch(`${this.controlUrl}/flush-trace`, {
+            method: "POST",
+        });
+        if (!res.ok) {
+            throw new Error(`Failed to flush trace: HTTP ${res.status}`);
+        }
+    }
     async injectDeadlock(options) {
         let pid = options.clientPid;
         if (!pid && options.client) {

package/package.json

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