From da2d1aab4447a9c066aadb9b99226d282d737250 Mon Sep 17 00:00:00 2001
From: Kould <kould2333@gmail.com>
Date: Wed, 24 Jul 2024 15:40:39 +0800
Subject: [PATCH] fix: projection miss _null and _ts

---
 src/lib.rs         |  2 +-
 src/transaction.rs | 13 ++++++++++++-
 2 files changed, 13 insertions(+), 2 deletions(-)

diff --git a/src/lib.rs b/src/lib.rs
index c70f9c48..b8fb68e8 100644
--- a/src/lib.rs
+++ b/src/lib.rs
@@ -295,7 +295,7 @@ where
         for p in &mut projection {
             *p += 2;
         }
-
+        projection.extend([0, 1, 2]);
         let mask = ProjectionMask::roots(
             &arrow_to_parquet_schema(R::arrow_schema()).unwrap(),
             projection,
diff --git a/src/transaction.rs b/src/transaction.rs
index 723c70b3..83114d05 100644
--- a/src/transaction.rs
+++ b/src/transaction.rs
@@ -92,7 +92,8 @@ where
         let streams = vec![TransactionScan {
             inner: self.local.range(range),
             ts: self.ts,
-        }.into()];
+        }
+        .into()];
         Scan::new(&self.share, range, self.ts, &self.version, streams)
     }
 
@@ -312,28 +313,38 @@ mod tests {
         let mut stream = txn
             .scan((Bound::Unbounded, Bound::Unbounded))
             .await
+            .projection(vec![1])
             .take()
             .await
             .unwrap();
 
         let entry_0 = stream.next().await.unwrap().unwrap();
         assert_eq!(entry_0.key().value, "1");
+        assert!(entry_0.value().vbool.is_none());
         let entry_1 = stream.next().await.unwrap().unwrap();
         assert_eq!(entry_1.key().value, "2");
+        assert!(entry_1.value().vbool.is_none());
         let entry_2 = stream.next().await.unwrap().unwrap();
         assert_eq!(entry_2.key().value, "3");
+        assert!(entry_2.value().vbool.is_none());
         let entry_3 = stream.next().await.unwrap().unwrap();
         assert_eq!(entry_3.key().value, "4");
+        assert!(entry_3.value().vbool.is_none());
         let entry_4 = stream.next().await.unwrap().unwrap();
         assert_eq!(entry_4.key().value, "5");
+        assert!(entry_4.value().vbool.is_none());
         let entry_5 = stream.next().await.unwrap().unwrap();
         assert_eq!(entry_5.key().value, "6");
+        assert!(entry_5.value().vbool.is_none());
         let entry_6 = stream.next().await.unwrap().unwrap();
         assert_eq!(entry_6.key().value, "7");
+        assert!(entry_6.value().vbool.is_none());
         let entry_7 = stream.next().await.unwrap().unwrap();
         assert_eq!(entry_7.key().value, "8");
+        assert!(entry_7.value().vbool.is_none());
         let entry_8 = stream.next().await.unwrap().unwrap();
         assert_eq!(entry_8.key().value, "9");
+        assert!(entry_8.value().vbool.is_none());
         let entry_9 = stream.next().await.unwrap().unwrap();
         assert_eq!(entry_9.key().value, "alice");
         let entry_10 = stream.next().await.unwrap().unwrap();