From 05eed776691c46f2dce5596a6c50ff6fb2713c51 Mon Sep 17 00:00:00 2001
From: Jean-Philippe Sirois <jp@veksen.net>
Date: Wed, 21 Jan 2026 12:36:09 +0400
Subject: [PATCH] feat: before and after explain plan

---
 src/remote/optimization.ts    | 1 +
 src/remote/query-optimizer.ts | 2 ++
 2 files changed, 3 insertions(+)

diff --git a/src/remote/optimization.ts b/src/remote/optimization.ts
index 48186ec..f2d7d46 100644
--- a/src/remote/optimization.ts
+++ b/src/remote/optimization.ts
@@ -29,6 +29,7 @@ export const LiveQueryOptimization = z.discriminatedUnion("state", [
     indexRecommendations: z.array(IndexRecommendation),
     indexesUsed: z.array(z.string()),
     explainPlan: z.custom<PostgresExplainStage>(),
+    optimizedExplainPlan: z.custom<PostgresExplainStage>(),
   }),
   z.object({
     state: z.literal("no_improvement_found"),
diff --git a/src/remote/query-optimizer.ts b/src/remote/query-optimizer.ts
index 35f6b53..f1bf79b 100644
--- a/src/remote/query-optimizer.ts
+++ b/src/remote/query-optimizer.ts
@@ -340,6 +340,7 @@ export class QueryOptimizer extends EventEmitter<EventMap> {
             indexRecommendations,
             indexesUsed,
             explainPlan,
+            optimizedExplainPlan: result.explainPlan,
           };
         }
       }
@@ -422,6 +423,7 @@ export class QueryOptimizer extends EventEmitter<EventMap> {
       indexRecommendations,
       indexesUsed,
       explainPlan,
+      optimizedExplainPlan: result.explainPlan,
     };
   }