From 3cb1b115bcc87e28ba5c320b4265d80109c55e77 Mon Sep 17 00:00:00 2001
From: Aaron3S <chenyang@fit2cloud.com>
Date: Fri, 19 Dec 2025 17:37:08 +0800
Subject: [PATCH] feat: flush last command before close window

---
 .../jumpserver/chen/framework/console/QueryConsole.java   | 8 ++++++++
 1 file changed, 8 insertions(+)

diff --git a/backend/framework/src/main/java/org/jumpserver/chen/framework/console/QueryConsole.java b/backend/framework/src/main/java/org/jumpserver/chen/framework/console/QueryConsole.java
index ee11d62..26c90d2 100644
--- a/backend/framework/src/main/java/org/jumpserver/chen/framework/console/QueryConsole.java
+++ b/backend/framework/src/main/java/org/jumpserver/chen/framework/console/QueryConsole.java
@@ -505,6 +505,14 @@ private void sendDataView(DataView dataView, boolean clearOthers) {
 
     @Override
     public void close() {
+        if (this.currentPlan != null) {
+            // flush
+            var session = SessionManager.getCurrentSession();
+            var lastCmd = this.currentPlan.getTargetSQL();
+            var cmdRecord = new CommandRecord(lastCmd);
+            cmdRecord.setError("Abnormal exit");
+            session.recordCommand(cmdRecord);
+        }
         log.info("console closed");
     }
 }