From f7c2eeecf3e435e3a8bf07e4a656dcfc07e4150c Mon Sep 17 00:00:00 2001
From: Eric Fu <fuyufjh@gmail.com>
Date: Thu, 2 Jan 2025 12:54:50 +0800
Subject: [PATCH 1/2] add back parser support of label

Signed-off-by: Eric Fu <fuyufjh@gmail.com>
---
 sqllogictest/src/parser.rs  | 32 ++++++++++++++++++++++++++++----
 sqllogictest/src/runner.rs  |  4 +++-
 tests/retry/query_retry.slt | 32 ++++++++++++++++++++++++++++++++
 3 files changed, 63 insertions(+), 5 deletions(-)

diff --git a/sqllogictest/src/parser.rs b/sqllogictest/src/parser.rs
index b069469..2426a0a 100644
--- a/sqllogictest/src/parser.rs
+++ b/sqllogictest/src/parser.rs
@@ -95,6 +95,7 @@ pub enum QueryExpect<T: ColumnType> {
         types: Vec<T>,
         sort_mode: Option<SortMode>,
         result_mode: Option<ResultMode>,
+        label: Option<String>,
         results: Vec<String>,
     },
     /// Query should fail with the given error message.
@@ -108,6 +109,7 @@ impl<T: ColumnType> QueryExpect<T> {
             types: Vec::new(),
             sort_mode: None,
             result_mode: None,
+            label: None,
             results: Vec::new(),
         }
     }
@@ -255,12 +257,18 @@ impl<T: ColumnType> std::fmt::Display for Record<T> {
                 write!(f, "query ")?;
                 match expected {
                     QueryExpect::Results {
-                        types, sort_mode, ..
+                        types,
+                        sort_mode,
+                        label,
+                        ..
                     } => {
                         write!(f, "{}", types.iter().map(|c| c.to_char()).join(""))?;
                         if let Some(sort_mode) = sort_mode {
                             write!(f, " {}", sort_mode.as_str())?;
                         }
+                        if let Some(label) = label {
+                            write!(f, " {label}")?;
+                        }
                     }
                     QueryExpect::Error(err) => err.fmt_inline(f)?,
                 }
@@ -281,6 +289,8 @@ impl<T: ColumnType> std::fmt::Display for Record<T> {
                         for result in results {
                             write!(f, "\n{result}")?;
                         }
+
+                        // query always ends with a blank line
                         writeln!(f)?
                     }
                     QueryExpect::Error(err) => err.fmt_multiline(f)?,
@@ -815,6 +825,7 @@ fn parse_inner<T: ColumnType>(loc: &Location, script: &str) -> Result<Vec<Record
                         }
                     }
                     [type_str, res @ ..] => {
+                        // query <type-string> [<sort-mode>] [<label>] [retry <attempts> backoff <backoff>]
                         let types = type_str
                             .chars()
                             .map(|ch| {
@@ -822,17 +833,30 @@ fn parse_inner<T: ColumnType>(loc: &Location, script: &str) -> Result<Vec<Record
                                     .ok_or_else(|| ParseErrorKind::InvalidType(ch).at(loc.clone()))
                             })
                             .try_collect()?;
-                        let sort_mode = res.first().and_then(|&s| SortMode::try_from_str(s).ok()); // Could be `retry`
+                        let sort_mode = res.first().and_then(|&s| SortMode::try_from_str(s).ok()); // Could be `retry` or label
+
+                        // To support `retry`, we assume the label must *not* be "retry"
+                        let label_start = if sort_mode.is_some() { 1 } else { 0 };
+                        let res = &res[label_start..];
+                        let label = res.first().and_then(|&s| {
+                            if s != "retry" {
+                                Some(s.to_owned())
+                            } else {
+                                None // `retry` is not a valid label
+                            }
+                        });
 
-                        let retry_start = if sort_mode.is_some() { 1 } else { 0 };
+                        let retry_start = if label.is_some() { 1 } else { 0 };
+                        let res = &res[retry_start..];
                         (
                             QueryExpect::Results {
                                 types,
                                 sort_mode,
                                 result_mode: None,
+                                label,
                                 results: Vec::new(),
                             },
-                            &res[retry_start..],
+                            res,
                         )
                     }
                     [] => (QueryExpect::empty_results(), &[][..]),
