From f3c5b1f6f14553fc9103bd2f4788f0edfd993def Mon Sep 17 00:00:00 2001
From: hanpen24 <ryo.hi79@gmail.com>
Date: Sat, 14 Sep 2024 16:48:42 +0900
Subject: [PATCH] cli: CLI prints out previous error message when executing a
 quit command

Previously, when executing the quit command in the CLI, the last
SQL error message would be displayed again. This was inadequate
because there is no need to show a previous SQL error when
executing a quit command. To address this, this patch modifies
the behavior so that when commands like quit or \q are executed,
any previous error messages are cleared.

Fixes: #85613

Release note (bug fix): Fixed a bug where the CLI would print out
the previous error message when executing the quit command.
---
 pkg/cli/clisqlshell/sql.go                        |  4 ++++
 pkg/cli/clisqlshell/sql_test.go                   | 14 ++++++++++++++
 pkg/cli/interactive_tests/test_last_statement.tcl |  4 ++--
 3 files changed, 20 insertions(+), 2 deletions(-)

diff --git a/pkg/cli/clisqlshell/sql.go b/pkg/cli/clisqlshell/sql.go
index 721661d9d148..1d31dbc56c31 100644
--- a/pkg/cli/clisqlshell/sql.go
+++ b/pkg/cli/clisqlshell/sql.go
@@ -1267,6 +1267,8 @@ func (c *cliState) doProcessFirstLine(startState, nextState cliStateEnum) cliSta
 		return startState
 
 	case "exit", "quit":
+		// When explicitly exiting, clear exitErr.
+		c.exitErr = nil
 		return cliStop
 	}
 
@@ -1412,6 +1414,8 @@ func (c *cliState) doHandleCliCmd(loopState, nextState cliStateEnum) cliStateEnu
 
 	switch cmd[0] {
 	case `\q`, `\quit`, `\exit`:
+		// When explicitly exiting, clear exitErr.
+		c.exitErr = nil
 		return cliStop
 
 	case `\`, `\?`, `\help`:
diff --git a/pkg/cli/clisqlshell/sql_test.go b/pkg/cli/clisqlshell/sql_test.go
index 83163bf9e323..8ca6a604f2bb 100644
--- a/pkg/cli/clisqlshell/sql_test.go
+++ b/pkg/cli/clisqlshell/sql_test.go
@@ -55,6 +55,12 @@ func Example_sql() {
 	c.RunWithArgs([]string{`sql`, `-e`, `select 1/(i-2) from generate_series(1,3) g(i)`})
 	c.RunWithArgs([]string{`sql`, `-e`, `SELECT '20:01:02+03:04:05'::timetz AS regression_65066`})
 
+	// Check that previous SQL error message is not displayed when the CLI is exited.
+	c.RunWithArgs([]string{`sql`, `-e`, `SELECT 1 FROM hoge`})
+	c.RunWithArgs([]string{`sql`, `-e`, `exit`})
+	c.RunWithArgs([]string{`sql`, `-e`, `SELECT 1 FROM hoge`})
+	c.RunWithArgs([]string{`sql`, `-e`, `\q`})
+
 	// Output:
 	// sql -e show application_name
 	// application_name
@@ -113,6 +119,14 @@ func Example_sql() {
 	// sql -e SELECT '20:01:02+03:04:05'::timetz AS regression_65066
 	// regression_65066
 	// 20:01:02+03:04:05
+	// sql -e SELECT 1 FROM hoge
+	// ERROR: relation "hoge" does not exist
+	// SQLSTATE: 42P01
+	// sql -e exit
+	// sql -e SELECT 1 FROM hoge
+	// ERROR: relation "hoge" does not exist
+	// SQLSTATE: 42P01
+	// sql -e \q
 }
 
 func Example_sql_config() {
diff --git a/pkg/cli/interactive_tests/test_last_statement.tcl b/pkg/cli/interactive_tests/test_last_statement.tcl
index 13f3825883c4..bab510fe0192 100644
--- a/pkg/cli/interactive_tests/test_last_statement.tcl
+++ b/pkg/cli/interactive_tests/test_last_statement.tcl
@@ -11,14 +11,14 @@ eexpect ":/# "
 send "$argv sql --no-line-editor\r"
 eexpect root@
 
-start_test "Check that an error in the last statement is propagated to the shell."
+start_test "Even if the last statement has a syntax error, if the user explicitly exits, the shell status should be normal exit."
 send "select ++;\r"
 eexpect "syntax error"
 eexpect root@
 send "\\q\r"
 eexpect ":/# "
 send "echo hello \$?\r"
-eexpect "hello 1"
+eexpect "hello 0"
 eexpect ":/# "
 end_test