fix: escape column names correctly

lancedb · Feb 27, 2024 · 639bdba · 639bdba
1 parent ea621cc
commit 639bdba
Showing 1 changed file with 8 additions and 1 deletion.
diff --git a/rust/lance/src/dataset/scanner.rs b/rust/lance/src/dataset/scanner.rs
@@ -187,6 +187,13 @@ pub struct Scanner {
     fragments: Option<Vec<Fragment>>,
 }
 
+fn escape_column_name(name: &str) -> String {
+    name.split(".")
+        .map(|s| format!("`{}`", s))
+        .collect::<Vec<_>>()
+        .join(".")
+}
+
 impl Scanner {
     pub fn new(dataset: Arc<Dataset>) -> Self {
         let projection = dataset.schema().clone();
@@ -254,7 +261,7 @@ impl Scanner {
         self.project_with_transform(
             &columns
                 .iter()
-                .map(|c| (c.as_ref(), format!("`{}`", c.as_ref())))
+                .map(|c| (c.as_ref(), escape_column_name(c.as_ref())))
                 .collect::<Vec<_>>(),
         )
     }