diff --git a/sqllogictest/src/runner.rs b/sqllogictest/src/runner.rs
index 88d2b59..c42fec8 100644
--- a/sqllogictest/src/runner.rs
+++ b/sqllogictest/src/runner.rs
@@ -1640,7 +1640,7 @@ pub fn update_record_with_output<T: ColumnType>(
                     expected: match expected {
                         QueryExpect::Results {
                             sort_mode,
-
+                            label,
                             result_mode,
                             ..
                         } => QueryExpect::Results {
@@ -1648,12 +1648,14 @@ pub fn update_record_with_output<T: ColumnType>(
                             types,
                             sort_mode,
                             result_mode,
+                            label,
                         },
                         QueryExpect::Error(_) => QueryExpect::Results {
                             results,
                             types,
                             sort_mode: None,
                             result_mode: None,
+                            label: None,
                         },
                     },
                     retry,
diff --git a/tests/retry/query_retry.slt b/tests/retry/query_retry.slt
index 73f450e..f69ca7e 100644
--- a/tests/retry/query_retry.slt
+++ b/tests/retry/query_retry.slt
@@ -1,3 +1,4 @@
+# SYNTAX: query <type-string> [<sort-mode>] [<label>] [retry <attempts> backoff <backoff>]
 query I retry 3 backoff 5s
 SELECT id FROM test;
 ----
@@ -15,6 +16,37 @@ SELECT id FROM test;
 ----
 1 
 
+query I my_label retry 1 backoff 500ms
+SELECT id FROM test;
+----
+1 
+
+query I rowsort my_label retry 1 backoff 500ms
+SELECT id FROM test;
+----
+1 
+
+query I rowsort my_label
+SELECT id FROM test;
+----
+1 
+
+query I rowsort
+SELECT id FROM test;
+----
+1 
+
+query I my_label
+SELECT id FROM test;
+----
+1 
+
+query I
+SELECT id FROM test;
+----
+1 
+
+
 query error retry 2 backoff 500ms
 SELECT id FROM test;
 ----

From 209b0508c9df10e880172a5f3985d34fd0dbad36 Mon Sep 17 00:00:00 2001
From: xxchan <xxchan22f@gmail.com>
Date: Mon, 6 Jan 2025 14:13:20 +0800
Subject: [PATCH 2/2] bump version

Signed-off-by: xxchan <xxchan22f@gmail.com>
---
 CHANGELOG.md                    | 5 +++++
 Cargo.lock                      | 8 ++++----
 Cargo.toml                      | 2 +-
 sqllogictest-bin/Cargo.toml     | 4 ++--
 sqllogictest-engines/Cargo.toml | 2 +-
 5 files changed, 13 insertions(+), 8 deletions(-)

diff --git a/CHANGELOG.md b/CHANGELOG.md
index 256c3ed..4fc7e7b 100644
--- a/CHANGELOG.md
+++ b/CHANGELOG.md
@@ -7,6 +7,11 @@ and this project adheres to [Semantic Versioning](https://semver.org/spec/v2.0.0
 
 ## Unreleased
 
+## [0.26.0] - 2025-01-06
+
+* paser: Add back `label` support, which was removed in 0.25.0.
+* parser/runner: support `[statement|query] error retry` (Only support multi-line error message)
+
 ## [0.25.0] - 2024-12-26
 
 * runner: Add `retry` clause to `statement ok` and `query ok|error`.
diff --git a/Cargo.lock b/Cargo.lock
index e6a8eee..27d6b17 100644
--- a/Cargo.lock
+++ b/Cargo.lock
@@ -1,6 +1,6 @@
 # This file is automatically @generated by Cargo.
 # It is not intended for manual editing.
-version = 3
+version = 4
 
 [[package]]
 name = "addr2line"
@@ -1884,7 +1884,7 @@ dependencies = [
 
 [[package]]
 name = "sqllogictest"
-version = "0.25.0"
+version = "0.26.0"
 dependencies = [
  "async-trait",
  "educe",
@@ -1907,7 +1907,7 @@ dependencies = [
 
 [[package]]
 name = "sqllogictest-bin"
-version = "0.25.0"
+version = "0.26.0"
 dependencies = [
  "anyhow",
  "async-trait",
@@ -1929,7 +1929,7 @@ dependencies = [
 
 [[package]]
 name = "sqllogictest-engines"
-version = "0.25.0"
+version = "0.26.0"
 dependencies = [
  "async-trait",
  "bytes",
diff --git a/Cargo.toml b/Cargo.toml
index 5113a9a..eab0e01 100644
--- a/Cargo.toml
+++ b/Cargo.toml
@@ -3,7 +3,7 @@ resolver = "2"
 members = ["sqllogictest", "sqllogictest-bin", "sqllogictest-engines", "tests"]
 
 [workspace.package]
-version = "0.25.0"
+version = "0.26.0"
 edition = "2021"
 homepage = "https://github.com/risinglightdb/sqllogictest-rs"
 keywords = ["sql", "database", "parser", "cli"]
diff --git a/sqllogictest-bin/Cargo.toml b/sqllogictest-bin/Cargo.toml
index f188c3a..f8aa4b4 100644
--- a/sqllogictest-bin/Cargo.toml
+++ b/sqllogictest-bin/Cargo.toml
@@ -23,8 +23,8 @@ glob = "0.3"
 itertools = "0.13"
 quick-junit = { version = "0.5" }
 rand = "0.8"
-sqllogictest = { path = "../sqllogictest", version = "0.25" }
-sqllogictest-engines = { path = "../sqllogictest-engines", version = "0.25" }
+sqllogictest = { path = "../sqllogictest", version = "0.26" }
+sqllogictest-engines = { path = "../sqllogictest-engines", version = "0.26" }
 tokio = { version = "1", features = [
     "rt",
     "rt-multi-thread",
diff --git a/sqllogictest-engines/Cargo.toml b/sqllogictest-engines/Cargo.toml
index e50446a..f6468c6 100644
--- a/sqllogictest-engines/Cargo.toml
+++ b/sqllogictest-engines/Cargo.toml
@@ -20,7 +20,7 @@ postgres-types = { version = "0.2.8", features = ["derive", "with-chrono-0_4"] }
 rust_decimal = { version = "1.36.0", features = ["tokio-pg"] }
 serde = { version = "1", features = ["derive"] }
 serde_json = "1"
-sqllogictest = { path = "../sqllogictest", version = "0.25" }
+sqllogictest = { path = "../sqllogictest", version = "0.26" }
 thiserror = "2"
 tokio = { version = "1", features = [
     "rt",