Generate less Parts when handling ??

ClickHouse · Sep 18, 2024 · 00044a2 · 00044a2
1 parent c0a2825
commit 00044a2
Showing 1 changed file with 7 additions and 5 deletions.
diff --git a/src/sql/mod.rs b/src/sql/mod.rs
@@ -30,14 +30,16 @@ impl<'parts> SqlBuilder<'parts> {
         let mut parts = Vec::new();
         let mut rest = template;
         while let Some(idx) = rest.find('?') {
-            if idx != 0 {
+            if rest[idx+1..].starts_with('?') {
+                parts.push(Part::Str(&rest[..idx+1]));
+                rest = &rest[idx + 2..];
+                continue
+            } else if idx != 0 {
                 parts.push(Part::Str(&rest[..idx]));
             }
+
             rest = &rest[idx + 1..];
-            if let Some(restqq) = rest.strip_prefix('?') {
-                parts.push(Part::Str("?"));
-                rest = restqq;
-            } else if let Some(restfields) = rest.strip_prefix("fields") {
+            if let Some(restfields) = rest.strip_prefix("fields") {
                 parts.push(Part::Fields);
                 rest = restfields;
             } else {